#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <strings.h>

#include "cards.h"

extern const char *__progname;

typedef struct ops OPS;

struct ops {
  void (*save_cmap)(int, unsigned short int (*)[3]);
  void (*save_bw)(void);
  void (*save_col)(int);
  } ;
#define SPLITOPS(name) {\
	&save_cmap_##name,				\
	&save_bw_##name,				\
	&save_col_##name,				\
	}

static char *deckdir;
static char *path;
static char *tmp;
/* last char of cvec is used for nonexistent colors */
static char cvec_[]
 = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ?";
#define MAXCOLORS ((sizeof(cvec_)/sizeof(cvec_[0]))-2)
static char *cvec;
static const char card_vals[] = "A23456789TJQK";
static const char card_suit[] = "CDHSRB";
static const char *card_xdeck[] = { "j" };

static void alloctemp(void)
{
 int len;

 len = strlen(deckdir);
 path = malloc(len+64);
 tmp = path + len;
 strcpy(path,deckdir);
 *tmp++ = '/';
}

static FILE *xfopen(const char *path, const char *mode)
{
 FILE *rv;

 rv = fopen(path,mode);
 if (rv) return(rv);
 fprintf(stderr,"%s: can't open %s for %s\n",__progname,path,('w'==*mode)?"write":"read");
 exit(1);
}

static void nlscan(FILE *f)
{
 int c;

 while (1)
  { c = getc(f);
    if (c == '\n') break;
    if (c == EOF)
     { fprintf(stderr,"%s: premature EOF on %s\n",__progname,path);
       exit(1);
     }
  }
}

static void make_bw(void)
{
 FILE *f;
 int x;
 int y;
 int c;
 int b;

 f = xfopen(path,"r");
 for (y=0;y<CARD_YSIZE;y++)
  { b = 0;
    for (x=0;x<CARD_XSIZE;x++)
     { c = getc(f);
       switch (c)
	{ case ' ': case '.': c = 1; break;
	  default: c = 0; break;
	}
       b = (b << 1) | c;
       if ((x & 7) == 7) putchar(b);
     }
#if CARD_XSIZE & 7
    putchar(b<<(8-(CARD_XSIZE&7)));
#endif
    nlscan(f);
  }
 fclose(f);
}

static void make_col(char *cv)
{
 FILE *f;
 int x;
 int y;
 int c;
 char *i;

 f = xfopen(path,"r");
 for (y=0;y<CARD_YSIZE;y++)
  { for (x=0;x<CARD_XSIZE;x++)
     { c = getc(f);
       i = index(cv,c);
       putchar(i?(i-cv):0);
     }
    nlscan(f);
  }
 fclose(f);
}

