/* drawf  DrawFile functions */

#include "oslib/os.h"

#include "oslib/osfile.h"

#include "oslib/drawfile.h"

#include "Python.h"

#include <limits.h>

#define PyDrawF_Check(op) ((op)->ob_type == &PyDrawFType)

#define HDRSIZE 40

#define GRPHDRSIZE 36

#define DRAWTYPE 0xaff

#define NEXT(d) ((drawfile_object*)((char*)(d)+(d)->size))

typedef struct

{ PyObject_HEAD

  drawfile_diagram *drawf;

  int size; /*length in bytes*/

  int nobjs;  /* no of objects */

} PyDrawFObject;

typedef struct dheader

{  char tag [4];

   int major_version;

   int minor_version;

   char source [12];

   os_box bbox;

} dheader;

static PyObject *DrawFError; /* Exception drawf.error */

static os_error *e;

static PyTypeObject PyDrawFType;

static dheader header=

{ {'D','r','a','w'},

  201,0,

  {'P','y','t','h','o','n',' ',' ',' ',' ',' ',' '},

  {INT_MAX,INT_MAX,INT_MIN,INT_MIN}

};

static PyObject *drawf_oserror(void)

{ PyErr_SetString(DrawFError,e->errmess);

  return 0;

}

static PyObject *drawf_error(char *s)

{ PyErr_SetString(DrawFError,s);

  return 0;

}

/* DrawFile commands */

static void countobjs(PyDrawFObject *pd)

{ int k=0,q;

  drawfile_diagram *dd=pd->drawf;

  drawfile_object *d=dd->objects;

  char *end=(char*)dd+pd->size;

  pd->nobjs=-1;

  while((char*)d<end)

  { k++;

    q=d->size;

    if(q<=0||q&3) return ;

    d=NEXT(d);

  }

  if ((char*)d==end) pd->nobjs=k;

}

static drawfile_object *findobj(PyDrawFObject *pd,Py_ssize_t n)

{ drawfile_diagram *dd=pd->drawf;

  drawfile_object *d=dd->objects;

  for(;n>0;n--) d=NEXT(d);

  return d;

}

static PyDrawFObject* new(int size)

{ PyDrawFObject *b=PyObject_NEW(PyDrawFObject,&PyDrawFType);

  if(!b) return NULL;

  size+=HDRSIZE;

  b->drawf=malloc(size);

  if(!b->drawf)

  { Py_DECREF(b);

    return (PyDrawFObject*)PyErr_NoMemory();

  }

  b->size=size;

  return b;

}

static void extend(os_box *b,os_box *c)

{ if(b->x0>c->x0) b->x0=c->x0;

  if(b->y0>c->y0) b->y0=c->y0;

  if(b->x1<c->x1) b->x1=c->x1;

  if(b->y1<c->y1) b->y1=c->y1;

}

static PyObject *DrawF_New(PyObject *self,PyObject *args)

{ PyDrawFObject *b;

  if(!PyArg_ParseTuple(args,"")) return NULL;

  b=new(0);

  if(!b) return NULL;

  *((dheader*)b->drawf)=header;

  b->nobjs=0;

  return (PyObject *)b;

}

static PyObject *DrawF_Load(PyObject *self,PyObject *args)

{ int size;

  char *fname;

  PyDrawFObject *b;

  fileswitch_object_type ot;

  if(!PyArg_ParseTuple(args,"s",&fname)) return NULL;

  e=xosfile_read_no_path(fname,&ot,0,0,&size,0);

  if(e) return drawf_oserror();

  size-=HDRSIZE;

  if(ot!=osfile_IS_FILE||size<0||size&3) return drawf_error("Bad draw file");

  b=new(size);

  if(!b) return NULL;

  e=xosfile_load_stamped_no_path(fname,(byte*)(b->drawf),0,0,0,0,0);

  if(e)

  { Py_DECREF(b);

    return drawf_oserror();

  }

  countobjs(b);

  if(b->nobjs>=0) return (PyObject *)b;

  Py_DECREF(b);

  return drawf_error("Corrupt draw file");

}

