/*
* in_xpm.cpp -- read an XPM file
* by [email protected] at Fri Mar  1 09:56:54 CET 2002
*/
/* Imp: test this code with various xpm files! */

#ifdef __GNUC__
#ifndef __clang__
#pragma implementation
#endif
#endif

#include "image.hpp"

#if USE_IN_XPM

#if OBJDEP
#  warning REQUIRES: mapping.o
#endif

#ifndef USE_BIG_MEMORY
#define USE_BIG_MEMORY 0
#endif

#include "mapping.hpp"
#include "error.hpp"
#include "xpmc.h"

#include <string.h> /* memchr() */
#include "gensio.hpp"

#define USGE(a,b) ((unsigned char)(a))>=((unsigned char)(b))

#if 0
/** @return true iff params are different strings (not respecting case) */
static int my_strcase_neq(char *s1, char *s2) {
 while ((USGE(*s1,'A') && USGE('Z',*s1) ? *s1+'a'-'A' : *s1) ==
        (USGE(*s2,'A') && USGE('Z',*s2) ? *s2+'a'-'A' : *s2)) {
   if (*s1=='\0') return 0;
   s1++; s2++;
 }
 return 1;
}
#endif

#define my_strcase_neq(s1,s2) GenBuffer::nocase_strcmp(s1,s2)

/** @ return RGB long */
static Image::Sampled::rgb_t parse_rgb(char const*s) {
 unsigned v=0, len;
 Image::Sampled::rgb_t ret;
 if (!s || !*s) return 0x2000000; /* not found */
 // fprintf(stderr, "'%s'\n", s);
 if (*s=='#') { /* an #RRGGBB web-style color spec; or #RRRRGGGGBBBB */
   unsigned dif=0;
   ++s;
   while (dif<13 && (USGE(5,(s[dif]|32)-'a') || USGE(9,s[dif]-'0'))) dif++; /* find at most 13 hex digits */
   if (s[dif]!='\0' || (dif!=12 && dif!=6)) return 0x2000000; /* not found, spec length error */
   dif=(dif==12) ? 3 : 1;
   // shr=24; while (shr!=0) {
   ret= (Image::Sampled::rgb_t)( USGE(9,*s-'0') ? *s-'0' : 10+(*s|32)-'a' )<<20;
   s++;
   ret|=(Image::Sampled::rgb_t)( USGE(9,*s-'0') ? *s-'0' : 10+(*s|32)-'a' )<<16;
   s+=dif; /* ignore lower two hex digits of 16-bit sample value */
   ret|=(Image::Sampled::rgb_t)( USGE(9,*s-'0') ? *s-'0' : 10+(*s|32)-'a' )<<12;
   s++;
   ret|=(Image::Sampled::rgb_t)( USGE(9,*s-'0') ? *s-'0' : 10+(*s|32)-'a' )<<8;
   s+=dif; /* ignore lower two hex digits of 16-bit sample value */
   ret|=(Image::Sampled::rgb_t)( USGE(9,*s-'0') ? *s-'0' : 10+(*s|32)-'a' )<<4;
   s++;
   ret|=(Image::Sampled::rgb_t)( USGE(9,*s-'0') ? *s-'0' : 10+(*s|32)-'a' );
   return ret;
 }
 /* vvv 223==255-32: ignore case when hashing */
 char const *p=s;
 while (*p!='\0') v=xpmColors_mul*v+(223&*(unsigned char const*)p++);
 p=xpmColors_dat+xpmColors_ofs[(v&65535)%xpmColors_mod];
 while (*p!='\0') {
   len=strlen(p);
   if (0==my_strcase_neq(p,s)) {
     p+=len;
     ret=(((Image::Sampled::rgb_t)((unsigned char const*)p)[1])<<16)+
         (((Image::Sampled::rgb_t)((unsigned char const*)p)[2])<<8)+
         (((Image::Sampled::rgb_t)((unsigned char const*)p)[3]));
     return (ret==0x30201) ? 0x1000000 : ret; /* transparent color */
   }
   p+=len+4;
 }
 return 0x2000000; /* not found */
}

