X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~yarrgweb/git?p=ypp-sc-tools.db-test.git;a=blobdiff_plain;f=pctb%2Fpages.c;h=1fba5a64fccc2b6ccbacc952bd9efb29daa86c1f;hp=39690388ebc41589c583653916aaaefc7a8adf3e;hb=5b8cb7299a9e60f7bfeb3fe5f3d7556e2357d485;hpb=ae2213ef391622c3365442baa088fa42bf7586b2 diff --git a/pctb/pages.c b/pctb/pages.c index 3969038..1fba5a6 100644 --- a/pctb/pages.c +++ b/pctb/pages.c @@ -1,168 +1,249 @@ /* - */ + * Interaction with the YPP client via X11 + */ +/* + * This is part of ypp-sc-tools, a set of third-party tools for assisting + * players of Yohoho Puzzle Pirates. + * + * Copyright (C) 2009 Ian Jackson + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + * + * Yohoho and Puzzle Pirates are probably trademarks of Three Rings and + * are used without permission. This program is not endorsed or + * sponsored by Three Rings. + */ + +/* + * Only this file #includes the X11 headers, as they're quite + * pollutant of the namespace. + */ -#include "ocr.h" +#include "structure.h" #include #include #include +#include + +#include +#include +#include CanonImage *page_images[MAX_PAGES]; int npages; +char *ocean, *pirate; + +static XWindowAttributes attr; static Window id; static Display *disp; static struct timeval tv_startup; +static unsigned wwidth, wheight; +static int max_relevant_y= -1; -static KeyCode keycode(KeySym sym) { - return XKeysymToKeycode(disp,sym); -} +static XImage *shmim; +static XShmSegmentInfo shminfo; -static void check_pclose(FILE *f, char *cmd) { - int r; - eassert(!ferror(f)); - r= fgetc(f); eassert(r==EOF); eassert(feof(f)); - r= pclose(f); eassert(r>=0); eassert(WIFEXITED(r) && !WEXITSTATUS(r)); - free(cmd); -} +DEBUG_DEFINE_DEBUGF(pages) -#if 0 -static CanonImage *screenshot_now(void) { - char *cmd; - CanonImage *ci; - int r; - - r= asprintf(&cmd, "xwd -silent -id 0x%lx | xwdtopnm", (unsigned long)id); - eassert(r>=0); - FILE *f= popen(cmd,"r"); eassert(f); - ci= file_read_image(f); - check_pclose(f, cmd); - return ci; -} -#endif +#define xassert(what) \ + ((what) ? (void)0 : \ + fatal("X11 operation unexpectedly failed." \ + " %s:%d: %s\n", __FILE__,__LINE__,#what)) -static void screenshot_startup(void) { - int r; - disp= XOpenDisplay(0); eassert(disp); - r= gettimeofday(&tv_startup,0); eassert(!r); +static KeyCode keycode(KeySym sym) { + return XKeysymToKeycode(disp,sym); } -#if 0 -static CanonImage *single_page(void) { - int r; - r= XRaiseWindow(disp, id); eassert(r); - r= XSync(disp, False); eassert(r); - return screenshot_now(); +void screenshot_startup(void) { + progress("starting..."); + disp= XOpenDisplay(0); + if (!disp) fatal("Unable to open X11 display."); + sysassert(! gettimeofday(&tv_startup,0) ); } -#endif /*---------- pager ----------*/ -typedef struct { - size_t len; - unsigned char d[]; -} Snapshot; +typedef XImage Snapshot; -static size_t snapshot_alloc= 1024; static double last_input; -static const double min_update_allowance= 0.5; +static const double min_update_allowance= 0.25; static double timestamp(void) { struct timeval tv; - int r; - r= gettimeofday(&tv,0); eassert(!r); + sysassert(! gettimeofday(&tv,0) ); double t= (tv.tv_sec - tv_startup.tv_sec) + (tv.tv_usec - tv_startup.tv_usec) * 1e-6; - fprintf(stderr,"%f\n",t); + debugf("PAGING %f\n",t); return t; } static void delay(double need_sleep) { - int r; - fprintf(stderr,"PAGING delay %f\n",need_sleep); - r= usleep(need_sleep * 1e6); eassert(!r); + debugf("PAGING delay %f\n",need_sleep); + sysassert(! usleep(need_sleep * 1e6) ); } static void sync_after_input(void) { - int r; - r= XSync(disp, False); eassert(r); + xassert( XSync(disp, False) ); last_input= timestamp(); } +static void translate_coords_toroot(int wx, int wy, int *rx, int *ry) { + Window dummy; + xassert( XTranslateCoordinates(disp, id,attr.root, wx,wy, rx,ry, &dummy) ); +} + +static void check_client_window_all_on_screen(void) { + Rect onroot; + unsigned rwidth, rheight; + Window dummy; + unsigned bd, depth; + int rxpos, rypos; + + xassert( XGetGeometry(disp,attr.root, &dummy, &rxpos,&rypos, + &rwidth, &rheight, + &bd,&depth) ); + + translate_coords_toroot(0,0, &onroot.tl.x,&onroot.tl.y); + translate_coords_toroot(wwidth-1,wheight-1, &onroot.br.x,&onroot.br.y); + if (!(onroot.tl.x >= 0 && + onroot.tl.y >= 0 && + onroot.br.x < rwidth && + onroot.br.y < rheight)) + fatal("YPP client window is not entirely on the screen."); +} + +static void check_not_disturbed(void) { + XEvent ev; + int r; + + for (;;) { + r= XCheckMaskEvent(disp, ~0, &ev); + if (r==False) return; + + switch (ev.type) { + case VisibilityNotify: + if (ev.xvisibility.state != VisibilityUnobscured) + fatal("YPP client window has become obscured."); + break; + case ConfigureNotify: + check_client_window_all_on_screen(); + break; + case FocusOut: + fatal("Focus left YPP client window."); + break; + case FocusIn: + warning("focus entered YPP client window ?!"); + break; + default: + fatal("Received unexpected X11 event (type code %d)!", ev.type); + } + } +} + static void send_key(KeySym sym) { + check_not_disturbed(); XTestFakeKeyEvent(disp, keycode(sym),1, 10); XTestFakeKeyEvent(disp, keycode(sym),0, 10); } +static int pgupdown; + static void send_pgup_many(void) { int i; - for (i=0; i<25; i++) + for (i=0; i<25; i++) { send_key(XK_Prior); - fprintf(stderr,"PAGING PageUp x %d\n",i); + pgupdown--; + } + debugf("PAGING PageUp x %d\n",i); sync_after_input(); } static void send_pgdown(void) { send_key(XK_Next); - fprintf(stderr,"PAGING PageDown\n"); + pgupdown++; + debugf("PAGING PageDown\n"); + sync_after_input(); +} +static void send_pgdown_torestore(void) { + debugf("PAGING PageDown x %d\n", -pgupdown); + while (pgupdown < 0) { + send_key(XK_Next); + pgupdown++; + } sync_after_input(); } +static void free_snapshot(Snapshot **io) { + if (*io) XDestroyImage(*io); + *io= 0; +} + static void snapshot(Snapshot **output) { - char *cmd; - int r; - - free(*output); *output=0; + free_snapshot(output); - fprintf(stderr,"PAGING snapshot\n"); + debugf("PAGING snapshot\n"); - r= asprintf(&cmd, "xwd -silent -id 0x%lx", (unsigned long)id); eassert(r>=0); - FILE *f= popen(cmd,"r"); eassert(f); + timestamp(); + *output= XGetImage(disp,id, 0,0, wwidth,wheight, AllPlanes, ZPixmap); + timestamp(); - int need_alloc=1; - size_t used=0; - for (;;) { - size_t allow= snapshot_alloc - used; - if (!allow) { - snapshot_alloc <<= 1; - need_alloc= 1; - fprintf(stderr,"PAGING snapshot grow %ld\n", (long)snapshot_alloc); - continue; - } - if (need_alloc) { - *output= realloc(*output, sizeof(**output) + snapshot_alloc); - eassert(*output); - need_alloc= 0; - } - size_t got= fread((*output)->d, 1, allow, f); - if (got==0) break; - used += got; - } - check_pclose(f,cmd); + check_not_disturbed(); - (*output)->len= used; - snapshot_alloc= used+1; - *output= realloc(*output, sizeof(**output) + snapshot_alloc); - eassert(*output); - - fprintf(stderr,"PAGING snapshot len=%ld\n", (long)used); + debugf("PAGING snapshot done.\n"); } static int identical(const Snapshot *a, const Snapshot *b) { - return !(memcmp(a, b, sizeof(*a)) || - memcmp(a->d, b->d, a->len)); + if (!(a->width == b->width && + a->height == b->height && + a->bytes_per_line == b->bytes_per_line && + a->format == b->format)) + return 0; + + int compare_to= a->height; + if (max_relevant_y && compare_to > max_relevant_y) + compare_to= max_relevant_y; + + return !memcmp(a->data, b->data, a->bytes_per_line * compare_to); } static void wait_for_stability(Snapshot **output, const Snapshot *previously, - void (*with_keypress)(void)) { + void (*with_keypress)(void), + const char *fmt, ...) + FMT(4,5); + +static void wait_for_stability(Snapshot **output, + const Snapshot *previously, + void (*with_keypress)(void), + const char *fmt, ...) { + va_list al; + va_start(al,fmt); + Snapshot *last=0; /* waits longer if we're going to return an image identical to previously * if previously==0, all images are considered identical to it */ - fprintf(stderr,"PAGING wait_for_stability" + debugf("PAGING wait_for_stability" " last_input=%f previously=%p\n", last_input, previously); + char *doing; + sysassert( vasprintf(&doing,fmt,al) >=0); + + progress("%s",doing); + for (;;) { double at_snapshot= timestamp(); double need_sleep= min_update_allowance - (at_snapshot - last_input); @@ -172,18 +253,20 @@ static void wait_for_stability(Snapshot **output, if (!with_keypress && !(previously && identical(*output,previously))) { - fprintf(stderr,"PAGING wait_for_stability simple\n"); + debugf("PAGING wait_for_stability simple\n"); break; } if (last && identical(*output,last)) { - fprintf(stderr,"PAGING wait_for_stability stabilised\n"); + debugf("PAGING wait_for_stability stabilised\n"); break; } - fprintf(stderr,"PAGING wait_for_stability retry\n"); + progress_spinner("%s",doing); - free(last); last=*output; *output=0; + debugf("PAGING wait_for_stability retry\n"); + + free_snapshot(&last); last=*output; *output=0; if (with_keypress) with_keypress(); @@ -191,79 +274,362 @@ static void wait_for_stability(Snapshot **output, delay(0.5); } - free(last); - fprintf(stderr,"PAGING wait_for_stability done.\n"); + free_snapshot(&last); + free(doing); + debugf("PAGING wait_for_stability done.\n"); + va_end(al); } -static void raise_and_set_focus(void) { - int r; - XWindowAttributes attr; - int xpos,ypos, evbase,errbase,majver,minver; - unsigned width,height,bd,depth; - Window dummy; +static void raise_and_get_details(void) { + int evbase,errbase,majver,minver; + int wxpos, wypos; + unsigned bd,depth; + + progress("raising and checking YPP client window..."); + + debugf("PAGING raise_and_get_details\n"); + + int xtest= XTestQueryExtension(disp, &evbase,&errbase,&majver,&minver); + if (!xtest) fatal("X server does not support the XTEST extension."); + + xassert( XRaiseWindow(disp, id) ); + /* in case VisibilityNotify triggers right away before we have had a + * change to raise; to avoid falsely detecting lowering in that case */ - fprintf(stderr,"PAGING raise_and_set_focus\n"); + xassert( XSelectInput(disp, id, + StructureNotifyMask| + VisibilityChangeMask + ) ); + + xassert( XRaiseWindow(disp, id) ); + /* in case the window was lowered between our Raise and our SelectInput; + * to avoid failing to detect that lowering */ + + xassert( XGetWindowAttributes(disp, id, &attr) ); + xassert( XGetGeometry(disp,id, &attr.root, + &wxpos,&wypos, &wwidth,&wheight, + &bd,&depth) ); + + if (!(wwidth >= 320 && wheight >= 320)) + fatal("YPP client window is implausibly small?"); + + check_client_window_all_on_screen(); + + int shm= XShmQueryExtension(disp); + if (shm) { + xassert( shmim= XShmCreateImage(disp, attr.visual, attr.depth, ZPixmap, + 0,&shminfo, wwidth,wheight) ); + + sigset_t oldset, all; + sigfillset(&all); + sysassert(! sigprocmask(SIG_BLOCK,&all,&oldset) ); + + int pfd[2]; + pid_t cleaner; + sysassert(! pipe(pfd) ); + sysassert( (cleaner= fork()) != -1 ); + if (!cleaner) { + sysassert(! close(pfd[1]) ); + for (;;) { + int r= read(pfd[0], &shminfo.shmid, sizeof(shminfo.shmid)); + if (!r) exit(0); + if (r==sizeof(shminfo.shmid)) break; + assert(r==-1 && errno==EINTR); + } + for (;;) { + char bc; + int r= read(pfd[0],&bc,1); + if (r>=0) break; + assert(r==-1 && errno==EINTR); + } + sysassert(! shmctl(shminfo.shmid,IPC_RMID,0) ); + exit(0); + } + sysassert(! close(pfd[0]) ); - r= XTestQueryExtension(disp, &evbase,&errbase,&majver,&minver); - eassert(r==True); + sysassert(! sigprocmask(SIG_SETMASK,&oldset,0) ); - r= XRaiseWindow(disp, id); eassert(r); + assert(shmim->height == wheight); + sysassert( (shminfo.shmid= + shmget(IPC_PRIVATE, shmim->bytes_per_line * wheight, + IPC_CREAT|0600)) >= 0 ); - r= XGetWindowAttributes(disp, id, &attr); eassert(r); - r= XGetGeometry(disp,id, &attr.root,&xpos,&ypos,&width,&height, &bd,&depth); - eassert(r); + sysassert( write(pfd[1],&shminfo.shmid,sizeof(shminfo.shmid)) == + sizeof(shminfo.shmid) ); + sysassert( shminfo.shmaddr= shmat(shminfo.shmid,0,0) ); + shmim->data= shminfo.shmaddr; + shminfo.readOnly= False; + xassert( XShmAttach(disp,&shminfo) ); - r= XTranslateCoordinates(disp, id,attr.root, 160,160, &xpos,&ypos, - &dummy); - eassert(r); + close(pfd[1]); /* causes IPC_RMID */ + } +} +static void set_focus(void) { int screen= XScreenNumberOfScreen(attr.screen); - XTestFakeMotionEvent(disp,screen, xpos, ypos, 0); + + progress("taking control of YPP client window..."); + + debugf("PAGING set_focus\n"); + + int xpos, ypos; + translate_coords_toroot(160,160, &xpos,&ypos); + XTestFakeMotionEvent(disp,screen, xpos,ypos, 0); XTestFakeButtonEvent(disp,1,1, 50); XTestFakeButtonEvent(disp,1,0, 50); sync_after_input(); - fprintf(stderr,"PAGING raise_and_set_focus done.\n"); + + delay(0.5); + xassert( XSelectInput(disp, id, + StructureNotifyMask| + VisibilityChangeMask| + FocusChangeMask + ) ); + + translate_coords_toroot(10,10, &xpos,&ypos); + XTestFakeMotionEvent(disp,screen, xpos,ypos, 0); + + sync_after_input(); + + debugf("PAGING raise_and_set_focus done.\n"); +} + +#define SAMPLEMASK 0xfful + +typedef struct { + int lshift, rshift; +} ShMask; + +static void compute_shift_mask(ShMask *sm, unsigned long ximage_mask) { + sm->lshift= 0; + sm->rshift= 0; + + for (;;) { + if (ximage_mask <= (SAMPLEMASK>>1)) { + sm->lshift++; ximage_mask <<= 1; + } else if (ximage_mask > SAMPLEMASK) { + sm->rshift++; ximage_mask >>= 1; + } else { + break; + } + assert(!(sm->lshift && sm->rshift)); + } + assert(sm->lshift < LONG_BIT); + assert(sm->rshift < LONG_BIT); } +static CanonImage *convert_page(Snapshot *sn) { + ShMask shiftmasks[3]; + CanonImage *im; + + fprintf(screenshot_file, + "P6\n" + "%d %d\n" + "255\n", sn->width, sn->height); + +#define COMPUTE_SHIFT_MASK(ix, rgb) \ + compute_shift_mask(&shiftmasks[ix], sn->rgb##_mask) + COMPUTE_SHIFT_MASK(0, red); + COMPUTE_SHIFT_MASK(1, green); + COMPUTE_SHIFT_MASK(2, blue); + + CANONICALISE_IMAGE(im, sn->width, sn->height, { + long xrgb= XGetPixel(sn, x, y); + int i; + rgb= 0; + for (i=0; i<3; i++) { + rgb <<= 8; + unsigned long sample= + ((xrgb << shiftmasks[i].lshift) + >> shiftmasks[i].rshift) & SAMPLEMASK; + rgb |= sample; + fputc(sample, screenshot_file); + } + }); + + sysassert(!ferror(screenshot_file)); + sysassert(!fflush(screenshot_file)); + return im; +} -static void read_pages(void) { +void take_screenshots(void) { Snapshot *current=0, *last=0; + CanonImage *test; - raise_and_set_focus(); + /* find the window and check it's on the right kind of screen */ + raise_and_get_details(); + wait_for_stability(¤t,0,0, "checking current YPP client screen..."); + test= convert_page(current); + find_structure(test, &max_relevant_y); + check_correct_commodities(); + free(test); /* page to the top - keep pressing page up until the image stops changing */ - wait_for_stability(&last,0, send_pgup_many); + set_focus(); + wait_for_stability(¤t,0, send_pgup_many, + "paging up to top of commodity list..."); /* now to actually page down */ for (;;) { - fprintf(stderr,"PAGING page %d\n",npages); - //eassert(npages < MAX_PAGES); - //page_images[npages++]= + debugf("paging page %d\n",npages); + + if (!(npages < MAX_PAGES)) + fatal("Paging down seems to generate too many pages - max is %d.", + MAX_PAGES); + + page_images[npages]= convert_page(current); + free_snapshot(&last); last=current; current=0; + + debugf("PAGING page %d converted\n",npages); + + wait_for_stability(¤t,last, 0, + "collecting screenshot of page %d...", + npages+1); - wait_for_stability(¤t,last, 0); if (npages && /* first pagedown doesn't do much */ identical(current,last)) { - free(current); + free_snapshot(¤t); break; } - free(last); last=current; current=0; - send_pgdown(); npages++; } - fprintf(stderr,"PAGING all done.\n"); -} + progress("finishing with the YPP client..."); + send_pgdown_torestore(); -int main(int argc, char **argv) { - screenshot_startup(); + debugf("PAGING all done.\n"); + progress_log("collected %d screenshots.",npages); +} - id= strtoul(*++argv,0,0); +void take_one_screenshot(void) { + Snapshot *current=0; + + raise_and_get_details(); + sync_after_input(); + wait_for_stability(¤t,0,0, "taking screenshot..."); + page_images[0]= convert_page(current); + npages= 1; + progress_log("collected single screenshot."); +} - read_pages(); - return 0; +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; + int nfound=0; + + if (id) return; + + progress("looking for YPP client window..."); + + 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); + xassert(wm_name != None); + + for (screen=0; screen=' ' && 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 { \ + sysassert( asprintf(&what, "%.*s", (end)-(start), start) >0 ); \ + }while(0) + ASSIGN(pirate, title + S(prefix), spc1); + ASSIGN(ocean, spc1 + S(onthe), (title + len) - S(suffix)); + + debugfind(" YES!\n"); + id= w2; + nfound++; + progress_log("found YPP client (0x%lx):" + " %s ocean - %s.", + (unsigned long)id, ocean, pirate); + } + if (children2) XFree(children2); + } + if (children1) XFree(children1); + } + if (nfound>1) + fatal("Found several YPP clients." + " Close one, or specify the windowid with --window-id.\n"); + if (nfound<1) + fatal("Did not find YPP client." + " Use --window-id and/or report this as a fault.\n"); +}