joystick und xorggrabber ausgelagert

1 files changed, 163 insertions, 0 deletions

diff --git a/xorg_grabber.cpp b/xorg_grabber.cpp
new file mode 100644
index 0000000..386cab6
--- /dev/null
+++ b/xorg_grabber.cpp
@@ -0,0 +1,163 @@
+#include "xorg_grabber.h"
+
+#include <iostream>
+using namespace std;
+
+XorgGrabber::XorgGrabber(const char* win_title)
+{
+	conn=xcb_connect(NULL,NULL);
+	
+	bool found_win=false;
+	grab_screen=NULL;
+	img=NULL;
+	
+	/* Find configured screen */
+	const xcb_setup_t* setup = xcb_get_setup(conn);
+	for (xcb_screen_iterator_t i = xcb_setup_roots_iterator(setup);
+		 i.rem > 0; xcb_screen_next (&i))
+	{
+		xcb_screen_t* scr = i.data;
+		xcb_query_tree_reply_t* reply = xcb_query_tree_reply( conn, xcb_query_tree(conn, scr->root), NULL);
+		if (reply)
+		{
+			int len = xcb_query_tree_children_length(reply);
+			xcb_window_t* children = xcb_query_tree_children(reply);
+			xcb_get_window_attributes_cookie_t* cookies = new xcb_get_window_attributes_cookie_t[len];
+			for (int i=0; i<len; ++i)
+				cookies[i]=xcb_get_window_attributes(conn, children[i]);
+			
+			for (int i=0; i<len; ++i)
+			{
+				xcb_get_window_attributes_reply_t* attr =
+					xcb_get_window_attributes_reply (conn, cookies[i], NULL);
+				
+				
+				xcb_get_property_reply_t* title_reply = xcb_get_property_reply(conn, 
+					xcb_get_property(conn, false, children[i], XCB_ATOM_WM_NAME, XCB_GET_PROPERTY_TYPE_ANY, 0, 128),
+					NULL );
+				
+				if (!attr->override_redirect && attr->map_state == XCB_MAP_STATE_VIEWABLE)
+				{
+					char* title=(char*)(title_reply+1);
+					cout << title << endl;
+					if (strstr(title, win_title))
+					{
+						xcb_get_geometry_reply_t* geo;
+						geo = xcb_get_geometry_reply (conn, xcb_get_geometry (conn, children[i]), NULL);
+						if (geo)
+						{
+							grab_width=geo->width;
+							grab_height=geo->height;
+							
+							free(geo);
+
+							grabbed_win=children[i];
+							found_win=true;
+
+							grab_screen=scr;						
+						}
+						else
+						{
+							cerr << "geo==NULL!" << endl;
+						}
+					}
+				}
+
+				free(title_reply);
+				free(attr);
+			}
+			
+			free(reply);
+			delete[] cookies;
+		}
+		else
+		{
+			cout << "xcb_get_setup failed" << endl;
+		}
+	}
+	
+	if (found_win)
+	{
+		xcb_get_image_reply_t* img = xcb_get_image_reply (conn,
+			xcb_get_image (conn, XCB_IMAGE_FORMAT_Z_PIXMAP, grabbed_win,
+				0, 0, grab_width, grab_height, ~0), NULL);
+		
+
+
+		xcb_depth_iterator_t depth_iterator = xcb_screen_allowed_depths_iterator(grab_screen);
+		
+		int ndepths=xcb_screen_allowed_depths_length(grab_screen);
+		
+		for (int i=0; i<ndepths; ++i)
+		{
+			if (depth_iterator.data->depth == img->depth)
+			{
+				xcb_visualtype_t* visuals = xcb_depth_visuals(depth_iterator.data);
+				int nvisuals = xcb_depth_visuals_length(depth_iterator.data);
+				
+				for (int j=0;j<nvisuals;j++)
+				{
+					if (visuals[j].visual_id==img->visual)
+					{
+						assert(visuals[j]._class==XCB_VISUAL_CLASS_TRUE_COLOR || visuals[j]._class==XCB_VISUAL_CLASS_DIRECT_COLOR);
+						cout << (int)visuals[j]._class << "," << XCB_VISUAL_CLASS_TRUE_COLOR << endl;
+						cout << visuals[j].red_mask << endl;
+						cout << visuals[j].green_mask << endl;
+						cout << visuals[j].blue_mask << endl;
+						break;
+					}
+				}
+				
+				break;
+			}
+			
+			xcb_depth_next(&depth_iterator);
+		}
+		
+		
+		
+		int nformats = xcb_setup_pixmap_formats_length(setup);
+		xcb_format_t* formats = xcb_setup_pixmap_formats(setup);
+		for (int i=0;i<nformats;i++)
+		{
+			if (formats[i].depth==img->depth)
+			{
+				cout << (int)formats[i].bits_per_pixel << endl;
+				cout << (int)formats[i].scanline_pad << endl;
+				break;
+			}
+		}
+		
+		cout << grab_width << "x" << grab_height << endl;
+		
+		free(img);
+		
+	}
+	else
+	{
+		throw string("FATAL: did not find window, exiting.");
+	}
+	
+	
+}
+
+XorgGrabber::~XorgGrabber()
+{
+	if (img) free(img);
+	xcb_disconnect(conn);
+}
+
+void XorgGrabber::read(Mat& mat)
+{
+	if (img) free(img);
+
+
+	// mat gets invalid when the next read() is called!
+	img = xcb_get_image_reply (conn,
+	xcb_get_image (conn, XCB_IMAGE_FORMAT_Z_PIXMAP, grabbed_win,
+		0, 0, grab_width, grab_height, ~0), NULL);
+	
+	mat = Mat(grab_height, grab_width, CV_8UC4, xcb_get_image_data(img));
+	mat.addref();
+}
+