#include <tcl.h>

#include <tk.h>

#include <stdlib.h>

#include <string.h>

#include "link.h"

#include "tcl_interface.h"

#include "stdio.h"

#include "Page.h"

extern Page* global_page;

extern Page* active_page;

extern Page* zoned_page;

extern Tcl_Interp* TCL_ip;

extern Tk_Window main_window;

extern double SCALE_FACTOR;

extern int DISPLAY_SPELLING_MISTAKES;

extern TclMode mode;

static int page_currently_open = 0;

Component * curCompSelected;

Component * prevCompSelected;

void scale(int& coordinate)

{

  coordinate = (int)(coordinate * SCALE_FACTOR);

}

void scale(int& coordinate, double scaleFactor)

{

 coordinate = (int)(coordinate * scaleFactor);

}

int error(char* s)

{

/* would like to make this take var num args */

  printf("Error: %s", s);

}

int quit_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  printf("Thank you for using OCRchie.\n");

  exit(0);

}

int get_linenum_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  // argv[1] - x coord   argv[2] - y coord

  global_page->get_linenum(atoi (argv[1]), atoi (argv[2]));

  return TCL_OK;

}

int add_equation_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  // argv[1]  starting line of equation, [2] startcol [3] endline [4] endcol

  global_page->addEquation(atoi (argv[1]), atoi (argv[2]), atoi(argv[3]), 

			   atoi(argv[4]));

  return TCL_OK;

}

int delete_equation_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  global_page->deleteEquation(atoi(argv[1]), atoi(argv[2]));

  return TCL_OK;

}

int auto_zone_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  ((ZonedPage *) global_page)->autoZone(atoi(argv[1]), atoi(argv[2]));

  return TCL_OK;

}

int add_zone_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  // first we need to unscale from scaled immage

  int ulx = (int)(atof(argv[1])/ZONING_SCALE_FACTOR);

  int uly = (int)(atof(argv[2])/ZONING_SCALE_FACTOR);

  int lrx = (int)(atof(argv[3])/ZONING_SCALE_FACTOR);

  int lry = (int)(atof(argv[4])/ZONING_SCALE_FACTOR);

 Point ul = Point(ulx,uly);

 Point lr = Point(lrx, lry);

 if ( ul != lr)

   {

     Zone * newzone = new Zone(ul,lr);

     newzone->buildPage((ZonedPage *) global_page);

     ((ZonedPage *)global_page)->zones()->Append(newzone);

   }

  return TCL_OK;

}

int remove_zone_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  int x =  (int)(atof(argv[1])/ZONING_SCALE_FACTOR);

  int y =  (int)(atof(argv[2])/ZONING_SCALE_FACTOR);

  Zones * z = ((ZonedPage *) global_page)->zones();

  z->removeElement(z->findZone(x,y));

  return TCL_OK;

}

int select_comp_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  prevCompSelected = curCompSelected;

  //argv[1] is x coord argv[2] is y

  curCompSelected = global_page->compAt(Point(atoi(argv[1]), atoi(argv[2])));

  if (curCompSelected == NULL) 

    {

      docommand("set validComponent 0");

    }

  else

    {

      docommand("set validComponent 1");

      docommand("set curCompId %s",curCompSelected->fasciiId);

      docommand("puts \"curCompId: $curCompId\"");

    }

  return TCL_OK;

}

int join_comp_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if (prevCompSelected != NULL && curCompSelected != NULL &&

      prevCompSelected != curCompSelected && 

      global_page->get_linenum(prevCompSelected) == 

      global_page->get_linenum(curCompSelected))

    {

      global_page->join(prevCompSelected,curCompSelected);

      printf("Joined. cur %u prev %u \n", (long) curCompSelected,

	   (long) prevCompSelected);

      prevCompSelected = NULL;

      curCompSelected = NULL;

    }

  else

    printf("not Joined. cur %u prev %u\n", (long) curCompSelected,

	   (long) prevCompSelected);

  {

      prevCompSelected = NULL;

      curCompSelected = NULL;

  }

  return TCL_OK;

}

int learn_comp_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  learn(curCompSelected, argv[1], 256);

  return TCL_OK;	

}

int split_comp_horiz_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  global_page->thinnestHorizontalSplit(curCompSelected);

  return TCL_OK;

}

int get_skew_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  get_skew(global_page->rmap());

  return TCL_OK;

}