class XPMTok {
public:
 /* Imp: report line numbers on errors */
 BEGIN_STATIC_ENUM1(int)
   T_NO_UNGOT=-1,
   T_COMMA=257 /* The comma token, outside strings. */
 END_STATIC_ENUM()
 /* XPM states */
 BEGIN_STATIC_ENUM1(char)
   ST_OUT=0, /* outside s string */
   ST_STR=1, /* inside a string */
   ST_EOF=2 /* EOF */
 END_STATIC_ENUM()
 int getcc();
 /** Reads an unsigned int. */
 Image::Sampled::dimen_t getDimen();
 Image::Sampled::rgb_t getColor();
 void read(char *buf, unsigned len);
 inline void readInStr(char *buf, unsigned len);
 inline XPMTok(FILE *f_): f(f_), state(ST_OUT), ungot(T_NO_UNGOT) {}
 inline void ungetcc(int c) { if (c>=0) ungot=c; }
 void getComma();
protected:
 FILE *f;
 /** Current state. */
 char state;
 int ungot;
};

int XPMTok::getcc() {
 int i;
 if (ungot>=0 /*T_NO_UNGOT<0*/) { i=ungot; ungot=T_NO_UNGOT; return i; }
 /* Imp: ignore C and C++ style comments (so we won't recognise strings inside comments) */
 switch (state) {
  case ST_OUT: st_out:
   while (1) {
     switch ((i=MACRO_GETC(f))) {
      case -1: state=ST_EOF; return -1;
      case ',': return T_COMMA;
      case '"': state=ST_STR; goto st_str;
      default: /* ignore outside strings */ break;
     }
   }
  case ST_STR: st_str:
   i=MACRO_GETC(f);
   if (i==-1) { ue: Error::sev(Error::EERROR) << "XPM: unexpected EOF" << (Error*)0; }
   else if (i=='"') { state=ST_OUT; goto st_out; }
   else if (i=='\\') {
     if ((i=MACRO_GETC(f))==-1) goto ue;
     if (i=='\n') goto st_str;
     /* Imp: handle octal, hexadecimal, \n etc. */
   }
   return i&255;
  default: return -1;
 }
}
Image::Sampled::dimen_t XPMTok::getDimen() {
 Image::Sampled::dimen_t ret=0, bak;
 int i;
 while ((i=getcc())==' ' || i=='\t') ;
 if (USGE(i,'0') && USGE('9',i)) {
   ret=i-'0';
   while (USGE((i=getcc()),'0') && USGE('9',i)) {
     bak=ret;
     ret=ret*10+(i-'0');
     if (ret/10!=bak) Error::sev(Error::EERROR) << "XPM: dimen overflow" << (Error*)0;
   }
   ungetcc(i);
   return ret;
 } else Error::sev(Error::EERROR) << "XPM: dimen expected" << (Error*)0;
 return 0; /*notreached*/
}
void XPMTok::getComma() {
 if (getcc()!=T_COMMA) Error::sev(Error::EERROR) << "XPM: comma expected at " << ftell(f) << (Error*)0;
}
void XPMTok::read(char *buf, unsigned len) {
 int i;
 while (len--!=0) {
   i=getcc();
   // fprintf(stderr,"i=%d\n", i);
   if (i>=0 && i<=255) *buf++=i; else Error::sev(Error::EERROR) << "XPM: data expected" << (Error*)0;
 }
}
void XPMTok::readInStr(char *buf, unsigned len) {
 /* Dat: this is OK noew */
 // assert(state==ST_STR);
 assert(ungot<0);
 param_assert(len>=1);
 if (state!=ST_STR) { len--; goto real; }
 int i;
 while (len--!=0) {
   if ((i=MACRO_GETC(f))>=0 && i<=255 && i!='"' && i!='\\') *buf++=i;
   else { assert(i>=0); ungetc(i,f); real:
     i=getcc();
     // fprintf(stderr,"i=%d\n", i);
     if (i>=0 && i<=255) *buf++=i; else Error::sev(Error::EERROR) << "XPM: data expected" << (Error*)0;
   }
 }
}
Image::Sampled::rgb_t XPMTok::getColor() {
 static char tmp[32];
 int i;
 while ((i=getcc())==' ' || i=='\t') ;
 if (i=='g') { at_g:
   i=getcc();
   if (i!='4') ungetcc(i);
   goto at_col;
 } else if (i=='c' || i=='m' || i=='b' || i=='s') { at_col:
   while ((i=getcc())==' ' || i=='\t') ;
   char *p=tmp, *pend=tmp+sizeof(tmp)-1;
   while (i>=33 && i<=126) {
     if (p==pend) goto cexp; /* color name too long */
     *p++=i;
     i=getcc();
   }
   *p='\0';
   if (i==' ' || i=='\t') { /* Maybe another color will come */
     while ((i=getcc())==' ' || i=='\t') ;
     if (i=='g') goto at_g;
     else if (i=='c' || i=='m' || i=='b' || i=='s') goto at_col;
   }
   if (i!=T_COMMA) goto cexp;
   Image::Sampled::rgb_t ret=parse_rgb(tmp);
   if (ret==0x2000000) Error::sev(Error::EERROR) << "XPM: unknown color: " << tmp << (Error*)0;
   return ret;
 } else { cexp: Error::sev(Error::EERROR) << "XPM: color expected" << (Error*)0; }
 return 0; /*notreached*/
}

