chiark / gitweb /
Merge ijackson@chiark:/home/ijackson/things/ypp-sc-tools.pctb-dict-test
[ypp-sc-tools.main.git] / pctb / rgbimage.c
index e49044f..d0ba6b7 100644 (file)
 
 #include "convert.h"
 
-#define dbassert(x)                                                    \
-  ((x) ? (void)0 :                                                     \
-     fatal("Error in pixmap database.\n"                               \
-          " Requirement not met: %s:%d: %s", __FILE__,__LINE__, #x))
+static int identify1(const RgbImage *base, Rect portion,
+                   char result[MAXIMGIDENT], const char *what,
+                   const char *which) {
+  char *dbfile_name= masprintf("#%s-pixmap#.txt",which);
+  if (!dbfile_open(dbfile_name))
+    goto not_found;
 
-static int identify(const RgbImage *base, Rect portion,
-                   char result[MAXIMGIDENT]) {
-  FILE *f;
-  sysassert( f= fopen("pixmaps.txt","r") );
-  struct pam inpam;
-
-#define FGETSLINE (fgetsline(f,result,MAXIMGIDENT))
+#define FGETSLINE (dbfile_getsline(result,MAXIMGIDENT,__FILE__,__LINE__))
 
   FGETSLINE;
-fprintf(stderr,">%s<\n",result);
   dbassert(!strcmp(result,"# ypp-sc-tools pctb pixmaps v1"));
   
   void *row= 0;
@@ -70,57 +65,96 @@ fprintf(stderr,">%s<\n",result);
   
   for (;;) {
     FGETSLINE;
-    if (!result || result[0]=='#') continue;
+    if (!result[0] || result[0]=='#') continue;
     if (!strcmp(result,".")) break;
 
-    pnm_readpaminit(f, &inpam, sizeof(inpam));
-    dbassert(inpam.maxval == 255);
-    dbassert(inpam.bytes_per_sample == 1);
-    dbassert(inpam.format == PPM_FORMAT);
+    char magic[10];
+    dbfile_getsline(magic,sizeof(magic),__FILE__,__LINE__);
 
-    if (rowa < inpam.width) {
-      rowa= inpam.width;
+    dbassert(!strcmp(magic,"P3"));
+    int w,h,maxval;
+    dbassert( dbfile_scanf("%d %d %d",&w,&h,&maxval) == 3);
+    dbassert(w>0); dbassert(h>0); dbassert(maxval==255);
+
+    if (rowa < w) {
+      rowa= w;
       row= mrealloc(row, rowa*3);
     }
-    int diff=
-      inpam.width  != RECT_W(portion) ||
-      inpam.height != RECT_H(portion);
-
-    int y;
-    for (y=0; y<inpam.height; y++) {
-      int r= fread(row,1, 3*inpam.width, f);
-      sysassert(!ferror(f));
-      dbassert(!feof(f));
-      assert(r == 3*inpam.width);
-      if (diff) continue;
-
-      diff= memcmp(RI_PIXEL(base, portion.tl.x, portion.tl.y + y),
-                  row, 3*inpam.width);
+    int diff= w != RECT_W(portion) || h != RECT_H(portion);
+
+    int x,y,i;
+    for (y=0; y<h; y++) {
+      for (x=0; x<w; x++) {
+       for (i=0; i<3; i++) {
+         int c;
+         dbassert( dbfile_scanf("%d",&c) == 1);
+         dbassert(c>=0 && c<=255);
+         diff |= (c != RI_PIXEL(base, portion.tl.x + x, portion.tl.y + y)[i]);
+       }
+      }
+    }
+    if (!diff) {
+      progress_log("Identified %s image: %s.",what,result);
+      goto found;
     }
-    if (!diff)
-      return 1;
   }
-  return 0;
+ not_found:
+  result[0]= 0;
+
+ found:
+  dbfile_close();
+  free(dbfile_name);
+  return !!result[0];
 }
-void identify_rgbimage(const RgbImage *base, Rect portion,
-                      char result[MAXIMGIDENT]) {
-  int ok= identify(base, portion, result);
-  if (ok) return;
 
-  if (DEBUGP(pixmap)) {
-    int x,y,i;
-    fprintf(stderr,"P3\n%d %d\n255\n", RECT_W(portion), RECT_H(portion));
-    for (y=portion.tl.y; y<=portion.br.y; y++) {
-      for (x=portion.tl.x; x<=portion.br.x; x++) {
-       putc(' ',stderr);
-       for (i=0; i<3; i++)
-         fprintf(stderr," %3d", RI_PIXEL(base,x,y)[i]);
-      }
-      putc('\n',stderr);
+static int identify(const RgbImage *base, Rect portion,
+                   char result[MAXIMGIDENT], const char *what) {
+  return identify1(base,portion,result,what, "master") ||
+         identify1(base,portion,result,what, "local");
+}
+
+static void fwrite_ppm(FILE *f, const RgbImage *base, Rect portion) {
+  int x,y,i;
+  fprintf(f,"P3\n%d %d\n255\n", RECT_W(portion), RECT_H(portion));
+  for (y=portion.tl.y; y<=portion.br.y; y++) {
+    for (x=portion.tl.x; x<=portion.br.x; x++) {
+      putc(' ',f);
+      for (i=0; i<3; i++)
+       fprintf(f," %3d", RI_PIXEL(base,x,y)[i]);
     }
+    putc('\n',f);
+  }
+  sysassert(!ferror(f));
+  sysassert(!fflush(f));
+}
+
+void identify_rgbimage(const RgbImage *base, Rect portion,
+                      char result[MAXIMGIDENT], const char *what) {
+  static int synced;
+
+  if (!synced) {
+    fetch_with_rsync("pixmap");
+    synced++;
+  }
+  
+  for (;;) {
+    int ok= identify(base, portion, result, what);
+    if (ok) return;
+    
+    if (DEBUGP(pixmap))
+      fwrite_ppm(stderr,base,portion);
+
+    FILE *resolver= resolve_start();
+    if (!resolver)
+       fatal("Image recognition failed - unrecognised island.\n"
+             "See FIXME.FIXME\n");
+
+    fprintf(resolver, "pixmap\n" "%s\n", what);
+    fwrite_ppm(resolver, base, portion);
+    putc('\n',resolver);
+
+    resolve_finish();
   }
-  fatal("Unrecognised pixmap.");
 }
 
 RgbImage *alloc_rgb_image(int w, int h) {