static PyObject *DrawF_Save(PyDrawFObject *self,PyObject *arg)

{ char *fname;

  if(!PyArg_ParseTuple(arg,"s",&fname)) return NULL;

  e=xosfile_save_stamped(fname,DRAWTYPE,

  (byte*)(self->drawf),(byte*)(self->drawf)+self->size);

  if (e) return drawf_oserror();

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_Render(PyDrawFObject *self,PyObject *arg)

{ int flags,trans,clip,flat;

  if(!PyArg_ParseTuple(arg,"iiii",&flags,&trans,&clip,&flat)) return NULL;

  e=xdrawfile_render((drawfile_render_flags)flags,self->drawf,self->size,

  (os_trfm*)trans,(os_box*)clip,flat);

  if(e) return drawf_oserror();

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_Path(PyDrawFObject *self,PyObject *arg)

{ PyObject *pl;

  PyObject *dp=0;

  os_colour fill;

  os_colour outline;

  int width,start=0;

  drawfile_path_style style;

  int size=40;

  int n,i,element_count;

  drawfile_diagram *diag;

  drawfile_object *dobj;

  drawfile_path *dpath;

  draw_path *thepath;

  draw_line_style line_style;

  draw_dash_pattern *dash_pattern=0;

  os_box *box;

  long *pe;

  if(!PyArg_ParseTuple(arg,"O!(iiii)|O!i",&PyList_Type,&pl,(int*)&fill,

     (int*)&outline,&width,(int*)&style,&PyTuple_Type,&dp,&start)) return NULL;

  if(dp)

  { element_count=PyTuple_Size(dp);

    size+=4*element_count+8;

    style.flags|=drawfile_PATH_DASHED;

  }

  else style.flags&=~drawfile_PATH_DASHED;

  n=PyList_Size(pl);

  size+=4*n+8;

  for(i=0;i<n;i++) if(!PyInt_Check(PyList_GetItem(pl,i))) size+=4;

  diag=realloc(self->drawf,self->size+size);

  if(!diag) return PyErr_NoMemory();

  self->drawf=diag;

  dobj=(drawfile_object*)((char*)diag+self->size);

  dobj->type=drawfile_TYPE_PATH;

  dobj->size=size;

  dpath=&dobj->data.path;

  self->size+=size;

  self->nobjs++;

  box=&dpath->bbox;

  dpath->fill=fill;dpath->outline=outline;

  dpath->width=width;dpath->style=style;

  pe=(long*)&(dpath->path);

  if(dp)

  { dash_pattern=&(((drawfile_path_with_pattern*)dpath)->pattern);

    dash_pattern->start=start;

    dash_pattern->element_count=element_count;

    for(i=0;i<element_count;i++)

    { dash_pattern->elements[i]=(int)PyInt_AsLong(PyTuple_GetItem(dp,i));

    }

    pe+=element_count+2;

  }

  thepath=(draw_path*)pe;

  *pe++=draw_MOVE_TO;

  for(i=0;i<n;i++)

  { PyObject *p=PyList_GetItem(pl,i);

    if(PyInt_Check(p))

      *pe++=PyInt_AsLong(p);

    else

    {

      *pe++=PyInt_AsLong(PyTuple_GetItem(p,0));

      *pe++=PyInt_AsLong(PyTuple_GetItem(p,1));

    }

  }

  *pe=draw_END_PATH;

  line_style.join_style=style.flags&3;

  line_style.end_cap_style=(style.flags&3)>>2;

  line_style.start_cap_style=(style.flags&3)>>4;

  line_style.reserved=0;

  line_style.mitre_limit=10;

  line_style.start_cap_width=style.cap_width;

  line_style.end_cap_width=style.cap_width;

  line_style.start_cap_length=style.cap_length;

  line_style.end_cap_length=style.cap_length;

  e=xdraw_process_path(thepath,0x70000000,0,0,width,&line_style,dash_pattern,

  (draw_output_path)((char*)box+0x80000000),0);

  if(e) return drawf_oserror();

  /* draw_process_path seems to have a bug:

     If the bounding box size is zero, it returns 0x7FFFFFFF, ..., 0x80000000 instead of the

     correct size. */

  if (box->x0==0x7FFFFFFF)

  {

      /* path has zero extension, so forget it; it would be invisible anyway */

      self->size-=size;

      self->nobjs--;

      diag=realloc(self->drawf,self->size);

      if(!diag) return PyErr_NoMemory();

      self->drawf=diag;

  }

  else

      extend(&(diag->bbox),box);

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_Text(PyDrawFObject *self,PyObject *arg)

{ os_colour fill,bg_hint;

  drawfile_text_style style;

  int xsize,ysize,x,y;

  int size,len;

  char *text;

  drawfile_diagram *diag;

  drawfile_object *dobj;

  drawfile_text *dtext;

  os_box *box;

  if(!PyArg_ParseTuple(arg,"(ii)s(iiiii)",&x,&y,&text,

     (int*)&fill,(int*)&bg_hint,(int*)&style,&xsize,&ysize)) return NULL;

  len=strlen(text);

  size=((len+4)&(~3))+52;

  diag=realloc(self->drawf,self->size+size);

  if(!diag) return PyErr_NoMemory();

  self->drawf=diag;

  dobj=(drawfile_object*)((char*)diag+self->size);

  dobj->type=drawfile_TYPE_TEXT;

  dobj->size=size;

  dtext=&dobj->data.text;

  self->size+=size;

  self->nobjs++;

  dtext->fill=fill;

  dtext->bg_hint=bg_hint;

  dtext->style=style;

  dtext->xsize=xsize;

  dtext->ysize=ysize;

  dtext->base.x=x;

  dtext->base.y=y;

  memset(dtext->text,0,(len+4)&(~3));

  sprintf(dtext->text,"%s",text);

  box=&(dtext->bbox);

  box->x0=x;box->y0=y;box->x1=x+len*xsize;box->y1=y+ysize;

  extend(&(diag->bbox),box);

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_TText(PyDrawFObject *self,PyObject *arg)

{ os_colour fill,bg_hint;

  drawfile_text_style style;

  int xsize,ysize,x,y;

  int t1,t2,t3,t4,t5,t6;

  int size,len;

  char *text;

  drawfile_diagram *diag;

  drawfile_object *dobj;

  drawfile_trfm_text *dtext;

  os_box *box;

  t1=1<<16;t2=0;

  t3=0;t4=1<<16;

  t5=t6=0;

  if(!PyArg_ParseTuple(arg,"(ii)s(iiiii)|(iiiiii)",&x,&y,&text,

     (int*)&fill,(int*)&bg_hint,(int*)&style,&xsize,&ysize,&t1,&t2,&t3,&t4,&t5,&t6)) return NULL;

  len=strlen(text);

  size=((len+4)&(~3))+52+28;

  diag=realloc(self->drawf,self->size+size);

  if(!diag) return PyErr_NoMemory();

  self->drawf=diag;

  dobj=(drawfile_object*)((char*)diag+self->size);

  dobj->type=drawfile_TYPE_TRFM_TEXT;

  dobj->size=size;

  dtext=&dobj->data.trfm_text;

  self->size+=size;

  self->nobjs++;

  dtext->trfm.entries[0][0]=t1;

  dtext->trfm.entries[0][1]=t2;

  dtext->trfm.entries[1][0]=t3;

  dtext->trfm.entries[1][1]=t4;

  dtext->trfm.entries[2][0]=t5;

  dtext->trfm.entries[2][1]=t6;

  dtext->flags=0;

  dtext->fill=fill;

  dtext->bg_hint=bg_hint;

  dtext->style=style;

  dtext->xsize=xsize;

  dtext->ysize=ysize;

  dtext->base.x=x;

  dtext->base.y=y;

  memset(dtext->text,0,(len+4)&(~3));

  sprintf(dtext->text,"%s",text);

  box=&(dtext->bbox);

  box->x0=x;box->y0=y;box->x1=x+len*xsize;box->y1=y+ysize;

  extend(&(diag->bbox),box);

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_FontTable(PyDrawFObject *self,PyObject *arg)

{ int size=8,n=0;

  PyObject *d,*key,*value;

  drawfile_diagram *diag;

  drawfile_object *dobj;

  char *dtable;

  if(!PyArg_ParseTuple(arg,"O!",&PyDict_Type,&d)) return NULL;

  while(PyDict_Next(d,&n,&key,&value))

  { int m=PyString_Size(value);

    if(m<0||!PyInt_Check(key)) return NULL;

    size+=m+2;

  }

  size=(size+3)&(~3);

  diag=realloc(self->drawf,self->size+size);

  if(!diag) return PyErr_NoMemory();

  self->drawf=diag;

  dobj=(drawfile_object*)((char*)diag+self->size);

  dobj->type=drawfile_TYPE_FONT_TABLE;

  dobj->size=size;

  dtable=(char*)(&dobj->data.font_table);

  self->size+=size;

  self->nobjs++;

  memset(dtable,0,size-8);

  n=0;

  while(PyDict_Next(d,&n,&key,&value))

  { int m=PyString_Size(value);

    *dtable=(char)PyInt_AsLong(key);

    strcpy(dtable+1,PyString_AsString(value));

    dtable+=m+2;

  }

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_Group(PyDrawFObject *self,PyObject *arg)

{ int size,n;

  PyDrawFObject *g;

  char *name="";

  drawfile_diagram *diag;

  drawfile_object *dobj;

  drawfile_group *dgroup;

  if(!PyArg_ParseTuple(arg,"O!|s",&PyDrawFType,(PyObject*)&g,&name))

       return NULL;

  size=g->size-4;

  diag=realloc(self->drawf,self->size+size);

  if(!diag) return PyErr_NoMemory();

  self->drawf=diag;

  self->nobjs++;

  dobj=(drawfile_object*)((char*)diag+self->size);

  self->size+=size;

  dobj->type=drawfile_TYPE_GROUP;

  dobj->size=g->size-4;

  dgroup=&dobj->data.group;

  dgroup->bbox=g->drawf->bbox;

  memset(dgroup->name,' ',12);

  n=strlen(name);

  if(n>12) n=12;

  memcpy(dgroup->name,name,n);

  memcpy((char*)dgroup->objects,(char*)g->drawf+40,g->size-40);

  extend(&(diag->bbox),&(dgroup->bbox));

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_Find(PyDrawFObject *self,PyObject *arg)

{ int x,y,n=0;

  int r=-1;

  drawfile_object *d;

  if(!PyArg_ParseTuple(arg,"ii|i",&x,&y,&n)) return NULL;

  if(n<self->nobjs&&n>=0)

  { d=findobj(self,n);

    while(n<self->nobjs)

    { if(x>=d->data.text.bbox.x0&&x<=d->data.text.bbox.x1&&

         y>=d->data.text.bbox.y0&&y<=d->data.text.bbox.y1) { r=n;break;}

      n++;

      d=NEXT(d);

    }

  }

  return PyInt_FromLong(r);

}

static PyObject *DrawF_Box(PyDrawFObject *self,PyObject *arg)

{ int n=-1;

  os_box *box;

  if(!PyArg_ParseTuple(arg,"|i",&n)) return NULL;

  if(n>=self->nobjs|n<0) box=&(self->drawf->bbox);

  else box=&(findobj(self,n)->data.text.bbox);

  return Py_BuildValue("iiii",box->x0,box->y0,box->x1,box->y1);

}

static void SetBlock(drawfile_object *d,int size,int type,int offset,int value)

{ char *end=(char*)d+size;

  printf("s%d t%d o%d v%d\n",size,type,offset,value);

  for(;(char*)d<end;d=NEXT(d))

    if(d->type==type) ((int*)(d))[offset]=value;

    else if(d->type==drawfile_TYPE_GROUP)

           SetBlock((drawfile_object*)&d->data.group.objects,

                    d->size,type,offset,value);

  printf("SetBlock Done\n");

}

static PyObject *SetWord(PyDrawFObject *self,PyObject *arg,int type,int offset)

{ int n=PyTuple_Size(arg);

  int m,value,q;

  PyObject *par;

  drawfile_object *e,*d=self->drawf->objects;

  if(n==0) return  drawf_error("Value Required");

  par=PyTuple_GetItem(arg,0);

  if(!PyInt_Check(par))

  { PyErr_SetString(PyExc_TypeError,"Int Required");

    return 0;

  }

  value=(int)PyInt_AsLong(par);

  if(n==1) SetBlock(d,self->size-HDRSIZE,type,offset,value);

  else

  { for(m=1;m<n;m++)

    { par=PyTuple_GetItem(arg,m);

      if(!PyInt_Check(par))

      { PyErr_SetString(PyExc_TypeError,"Int Required");

        return 0;

      }

      q=(int)PyInt_AsLong(par);

      if(q<0||q>=self->nobjs)

      { PyErr_SetString(PyExc_ValueError,"Object out of range");

        return 0;

      }

      e=d;

      for(;q>0;q--) e=NEXT(e);

      if(e->type==type)

      { ((int*)(e))[offset]=value;

      }

      else if(e->type==drawfile_TYPE_GROUP)

             SetBlock((drawfile_object*)&e->data.group.objects,

                      e->size-GRPHDRSIZE,type,offset,value);

    }

  }

  Py_INCREF(Py_None);return Py_None;

}

static PyObject *DrawF_PathFill(PyDrawFObject *self,PyObject *arg)

{ return SetWord(self,arg,drawfile_TYPE_PATH,6);

}

static PyObject *DrawF_PathColour(PyDrawFObject *self,PyObject *arg)

{ return SetWord(self,arg,drawfile_TYPE_PATH,7);

}

static PyObject *DrawF_TextColour(PyDrawFObject *self,PyObject *arg)

{ return SetWord(self,arg,drawfile_TYPE_TEXT,6);

}

static PyObject *DrawF_TextBackground(PyDrawFObject *self,PyObject *arg)

{ return SetWord(self,arg,drawfile_TYPE_TEXT,7);

}

static struct PyMethodDef PyDrawF_Methods[]=

{

  { "render",(PyCFunction)DrawF_Render,1},

  { "save",(PyCFunction)DrawF_Save,1},

  { "path",(PyCFunction)DrawF_Path,1},

  { "text",(PyCFunction)DrawF_Text,1},

  { "ttext",(PyCFunction)DrawF_TText,1},

  { "fonttable",(PyCFunction)DrawF_FontTable,1},

  { "group",(PyCFunction)DrawF_Group,1},

  { "find",(PyCFunction)DrawF_Find,1},

  { "box",(PyCFunction)DrawF_Box,1},

  { "pathfill",(PyCFunction)DrawF_PathFill,1},

  { "pathcolour",(PyCFunction)DrawF_PathColour,1},

  { "textcolour",(PyCFunction)DrawF_TextColour,1},

  { "textbackground",(PyCFunction)DrawF_TextBackground,1},

  { NULL,NULL}		/* sentinel */

};

static int drawf_len(PyDrawFObject *b)

{ return b->nobjs;

}

static PyObject *drawf_concat(PyDrawFObject *b,PyDrawFObject *c)

{ int size=b->size+c->size-2*HDRSIZE;

  PyDrawFObject *p=new(size);

  drawfile_diagram *dd;

  drawfile_object *d;

  int n;

  if(!p) return NULL;

  dd=p->drawf;

  d=(drawfile_object*)((char*)dd+b->size);

  memcpy((char*)dd,(char*)(b->drawf),b->size);

  memcpy(d,(char*)(c->drawf)+HDRSIZE,c->size-HDRSIZE);

  p->nobjs=b->nobjs+c->nobjs;

  for(n=c->nobjs;n>0;n--)

  { extend(&(dd->bbox),&(d->data.path.bbox));

    d=NEXT(d);

  }

  return (PyObject*)p;

}

static PyObject *drawf_repeat(PyDrawFObject *b,Py_ssize_t i)

{ PyErr_SetString(PyExc_IndexError,"drawf repetition not implemented");

  return NULL;

}

static PyObject *drawf_item(PyDrawFObject *b,Py_ssize_t i)

{ PyDrawFObject *c;