#if 0
gcc ${CFLAGS:--s -O2} -c -Wno-multichar -Wno-unused-result picedit.c `sdl-config --cflags`
exit
#endif
/*
This program is part of Free Hero Mesh and is public domain.
*/
#include "SDL.h"
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "sqlite3.h"
#include "smallxrm.h"
#include "heromesh.h"
#include "quarks.h"
#include "cursorshapes.h"
typedef struct {
Uint8 size,meth;
Uint8 data[0]; // the first row is all 0, since the compression algorithm requires this
} Picture;
typedef struct {
Uint8 colors[64];
Uint8 ncolors;
} ColorFilter;
typedef struct {
char name[64];
} OverlayFilter;
typedef struct {
Uint8 shift[32];
Uint8 size[32];
Uint8 nshift;
} ShiftFilter;
typedef struct {
Uint8 code;
// 0-7 = flip/rotations
// 8-10 = change colours (8 is *->* *->* *->*; 9 is *->*->*->*->*; 10 is *<->* *<->* *<->*)
// 11 = overlay
// 12-15 = shift (up, down, right, left)
union {
ColorFilter color;
OverlayFilter overlay;
ShiftFilter shift;
};
} Filter;
typedef struct {
char basename[64];
Filter filters[64];
Uint8 nfilters;
} DependentPicture;
typedef struct {
sqlite3_vtab_cursor super;
int pos;
} Cursor;
static Uint8 cur_type;
static Uint8 gsizes[16];
static Picture*cur_pic;
static Picture*spare_page;
static sqlite3_stmt*macro[26];
static void fn_valid_name(sqlite3_context*cxt,int argc,sqlite3_value**argv) {
const char*s=sqlite3_value_text(*argv);
if(!s || !*s || s[strspn(s,"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz_-0123456789")]) {
sqlite3_result_error(cxt,"Invalid name",-1);
return;
}
sqlite3_result_value(cxt,*argv);
}
static int vt_graph_close(sqlite3_vtab_cursor*cur) {
sqlite3_free(cur);
return SQLITE_OK;
}
static int vt_graph_eof(sqlite3_vtab_cursor*cc) {
Cursor*cur=(void*)cc;
return (!cur_pic || cur->pos>=cur_pic->size*cur_pic->size);
}
static int vt_graph_open(sqlite3_vtab*vt,sqlite3_vtab_cursor**cur) {
*cur=sqlite3_malloc(sizeof(Cursor));
return *cur?SQLITE_OK:SQLITE_NOMEM;
}
static int vt_graph_connect(sqlite3*db,void*aux,int argc,const char*const*argv,sqlite3_vtab**vt,char**ex) {
*vt=sqlite3_malloc(sizeof(sqlite3_vtab));
if(!*vt) return SQLITE_NOMEM;
sqlite3_declare_vtab(db,"CREATE TABLE `GRAPH`(`X` INT, `Y` INT, `C` INT);");
return SQLITE_OK;
}
static int vt_graph_disconnect(sqlite3_vtab*vt) {
sqlite3_free(vt);
return SQLITE_OK;
}
static int vt_graph_filter(sqlite3_vtab_cursor*cc,int n,const char*s,int c,sqlite3_value**v) {
Cursor*cur=(void*)cc;
cur->pos=0;
return SQLITE_OK;
}
static int vt_graph_index(sqlite3_vtab*vt,sqlite3_index_info*inf) {
return SQLITE_OK;
}
static int vt_graph_column(sqlite3_vtab_cursor*cc,sqlite3_context*cxt,int n) {
Cursor*cur=(void*)cc;
sqlite3_result_int(cxt,n==0?cur->pos%cur_pic->size:n==1?cur->pos/cur_pic->size:cur_pic->data[cur->pos+cur_pic->size]);
return SQLITE_OK;
}
static int vt_graph_next(sqlite3_vtab_cursor*cc) {
Cursor*cur=(void*)cc;
cur->pos++;
return SQLITE_OK;
}
static int vt_graph_rowid(sqlite3_vtab_cursor*cc,sqlite3_int64*p) {
Cursor*cur=(void*)cc;
*p=cur->pos;
return SQLITE_OK;
}
static int vt_graph_update(sqlite3_vtab*vt,int argc,sqlite3_value**argv,sqlite3_int64*rowid) {
int x,y;
if(argc!=5) return SQLITE_CONSTRAINT_VTAB;
if(sqlite3_value_type(argv[0])==SQLITE_NULL) {
x=sqlite3_value_int(argv[2]);
y=sqlite3_value_int(argv[3]);
if(x<0 || x>=cur_pic->size || y<0 || y>=cur_pic->size) return SQLITE_CONSTRAINT_VTAB;
x+=y*cur_pic->size;
} else {
x=sqlite3_value_int(argv[0]);
}
cur_pic->data[cur_pic->size+x]=sqlite3_value_int(argv[4]);
return SQLITE_OK;
}
static const sqlite3_module vt_graph={
.iVersion=1,
.xBestIndex=vt_graph_index,
.xClose=vt_graph_close,
.xColumn=vt_graph_column,
.xConnect=vt_graph_connect,
.xDisconnect=vt_graph_disconnect,
.xEof=vt_graph_eof,
.xFilter=vt_graph_filter,
.xNext=vt_graph_next,
.xOpen=vt_graph_open,
.xRowid=vt_graph_rowid,
.xUpdate=vt_graph_update,
};
static void*load_macros_1(xrm_db*db,void*usr,int loose,xrm_quark q) {
const char*txt;
if(q<Q_A || q>Q_Z) return 0;
if(!(db=xrm_sub(db,loose,q)) || !(txt=xrm_get(db))) return 0;
if(sqlite3_prepare_v3(userdb,txt,-1,SQLITE_PREPARE_PERSISTENT,macro+(q-Q_A),0)) fatal("Error in macro: %s\n%s\n",txt,sqlite3_errmsg(userdb));
return 0;
}
static void*load_macros(xrm_db*db,void*usr) {
xrm_enumerate(db,load_macros_1,0);
return 0;
}
static int load_picture_file(void) {
sqlite3_stmt*st=0;
FILE*fp;
char*nam;
char*buf=0;
int r=0;
int i,j;
i=sqlite3_exec(userdb,
"BEGIN;"
"CREATE TEMPORARY TABLE `PICEDIT`(`ID` INTEGER PRIMARY KEY,`NAME` TEXT NOT NULL COLLATE NOCASE,`TYPE` INT,`DATA` BLOB);"
"CREATE INDEX `PICEDIT_I1` ON `PICEDIT`(`NAME`,`TYPE`);"
,0,0,0);
if(i) fatal("SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
nam=sqlite3_mprintf("%s.xclass",basefilename);
if(!nam) fatal("Allocation failed\n");
fprintf(stderr,"Loading pictures...\n");
fp=fopen(nam,"r");
sqlite3_free(nam);
if(!fp) {
nam=0;
fprintf(stderr,"%m\n");
goto done;
}
nam=malloc(256);
if(!nam) fatal("Allocation failed\n");
i=sqlite3_prepare_v2(userdb,"INSERT INTO `PICEDIT`(`NAME`,`TYPE`,`DATA`) VALUES(?1,?2,?3);",-1,&st,0);
if(i) fatal("SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
while(!feof(fp)) {
i=0;
while(j=fgetc(fp)) {
if(j==EOF) goto done;
if(i<255) nam[i++]=j;
}
nam[i]=0;
sqlite3_reset(st);
sqlite3_bind_text(st,1,nam,i,SQLITE_TRANSIENT);
sqlite3_bind_int(st,2,j=(i>4 && !memcmp(".IMG",nam+i-4,4)?1:!memcmp(".DEP",nam+i-4,4)?2:0));
if(j) r++;
i=fgetc(fp)<<16;
i|=fgetc(fp)<<24;
i|=fgetc(fp)<<0;
i|=fgetc(fp)<<8;
if(!i) continue;
if(i<0) fatal("Bad or too big lump in .xclass file\n");
buf=realloc(buf,i);
if(!buf) fatal("Allocation failed\n");
fread(buf,1,i,fp);
if(!sqlite3_stricmp(nam,"PICEDIT.CFG")) {
memcpy(gsizes,buf,i<15?i:15);
if(!*gsizes) *gsizes=picture_size;
}
sqlite3_bind_blob(st,3,buf,i,SQLITE_TRANSIENT);
while((i=sqlite3_step(st))==SQLITE_ROW);
if(i!=SQLITE_DONE) fatal("SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
}
done:
if(st) sqlite3_finalize(st);
if(fp) fclose(fp);
free(nam);
free(buf);
fprintf(stderr,"Done\n");
sqlite3_exec(userdb,
"CREATE TRIGGER `PICEDIT_T1` BEFORE INSERT ON `PICEDIT` BEGIN"
" SELECT RAISE(FAIL,'Duplicate name') FROM `PICEDIT` WHERE REPLACE(`NAME`||'*','.IMG*','.DEP*')=REPLACE(NEW.`NAME`||'*','.IMG*','.DEP*');"
"END;"
"CREATE TRIGGER `PICEDIT_T2` BEFORE UPDATE OF `NAME` ON `PICEDIT` BEGIN"
" SELECT RAISE(FAIL,'Duplicate name') FROM `PICEDIT` WHERE REPLACE(`NAME`||'*','.IMG*','.DEP*')=REPLACE(NEW.`NAME`||'*','.IMG*','.DEP*');"
"END;"
,0,0,0);
return r;
}
static void save_picture_file(void) {
sqlite3_stmt*st;
FILE*fp;
char*s=sqlite3_mprintf("%s.xclass",basefilename);
int i;
const char*nam;
const char*buf;
if(!s) fatal("Allocation failed\n");
fprintf(stderr,"Saving pictures...\n");
fp=fopen(s,"w");
if(!fp) fatal("Cannot open picture file for writing: %m\n");
sqlite3_free(s);
i=sqlite3_prepare_v2(userdb,"SELECT `NAME`,`DATA` FROM `PICEDIT`;",-1,&st,0);
if(i) fatal("SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
while((i=sqlite3_step(st))==SQLITE_ROW) {
nam=sqlite3_column_text(st,0);
buf=sqlite3_column_blob(st,1);
i=sqlite3_column_bytes(st,1);
if(!nam) continue;
fwrite(nam,1,strlen(nam)+1,fp);
fputc(i>>16,fp);
fputc(i>>24,fp);
fputc(i>>0,fp);
fputc(i>>8,fp);
if(i) fwrite(buf,1,i,fp);
}
if(i!=SQLITE_DONE) fprintf(stderr,"SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
done:
if(st) sqlite3_finalize(st);
if(fp) fclose(fp);
fprintf(stderr,"Done\n");
}
static sqlite3_int64 ask_picture_id(const char*t) {
sqlite3_stmt*st;
const char*r=screen_prompt(t);
int i;
sqlite3_int64 id=0;
if(!r || !*r) return 0;
i=sqlite3_prepare_v2(userdb,"SELECT `ID`, `TYPE` FROM `PICEDIT` WHERE `NAME` = (?1 || '.IMG') OR `NAME` = (?1 || '.DEP');",-1,&st,0);
if(i) {
screen_message(sqlite3_errmsg(userdb));
return 0;
}
sqlite3_bind_text(st,1,r,-1,0);
i=sqlite3_step(st);
if(i==SQLITE_ROW) id=sqlite3_column_int64(st,0),cur_type=sqlite3_column_int(st,1);
if(i==SQLITE_DONE) screen_message("Picture not found"),cur_type=0;
sqlite3_finalize(st);
return id;
}
static void uncompress_picture(FILE*fp,Picture*pic) {
Uint8*p=pic->data+pic->size;
int c,n,t,x,y;
n=t=0;
y=pic->size*pic->size;
while(y--) {
if(!n) {
n=fgetc(fp);
if(n<85) {
// Homogeneous run
n++;
x=fgetc(fp);
if(t==1 && x==c) n*=85; else n++;
c=x;
t=1;
} else if(n<170) {
// Heterogeneous run
n-=84;
t=2;
} else {
// Copy-above run
n-=169;
if(t==3) n*=85;
t=3;
}
}
n--;
if(t==2) c=fgetc(fp);
if(t==3) c=p[-pic->size];
*p++=c;
}
}
static void load_rotate(Picture*pic) {
int x,y;
int m=pic->meth;
int s=pic->size;
Uint8*d=pic->data+s;
static Uint8*buf;
Uint8*p;
if(!buf) {
buf=malloc(255*255);
if(!buf) fatal("Allocation failed\n");
}
p=buf;
for(y=0;y<s;y++) for(x=0;x<s;x++) {
if(m&1) x=s-x-1;
if(m&2) y=s-y-1;
*p++=d[m&4?x*s+y:y*s+x];
if(m&1) x=s-x-1;
if(m&2) y=s-y-1;
}
memcpy(d,buf,s*s);
}
static void out_run(FILE*fp,int ch,int am,const Uint8*d,int le) {
int n=am%85?:85;
fputc(ch+n-1,fp);
if(le) fwrite(d,1,le<n?le:n,fp);
if(ch) d+=n,le-=n;
while(am-=n) {
n=am>7225?7225:am;
fputc(ch+n/85-1,fp);
if(le>0) fwrite(d,1,le<n?le:n,fp);
if(ch) d+=n,le-=n;
}
}
static void compress_picture_1(FILE*fp,Picture*pic) {
int ps=pic->size;
int ms=ps*ps;
const Uint8*d=pic->data+ps;
int i=0;
int ca,homo,hetero;
while(i<ms) {
ca=0;
while(i+ca<ms && d[i+ca]==d[i+ca-ps]) ca++;
homo=1;
while(i+homo<ms && d[i+homo]==d[i]) homo++;
hetero=1;
while(i+hetero<ms) {
if(d[i+hetero]==d[i+hetero-ps] && d[i+hetero+1]==d[i+hetero+1-ps]) break;
if(d[i+hetero]==d[i+hetero+1] && i+hetero==ms-1) break;
if(d[i+hetero]==d[i+hetero+1] && d[i+hetero]==d[i+hetero+2]) break;
hetero++;
}
if(ca>=homo && (ca>=hetero || homo>1)) {
out_run(fp,170,ca,0,0);
i+=ca;
} else if(homo>1) {
out_run(fp,0,homo-1,d+i,1);
i+=homo;
} else {
if(hetero>85) hetero=85;
out_run(fp,85,hetero,d+i,hetero);
i+=hetero;
}
}
}
static void compress_picture(FILE*out,Picture*pic) {
int bm=15;
int bs=pic->size*pic->size;
FILE*fp=fmemopen(0,bs,"w");
int i,j;
if(!fp) fatal("Error with fmemopen\n");
for(i=0;i<8;i++) {
compress_picture_1(fp,pic);
if(!ferror(fp)) {
j=ftell(fp);
if(j>0 && j<bs) bs=j,bm=i;
}
rewind(fp);
pic->meth="\1\3\1\7\1\3\1\7"[i];
load_rotate(pic);
}
pic->meth=bm;
if(bm && bm!=15) load_rotate(pic);
fclose(fp);
if(bm==15) fwrite(pic->data+pic->size,pic->size,pic->size,out); else compress_picture_1(out,pic);
}
static void block_rotate(Uint8*d,Uint8 s,Uint8 w,Uint8 h,Uint8 m) {
Uint8*b=malloc(w*h);
Uint8*p=b;
int x,y;
if(!b) fatal("Allocation failed\n");
if(w!=h && (m&4)) goto end;
for(y=0;y<h;y++) for(x=0;x<w;x++) {
if(m&1) x=w-x-1;
if(m&2) y=h-y-1;
*p++=d[m&4?x*s+y:y*s+x];
if(m&1) x=w-x-1;
if(m&2) y=h-y-1;
}
p=b;
for(y=0;y<h;y++) {
memcpy(d,p,w);
p+=w;
d+=s;
}
end: free(b);
}
static void load_picture_lump(const unsigned char*data,int len,Picture**pict) {
Uint8 buf[32];
FILE*fp;
int i,j,n;
if(!len) return;
fp=fmemopen((unsigned char*)data,len,"r");
if(!fp) fatal("Failed to open in-memory stream of size %d\n",len);
*buf=fgetc(fp);
n=*buf&15;
fread(buf+1,1,n+(n>>1),fp);
for(i=0;i<n;i++) {
j=buf[i+1];
pict[i]=malloc(sizeof(Picture)+(j+1)*j);
if(!pict[i]) fatal("Allocation failed\n");
memset(pict[i]->data,0,pict[i]->size=j);
j=(i?buf[n+1+((i-1)>>1)]>>(i&1?0:4):*buf>>4)&15;
pict[i]->meth=j^((j==5 || j==6)?3:0);
}
for(i=0;i<n;i++) {
j=pict[i]->size;
if(pict[i]->meth==15) fread(pict[i]->data+j,j,j,fp),pict[i]->meth=0; else uncompress_picture(fp,pict[i]);
if(pict[i]->meth) load_rotate(pict[i]);
}
fclose(fp);
}
static inline void show_cursor_xy(int x,int y,int xx,int yy,const Picture*p) {
char buf[64];
if(x>=0 && xx>=0) snprintf(buf,64,"[%d,%d]:(%d,%d)=%d %+d%+d%63s",xx,yy,x,y,p->data[(y+1)*p->size+x],x-xx,y-yy,"");
else if(x>=0) snprintf(buf,64,"(%d,%d)=%d%63s",x,y,p->data[(y+1)*p->size+x],"");
else if(xx>=0) snprintf(buf,64,"[%d,%d]%63s",xx,yy,"");
else snprintf(buf,64,"%63s","");
SDL_LockSurface(screen);
draw_text(0,32,buf,0xF0,0xF9);
SDL_UnlockSurface(screen);
SDL_Flip(screen);
}
static void draw_line(Uint8*p,Uint8 s,Uint8 x0,Uint8 y0,Uint8 x1,Uint8 y1,Uint8 c) {
int dx=abs(x1-x0);
int sx=x0<x1?1:-1;
int dy=-abs(y1-y0);
int sy=y0<y1?1:-1;
int e=dx+dy;
int e2;
for(;;) {
p[y0*s+x0]=c;
if(x0==x1 && y0==y1) break;
e2=e+e;
if(e2>=dy) e+=dy,x0+=sx;
if(e2<=dx) e+=dx,y0+=sy;
}
}
static void draw_circle(Uint8*p,Uint8 s,Uint8 x0,Uint8 y0,Uint32 r,Uint8 c) {
int i,j,x,y;
for(i=0;i<s && r>=0;i++) {
j=sqrt(r);
r-=i+i+1;
x=x0+i,y=y0+j; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
x=x0-i,y=y0+j; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
x=x0+i,y=y0-j; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
x=x0-i,y=y0-j; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
x=x0+j,y=y0+i; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
x=x0-j,y=y0+i; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
x=x0+j,y=y0-i; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
x=x0-j,y=y0-i; if(x>=0 && x<s && y>=0 && y<s) p[y*s+x]=c;
}
}
static void fill_circle(Uint8*p,Uint8 s,Uint8 x0,Uint8 y0,Uint32 r,Uint8 c) {
int x,y;
for(y=0;y<s;y++) for(x=0;x<s;x++) {
if((x-x0)*(x-x0)+(y-y0)*(y-y0)<=r) p[y*s+x]=c;
}
}
static void draw_ellipse(Uint8*p,Uint8 s,Uint8 x0,Uint8 y0,Uint8 x1,Uint8 y1,Uint8 c) {
double cx,cy,rx,ry,v;
int x,y;
if(x1<x0) x=x1,x1=x0,x0=x;
if(y1<y0) y=y1,y1=y0,y0=y;
cx=0.5*(x1+x0);
cy=0.5*(y1+y0);
rx=0.5*(x1-x0);
ry=0.5*(y1-y0);
for(y=y0;y<=y1;y++) {
v=(y-cy)/ry; v*=v;
if(v>1.0) continue;
v=rx*sqrt(1.0-v);
x=round(cx-v); if(x>=0 && x<s) p[y*s+x]=c;
x=round(cx+v); if(x>=0 && x<s) p[y*s+x]=c;
}
for(x=x0;x<=x1;x++) {
v=(x-cx)/rx; v*=v;
if(v>1.0) continue;
v=ry*sqrt(1.0-v);
y=round(cy-v); if(y>=0 && y<s) p[y*s+x]=c;
y=round(cy+v); if(y>=0 && y<s) p[y*s+x]=c;
}
}
static void fill_ellipse(Uint8*p,Uint8 s,Uint8 x0,Uint8 y0,Uint8 x1,Uint8 y1,Uint8 c) {
double cx,cy,rx,ry;
int x,y;
if(x1<x0) x=x1,x1=x0,x0=x;
if(y1<y0) y=y1,y1=y0,y0=y;
cx=0.5*(x1+x0);
cy=0.5*(y1+y0);
rx=0.5*(x1-x0);
ry=0.5*(y1-y0);
for(y=y0;y<=y1;y++) for(x=x0;x<=x1;x++) {
if(pow((x-cx)/rx,2.0)+pow((y-cy)/ry,2.0)<=1.0) p[y*s+x]=c;
}
}
static void flood(Uint8 x,Uint8 y,Uint8 mx,Uint8 my,Uint8 mw,Uint8 mh,Uint8 c,Uint8 c0,Uint8 b) {
sqlite3_stmt*st;
if(c0==c) return;
if(sqlite3_prepare_v2(userdb,
// This isn't as efficient as it should be, since it has to scan the entire table every time.
"INSERT INTO GRAPH(X,Y,C) WITH R(XX,YY) AS ("
"SELECT ?1,?2 UNION SELECT X,Y FROM R,GRAPH WHERE C=?4"
" AND ABS(X-XX)+ABS(Y-YY)<>?5 AND ABS(X-XX)<2 AND ABS(Y-YY)<2"
" AND X>=?6 AND X<?8 AND Y>=?7 AND Y<?9"
") SELECT XX,YY,?3 FROM R;"
,-1,&st,0)) {
screen_message(sqlite3_errmsg(userdb));
return;
}
sqlite3_bind_int(st,1,x);
sqlite3_bind_int(st,2,y);
sqlite3_bind_int(st,3,c);
sqlite3_bind_int(st,4,c0);
sqlite3_bind_int(st,5,3-b);
sqlite3_bind_int(st,6,mx);
sqlite3_bind_int(st,7,my);
sqlite3_bind_int(st,8,mx+mw);
sqlite3_bind_int(st,9,my+mh);
sqlite3_step(st);
sqlite3_finalize(st);
}
static Picture*do_import(Picture*pic) {
SDL_Color*pal=screen->format->palette->colors;
int a,b,i,j,x,y;
Uint8 buf[16];
Uint8*q;
const char*cmd=screen_prompt("Import?");
FILE*fp;
if(!cmd || !*cmd) return pic;
fp=popen(cmd,"r");
if(!fp) {
screen_message("Import failed");
return pic;
}
if(fread(buf,1,16,fp)!=16 || memcmp("farbfeld",buf,8)) {
screen_message("Invalid format");
goto end;
}
if(buf[8] || buf[9] || buf[10] || buf[12] || buf[13] || buf[14] || !buf[15] || buf[11]!=buf[15]) {
screen_message("Invalid size");
goto end;
}
if(pic->size!=buf[15]) {
pic=realloc(pic,sizeof(Picture)+(buf[15]+1)*buf[15]);
if(!pic) fatal("Allocation failed\n");
pic->size=buf[15];
memset(pic->data+pic->size,0,pic->size);
}
q=pic->data+pic->size;
for(y=0;y<pic->size;y++) for(x=0;x<pic->size;x++) {
fread(buf,1,8,fp);
if(buf[6]&0x80) {
for(i=1;i<255;i++) if(buf[0]==pal[i].r && buf[2]==pal[i].g && buf[4]==pal[i].b) goto found;
i=1;
a=0x300000;
for(j=1;j<255;j++) {
b=abs(buf[0]-pal[j].r)+abs(buf[2]-pal[j].g)+abs(buf[4]-pal[j].b);
if(b<=a) a=b,i=j;
}
found: *q++=i;
} else {
*q++=0;
}
}
end:
pclose(fp);
return pic;
}
static void do_export(Picture*pic) {
SDL_Color*pal=screen->format->palette->colors;
Uint8*q=pic->data+pic->size;
int c,x,y;
const char*cmd=screen_prompt("Export?");
FILE*fp;
if(!cmd || !*cmd) return;
fp=popen(cmd,"w");
fwrite("farbfeld\0\0\0",1,11,fp); fputc(pic->size,fp);
fwrite("\0\0\0",1,3,fp); fputc(pic->size,fp);
for(y=0;y<pic->size;y++) for(x=0;x<pic->size;x++) {
c=*q++;
fputc(pal[c].r,fp); fputc(pal[c].r,fp);
fputc(pal[c].g,fp); fputc(pal[c].g,fp);
fputc(pal[c].b,fp); fputc(pal[c].b,fp);
fputc(c?255:0,fp); fputc(c?255:0,fp);
}
pclose(fp);
}
static int response_cb(void*aux,int argc,char**argv,char**names) {
if(argc && *argv) screen_message(*argv);
return 1;
}
static inline void edit_picture_1(Picture**pict,const char*name) {
static const Uint8 shade[64]=
"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F"
"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F"
"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F"
"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F"
;
static const Uint8 gridlines[32]=
"\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF"
"\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF\xF8\xFF"
;
static const char*const tool[11]={
"Draw",
"Mark",
"Pick",
"Line",
"Rect",
"Fillrect",
"Circle",
"fIllcirc",
"Ellipse",
"fillellipSe",
"flOod",
};
static Picture*pclip=0;
Uint8*p;
Uint8*q;
Uint8 sel=0;
Uint8 cc=0;
Uint8 t=2;
SDL_Rect r,m;
SDL_Event ev;
int i,j,x,y,z;
int xx=-1;
int yy=-1;
unsigned char buf[256];
m.x=m.y=m.w=m.h=0;
set_cursor(XC_arrow);
redraw:
if((sel&~15) || !pict[sel]) sel=0;
cur_pic=pict[sel];
z=screen->w-pict[sel]->size-169;
if(z>screen->h-49) z=screen->h-49;
z/=pict[sel]->size;
if(z<3) return;
if(z>32) z=32;
SDL_LockSurface(screen);
p=screen->pixels;
r.x=r.y=0; r.w=screen->w; r.h=screen->h;
SDL_FillRect(screen,&r,0xF0);
draw_text(0,0,name,0xF0,0xF5);
x=strlen(name)+1;
for(i=0;i<16;i++) if(pict[i]) {
j=snprintf(buf,255,"%c%d%c",i==sel?'<':' ',pict[i]->size,i==sel?'>':' ');
draw_text(x<<3,0,buf,0xF0,i==sel?0xFF:0xF8);
x+=j;
}
draw_text(0,8,"<ESC> Exit <1-9> Sel <SP> Unmark <RET> Copy <INS> Paste <DEL> Erase <=> SparePage",0xF0,0xFB);
draw_text(0,16,"<F1> CW <F2> CCW <F3> \x12 <F4> \x1D <F5> Size <F6> Add <F7> Del <F8> Mark <F9> In <F10> Out",0xF0,0xFB);
x=0;
for(i=0;i<11;i++) {
j=snprintf(buf,255,"%c%s%c",i==t?'<':' ',tool[i],i==t?'>':' ');
draw_text(x<<3,24,buf,0xF0,i==t?0xFE:0xF8);
x+=j;
}
p=screen->pixels+40*screen->pitch;
q=pict[sel]->data+pict[sel]->size;
for(y=0;y<pict[sel]->size;y++) {
memcpy(p,q,pict[sel]->size);
p+=screen->pitch;
q+=pict[sel]->size;
}
p=screen->pixels+40*screen->pitch-screen->w-161;
for(x=0;x<256;x++) {
for(y=0;y<10;y++) memset(p+y*screen->pitch,x,10);
if(x==cc) {
memset(p+1,0xF0,8);
for(y=1;y<9;y++) {
p[y*screen->pitch]=0xF0;
p[y*screen->pitch+(y&1)+4]=(y&1)+0xF7;
p[y*screen->pitch+9]=0xFF;
}
memset(p+9*screen->pitch+1,0xFF,8);
}
if(15&~x) p+=10; else p+=10*screen->pitch-150;
}
p=screen->pixels+41*screen->pitch+pict[sel]->size+2;
q=pict[sel]->data+pict[sel]->size;
for(y=0;y<pict[sel]->size;y++) {
for(x=0;x<pict[sel]->size;x++) {
memcpy(p,gridlines,z);
for(i=1;i<z;i++) {
p[i*screen->pitch]=i&1?0xF8:0xFF;
if(*q) memset(p+i*screen->pitch+1,*q,z-1);
else memcpy(p+i*screen->pitch+1,shade+(i&15),z-1);
}
if(xx==x && yy==y) {
memset(p+(z/2)*screen->pitch,0xFA,z);
memset(p+(z/2+1)*screen->pitch,0xF2,z);
for(i=1;i<z;i++) memcpy(p+i*screen->pitch+z/2,"\xF2\xFA",2);
}
p+=z;
q++;
}
for(i=1;i<z;i++) p[i*screen->pitch]=i&1?0xF8:0xFF;
p+=z*(screen->pitch-pict[sel]->size);
}
for(x=0;x<pict[sel]->size;x++) {
memcpy(p,gridlines,z);
p+=z;
}
if(m.h) {
r.x=m.x*z+pict[sel]->size+4;
r.y=m.y*z+43;
r.w=m.w*z-4;
r.h=m.h*z-4;
p=screen->pixels+r.y*screen->pitch+r.x;
for(i=0;i<r.w;i++) p[i]=i&1?0xFE:0xF6;
for(i=0;i<r.h;i++) {
*p=p[r.w]=i&1?0xF6:0xFE;
p+=screen->pitch;
}
for(i=0;i<r.w;i++) p[i]=i&1?0xFE:0xF6;
}
SDL_UnlockSurface(screen);
SDL_Flip(screen);
while(SDL_WaitEvent(&ev)) {
switch(ev.type) {
case SDL_QUIT:
exit(0);
return;
case SDL_MOUSEMOTION:
if(ev.motion.x<pict[sel]->size+2 || ev.motion.x>=pict[sel]->size*(z+1)+2 || ev.motion.y<41 || ev.motion.y>=z*pict[sel]->size+41) {
x=-1;
set_cursor(XC_arrow);
} else {
x=(ev.motion.x-pict[sel]->size-2)/z;
y=(ev.motion.y-41)/z;
set_cursor(XC_tcross);
if(ev.motion.state && !t) {
if(ev.motion.state&SDL_BUTTON(2)) break;
pict[sel]->data[(y+1)*pict[sel]->size+x]=(ev.motion.state&SDL_BUTTON(1)?cc:0);
goto redraw;
}
}
show_cursor_xy(x,y,xx,yy,pict[sel]);
break;
case SDL_MOUSEBUTTONDOWN:
if(ev.button.x>=screen->w-161 && ev.button.x<screen->w-1 && ev.button.y>=40 && ev.button.y<200) {
x=(ev.button.x+161-screen->w)/10;
y=(ev.button.y-40)/10;
i=y*16+x;
pick:
y=SDL_GetModState()&0xFFF;
switch(ev.button.button) {
case 1: cc=i; break;
case 2:
for(x=(m.y+1)*pict[sel]->size+m.x;x<pict[sel]->size*(pict[sel]->size+1);x++) {
if(!y || ((y&(KMOD_CTRL|KMOD_SHIFT)) && (((x-x/pict[sel]->size)^(y&KMOD_CTRL?1:0))&1))) {
if(pict[sel]->data[x]==cc) pict[sel]->data[x]=i;
}
if(m.w && x%pict[sel]->size==m.x+m.w-1) x+=pict[sel]->size-m.w;
if(m.h && x/pict[sel]->size>m.h+m.y) break;
}
break;
case 3:
for(x=(m.y+1)*pict[sel]->size+m.x;x<pict[sel]->size*(pict[sel]->size+1);x++) {
if(!y || ((y&(KMOD_CTRL|KMOD_SHIFT)) && (((x-x/pict[sel]->size)^(y&KMOD_CTRL?1:0))&1))) {
if(pict[sel]->data[x]==cc) pict[sel]->data[x]=i;
else if(pict[sel]->data[x]==i) pict[sel]->data[x]=cc;
}
if(m.w && x%pict[sel]->size==m.x+m.w-1) x+=pict[sel]->size-m.w;
if(m.h && x/pict[sel]->size>m.h+m.y) break;
}
break;
}
goto redraw;
} else if(ev.button.x>=pict[sel]->size+2 && ev.button.x<pict[sel]->size*(z+1)+2 && ev.button.y>=41 && ev.button.y<z*pict[sel]->size+41) {
x=(ev.button.x-pict[sel]->size-2)/z;
y=(ev.button.y-41)/z;
i=ev.button.button;
if(i>3) break;
switch(t) {
case 0: // Draw
if(i==2) cc=pict[sel]->data[(y+1)*pict[sel]->size+x];
else pict[sel]->data[(y+1)*pict[sel]->size+x]=(i==1?cc:0);
break;
case 1: // Mark
if(i==1) {
xx=x; yy=y;
} else if(i==3) {
m.x=(x<xx?x:xx); m.y=(y<yy?y:yy);
m.w=abs(x-xx)+1; m.h=abs(y-yy)+1;
}
break;
case 2: // Pick
i=pict[sel]->data[(y+1)*pict[sel]->size+x];
goto pick;
case 3: // Line
if((i&2) && xx!=-1) draw_line(pict[sel]->data+pict[sel]->size,pict[sel]->size,x,y,xx,yy,cc);
if(i&1) xx=x,yy=y;
break;
case 4: // Rectangle
if(i==1) {
xx=x; yy=y;
} else if(xx!=-1) {
p=pict[sel]->data+(j=pict[sel]->size);
if(xx<x) i=x,x=xx,xx=i;
if(yy<y) i=y,y=yy,yy=i;
memset(p+y*j+x,cc,xx+1-x);
memset(p+yy*j+x,cc,xx+1-x);
while(y<yy) p[y*j+x]=p[y*j+xx]=cc,y++;
xx=yy=-1;
}
break;
case 5: // Fill rectangle
if(i==1) {
xx=x; yy=y;
} else if(xx!=-1) {
p=pict[sel]->data+(j=pict[sel]->size);
if(xx<x) i=x,x=xx,xx=i;
if(yy<y) i=y,y=yy,yy=i;
while(y<=yy) memset(p+y*j+x,cc,xx+1-x),y++;
xx=yy=-1;
}
break;
case 6: // Circle
if(i==1) {
xx=x; yy=y;
} else if(i==2) {
draw_circle(pict[sel]->data+pict[sel]->size,pict[sel]->size,x,y,(x-xx)*(x-xx)+(y-yy)*(y-yy),cc);
} else if(i==3) {
draw_circle(pict[sel]->data+pict[sel]->size,pict[sel]->size,xx,yy,(x-xx)*(x-xx)+(y-yy)*(y-yy),cc);
}
break;
case 7: // Fill circle
if(i==1) {
xx=x; yy=y;
} else if(i==2) {
fill_circle(pict[sel]->data+pict[sel]->size,pict[sel]->size,x,y,(x-xx)*(x-xx)+(y-yy)*(y-yy),cc);
} else if(i==3) {
fill_circle(pict[sel]->data+pict[sel]->size,pict[sel]->size,xx,yy,(x-xx)*(x-xx)+(y-yy)*(y-yy),cc);
}
break;
case 8: // Ellipse
if(i==1) {
xx=x; yy=y;
} else if(xx!=-1) {
draw_ellipse(pict[sel]->data+pict[sel]->size,pict[sel]->size,x,y,xx,yy,cc);
}
break;
case 9: // Fill ellipse
if(i==1) {
xx=x; yy=y;
} else if(xx!=-1) {
fill_ellipse(pict[sel]->data+pict[sel]->size,pict[sel]->size,x,y,xx,yy,cc);
draw_ellipse(pict[sel]->data+pict[sel]->size,pict[sel]->size,x,y,xx,yy,cc);
}
break;
case 10: // Flood
if(!m.w) m.x=m.y=m.h=0;
flood(x,y,m.x,m.y,m.w?:pict[sel]->size,m.h?:pict[sel]->size,cc,pict[sel]->data[(y+1)*pict[sel]->size+x],i);
}
goto redraw;
}
break;
case SDL_KEYDOWN:
if((ev.key.keysym.mod&(KMOD_ALT|KMOD_META)) && ev.key.keysym.sym>=SDLK_a && ev.key.keysym.sym<=SDLK_z) {
sqlite3_stmt*st=macro[ev.key.keysym.sym-SDLK_a];
if(!st) goto redraw;
sqlite3_reset(st);
sqlite3_bind_int(st,sqlite3_bind_parameter_index(st,"$cc"),cc);
sqlite3_bind_int(st,sqlite3_bind_parameter_index(st,"$xx"),xx);
sqlite3_bind_int(st,sqlite3_bind_parameter_index(st,"$yy"),yy);
sqlite3_bind_int(st,sqlite3_bind_parameter_index(st,"$mx"),m.x);
sqlite3_bind_int(st,sqlite3_bind_parameter_index(st,"$my"),m.y);
sqlite3_bind_int(st,sqlite3_bind_parameter_index(st,"$mw"),m.w);
sqlite3_bind_int(st,sqlite3_bind_parameter_index(st,"$mh"),m.h);
while((i=sqlite3_step(st))==SQLITE_ROW) {
for(j=0;j<sqlite3_data_count(st);) switch(sqlite3_column_int(st,j++)) {
case 1:
xx=sqlite3_column_int(st,j++);
yy=sqlite3_column_int(st,j++);
break;
case 2:
m.x=sqlite3_column_int(st,j++);
m.y=sqlite3_column_int(st,j++);
m.w=sqlite3_column_int(st,j++);
m.h=sqlite3_column_int(st,j++);
break;
case 3:
cc=sqlite3_column_int(st,j++);
break;
}
}
if(i!=SQLITE_DONE) screen_message(sqlite3_errmsg(userdb));
goto redraw;
}
switch(ev.key.keysym.sym) {
case SDLK_ESCAPE: return;
case SDLK_LEFTBRACKET: case SDLK_LEFTPAREN: --sel; m.x=m.y=m.w=m.h=0; goto redraw;
case SDLK_RIGHTBRACKET: case SDLK_RIGHTPAREN: ++sel; m.x=m.y=m.w=m.h=0; goto redraw;
case SDLK_1 ... SDLK_9: sel=ev.key.keysym.sym-SDLK_1; m.x=m.y=m.w=m.h=0; goto redraw;
case SDLK_c: t=6; xx=yy=-1; goto redraw;
case SDLK_d: t=0; xx=yy=-1; goto redraw;
case SDLK_e: t=8; xx=yy=-1; goto redraw;
case SDLK_f: t=5; xx=yy=-1; goto redraw;
case SDLK_i: t=7; xx=yy=-1; goto redraw;
case SDLK_l: t=3; xx=yy=-1; goto redraw;
case SDLK_m: t=1; xx=yy=-1; goto redraw;
case SDLK_o: t=10; xx=yy=-1; goto redraw;
case SDLK_p: t=2; xx=yy=-1; goto redraw;
case SDLK_r: t=4; xx=yy=-1; goto redraw;
case SDLK_s: t=9; xx=yy=-1; goto redraw;
case SDLK_TAB: if(ev.key.keysym.mod&KMOD_SHIFT) --sel; else ++sel; m.x=m.y=m.w=m.h=0; goto redraw;
case SDLK_SPACE:
m.x=m.y=m.w=m.h=0;
xx=yy=-1;
goto redraw;
case SDLK_RETURN: case SDLK_KP_ENTER:
if(!m.w) m.w=m.h=pict[sel]->size;
free(pclip);
pclip=malloc(sizeof(Picture)+(m.h+1)*m.w);
if(!pclip) fatal("Allocation failed\n");
pclip->meth=m.w;
pclip->size=m.h;
memset(pclip->data,0,m.w);
p=pict[sel]->data+(m.y+1)*pict[sel]->size+m.x;
q=pclip->data+m.w;
for(y=0;y<m.h;y++) {
memcpy(q,p,m.w);
p+=pict[sel]->size;
q+=m.w;
}
goto redraw;
case SDLK_INSERT: case SDLK_KP0: paste:
if(!pclip) break;
if(!m.w) {
if(xx!=-1) {
if(xx+pclip->meth>pict[sel]->size || yy+pclip->size>pict[sel]->size) break;
m.x=xx;
m.y=yy;
}
m.w=pclip->meth;
m.h=pclip->size;
if(m.w>pict[sel]->size) m.w=pict[sel]->size;
if(m.h>pict[sel]->size) m.h=pict[sel]->size;
}
p=pclip->data+pclip->meth;
q=pict[sel]->data+(m.y+1)*pict[sel]->size+m.x;
for(y=0;y<m.h;y++) {
for(x=0;x<m.w;x++) {
i=(x*pclip->meth)/m.w;
j=(y*pclip->size)/m.h;
q[x]=p[j*pclip->meth+i];
}
q+=pict[sel]->size;
}
goto redraw;
case SDLK_DELETE: case SDLK_KP_PERIOD:
if(!m.w) break;
p=pict[sel]->data+(m.y+1)*pict[sel]->size+m.x;
for(y=0;y<m.h;y++) memset(p,0,m.w),p+=pict[sel]->size;
goto redraw;
case SDLK_F1:
if(!m.w) m.w=m.h=pict[sel]->size;
block_rotate(pict[sel]->data+(m.y+1)*pict[sel]->size+m.x,pict[sel]->size,m.w,m.h,5);
goto redraw;
case SDLK_F2:
if(!m.w) m.w=m.h=pict[sel]->size;
block_rotate(pict[sel]->data+(m.y+1)*pict[sel]->size+m.x,pict[sel]->size,m.w,m.h,6);
goto redraw;
case SDLK_F3:
if(!m.w) m.w=m.h=pict[sel]->size;
block_rotate(pict[sel]->data+(m.y+1)*pict[sel]->size+m.x,pict[sel]->size,m.w,m.h,2);
goto redraw;
case SDLK_F4:
if(!m.w) m.w=m.h=pict[sel]->size;
block_rotate(pict[sel]->data+(m.y+1)*pict[sel]->size+m.x,pict[sel]->size,m.w,m.h,1);
goto redraw;
case SDLK_F5: resize:
m.x=m.y=m.w=m.h=0; xx=yy=-1;
p=(Uint8*)screen_prompt("Size? (1 to 255, or P to paste)");
if(!p || !*p) goto redraw;
if(*p=='p' || *p=='P') {
if(!pclip || pclip->meth!=pclip->size) goto redraw;
i=pclip->size;
} else {
i=strtol(p,0,10);
}
if(i<1 || i>255) goto redraw;
free(pict[sel]);
pict[sel]=malloc(sizeof(Picture)+(i+1)*i);
if(!pict[sel]) fatal("Allocation failed\n");
pict[sel]->size=i;
if(*p=='p' || *p=='P') memcpy(pict[sel]->data,pclip->data,(i+1)*i);
else memset(pict[sel]->data,0,(i+1)*i);
goto redraw;
case SDLK_F6:
if(pict[14]) break;
for(sel=0;sel<15;sel++) if(!pict[sel]) break;
goto resize;
case SDLK_F7:
if(!sel && !pict[1]) break;
free(pict[sel]);
for(i=sel;i<15;i++) pict[i]=pict[i+1];
pict[15]=0;
goto redraw;
case SDLK_F8: case SDLK_KP_MULTIPLY:
m.x=m.y=0;
m.w=m.h=pict[sel]->size;
goto redraw;
case SDLK_F9:
xx=yy=-1;
m.x=m.y=m.w=m.h=0;
pict[sel]=do_import(pict[sel]);
goto redraw;
case SDLK_F10:
do_export(pict[sel]);
goto redraw;
case SDLK_LEFT: case SDLK_KP4: --cc; goto redraw;
case SDLK_RIGHT: case SDLK_KP6: ++cc; goto redraw;
case SDLK_UP: case SDLK_KP8: cc-=16; goto redraw;
case SDLK_DOWN: case SDLK_KP2: cc+=16; goto redraw;
case SDLK_HOME: case SDLK_KP7: cc=0; goto redraw;
case SDLK_F12:
p=(Uint8*)screen_prompt("<SQL>");
if(p) sqlite3_exec(userdb,p,response_cb,0,0);
goto redraw;
case SDLK_EQUALS: case SDLK_BACKQUOTE:
if(!spare_page) {
spare_page=malloc(sizeof(Picture)+(picture_size+1)*picture_size);
if(!spare_page) fatal("Allocation failed\n");
spare_page->size=picture_size;
memset(spare_page->data,0,(picture_size+1)*picture_size);
}
if(spare_page->size!=cur_pic->size) {
m.x=m.y=m.w=m.h=0;
xx=yy=-1;
}
pict[sel]=spare_page;
spare_page=cur_pic;
goto redraw;
}
break;
case SDL_VIDEOEXPOSE:
goto redraw;
}
}
}
static void load_dependent_picture(const Uint8*data,int size,DependentPicture*dp) {
FILE*fp;
int i,j,k;
memset(dp,0,sizeof(DependentPicture));
if(!size) {
const char*s=screen_prompt("Name of base picture:");
if(s) strncpy(dp->basename,s,63);
return;
}
fp=fmemopen((Uint8*)data,size,"r");
if(!fp) fatal("Error with fmemopen\n");
i=0;
for(;;) {
j=fgetc(fp);
if(j<32) {
if(j>0) ungetc(j,fp);
break;
}
if(i<63) dp->basename[i++]=j;
}
while(dp->nfilters<64) {
if((j=fgetc(fp))<0) break;
switch(dp->filters[i=dp->nfilters++].code=j) {
case 8 ... 10:
j=dp->filters[i].color.ncolors=fgetc(fp);
if(j<0 || j>64) j=64;
fread(dp->filters[i].color.colors,1,j,fp);
break;
case 11:
k=0;
for(;;) {
j=fgetc(fp);
if(j<32) {
if(j>0) ungetc(j,fp);
break;
}
if(k<63) dp->filters[i].overlay.name[k++]=j;
}
break;
case 12 ... 15:
for(k=0;k<64;k++) {
dp->filters[i].shift.shift[k]=(j=fgetc(fp))&127;
dp->filters[i].shift.size[k]=fgetc(fp);
if(j&128) break;
}
dp->filters[i].shift.nshift=k+1;
break;
}
}
fclose(fp);
}
static void save_dependent_picture(FILE*fp,DependentPicture*dp) {
int i,j;
fwrite(dp->basename,1,strlen(dp->basename)+(dp->filters[0].code<32?0:1),fp);
for(i=0;i<dp->nfilters;i++) {
fputc(dp->filters[i].code,fp);
switch(dp->filters[i].code) {
case 8 ... 10:
fputc(dp->filters[i].color.ncolors,fp);
fwrite(dp->filters[i].color.colors,1,dp->filters[i].color.ncolors,fp);
break;
case 11:
fwrite(dp->filters[i].overlay.name,1,strlen(dp->filters[i].overlay.name)+(i<dp->nfilters-1&&dp->filters[i+1].code<32?0:1),fp);
break;
case 12 ... 15:
if(!dp->filters[i].shift.nshift) fwrite("\x80",1,2,fp);
for(j=0;j<dp->filters[i].shift.nshift;j++) {
fputc(dp->filters[i].shift.shift[j]|(j==dp->filters[i].shift.nshift-1?128:0),fp);
fputc(dp->filters[i].shift.size[j],fp);
}
break;
}
}
}
static int add_filter(DependentPicture*dp,const char*const*const txt,Sint8 c) {
SDL_Rect r;
SDL_Event ev;
Uint8 f;
char buf[4]="<?>";
if(c<0 || dp->nfilters>63) return;
r.x=r.y=12; r.w=200; r.h=136;
set_cursor(XC_iron_cross);
redraw:
SDL_LockSurface(screen);
SDL_FillRect(screen,&r,0xF8);
for(f=0;f<16;f++) {
buf[1]=f+'a';
draw_text(16,(f+2)<<3,buf,0xF8,0xFB);
draw_text(48,(f+2)<<3,txt[f],0xF8,0xFF);
}
SDL_UnlockSurface(screen);
SDL_Flip(screen);
while(SDL_WaitEvent(&ev)) switch(ev.type) {
case SDL_QUIT: exit(0); return 0;
case SDL_KEYDOWN:
if(ev.key.keysym.sym==SDLK_ESCAPE || ev.key.keysym.sym==SDLK_z) return 0;
if(ev.key.keysym.sym<SDLK_a || ev.key.keysym.sym>SDLK_p) break;
f=ev.key.keysym.sym-SDLK_a;
goto found;
case SDL_VIDEOEXPOSE: goto redraw;
}
found:
++dp->nfilters;
if(c!=63) memmove(dp->filters+c+1,dp->filters+c,(63-c)*sizeof(Filter));
memset(dp->filters+c,0,sizeof(Filter));
dp->filters[c].code=f;
return 1;
}
static void edit_color_filter(ColorFilter*f) {
SDL_Rect r;
SDL_Event ev;
char buf[8];
int i;
Uint8 c=0;
redraw:
if(c>=f->ncolors) c=0;
SDL_LockSurface(screen);
r.x=r.y=0; r.w=screen->w; r.h=screen->h;
SDL_FillRect(screen,&r,0xF0);
if(c<f->ncolors) {
r.w=12;
SDL_FillRect(screen,&r,f->colors[c]);
}
draw_text(16,0,"<ESC> Exit <\x18\x19\x1A\x1B> Cursor <-/+> Adjust <INS> Append <DEL> Delete",0xF0,0xFB);
for(i=0;i<f->ncolors;i++) {
snprintf(buf,8,"%c%3d%c",c==i?'<':' ',f->colors[i],c==i?'>':' ');
draw_text((i&1)*50+20,((i&~1)<<2)+12,buf,0xF0,c==i?0xFF:0xF7);
}
SDL_UnlockSurface(screen);
SDL_Flip(screen);
while(SDL_WaitEvent(&ev)) switch(ev.type) {
case SDL_QUIT: exit(0); return;
case SDL_KEYDOWN:
switch(ev.key.keysym.sym) {
case SDLK_ESCAPE: case SDLK_RETURN: case SDLK_KP_ENTER: return;
case SDLK_LEFT: case SDLK_KP4: case SDLK_h: c--; break;
case SDLK_RIGHT: case SDLK_KP6: case SDLK_l: case SDLK_TAB: c++; break;
case SDLK_UP: case SDLK_KP8: case SDLK_k: c-=2; break;
case SDLK_DOWN: case SDLK_KP2: case SDLK_j: c+=2; break;
case SDLK_HOME: case SDLK_KP7: c=0; break;
case SDLK_END: case SDLK_KP1: c=f->ncolors-1; break;
case SDLK_INSERT: case SDLK_KP0:
if(f->ncolors==64) break;
f->colors[c=f->ncolors++]=0;
break;
case SDLK_DELETE: case SDLK_KP_PERIOD:
if(!f->ncolors) break;
f->ncolors--;
c=f->ncolors-1;
break;
case SDLK_KP_PLUS: case SDLK_PLUS: case SDLK_EQUALS:
f->colors[c]+=ev.key.keysym.mod&KMOD_SHIFT?16:1;
break;
case SDLK_KP_MINUS: case SDLK_MINUS: case SDLK_UNDERSCORE:
f->colors[c]-=ev.key.keysym.mod&KMOD_SHIFT?16:1;
break;
case SDLK_SPACE:
f->colors[c]=0;
break;
case SDLK_0 ... SDLK_9:
f->colors[c]=10*f->colors[c]+ev.key.keysym.sym-SDLK_0;
break;
}
goto redraw;
case SDL_VIDEOEXPOSE: goto redraw;
}
}
static void edit_shift_filter(ShiftFilter*f) {
SDL_Rect r;
SDL_Event ev;
const char*s;
char buf[32];
int i;
Uint8 c=0;
redraw:
r.x=r.y=0; r.w=screen->w; r.h=screen->h;
SDL_LockSurface(screen);
SDL_FillRect(screen,&r,0xF0);
draw_text(0,0,"<ESC> Exit <\x18/\x19> Cursor <SP> Edit <INS> Insert <DEL> Delete",0xF0,0xFB);
if(c>f->nshift+1) c=f->nshift+1;
for(i=0;i<f->nshift;i++) {
snprintf(buf,32,"%c %d: %d",c==i?0x1A:0xFA,f->size[i],f->shift[i]);
draw_text(4,(i+2)<<3,buf,0xF0,c==i?0xFF:0xF7);
}
snprintf(buf,32,"%c <End>",c==i?0x1A:0xFA);
draw_text(4,(i+2)<<3,buf,0xF0,c==i?0xFF:0xF7);
SDL_UnlockSurface(screen);
SDL_Flip(screen);
while(SDL_WaitEvent(&ev)) switch(ev.type) {
case SDL_QUIT: exit(0); return;
case SDL_KEYDOWN:
switch(ev.key.keysym.sym) {
case SDLK_ESCAPE: return;
case SDLK_HOME: case SDLK_KP7: c=0; break;
case SDLK_END: case SDLK_KP1: c=f->nshift; break;
case SDLK_UP: case SDLK_KP8: case SDLK_k: c--; break;
case SDLK_DOWN: case SDLK_KP2: case SDLK_j: c++; break;
case SDLK_INSERT: case SDLK_KP0:
if(f->nshift==64) break;
if(c<f->nshift) {
memmove(f->shift+c+1,f->shift+c,63-c);
memmove(f->size+c+1,f->size+c,63-c);
}
f->nshift++;
//fallthru
case SDLK_SPACE:
if(c>=f->nshift) break;
if((s=screen_prompt("Size?")) && (i=strtol(s,0,10)) && (s=screen_prompt("Shift?")) && *s) {
f->size[c]=i;
f->shift[c]=strtol(s,0,10)&127;
}
break;
case SDLK_DELETE: case SDLK_KP_PERIOD:
if(c>=f->nshift) break;
if(c<f->nshift-1) {
memmove(f->shift+c,f->shift+c+1,63-c);
memmove(f->size+c,f->size+c+1,63-c);
}
f->nshift--;
break;
}
goto redraw;
case SDL_VIDEOEXPOSE: goto redraw;
}
}
static void edit_dependent_picture(DependentPicture*dp,const char*name) {
static const char*const txt[16]={
"Identity",
"Flip \x1D",
"Flip \x12",
"Rotate 180\xF8",
"Transpose",
"Transposed flip \x1D",
"Transposed flip \x12",
"Reverse Transpose",
"Replace colors",
"Advance colors",
"Exchange colors",
"Overlay",
"Shift \x18",
"Shift \x19",
"Shift \x1A",
"Shift \x1B",
};
const char*s;
char buf[64];
Sint8 c=-1;
int i,j,y;
SDL_Rect r;
SDL_Event ev;
set_cursor(XC_arrow);
redraw:
r.x=r.y=0; r.w=screen->w; r.h=screen->h;
SDL_LockSurface(screen);
SDL_FillRect(screen,&r,0xF0);
draw_text(0,0,"Dependent Image:",0xF0,0xF7);
draw_text(136,0,name,0xF0,0xF5);
draw_text(0,8,"<ESC> Exit <\x18/\x19> Cursor <SP> Edit <INS> Insert <DEL> Delete",0xF0,0xFB);
draw_text(0,16,c<0?"\x1A":"\xFA",0xF0,c<0?0xFA:0xF2);
draw_text(16,16,"Base:",0xF0,0xF7);
draw_text(64,16,dp->basename,0xF0,0xFE);
for(i=0,y=24;i<dp->nfilters;i++) {
draw_text(0,y,c==i?"\x1A":"\xFA",0xF0,c==i?0xFA:0xF2);
switch(j=dp->filters[i].code) {
case 0 ... 7:
draw_text(16,y,txt[j],0xF0,0xF7);
y+=8;
break;
case 8 ... 10:
draw_text(16,y,txt[j],0xF0,0xF7);
y+=8;
for(j=0;j<dp->filters[i].color.ncolors;j+=2) {
if(j==dp->filters[i].color.ncolors-1) snprintf(buf,64,"%3d",dp->filters[i].color.colors[j]);
else snprintf(buf,64,"%3d %3d",dp->filters[i].color.colors[j],dp->filters[i].color.colors[j+1]);
draw_text(32,y,buf,0xF0,0xFE);
y+=8;
}
break;
case 11:
draw_text(16,y,"Overlay: ",0xF0,0xF7);
draw_text(88,y,dp->filters[i].overlay.name,0xF0,0xFE);
y+=8;
break;
case 12 ... 15:
draw_text(16,y,txt[j],0xF0,0xF7);
y+=8;
for(j=0;j<dp->filters[i].shift.nshift;j++) {
snprintf(buf,64,"%d: %d",dp->filters[i].shift.size[j],dp->filters[i].shift.shift[j]);
draw_text(32,y,buf,0xF0,0xFE);
y+=8;
}
break;
default:
draw_text(16,y,"???",0xF0,0xFC);
y+=8;
break;
}
}
if(c>dp->nfilters) c=dp->nfilters;
draw_text(0,y,c==dp->nfilters?"\x1A":"\xFA",0xF0,c==dp->nfilters?0xFA:0xF2);
draw_text(16,y,"<End>",0xF0,0xF8);
SDL_UnlockSurface(screen);
SDL_Flip(screen);
while(SDL_WaitEvent(&ev)) {
switch(ev.type) {
case SDL_QUIT:
exit(0);
break;
case SDL_KEYDOWN:
switch(ev.key.keysym.sym) {
case SDLK_ESCAPE: return;
case SDLK_UP: case SDLK_KP8: case SDLK_a: case SDLK_k: if(c>=0) --c; break;
case SDLK_DOWN: case SDLK_KP2: case SDLK_z: case SDLK_j: ++c; break;
case SDLK_INSERT: case SDLK_KP0:
if(add_filter(dp,txt,c)) // maybe fallthrough
case SDLK_SPACE:
if(c==-1) {
s=screen_prompt("Name of base picture:");
if(s && *s) strncpy(dp->basename,s,63);
} else if(c<dp->nfilters) {
switch(dp->filters[c].code) {
case 8 ... 10:
edit_color_filter(&dp->filters[c].color);
break;
case 11:
s=screen_prompt("Name of overlay picture:");
if(s && *s) strncpy(dp->filters[c].overlay.name,s,63);
break;
case 12 ... 15:
edit_shift_filter(&dp->filters[c].shift);
break;
}
}
break;
case SDLK_DELETE: case SDLK_KP_PERIOD:
if(c<0 || c>=dp->nfilters) break;
--dp->nfilters;
if(c<dp->nfilters) memmove(dp->filters+c,dp->filters+c+1,(dp->nfilters-c)*sizeof(Filter));
break;
case SDLK_HOME: case SDLK_KP7: c=-1; break;
case SDLK_END: case SDLK_KP1: c=dp->nfilters; break;
}
goto redraw;
case SDL_VIDEOEXPOSE:
goto redraw;
}
}
}
static void edit_picture(sqlite3_int64 id) {
sqlite3_stmt*st;
Picture*pict[16]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
DependentPicture*dpict;
char*name;
const unsigned char*data;
Uint8*buf=0;
size_t size=0;
FILE*fp;
int i,n;
if(i=sqlite3_prepare_v2(userdb,"SELECT SUBSTR(`NAME`,1,LENGTH(`NAME`)-4),`DATA`,`TYPE` FROM `PICEDIT` WHERE `ID`=?1;",-1,&st,0)) {
screen_message(sqlite3_errmsg(userdb));
return;
}
sqlite3_bind_int64(st,1,id);
i=sqlite3_step(st);
if(i!=SQLITE_ROW) {
screen_message(i==SQLITE_DONE?"No such ID":sqlite3_errmsg(userdb));
sqlite3_finalize(st);
return;
}
data=sqlite3_column_blob(st,1);
n=sqlite3_column_bytes(st,1);
i=sqlite3_column_int(st,2);
name=strdup(sqlite3_column_text(st,0)?:(const unsigned char*)"???");
if(!name) fatal("Allocation failed\n");
if(i==1) {
load_picture_lump(data,n,pict);
sqlite3_finalize(st);
if(!*pict) {
if(!*gsizes) *gsizes=picture_size;
for(i=0;gsizes[i];i++) {
pict[i]=malloc(sizeof(Picture)+(gsizes[i]+1)*gsizes[i]);
if(!pict[i]) fatal("Allocation failed\n");
pict[i]->size=gsizes[i];
memset(pict[i]->data,0,(gsizes[i]+1)*gsizes[i]);
}
}
edit_picture_1(pict,name);
cur_pic=0;
fp=open_memstream((char**)&buf,&size);
if(!fp) fatal("Cannot open memory stream\n");
for(i=n=0;i<16;i++) if(pict[i]) n++;
fputc(n,fp);
for(i=0;i<n;i++) fputc(pict[i]->size,fp);
for(i=0;i<n>>1;i++) fputc(0,fp);
for(i=0;i<n;i++) compress_picture(fp,pict[i]);
fclose(fp);
if(!buf || !size) fatal("Allocation failed\n");
*buf|=pict[0]->meth<<4;
for(i=1;i<n;i++) buf[n+1+((i-1)>>1)]|=pict[i]->meth<<(i&1?0:4);
for(i=0;i<16;i++) free(pict[i]);
} else if(i==2) {
dpict=malloc(sizeof(DependentPicture));
if(!dpict) fatal("Allocation failed\n");
load_dependent_picture(data,n,dpict);
fp=open_memstream((char**)&buf,&size);
if(!fp) fatal("Cannot open memory stream\n");
edit_dependent_picture(dpict,name);
save_dependent_picture(fp,dpict);
free(dpict);
fclose(fp);
if(!buf || !size) fatal("Allocation failed\n");
}
free(name);
if(i=sqlite3_prepare_v2(userdb,"UPDATE `PICEDIT` SET `DATA`=?2 WHERE ID=?1;",-1,&st,0)) {
screen_message(sqlite3_errmsg(userdb));
free(buf);
return;
}
sqlite3_bind_int64(st,1,id);
sqlite3_bind_blob(st,2,buf,size,free);
i=sqlite3_step(st);
if(i!=SQLITE_DONE) screen_message(sqlite3_errmsg(userdb));
sqlite3_finalize(st);
}
static int add_picture(int t) {
sqlite3_stmt*st;
const char*s=screen_prompt("Enter name of new picture:");
int i;
if(!s || !*s) return 0;
if(sqlite3_prepare_v2(userdb,"INSERT INTO `PICEDIT`(`NAME`,`TYPE`,`DATA`) SELECT VALID_NAME(?1)||CASE ?2 WHEN 1 THEN '.IMG' ELSE '.DEP' END,?2,X'';",-1,&st,0)) {
screen_message(sqlite3_errmsg(userdb));
return 0;
}
sqlite3_bind_text(st,1,s,-1,0);
sqlite3_bind_int(st,2,t);
i=sqlite3_step(st);
sqlite3_finalize(st);
if(i!=SQLITE_DONE) {
screen_message(sqlite3_errmsg(userdb));
return 0;
}
edit_picture(sqlite3_last_insert_rowid(userdb));
return 1;
}
static int delete_picture(void) {
sqlite3_stmt*st;
const char*s=screen_prompt("Enter name of picture to delete:");
int i;
if(!s || !*s) return 0;
if(sqlite3_prepare_v2(userdb,"DELETE FROM `PICEDIT` WHERE `NAME`=?1||'.IMG' OR `NAME`=?1||'.DEP';",-1,&st,0)) {
screen_message(sqlite3_errmsg(userdb));
return 0;
}
sqlite3_bind_text(st,1,s,-1,0);
i=sqlite3_step(st);
sqlite3_finalize(st);
if(i!=SQLITE_DONE) {
screen_message(sqlite3_errmsg(userdb));
return 0;
}
return 1;
}
static void rename_picture(void) {
sqlite3_stmt*st;
const char*s=screen_prompt("Old name:");
int i;
if(!s || !*s) return;
if(sqlite3_prepare_v2(userdb,"UPDATE `PICEDIT` SET `NAME`=VALID_NAME(?2)||SUBSTR(`NAME`,-4) WHERE `NAME`=?1||'.IMG' OR `NAME`=?1||'.DEP';",-1,&st,0)) {
screen_message(sqlite3_errmsg(userdb));
return;
}
sqlite3_bind_text(st,1,s,-1,SQLITE_TRANSIENT);
s=screen_prompt("New name:");
if(!s || !*s) {
sqlite3_finalize(st);
return;
}
sqlite3_bind_text(st,2,s,-1,SQLITE_TRANSIENT);
i=sqlite3_step(st);
sqlite3_finalize(st);
if(i!=SQLITE_DONE) screen_message(sqlite3_errmsg(userdb));
}
static sqlite3_int64 copy_picture(void) {
sqlite3_stmt*st;
const char*s=screen_prompt("Copy from:");
int i;
sqlite3_set_last_insert_rowid(userdb,0);
if(!s || !*s) return 0;
if(sqlite3_prepare_v2(userdb,"INSERT INTO `PICEDIT`(`NAME`,`TYPE`,`DATA`) SELECT VALID_NAME(?2)||SUBSTR(`NAME`,-4),`TYPE`,`DATA` "
"FROM `PICEDIT` WHERE SUBSTR(`NAME`,1,LENGTH(`NAME`)-4)=?1 AND `TYPE`<>0;",-1,&st,0)) {
screen_message(sqlite3_errmsg(userdb));
return 0;
}
sqlite3_bind_text(st,1,s,-1,SQLITE_TRANSIENT);
s=screen_prompt("Copy to:");
if(!s || !*s) {
sqlite3_finalize(st);
return 0;
}
sqlite3_bind_text(st,2,s,-1,SQLITE_TRANSIENT);
i=sqlite3_step(st);
sqlite3_finalize(st);
if(i==SQLITE_DONE) {
return sqlite3_last_insert_rowid(userdb);
} else {
screen_message(sqlite3_errmsg(userdb));
return 0;
}
}
static void do_config(void) {
sqlite3_stmt*st;
const char*s;
char buf[16];
int i,n;
for(n=0;n<15;n++) {
snprintf(buf,15,"%d%c%c variant:",n+1,"snrt"[n>3?3:n],"tddh"[n>3?3:n]);
s=screen_prompt(buf);
if(!s || !*s) break;
gsizes[n]=strtol(s,0,10);
}
if(!n) return;
gsizes[n]=0;
i=sqlite3_exec(userdb,
"INSERT INTO `PICEDIT`(`NAME`,`TYPE`,`DATA`) SELECT 'PICEDIT.CFG',0,'' EXCEPT SELECT `NAME`,0,'' FROM `PICEDIT`;"
,0,0,0);
if(i) fatal("SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
i=sqlite3_prepare_v2(userdb,"UPDATE `PICEDIT` SET `DATA`=?1 WHERE `NAME`='PICEDIT.CFG';",-1,&st,0);
if(i) fatal("SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
sqlite3_bind_blob(st,1,gsizes,n,0);
if(sqlite3_step(st)!=SQLITE_DONE) fatal("SQL error: %s\n",sqlite3_errmsg(userdb));
sqlite3_finalize(st);
}
static void set_caption(void) {
char buf[256];
snprintf(buf,255,"Free Hero Mesh - %s - Picture",basefilename);
SDL_WM_SetCaption(buf,buf);
}
void run_picture_editor(void) {
sqlite3_int64*ids;
SDL_Event ev;
SDL_Rect r;
sqlite3_stmt*st;
int sc=0;
int max=load_picture_file();
int i,n;
sqlite3_create_function(userdb,"VALID_NAME",1,SQLITE_UTF8|SQLITE_DETERMINISTIC,0,fn_valid_name,0,0);
sqlite3_create_module(userdb,"GRAPH",&vt_graph,0);
init_palette();
optionquery[1]=Q_imageSize;
picture_size=strtol(xrm_get_resource(resourcedb,optionquery,optionquery,2)?:"16",0,10);
optionquery[1]=Q_picedit;
optionquery[2]=Q_macro;
xrm_search(resourcedb,optionquery,optionquery,3,load_macros,0);
if(!*gsizes) *gsizes=picture_size;
set_cursor(XC_arrow);
set_caption();
i=sqlite3_prepare_v3(userdb,"SELECT `ID`,SUBSTR(`NAME`,1,LENGTH(`NAME`)-4),`TYPE` FROM `PICEDIT` WHERE `TYPE` ORDER BY `NAME` LIMIT ?1 OFFSET ?2;",-1,SQLITE_PREPARE_PERSISTENT,&st,0);
if(i) fatal("SQL error (%d): %s\n",i,sqlite3_errmsg(userdb));
ids=calloc(screen->h/8,sizeof(sqlite3_int64));
if(!ids) fatal("Allocation failed\n");
redraw:
sqlite3_reset(st);
if(sc>max-screen->h/8+1) sc=max-screen->h/8+1;
if(sc<0) sc=0;
sqlite3_bind_int(st,1,screen->h/8-1);
sqlite3_bind_int(st,2,sc);
SDL_LockSurface(screen);
r.x=r.y=0; r.w=screen->w; r.h=screen->h;
SDL_FillRect(screen,&r,0xF0);
draw_text(0,0,"<ESC> Save/Quit <F1> Add <F2> Delete <F3> Edit <F4> Rename <F5> AddDependent <F6> Config <F7> Copy",0xF0,0xFB);
n=0;
while((i=sqlite3_step(st))==SQLITE_ROW) {
ids[n++]=sqlite3_column_int64(st,0);
draw_text(16,8*n,sqlite3_column_text(st,1),0xF0,sqlite3_column_int(st,2)==1?0xF7:0xF2);
if(8*n+8>screen->h-8) break;
}
SDL_UnlockSurface(screen);
sqlite3_reset(st);
r.y=8; r.h-=8;
scrollbar(&sc,r.h/8,max,0,&r);
SDL_Flip(screen);
while(SDL_WaitEvent(&ev)) {
if(ev.type!=SDL_VIDEOEXPOSE && scrollbar(&sc,screen->h/8,max,&ev,&r)) goto redraw;
switch(ev.type) {
case SDL_QUIT:
return;
case SDL_KEYDOWN:
switch(ev.key.keysym.sym) {
case SDLK_ESCAPE:
if(!(ev.key.keysym.mod&(KMOD_SHIFT|KMOD_CTRL))) save_picture_file();
return;
case SDLK_q:
if(!(ev.key.keysym.mod&KMOD_CTRL)) break;
if(!(ev.key.keysym.mod&KMOD_SHIFT)) save_picture_file();
return;
case SDLK_HOME: case SDLK_KP7: sc=0; goto redraw;
case SDLK_UP: case SDLK_KP8: if(sc) --sc; goto redraw;
case SDLK_DOWN: case SDLK_KP2: ++sc; goto redraw;
case SDLK_END: case SDLK_KP1: sc=max-screen->h/8+1; goto redraw;
case SDLK_PAGEUP: case SDLK_KP9: sc-=screen->h/8-1; goto redraw;
case SDLK_PAGEDOWN: case SDLK_KP3: sc+=screen->h/8-1; goto redraw;
case SDLK_F1:
if(max<32767) max+=add_picture(1);
else screen_message("Too many pictures");
goto redraw;
case SDLK_F2:
max-=delete_picture();
goto redraw;
case SDLK_F3:
*ids=ask_picture_id("Edit:");
if(*ids) edit_picture(*ids);
goto redraw;
case SDLK_F4:
rename_picture();
goto redraw;
case SDLK_F5:
if(max<32767) max+=add_picture(2);
else screen_message("Too many pictures");
goto redraw;
case SDLK_F6:
do_config();
goto redraw;
case SDLK_F7:
*ids=copy_picture();
if(*ids) max++,edit_picture(*ids);
goto redraw;
case SDLK_F12:
sqlite3_exec(userdb,screen_prompt("<SQL>")?:"",response_cb,0,0);
goto redraw;
}
break;
case SDL_MOUSEMOTION:
set_cursor(XC_arrow);
break;
case SDL_VIDEOEXPOSE:
goto redraw;
}
}
}