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           diff |= (c != RI_PIXEL(base, portion.tl.x + x, portion.tl.y + y)[i]);
  93         }
  94       }
  95     }
  96     if (!diff) {
  97       progress_log("Identified %s image: %s.",what,result);
  98       goto found;
  99     }
 100   }
 101  not_found:
 102   result[0]= 0;
 103
 104  found:
 105   dbfile_close();
 106   free(dbfile_name);
 107   return !!result[0];
 108 }
 109
 110 static int identify(const RgbImage *base, Rect portion,
 111                     char result[MAXIMGIDENT], const char *what) {
 112   return identify1(base,portion,result,what, "master") ||
 113          identify1(base,portion,result,what, "local");
 114 }
 115
 116 static void fwrite_ppm(FILE *f, const RgbImage *base, Rect portion) {
 117   int x,y,i;
 118   fprintf(f,"P3\n%d %d\n255\n", RECT_W(portion), RECT_H(portion));
 119   for (y=portion.tl.y; y<=portion.br.y; y++) {
 120     for (x=portion.tl.x; x<=portion.br.x; x++) {
 121       putc(' ',f);
 122       for (i=0; i<3; i++)
 123         fprintf(f," %3d", RI_PIXEL(base,x,y)[i]);
 124     }
 125     putc('\n',f);
 126   }
 127   sysassert(!ferror(f));
 128   sysassert(!fflush(f));
 129 }
 130
 131 void identify_rgbimage(const RgbImage *base, Rect portion,
 132                        char result[MAXIMGIDENT], const char *what) {
 133   static int synced;
 134
 135   if (!synced) {
 136     fetch_with_rsync("pixmap");
 137     synced++;
 138   }
 139
 140   for (;;) {
 141     int ok= identify(base, portion, result, what);
 142     if (ok) return;
 143
 144     if (DEBUGP(pixmap))
 145       fwrite_ppm(stderr,base,portion);
 146
 147     FILE *resolver= resolve_start();
 148     if (!resolver)
 149         fatal("Image recognition failed - unrecognised island.\n"
 150               "See FIXME.FIXME\n");
 151
 152     fprintf(resolver, "pixmap\n" "%s\n", what);
 153     fwrite_ppm(resolver, base, portion);
 154     putc('\n',resolver);
 155
 156     resolve_finish();
 157   }
 158 }
 159
 160 RgbImage *alloc_rgb_image(int w, int h) {
 161   RgbImage *ri;
 162   ri= mmalloc(sizeof(*ri) + w*h*3);
 163   ri->w= w;
 164   ri->h= h;
 165   return ri;
 166 }