static Image::Sampled *in_xpm_reader(Image::Loader::UFD *ufd, SimBuffer::Flat const&) {
 // Error::sev(Error::EERROR) << "Cannot load XPM images yet." << (Error*)0;
 XPMTok tok(((Filter::UngetFILED*)ufd)->getFILE(/*seekable:*/false));
 Image::Sampled::dimen_t wd=tok.getDimen();
 Image::Sampled::dimen_t ht=tok.getDimen();
 Image::Sampled::dimen_t colors=tok.getDimen();
 Image::Sampled::dimen_t cpp=tok.getDimen(); /* chars per pixel */

 /* width height ncolors cpp [x_hot y_hot] */
 int i; /* multiple purpose */
 while ((i=tok.getcc())==' ' || i=='\t' || USGE(9,i-'0')) ;
 tok.ungetcc(i); tok.getComma();

 // Error::sev(Error::DEBUG) << "wd="<<wd<<" ht="<<ht<<" colors="<<colors<<" cpp="<<cpp << (Error*)0;
 if (1UL*cpp*colors>65535) Error::sev(Error::EERROR) << "XPM: too many colors" << (Error*)0;
 // if (cpp==1) {
 // }
 Image::Sampled::dimen_t transp=colors; /* No transparent colors yet. */
 /* vvv Dat: last cpp bytes of tab are used for storing current pixel. */
 char *tab=new char[cpp*(colors+1)], *p, *pend; /* BUGFIX: unsinged at Fri Nov 26 12:18:21 CET 2004 */
 Image::Sampled::rgb_t *rgb=new Image::Sampled::rgb_t[colors], *rp;
 for (rp=rgb,p=tab,pend=tab+cpp*colors; p!=pend; p+=cpp,rp++) {
   tok.read(p,cpp);
   *rp=tok.getColor();
   if (*rp==0x1000000) {
     if (transp!=colors) Error::sev(Error::WARNING) << "XPM: subsequent transparency might be blacked" << (Error*)0;
                    else transp=rp-rgb;
   }
 }
 /* Now: transp: index of the last transparent color */
 /* Dat: since (0x1000000&0xffffff)==0, transparent colors will have palette enrty black */

 char *outbuf=0; /* avoid gcc warning */
 Image::Sampled *ret;
 Image::Indexed *iimg=(Image::Indexed*)NULLP;
 tok.ungetcc(tok.T_COMMA);

 if (colors<=256) {
   ret=iimg=new Image::Indexed(wd,ht,colors,8);
   if (transp!=colors) iimg->setTransp(transp);
 } else {
   ret=new Image::RGB(wd,ht,8);
   if (transp!=colors) Error::sev(Error::WARNING) << "XPM: too many colors, transparency blacked" << (Error*)0;
 }
 outbuf=ret->getRowbeg();

 if (cpp==1) { /* Easy job: make an Indexed image; defer .packPal() */
   assert(colors<=256);
   signed short bin[256], s;
   memset(bin, 255, sizeof(bin)); /* Make bin[*]=-1 */
   for (i=0;(unsigned)i<colors;i++) {
     iimg->setPal(i, rgb[i]);
     bin[(unsigned char)tab[i]]=i;
   }
   assert(p==pend);
   while (ht--!=0) {
     tok.getComma();
     for (p=outbuf+ret->getRlen(); outbuf!=p; ) {
       if ((i=tok.getcc())<0 || i>255) Error::sev(Error::EERROR) << "XPM: data expected" << (Error*)0;
       if ((s=bin[i])<0) Error::sev(Error::EERROR) << "XPM: unpaletted color" << (Error*)0;
       *outbuf++=s;
     }
   }
 #if USE_BIG_MEMORY
  } else if (cpp==2 && colors<=256) { /* Similarly easy job: make an Indexed image; defer .packPal() */
   signed short *bin=new short[65536], s;
   memset(bin, 255, sizeof(*bin) * 65536); /* Make bin[*]=-1 */
   for (i=0,p=tab; (unsigned)i<colors; i++, p+=2) {
     iimg->setPal(i, rgb[i]);
     bin[(((unsigned char*)p)[0]<<8)+((unsigned char*)p)[1]]=i;
   }
   assert(p==pend);
   while (ht--!=0) {
     tok.getComma();
     for (p=outbuf+ret->getRlen(); outbuf!=p; ) {
       tok.readInStr(pend,2);
       if ((s=bin[(((unsigned char*)pend)[0]<<8)+((unsigned char*)pend)[1]])<0) Error::sev(Error::EERROR) << "XPM: unpaletted color" << (Error*)0;
       *outbuf++=s;
     }
   }
   delete [] bin;
 } else if (cpp==2 && colors<=65535) {
   Image::Sampled::rgb_t rgb1;
   unsigned short *bin=new unsigned short[65536], s;
   memset(bin, 255, sizeof(*bin) * 65536); /* Make bin[*]=max */
   for (i=0,p=tab; (unsigned)i<colors; i++, p+=2) bin[(((unsigned char*)p)[0]<<8)+((unsigned char*)p)[1]]=i;
   while (ht--!=0) {
     tok.getComma();
     for (p=outbuf+ret->getRlen(); outbuf!=p; ) {
       tok.readInStr(pend,2);
       if ((s=bin[(((unsigned char*)pend)[0]<<8)+((unsigned char*)pend)[1]])==(unsigned short)-1) Error::sev(Error::EERROR) << "XPM: unpaletted color" << (Error*)0;
       *outbuf++=(rgb1=rgb[s])>>16;
       *outbuf++=rgb1>>8;
       *outbuf++=rgb1;
     }
   }
   delete [] bin;
  #endif /* USE_BIG_MEMORY */
 } else { /* Now comes the slow, but general solution */
  #if USE_IN_XPM_MAPPING /* use Mapping */
   if (colors<=256) {
     Mapping::H h(1);
     char c;
     /* vvv `c' might become negative, but it is harmless */
     for (p=tab,pend=tab+cpp*colors,c=0; p!=pend; p+=cpp,c++)
       h.set(p, cpp, &c); /* every color-string should be unique; but no error message if it isn't */
     while (ht--!=0) {
       tok.getComma();
       for (p=outbuf+ret->getRlen(); outbuf!=p; ) {
         tok.readInStr(pend,cpp);
         if (NULLP==(pend=h.get(pend, cpp))) Error::sev(Error::EERROR) << "XPM: unpaletted color" << (Error*)0;
         *outbuf++=*pend;
       }
     }
   } else { /* most general case with Mapping */
     if (transp!=colors) Error::sev(Error::WARNING) << "XPM: too many colors, transparency blacked" << (Error*)0;
     /* Dat: reading a JPEG downsampled to a 256-color XPM takes 3000 ms
      * with Mapping, and 31000 ms without mapping. Nice speed increase.
      */
     Mapping::H h(3);
     char tmpcol[3];
     for (p=tab,pend=tab+cpp*colors,rp=rgb; p!=pend; p+=cpp,rp++) {
       tmpcol[0]=rp[0]>>16;
       tmpcol[1]=rp[0]>>8;
       tmpcol[2]=rp[0];
       h.set(p, cpp, tmpcol); /* every color-string should be unique; but no error message if it isn't */
     }
     while (ht--!=0) {
       tok.getComma();
       for (p=outbuf+ret->getRlen(); outbuf!=p; outbuf+=3) {
         tok.readInStr(pend,cpp);
         if (NULLP==(pend=h.get(pend, cpp))) Error::sev(Error::EERROR) << "XPM: unpaletted color" << (Error*)0;
         memcpy(outbuf, pend, 3);
       }
     }
   }
  #else /* don't USE_IN_XPM_MAPPING */
   Image::Sampled::dimen_t lastcol=0, x;
   p=tab; /* cache pointer for the last color (lastcol) */
   if (colors<=256) {
     assert(cpp>1);
     for (lastcol=0;lastcol<colors;lastcol++) iimg->setPal(lastcol, rgb[lastcol]);
     lastcol=0;
     while (ht--!=0) {
       // putchar('.');
       tok.getComma();
       for (x=0;x<wd;x++) {
         tok.read(pend,cpp);
         if (0!=memcmp(p,pend,cpp)) {
           p=tab; lastcol=0; while (p!=pend && 0!=memcmp(p,pend,cpp)) { p+=cpp; lastcol++; }
           if (p==pend) Error::sev(Error::EERROR) << "XPM: unpaletted color" << (Error*)0;
           if (rgb[lastcol]==0x1000000) p=tab+cpp*(lastcol=transp); /* fix single transp */
         }
         *outbuf++=lastcol;
       }
     }
   } else { /* colors>256 */
     while (ht--!=0) {
       tok.getComma();
       for (x=0;x<wd;x++) {
         tok.read(pend,cpp);
         if (0!=memcmp(p,pend,cpp)) {
           p=tab; lastcol=0; while (p!=pend && 0!=memcmp(p,pend,cpp)) { p+=cpp; lastcol++; }
           if (p==pend) Error::sev(Error::EERROR) << "XPM: unpaletted color" << (Error*)0;
         }
         *outbuf++=rgb[lastcol]>>16;
         *outbuf++=rgb[lastcol]>>8;
         *outbuf++=rgb[lastcol];
       }
     }
   }
  #endif
 }
 delete [] tab;
 delete [] rgb;
 /* Dat: we don't check for EOF. Imp: emit a warning? */
 // Error::sev(Error::DEBUG) << "rp[-1]=" << rp[-1] << (Error*)0;
 // while (-1!=(i=tok.getcc())) { putchar(i); }
 /* fclose((FILE*)file_); */
 return ret;
}

static Image::Loader::reader_t in_xpm_checker(char buf[Image::Loader::MAGIC_LEN], char [Image::Loader::MAGIC_LEN], SimBuffer::Flat const&, Image::Loader::UFD*) {
 return (0==memcmp(buf, "/* XPM */", 9)) ? in_xpm_reader : 0;
}

#else
#  define in_xpm_checker NULLP
#endif /* USE_IN_XPM */

Image::Loader in_xpm_loader = { "XPM", in_xpm_checker, 0 };