static void domake(void)
{
 int nc;
 int nb;
 int i;
 FILE *f;
 int c;
 int r;
 int g;
 int b;
 unsigned short int (*colors)[3];

 nc = 0;
 i = 8;
 cvec = malloc(i+1);
 colors = malloc(i*sizeof(*colors));
 strcpy(tmp,"c");
 f = xfopen(path,"r");
 while (1)
  { c = getc(f);
    if (c == EOF) break;
    if (nc >= i)
     { i += 8;
       cvec = realloc(cvec,i+1);
       colors = realloc(colors,i*sizeof(*colors));
     }
    cvec[nc] = c;
    if (fscanf(f,"%x%x%x",&r,&g,&b) != 3) break;
    colors[nc][0] = r;
    colors[nc][1] = g;
    colors[nc][2] = b;
    nc ++;
    nlscan(f);
  }
 fclose(f);
 cvec[nc] = '\0';
 if (nc < 1)
  { fprintf(stderr,"%s: no colors\n",__progname);
    exit(1);
  }
 nb = 0;
 while (1)
  { sprintf(tmp,"bb%d",nb+1);
    if (access(path,R_OK) < 0) break;
    nb ++;
  }
 if (nb < 1)
  { fprintf(stderr,"%s: no backs\n",__progname);
    exit(1);
  }
 putchar(nc);
 putchar(nb);
 for (i=0;i<nc;i++)
  { for (c=0;c<3;c++)
     { putchar(colors[i][c]>>8);
       putchar(colors[i][c]&0xff);
     }
  }
 for (i=0;i<nb;i++)
  { sprintf(tmp,"bb%d",i+1);
    make_bw();
  }
 for (i=0;i<CARD_DECK;i++)
  { sprintf(tmp,"b%c%c",card_vals[CARD_VALUE(i)],card_suit[CARD_SUIT(i)]);
    make_bw();
  }
 for (;i<CARD_XDECK;i++)
  { sprintf(tmp,"b%s",card_xdeck[i-CARD_DECK]);
    make_bw();
  }
 for (i=0;i<nb;i++)
  { sprintf(tmp,"cb%d",i+1);
    make_col(cvec);
  }
 for (i=0;i<CARD_DECK;i++)
  { sprintf(tmp,"c%c%c",card_vals[CARD_VALUE(i)],card_suit[CARD_SUIT(i)]);
    make_col(cvec);
  }
 for (;i<CARD_XDECK;i++)
  { sprintf(tmp,"c%s",card_xdeck[i-CARD_DECK]);
    make_col(cvec);
  }
}

static int get(void)
{
 int c;

 c = getchar();
 if (c == EOF)
  { fprintf(stderr,"%s: premature EOF\n",__progname);
    exit(1);
  }
 return(c);
}

static void dosplit(const OPS *ops)
{
 int ncol;
 int nbacks;
 unsigned short int (*colours)[3];
 int i;
 int j;

 cvec = &cvec_[0];
 ncol = get();
 nbacks = get();
 printf("%d color%s, %d back design%s\n",ncol,(ncol==1)?"":"s",nbacks,(nbacks==1)?"":"s");
 if ((ncol < 1) || (nbacks < 1))
  { fprintf(stderr,"%s: invalid deck file\n",__progname);
    exit(1);
  }
 if (ncol > MAXCOLORS)
  { fprintf(stderr,"%s: can't handle decks with more than %d colors\n",__progname,(int)MAXCOLORS);
    exit(1);
  }
 colours = malloc(ncol*sizeof(*colours));
 for (i=0;i<ncol;i++)
  { for (j=0;j<3;j++)
     { int c1;
       int c2;
       c1 = get();
       c2 = get();
       colours[i][j] = (c1 << 8) | c2;
     }
  }
 (*ops->save_cmap)(ncol,colours);
 free(colours);
 for (i=0;i<nbacks;i++)
  { sprintf(tmp,"bb%d",i+1);
    (*ops->save_bw)();
  }
 sprintf(tmp,"bb%d",i+1);
 unlink(path);
 for (i=0;i<CARD_DECK;i++)
  { sprintf(tmp,"b%c%c",card_vals[CARD_VALUE(i)],card_suit[CARD_SUIT(i)]);
    (*ops->save_bw)();
  }
 for (;i<CARD_XDECK;i++)
  { sprintf(tmp,"b%s",card_xdeck[i-CARD_DECK]);
    (*ops->save_bw)();
  }
 for (i=0;i<nbacks;i++)
  { sprintf(tmp,"cb%d",i+1);
    (*ops->save_col)(ncol);
  }
 sprintf(tmp,"cb%d",i+1);
 unlink(path);
 for (i=0;i<CARD_DECK;i++)
  { sprintf(tmp,"c%c%c",card_vals[CARD_VALUE(i)],card_suit[CARD_SUIT(i)]);
    (*ops->save_col)(ncol);
  }
 for (;i<CARD_XDECK;i++)
  { sprintf(tmp,"c%s",card_xdeck[i-CARD_DECK]);
    (*ops->save_col)(ncol);
  }
}