int deskew_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if(DESKEW_METHOD != -1)

    {

      if(global_page->deskew(DESKEW_METHOD))

	global_page->rmap()->display_intervals("black");

      return TCL_OK;

    }

  else

    return TCL_OK;

}

int display_intervals_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if (ac == 1)

    global_page->rmap()->display_intervals("black");

  else

    global_page->rmap()->display_intervals(argv[1], atof(argv[2]), "black");

  return TCL_OK;

}

int page_open_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  mode = REGULAR;

  if(ac != 2)

    return TCL_ERROR;

  printf("Opening %s\n", argv[1]);

  if(page_currently_open == 1)

    {

      /* should print some message about closing the current one first */

      return TCL_OK;

    }

  global_page = new ZonedPage;

  if(global_page->readMap(argv[1]) != VALID)

    interp->result = "0";

  else

    {

         interp->result = "1";

    }

  active_page = global_page;

  page_currently_open = 1;

  return TCL_OK;

}

int zoned_page_open_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  mode = ZONING;

  if(ac != 2)

    return TCL_ERROR;

  printf("Opening %s\n", argv[1]);

  if(page_currently_open == 1)

    {

      /* should print some message about closing the current one first */

      return TCL_OK;

    }

  global_page = new ZonedPage;

  if(global_page->readMap(argv[1]) != VALID)

    interp->result = "0";

  else

    {

         interp->result = "1";

    }

  zoned_page = global_page;

  page_currently_open = 1;

  return TCL_OK;

}

int extract_comp_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

 global_page->extractComponents(MinHorizSeparation);

 return TCL_OK;

}

int get_page_height_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  sprintf(interp->result, "%d", global_page->get_height());

  /*  printf("Interpereter height = %s\n", interp->result); */

  return TCL_OK;

}

int get_page_width_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  sprintf(interp->result, "%d",  global_page->get_width());

  /* printf("Interpereter width = %s\n", interp->result); */

  return TCL_OK;

}

int zoom_in_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if(ac == 1)

    {

      SCALE_FACTOR = SCALE_FACTOR * 2;

      global_page->rmap()->display_intervals("black");

    }

  else

    {

      //argv[1] window argv[2] scalefactor

      global_page->rmap()->display_intervals(argv[1], atof(argv[2]),"black");

    }

  return TCL_OK;

}

int zoom_out_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if (ac == 1)

    {

      SCALE_FACTOR = SCALE_FACTOR * 0.5;

      global_page->rmap()->display_intervals("black");

    }

  else

      global_page->rmap()->display_intervals(argv[1], atof(argv[2]),"black");

  return TCL_OK;

}

int deallocate_page_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if(page_currently_open == 0)

    return TCL_OK;  /* don't do anything if their isn't anything open */

  delete global_page;

  page_currently_open = 0;

  return TCL_OK;

}

int learn_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  // argv1 is tif argv2 is txt and argv3 is a bool for word synchronization

  printf("Learning from %s and %s sych %d\n", argv[1], argv[2], atoi(argv[3]));

  learn(argv[1], argv[2], atoi(argv[3]));

  return TCL_OK;

}

int learn_page_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  // argv1 is txt and argv2 is a bool for word synchronization

  printf("Learning from %s and %s sych %d\n", argv[1], argv[2], atoi(argv[3]));

  learn(global_page, argv[1], atoi(argv[2]));

  return TCL_OK;

}

int learn_data_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  printf("Learning data from %s\n", argv[1]);

  readLearnedGroups(argv[1]);

  return TCL_OK;

}

int write_word_pos_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  global_page->writeWordPos(argv[1]);

  return TCL_OK;

}

int write_wordbox_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  // arguments are output file, x offset,  y offset and WbxEquationsOnly variable

  global_page->writeWordbox(argv[1], atoi(argv[2]), atoi(argv[3]), atoi(argv[4]));

  return TCL_OK;

}

int write_equations_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  // arguments are output file, linenumber offset

  global_page->writeEquations(argv[1], atoi(argv[2]));

  return TCL_OK;

}

int write_learned_chars_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  writeLearnedGroups(argv[1]);

  return TCL_OK;

}

int find_lines_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  deskew_cmd(clientData, interp, ac, argv);

  /*  printf("Calling setlines\n"); */

  global_page->setLines();

  return TCL_OK;

}

