pctb/rgbimage.c

   1 /*
   2  * Handling of colour (RGB) images
   3  */
   4 /*
   5  *  This is part of ypp-sc-tools, a set of third-party tools for assisting
   6  *  players of Yohoho Puzzle Pirates.
   7  *
   8  *  Copyright (C) 2009 Ian Jackson <ijackson@chiark.greenend.org.uk>
   9  *
  10  *  This program is free software: you can redistribute it and/or modify
  11  *  it under the terms of the GNU General Public License as published by
  12  *  the Free Software Foundation, either version 3 of the License, or
  13  *  (at your option) any later version.
  14  *
  15  *  This program is distributed in the hope that it will be useful,
  16  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  *  GNU General Public License for more details.
  19  *
  20  *  You should have received a copy of the GNU General Public License
  21  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22  *
  23  *  Yohoho and Puzzle Pirates are probably trademarks of Three Rings and
  24  *  are used without permission.  This program is not endorsed or
  25  *  sponsored by Three Rings.
  26  */
  27
  28 #include "convert.h"
  29
  30 /*
  31  *
  32  *
  33  * widgets.txt
  34  * format
  35  *
  36  *  <magic>
  37  *
  38  *  <context>
  39  *  <string>
  40  *  ppmnoraw depth 8
  41  *
  42  *  <context>
  43  *  <string>
  44  *  ppmnoraw depth 8
  45  *  .
  46  *
  47  */
  48
  49 #include "convert.h"
  50
  51 static int identify1(const RgbImage *base, Rect portion,
  52                     char result[MAXIMGIDENT], const char *what,
  53                     const char *which) {
  54   char *dbfile_name= masprintf("#%s-pixmap#.txt",which);
  55   if (!dbfile_open(dbfile_name))
  56     goto not_found;
  57
  58 #define FGETSLINE (dbfile_getsline(result,MAXIMGIDENT,__FILE__,__LINE__))
  59
  60   FGETSLINE;
  61   dbassert(!strcmp(result,"# ypp-sc-tools pctb pixmaps v1"));
  62
  63   void *row= 0;
  64   size_t rowa= 0;
  65
  66   for (;;) {
  67     FGETSLINE;
  68     if (!result[0] || result[0]=='#') continue;
  69     if (!strcmp(result,".")) break;
  70
  71     char magic[10];
  72     dbfile_getsline(magic,sizeof(magic),__FILE__,__LINE__);
  73
  74     dbassert(!strcmp(magic,"P3"));
  75     int w,h,maxval;
  76     dbassert( dbfile_scanf("%d %d %d",&w,&h,&maxval) == 3);
  77     dbassert(w>0); dbassert(h>0); dbassert(maxval==255);
  78
  79     if (rowa < w) {
  80       rowa= w;
  81       row= mrealloc(row, rowa*3);
  82     }
  83     int diff= w != RECT_W(portion) || h != RECT_H(portion);
  84
  85     int x,y,i;
  86     for (y=0; y<h; y++) {
  87       for (x=0; x<w; x++) {
  88         for (i=0; i<3; i++) {
  89           int c;
  90           dbassert( dbfile_scanf("%d",&c) == 1);
  91           dbassert(c>=0 && c<=255);
  92           int px= portion.tl.x + x, py= portion.tl.y + y;
  93           diff |= px > portion.br.x || py > portion.br.y ||
  94                   (c != RI_PIXEL(base,px,py)[i]);
  95         }
  96       }
  97     }
  98     if (!diff) {
  99       progress_log("Identified %s image: %s.",what,result);
 100       goto found;
 101     }
 102   }
 103  not_found:
 104   result[0]= 0;
 105
 106  found:
 107   dbfile_close();
 108   free(dbfile_name);
 109   return !!result[0];
 110 }
 111
 112 static int identify(const RgbImage *base, Rect portion,
 113                     char result[MAXIMGIDENT], const char *what) {
 114   return identify1(base,portion,result,what, "master") ||
 115          identify1(base,portion,result,what, "local");
 116 }
 117
 118 void fwrite_ppmraw(FILE *f, const RgbImage *ri) {
 119   fprintf(f,
 120           "P6\n"
 121           "%d %d\n"
 122           "255\n", ri->w, ri->h);
 123   int count= ri->w * ri->h * 3;
 124   sysassert( fwrite(ri->data, 1, count, f) == count );
 125   sysassert(!ferror(f));
 126   sysassert(!fflush(f));
 127 }
 128
 129 static void fwrite_ppm(FILE *f, const RgbImage *base, Rect portion) {
 130   int x,y,i;
 131   fprintf(f,"P3\n%d %d\n255\n", RECT_W(portion), RECT_H(portion));
 132   for (y=portion.tl.y; y<=portion.br.y; y++) {
 133     for (x=portion.tl.x; x<=portion.br.x; x++) {
 134       putc(' ',f);
 135       for (i=0; i<3; i++)
 136         fprintf(f," %3d", RI_PIXEL(base,x,y)[i]);
 137     }
 138     putc('\n',f);
 139   }
 140   sysassert(!ferror(f));
 141   sysassert(!fflush(f));
 142 }
 143
 144 void identify_rgbimage(const RgbImage *base, Rect portion,
 145                        char result[MAXIMGIDENT], const char *what) {
 146   static int synced;
 147
 148   if (!synced) {
 149     fetch_with_rsync("pixmap");
 150     synced++;
 151   }
 152
 153   for (;;) {
 154     int ok= identify(base, portion, result, what);
 155     if (ok) return;
 156
 157     if (DEBUGP(pixmap))
 158       fwrite_ppm(stderr,base,portion);
 159
 160     FILE *resolver= resolve_start();
 161     if (!resolver)
 162         fatal("Image recognition failed - unrecognised island.\n"
 163               "See FIXME.FIXME\n");
 164
 165     fprintf(resolver, "pixmap\n" "%s\n", what);
 166     fwrite_ppm(resolver, base, portion);
 167     putc('\n',resolver);
 168
 169     resolve_finish();
 170   }
 171 }
 172
 173 RgbImage *alloc_rgb_image(int w, int h) {
 174   RgbImage *ri;
 175   ri= mmalloc(sizeof(*ri) + w*h*3);
 176   ri->w= w;
 177   ri->h= h;
 178   return ri;
 179 }