m + fix multiscreen bug

Signed-off-by: Mattias Andrée <maandree@operamail.com>

3 files changed, 14 insertions, 11 deletions

diff --git a/src/blueshift_iccprofile.c b/src/blueshift_iccprofile.c
index 99d5c42..e87a5c1 100644
--- a/src/blueshift_iccprofile.c
+++ b/src/blueshift_iccprofile.c
@@ -50,7 +50,6 @@ int main(int argc, char **argv)
   char* display = NULL;
   xcb_screen_iterator_t iter;
   int screen_count;
-  xcb_screen_t* screens;
   int screen_i;
   
   
@@ -83,21 +82,22 @@ int main(int argc, char **argv)
   
   /* Acquire a list of all screens in the display, */
   iter = xcb_setup_roots_iterator(xcb_get_setup(connection));
-  /* count the list, */
+  /* count the list. */
   screen_count = iter.rem;
-  /* and start at the first screen. */
-  screens = iter.data;
   
   for (screen_i = 0; screen_i < screen_count; screen_i++)
     {
       /* For each screen */
-      xcb_screen_t* screen = screens + screen_i;
+      xcb_screen_t* screen = iter.data;
       
       xcb_list_properties_cookie_t list_cookie;
       xcb_list_properties_reply_t* list_reply;
       xcb_atom_t* atoms;
       xcb_atom_t* atoms_end;
       
+      /* We have acquired the screen, got to next in preperate for next iteration. */
+      xcb_screen_next(&iter);
+      
       
       /* Get root window properties */
       
diff --git a/src/blueshift_idcrtc.c b/src/blueshift_idcrtc.c
index 6bb0fff..6306096 100644
--- a/src/blueshift_idcrtc.c
+++ b/src/blueshift_idcrtc.c
@@ -66,7 +66,6 @@ int main(int argc, char** argv)
   xcb_randr_query_version_reply_t* randr_version;
   xcb_screen_iterator_t iter;
   int screen_count;
-  xcb_screen_t* screens;
   int screen_i;
   int i;
   
@@ -119,23 +118,24 @@ int main(int argc, char** argv)
   
   /* Acquire a list of all screens in the display, */
   iter = xcb_setup_roots_iterator(xcb_get_setup(connection));
-  /* count the list, */
+  /* count the list. */
   screen_count = iter.rem;
-  /* and start at the first screen. */
-  screens = iter.data;
   
   /* Print the number available screens. */
   printf("Screen count: %i\n", screen_count);
   for (screen_i = 0; screen_i < screen_count; screen_i++)
     {
       /* For each screen */
-      xcb_screen_t* screen = screens + screen_i;
+      xcb_screen_t* screen = iter.data;
       xcb_randr_get_screen_resources_current_cookie_t res_cookie;
       xcb_randr_get_screen_resources_current_reply_t* res_reply;
       xcb_randr_output_t* outputs;
       xcb_randr_crtc_t* crtcs;
       int output_i;
       
+      /* We have acquired the screen, got to next in preperate for next iteration. */
+      xcb_screen_next(&iter);
+      
       /* Print the screen index. */
       printf("Screen: %i\n", screen_i);
       
diff --git a/src/blueshift_randr_c.c b/src/blueshift_randr_c.c
index 0e38afc..eb3bdf1 100644
--- a/src/blueshift_randr_c.c
+++ b/src/blueshift_randr_c.c
@@ -65,6 +65,7 @@ int blueshift_randr_open(int use_screen, char* display)
   xcb_randr_get_crtc_gamma_size_reply_t* gamma_size_reply;
   xcb_randr_get_crtc_gamma_cookie_t gamma_get_cookie;
   xcb_randr_get_crtc_gamma_reply_t* gamma_get_reply;
+  int iter_i;
   
   
   
@@ -101,7 +102,9 @@ int blueshift_randr_open(int use_screen, char* display)
   /* Get X resources */
   
   iter = xcb_setup_roots_iterator(xcb_get_setup(connection));
-  screen = iter.data + use_screen;
+  for (iter_i = 0; iter_i < use_screen; iter_i++)
+    xcb_screen_next(&iter);
+  screen = iter.data;
   
   res_cookie = xcb_randr_get_screen_resources_current(connection, screen->root);
   res_reply = xcb_randr_get_screen_resources_current_reply(connection, res_cookie, &error);