static void save_cmap_text(int ncol, unsigned short int (*cols)[3])
{
 int i;
 FILE *f;

 strcpy(tmp,"c");
 f = xfopen(path,"w");
 for (i=0;i<ncol;i++)
  { fprintf(f,"%c %04x %04x %04x\n",cvec[i],cols[i][0],cols[i][1],cols[i][2]);
  }
 fclose(f);
}

static void save_bw_text(void)
{
 FILE *f;
 int x;
 int y;
 int c;
 int b;

 f = xfopen(path,"w");
 for (y=0;y<CARD_YSIZE;y++)
  { for (x=0;x<CARD_XSIZE;x++)
     { if ((x & 7) == 0)
	{ c = get();
	  b = 0x80;
	}
       putc((c&b)?'.':'*',f);
       b >>= 1;
     }
    putc('\n',f);
  }
 fclose(f);
}

static void save_col_text(int ncol)
{
 FILE *f;
 int x;
 int y;
 int c;

 f = xfopen(path,"w");
 for (y=0;y<CARD_YSIZE;y++)
  { for (x=0;x<CARD_XSIZE;x++)
     { c = get();
       putc((c<ncol)?cvec[c]:cvec[MAXCOLORS],f);
     }
    putc('\n',f);
  }
 fclose(f);
}

static const OPS splitops_text = SPLITOPS(text);

static unsigned char ppmcm_cols[256][3];

static void save_cmap_ppm(int ncol, unsigned short int (*cols)[3])
{
 int i;

 for (i=255;i>=ncol;i--)
  { ppmcm_cols[i][0] = 0;
    ppmcm_cols[i][1] = 0;
    ppmcm_cols[i][2] = 0;
  }
 for (i=ncol-1;i>=0;i--)
  { ppmcm_cols[i][0] = cols[i][0] >> 8;
    ppmcm_cols[i][1] = cols[i][1] >> 8;
    ppmcm_cols[i][2] = cols[i][2] >> 8;
  }
}

static void save_bw_ppm(void)
{
 FILE *f;
 int x;
 int y;
 int c;
 int b;

 f = xfopen(path,"w");
 fprintf(f,"P1\n%d %d\n",CARD_XSIZE,CARD_YSIZE);
 for (y=0;y<CARD_YSIZE;y++)
  { for (x=0;x<CARD_XSIZE;x++)
     { if ((x & 7) == 0)
	{ c = get();
	  b = 0x80;
	}
       putc((c&b)?'1':'0',f);
       b >>= 1;
     }
    putc('\n',f);
  }
 fclose(f);
}

static void save_col_ppm(int ncol)
{
 FILE *f;
 int x;
 int y;
 int c;

 f = xfopen(path,"w");
 fprintf(f,"P3\n%d %d\n255\n",CARD_XSIZE,CARD_YSIZE);
 for (y=0;y<CARD_YSIZE;y++)
  { for (x=0;x<CARD_XSIZE;x++)
     { c = get();
       if (c < ncol)
	{ fprintf(f,"%d %d %d\n",ppmcm_cols[c][0],ppmcm_cols[c][1],ppmcm_cols[c][2]);
	}
       else
	{ fprintf(f,"0 0 0\n");
	}
     }
  }
 fclose(f);
}

static const OPS splitops_ppm = SPLITOPS(ppm);

int main(int, char **);
int main(int ac, char **av)
{
 if (0)
  {
usage:;
    fprintf(stderr,"Usage: %s -[make|text|ppm] deck-directory\n",__progname);
    fprintf(stderr,"(-make uses the -text file format)\n");
    exit(1);
  }
 if (ac != 3) goto usage;
 deckdir = av[2];
 alloctemp();
      if (!strcmp(av[1],"-make" )) domake();
 else if (!strcmp(av[1],"-text")) dosplit(&splitops_text);
 else if (!strcmp(av[1],"-ppm")) dosplit(&splitops_ppm);
 else                              goto usage;
 exit(0);
}