chiark / gitweb /
wip finding window
authorIan Jackson <ian@liberator.relativity.greenend.org.uk>
Sun, 7 Jun 2009 12:50:20 +0000 (13:50 +0100)
committerIan Jackson <ian@liberator.relativity.greenend.org.uk>
Sun, 7 Jun 2009 12:50:20 +0000 (13:50 +0100)
pctb/convert.c
pctb/convert.h
pctb/ocr.h
pctb/pages.c

index cdd144a..b44c294 100644 (file)
@@ -11,8 +11,18 @@ void debug_flush(void) {
 const char *get_vardir(void) { return "."; }
 
 
-static enum { mode_all=03, mode_screenshot=01, mode_analyse=02 }
-    o_mode= mode_all;
+static enum {
+  mf_findwindow=    01,
+  mf_screenshot=    02,
+  mf_analyse=       04,
+  
+  mode_findwindow=  01,
+  mode_screenshot=  03,
+  mode_analyse=     04,
+
+  mode_all=         07,
+} o_mode= mode_all;
+
 static char *o_screenshots_fn;
 static int o_single_page;
 
@@ -23,7 +33,9 @@ int main(int argc, char **argv) {
   int r;
 
   while ((arg=*++argv)) {
-    if (!strcmp(arg,"--screenshot-only"))
+    if (!strcmp(arg,"--find-window-only"))
+      o_mode= mode_findwindow;
+    else if (!strcmp(arg,"--screenshot-only"))
       o_mode= mode_screenshot;
     else if (!strcmp(arg,"--analyse-only"))
       o_mode= mode_analyse;
@@ -51,9 +63,11 @@ int main(int argc, char **argv) {
     eassert(r>=0);  eassert(o_screenshots_fn);
   }
 
-  if (o_mode & mode_screenshot) {
+  if (o_mode & mf_findwindow) {
     screenshot_startup();
     find_yppclient_window();
+  }
+  if (o_mode & mf_screenshot) {
     screenshots_file= fopen(o_screenshots_fn, "w"); eassert(screenshots_file);
     if (o_single_page)
       take_one_screenshot();
@@ -63,7 +77,7 @@ int main(int argc, char **argv) {
     screenshots_file= fopen(o_screenshots_fn, "r"); eassert(screenshots_file);
     read_screenshots();
   }
-  if (o_mode & mode_analyse) {
+  if (o_mode & mf_analyse) {
     analyse();
     //output_tsv();
   }
index d532c2b..5bb4707 100644 (file)
@@ -34,5 +34,7 @@ void take_one_screenshot(void);
 extern CanonImage *page_images[MAX_PAGES];
 extern int npages;
 
+extern char *ocean, *pirate;
+
 
 #endif /*CONVERT_H*/
index e8b5b1a..5b5f697 100644 (file)
@@ -55,10 +55,11 @@ OcrResultGlyph *ocr(OcrReader *rd, OcrCellType, int w, Pixcol cols[]);
 /*----- debugging arrangements, rather contingent -----*/
 
 #define DEBUG_FLAG_LIST                                \
-   DF(ocr)                                     \
+   DF(findypp)                                 \
+   DF(pages)                                   \
    DF(rect)                                    \
-   DF(callout)                                 \
-   DF(pages)
+   DF(ocr)                                     \
+   DF(callout)
 
 enum {
 #define DF(f) dbg__shift_##f,
@@ -82,13 +83,15 @@ void debug_flush(void);
 
 const char *get_vardir(void);
 
-#define DEBUG_DEFINE_DEBUGF(f)                                         \
-  static void vdebugf(const char *fmt, va_list al) {                   \
-    if (DEBUGP(f))                                                     \
+#define DEBUG_DEFINE_SOME_DEBUGF(fl,funcf)                             \
+  static void v##funcf(const char *fmt, va_list al) {                  \
+    if (DEBUGP(fl))                                                    \
       vfprintf(debug,fmt,al);                                          \
   }                                                                    \
-  static void debugf(const char *fmt, ...) {                           \
-    va_list al;  va_start(al,fmt);  vdebugf(fmt,al);  va_end(al);      \
+  static void funcf(const char *fmt, ...) {                            \
+    va_list al;  va_start(al,fmt);  v##funcf(fmt,al);  va_end(al);     \
   }
 
+#define DEBUG_DEFINE_DEBUGF(fl) DEBUG_DEFINE_SOME_DEBUGF(fl,debugf)
+
 #endif /*OCR_H*/
index 2f4f669..b9e1ac1 100644 (file)
@@ -11,6 +11,8 @@
 CanonImage *page_images[MAX_PAGES];
 int npages;
 
+char *ocean, *pirate;
+
 static XWindowAttributes attr;
 static Window id;
 static Display *disp;
@@ -298,7 +300,109 @@ void set_yppclient_window(unsigned long wul) {
   id= wul;
 }
 
+DEBUG_DEFINE_SOME_DEBUGF(findypp,debugfind)
+
 void find_yppclient_window(void) {
+  Window root, gotroot, gotparent;
+  int screen, r;
+  int nfound=0;
+  
   if (id) return;
-  eassert(!"finding client window NYI");
+  
+  static const char prefix[]= "Puzzle Pirates - ";
+  static const char onthe[]= " on the ";
+  static const char suffix[]= " ocean";
+#define S(x) (sizeof((x))-1)
+
+  Atom wm_name= XInternAtom(disp,"WM_NAME",True);
+  eassert(wm_name != None);
+
+  for (screen=0; screen<ScreenCount(disp); screen++) {
+    debugfind("FINDYPP screen %d\n", screen);
+    root= RootWindow(disp,screen);
+    unsigned int nchildren1;
+    Window *children1=0;
+
+    r= XQueryTree(disp,root,
+                 &gotroot,&gotparent,
+                 &children1,&nchildren1);
+    eassert(r);
+    debugfind("FINDYPP screen %d nchildren1=%d\n", screen, nchildren1);
+
+    int i;
+    for (i=0; i<nchildren1; i++) {
+      Window w1= children1[i];
+      unsigned int nchildren2;
+      Window *children2=0;
+
+      r= XQueryTree(disp,w1,
+                   &gotroot,&gotparent,
+                   &children2,&nchildren2);
+      eassert(r);
+      debugfind("FINDYPP screen %d c1[%2d]=0x%08lx nchildren2=%d\n",
+               screen, i, (unsigned long)w1, nchildren2);
+
+      int j;
+      for (j=-1; j<(int)nchildren2; j++) {
+       Window w2= j<0 ? w1 : children2[j];
+       debugfind("FINDYPP screen %d c1[%2d]=0x%08lx c2[%2d]=0x%08lx",
+                 screen, i, (unsigned long)w1, j, (unsigned long)w2);
+
+       int gotfmt;
+       Atom gottype;
+       unsigned long len, gotbytesafter;
+       char *title;
+       unsigned char *gottitle=0;
+       r= XGetWindowProperty(disp,w2, wm_name,0,512, False,
+                             AnyPropertyType,&gottype, &gotfmt, &len,
+                             &gotbytesafter, &gottitle);
+       eassert(!r);
+       title= (char*)gottitle;
+
+       if (DEBUGP(findypp)) {
+         debugfind(" gf=%d len=%lu gba=%lu \"", gotfmt,len,gotbytesafter);
+         char *p;
+         for (p=title; p < title+len; p++) {
+           char c= *p;
+           if (c>=' ' && c<=126) fputc(c,debug);
+           else fprintf(debug,"\\x%02x",c & 0xff);
+         }
+         fputs("\": ",debug);
+       }
+
+#define REQUIRE(pred)                                                     \
+        if (!(pred)) { debugfind(" failed test  %s\n", #pred); continue; } \
+        else
+
+       REQUIRE( gottype!=None );
+       REQUIRE( len );
+       REQUIRE( gotfmt==8 );
+
+       REQUIRE( len >= S(prefix) + 1 + S(onthe) + 1 + S(suffix) );
+
+       char *spc1= strchr(  title        + S(prefix), ' ');  REQUIRE(spc1);
+       char *spc2= strrchr((title + len) - S(suffix), ' ');  REQUIRE(spc2);
+
+       REQUIRE( (title + len) - spc1  >= S(onthe)  + S(suffix) );
+       REQUIRE(  spc2         - title >= S(prefix) + S(onthe) );
+
+       REQUIRE( !memcmp(title,                   prefix, S(prefix)) );
+       REQUIRE( !memcmp(title + len - S(suffix), suffix, S(suffix))  );
+       REQUIRE( !memcmp(spc1,                    onthe,  S(onthe))  );
+
+#define ASSIGN(what, start, end) do {                                  \
+       r= asprintf(&what, "%.*s", end - start, start);  eassert(r>0);  \
+     }while(0)
+       ASSIGN(ocean,  title + S(prefix),  spc1);
+       ASSIGN(pirate, spc1 + S(onthe),   (title + len) - S(suffix));
+
+       debugfind(" YES!\n");
+       id= w2;
+       nfound++;
+      }
+      if (children2) XFree(children2);
+    }
+    if (children1) XFree(children1);
+  }
+  eassert(nfound==1);
 }