int recognize_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  /* just in case someone has left something sitting around */

  docommand(".main_window.edit_window.text_part delete 1.0 end");

  docommand("set COLORED_WORDS {}");

   /* printf("Calling extractComponents\n"); */

  if(global_page->components() == NULL) 

    global_page->extractComponents(MinHorizSeparation);

  /* printf("Calling recognize\n"); */

  global_page->recognize();

  /* printf("Calling extractwords\n"); */

  global_page->extractWords();

  if(SPELLCHECK)

    {

      /* printf("Spellchecking\n"); */

      global_page->spellcheck();

    }	     

  /* printf("Calling sendwordstotcl\n"); */

  global_page->send_words_to_tcl();

  return TCL_OK;

}

int find_lines_and_recognize_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  /* just in case someone has left something sitting around */

  docommand(".main_window.edit_window.text_part delete 1.0 end");

  docommand("set COLORED_WORDS {}");

  /* printf("Calling deskew\n"); */

  deskew_cmd(clientData, interp, ac, argv);

  /*  printf("Calling setlines\n"); */

  global_page->setLines();

  /* printf("Calling extractComponents\n"); */

  global_page->extractComponents(MinHorizSeparation);

  /* printf("Calling recognize\n"); */

  global_page->recognize();

  /* printf("Calling extractwords\n"); */

  global_page->extractWords();

  if(SPELLCHECK)

    {

      /* printf("Spellchecking\n"); */

      global_page->spellcheck();

    }	     

  /* printf("Calling sendwordstotcl\n"); */

  global_page->send_words_to_tcl();

  return TCL_OK;

}

int switch_to_active_page_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if (active_page != NULL)

    global_page = active_page;

  return TCL_OK;

}

int switch_to_zoned_page_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  if (zoned_page != NULL)

    global_page = zoned_page;

  return TCL_OK;

}

int set_active_page_cmd(ClientData clientData, Tcl_Interp *interp, int ac, char** argv)

{

  int x = atoi(argv[1]);

  int y = atoi(argv[2]);

  active_page = ((ZonedPage *) zoned_page)->activate(x,y);

  if (active_page != NULL)

    {

      docommand(".main_window.edit_window.text_part delete 1.0 end");

      docommand("set COLORED_WORDS {}");

      docommand("focus .main_window.display");

      docommand("grab current .main_window.display");

      global_page = active_page;

      docommand("DISPLAY_INTERVALS .main_window.display.work_space $SCALE_FACTOR");

      docommand("FIND_LINES");

      docommand("grab release .main_window.display"); 

      docommand("zone_message \"Active Zone at (%d,%d)\"", x, y);

    }

  else

    {

      docommand("zone_message \"No zone found here\"");

    }

  return TCL_OK;

}

static int

vdocommand1(char* s)

{

/* final function called to do a tcl docommand */

  int code;

  code = Tcl_Eval(TCL_ip, s);      

  if (code == TCL_ERROR)

    error(TCL_ip->result);

  return code;

}

void update()

{

  Tk_DoOneEvent(TK_ALL_EVENTS);   

}

static int

vdocommand(int record, char* fmt, va_list args)

{

/* helper for docommand */

  char buf[4097];

  int code;

  vsprintf(buf, fmt, args);

  if (strchr(buf, '\?'))

    error("Huh?");

  code = vdocommand1(buf);

  return code;

}

int

docommand(char* fmt, ...)

{

/* do a tcl command, var number of args */

  va_list args;

  va_start(args, fmt);

  vdocommand(0, fmt, args);

  va_end(args);

}

static int

vset_status1(char* s)

{

/* final function called to do a tcl docommand */

  int code;

	code = Tcl_Eval(TCL_ip, s);      

	if (code == TCL_ERROR)

	  error(TCL_ip->result);

	return code;

}

static int

vset_status(int record, char* fmt, va_list args)

{

/* helper for docommand */

  char buf[4097];

  char newbuf[4097];

  int code;

  vsprintf(buf, fmt, args);

  sprintf(newbuf, ".main_window.button_bar.msg configure -text \"%s\"", buf); 

  if (strchr(buf, '\?'))

    error("Huh?");

  code = vset_status1(newbuf);

  return code;

}

int set_status(char* fmt, ...)

{

  va_list args;

  va_start(args, fmt);

  vset_status(0, fmt, args);

  va_end(args);

  docommand("update");

}

int mispelled(char* word)

{