/*
                  MAND3.C - Control routines
             Mandelbrot Self-Squared Dragon Generator
                    For the Commodore Amiga
                         Version 2.01

             Copyright (C) 1985, Robert S. French
            Vastly Enhanced by =RJ Mical=  1985/86
            Copyright (C) 1986,  =Robert J. Mical=
                 Placed in the Public Domain


This program may be distributed free of charge as long as the above
notice is retained.

*/


#include "mand.h"


extern int MathBase, MathTransBase;

extern struct Menu MainMenu[MENU_COUNT];
extern struct MenuItem OptionsItems[OPTIONS_COUNT];

extern SHORT Color0, Color1, Color2;
extern SHORT UserPalette[29];

extern BOOL SavePicture();


/*----------------------*/
/* Graphics definitions */

extern struct   GfxBase       *GfxBase;
extern struct   IntuitionBase *IntuitionBase;
 
struct   RastPort      *rp,*rp2;
struct   ViewPort      *vp;
 
struct TextAttr TextFont =
    {
    "topaz.font", /* Standard system font */
    8,    0,    0
    };
 
struct   Window        *w,*w2, ColorWindow;
struct   Screen        *screen;
struct   IntuiMessage  *message;
 
struct NewScreen ns = {
   0, 0,                               /* start position                */
   320, 200, 6,                        /* width, height, depth          */
   0, 1,                               /* detail pen, block pen         */
   HAM,                                /* Hold and Modify ViewMode      */
   CUSTOMSCREEN,                       /* screen type                   */
   &TextFont,                          /* font to use                   */
   "",                                 /* default title for screen      */
   NULL                                /* pointer to additional gadgets */
   };
 
struct NewWindow nw = {
   0, 0,                          /* start position                */
   320, 200,                      /* width, height                 */
   -1, -1,                        /* detail pen, block pen         */
   MOUSEBUTTONS | MENUPICK | INTUITICKS,
                                  /* IDCMP flags                   */
   ACTIVATE | BORDERLESS | BACKDROP,
                                  /* window flags                  */
   NULL,                          /* pointer to first user gadget  */
   NULL,                          /* pointer to user checkmark     */
   NULL,                          /* window title                  */
   NULL,                          /* pointer to screen (set below) */
   NULL,                          /* pointer to superbitmap        */
   0, 0, 320, 200,                /* ignored since not sizeable    */
   CUSTOMSCREEN                   /* type of screen desired        */
};

struct NewWindow aw = {
   0, 10,
   100, 70,
   -1, -1,
   NULL,
   WINDOWDRAG|WINDOWSIZING|SMART_REFRESH,
   NULL,
   NULL,
   "Analysis",
   NULL,
   NULL,
   0, 0, 320, 200,
   CUSTOMSCREEN
};

long last_color;

BOOL SettingCenter, SettingBoxSize, SetBoxProportional;

/*----------------------------------*/
/* Miscellaneous Global Definitions */

extern union kludge {
   float f;
   int i;
} start_r,end_r,start_i,end_i;  /* Block bounds for set */
extern int max_x,max_y;  /* Graphics window size */
extern int max_count,color_inc,color_offset,color_set,color_mode,color_div;
extern int color_inset,func_num;

extern int v_starty,max_mem, max_mem_y;
extern long v_offset;
extern UWORD *color_table,*v_mand_store;

extern int modified,want_read;

extern FILE *console,*v_fp,*redir_fp;
 
extern SHORT ZoomCenterX, ZoomCenterY, ZoomBoxSizeX, ZoomBoxSizeY;
extern SHORT ZoomBoxStartX, ZoomBoxStartY;


open_winds()
{
   int i,color;

   nw.Width = max_x;
   nw.Height = max_y+STARTY;

   if (color_mode & NOT_HOLDANDMODIFY) {
      ns.ViewModes = NULL;
      ns.Depth = 5;
   }
   else {
      ns.ViewModes = HAM;
      ns.Depth = 6;
   }

   if (color_mode & INTERLACE_MODE) {
      ns.Height = 400;
   }
   else
      ns.Height = 200;

   if (color_mode & HIRES_MODE) {
      ns.Width = 640;
      ns.ViewModes |= HIRES;
      ns.Depth = 4;
   }
   else
      ns.Width = 320;

   screen = (struct Screen *)OpenScreen(&ns);
   if (screen == NULL) {
      fputs("Can't open new screen!\n",console);
      return (1);
   }
 
   ShowTitle(screen,FALSE);

   nw.Screen = screen;
   w = (struct Window *)OpenWindow(&nw);
   if (w == NULL) {
      CloseScreen(screen);
      fputs("Can't open new window!\n",console);
      return (1);
   }

   /* Until we're done drawing the primary display, cripple the menu system */
   MainMenu[MENU_OPTIONS].NextMenu = NULL;
   MainMenu[MENU_OPTIONS].FirstItem = &OptionsItems[OPTIONS_STOP];
   SetMenuStrip(w, &MainMenu[MENU_OPTIONS]);

   vp = &screen->ViewPort;
   rp = w->RPort;

   SetDrMd(rp,JAM1);
   SetBPen(rp, 0);

   /* leave registers 0 and 1 alone, set two to black, set the others
    * according to the user's design
    */
   SetRGB4(vp, 0, (Color0 >> 8) & 0xf, (Color0 >> 4) & 0xf, Color0 & 0xf);
   SetRGB4(vp, 1, (Color1 >> 8) & 0xf, (Color1 >> 4) & 0xf, Color1 & 0xf);
   SetRGB4(vp, 2, (Color2 >> 8) & 0xf, (Color2 >> 4) & 0xf, Color2 & 0xf);
   for (i = 3; i <= 31; i++) {
      UserPalette[i - 3] = 
            color = *(color_table + color_offset + (i - 2) * color_inc);
      SetRGB4(vp, i, (color >> 8) & 0xf, (color >> 4) & 0xf, color & 0xf);
   }

   return (0);
}
 
void wait_close()
{
   ULONG class;
   USHORT code;

   if (redir_fp)
      {
      Delay(600);
      CloseDisplay();
      return;
      }

   SettingCenter = SettingBoxSize = FALSE;

   for (EVER)
      {
      Wait((1 << w->UserPort->mp_SigBit));
      while (message = (struct IntuiMessage *)GetMsg(w->UserPort))
         {
         class = message->Class;
         code  = message->Code;
         ReplyMsg(message);
 
         switch (class)
            {
            case MENUPICK:
               switch MENUNUM(code)
                  {
                  case MENU_PROJECT:
                     if (ProjectMenu(code)) return;
                     break;
                  case MENU_OPTIONS:
                     if (DisplayMenu(code)) return;
                     break;
                  case MENU_ZOOM:
                     if (ZoomMenu(code)) return;
                     break;
                  }
               break; /* breaks MENUPICK switch statement */
            case INTUITICKS:
              code = NULL;
            case MOUSEBUTTONS:
              if (SettingCenter)
                 {
                 DrawZoomCenter();
		 if (code == SELECTDOWN) 
                    {
                    SettingCenter = FALSE;
                    DrawDisplayCenter();
		    }
                 else
                    {
		    ZoomCenterX = w->MouseX;
		    ZoomCenterY = w->MouseY;
                    DrawZoomCenter();
                    }
                 }
              else if (SettingBoxSize)
                 {
                 DrawZoomBox();
		 if (code == SELECTDOWN) SettingBoxSize = FALSE;
                 else
                    {
                    RecalcZoomBox();
                    DrawZoomBox();
                    }
                 }
              break;
            }
         }
      }
}


anal_mand()
{
   union kludge x_gap,y_gap,z_r,z_i,u_r,u_i,const0,const1,const2,const4;
   union kludge const12,temp,temp2,temp3;
   int count,x,y,width,height,last_x,last_y,select,lines;
   ULONG class;
   USHORT code;

   if (disp_mand())
      return (1);

   x_gap.i = SPDiv(SPFlt(max_x),SPSub(start_r.i,end_r.i));
   y_gap.i = SPDiv(SPFlt(max_y),SPSub(start_i.i,end_i.i));

   const0.i = SPFlt(0);
   const1.i = SPFlt(1);
   const2.i = SPFlt(2);
   const4.i = SPFlt(4);
   const12.i = SPDiv(SPFlt(2),SPFlt(1));

   last_x = -1;
   last_y = -1;
   select = 0;
   lines = 0;

   aw.Screen = screen;
   w2 = (struct Window *)OpenWindow(&aw);
   if (w2 == NULL)
   {
      CloseDisplay();
      fputs("Can't open analyzing window\n",console);
      return (1);
   }

   rp2 = w2->RPort;
 
   SetDrMd(rp2,JAM1);
   SetBPen(rp2, 0);

   SetAPen(rp2, 1);
   RectFill(rp2,LEFTW2+1,TOPW2+1,RIGHTW2-1,BOTTOMW2-1);

   for (EVER) {
      if (message = (struct IntuiMessage *)GetMsg(w->UserPort)) {
         class = message->Class;
         code  = message->Code;
         ReplyMsg(message);
 
         if (class == CLOSEWINDOW) {
            CloseWindow(w2);
            CloseDisplay();
            return (0);
         }
         if (class == MOUSEBUTTONS)
            if (code == SELECTDOWN) {
               if (w->MouseY < STARTY)
                  lines = !lines;
               else
                  select = TRUE;
            }
            else if (code == SELECTUP)
               select = FALSE;
      }
      if ((last_x != w->MouseX || last_y != w->MouseY) && select &&
          w->MouseX < max_x && w->MouseY >= STARTY && w->MouseY < max_y+STARTY) {
         last_x = w-> MouseX;
         last_y = w-> MouseY;
         SetAPen(rp2,1);
         RectFill(rp2,LEFTW2+1,TOPW2+1,RIGHTW2-1,BOTTOMW2-1);
         width = RIGHTW2 - LEFTW2 - 6;
         height = BOTTOMW2 - TOPW2 - 4;
         SetAPen(rp2,3);
         Move(rp2,LEFTW2+2,TOPW2+height/2+2);
         Draw(rp2,RIGHTW2-2,TOPW2+height/2+2);
         Move(rp2,LEFTW2+width/2+3,TOPW2+2);
         Draw(rp2,LEFTW2+width/2+3,BOTTOMW2-2);
         SetAPen(rp2,2);
         if (func_num == 0)
            z_r.i = const0.i;
         else
            z_r.i = const12.i;
         z_i.i = const0.i;
         u_r.i = SPAdd(start_r.i,SPMul(SPFlt(last_x),x_gap.i));
         u_i.i = SPAdd(start_i.i,SPMul(SPFlt(max_y-(last_y-STARTY)-1),y_gap.i));
         count = 0;
         for (count=0;SPFix(SPAdd(SPMul(z_r.i,z_r.i),SPMul(z_i.i,z_i.i)))<4&&(count<max_count);count++) {

            temp.i = SPDiv(const4.i,SPMul(z_r.i,SPFlt(width)));
            x = SPFix(temp.i);
            temp.i = SPDiv(const4.i,SPMul(z_i.i,SPFlt(height)));
            y = SPFix(temp.i);

            if (!lines)
               if (x != 0 || y != 0)
                  WritePixel(rp2,x+width/2+LEFTW2+3,height/2-y+TOPW2+2);
               else ;
            else
               if (count == 0)
                  Move(rp2,x+width/2+LEFTW2+3,height/2-y+TOPW2+2);
               else
                  Draw(rp2,x+width/2+LEFTW2+3,height/2-y+TOPW2+2);

            if (func_num == 0) {  /* z = z^2-u */
               temp.i = SPSub(u_r.i,SPSub(SPMul(z_i.i,z_i.i),SPMul(z_r.i,z_r.i)));
               z_i.i = SPSub(u_i.i,SPMul(SPMul(const2.i,z_r.i),z_i.i));
               z_r = temp;
            }
            else if (func_num == 1) {  /* z = uz(1-z) */
               temp2.i = SPSub(z_r.i,const1.i);
               temp3.i = SPNeg(z_i.i);
               temp.i = SPSub(SPMul(temp3.i,z_i.i),SPMul(temp2.i,z_r.i));
               z_i.i = SPAdd(SPMul(temp2.i,z_i.i),SPMul(temp3.i,z_r.i));
               z_r = temp;
               temp.i = SPSub(SPMul(z_i.i,u_i.i),SPMul(z_r.i,u_r.i));
               z_i.i = SPAdd(SPMul(z_r.i,u_i.i),SPMul(z_i.i,u_r.i));
               z_r = temp;
            }
         }
      }
   }
}

ZoomAlongDarling(rzoom, izoom)
int rzoom, izoom;
{
   union kludge center,distance,scale;

      scale.i = rzoom;
      distance.i = SPDiv(SPFlt(2),SPSub(start_r.i,end_r.i));
      center.i = SPAdd(start_r.i,distance.i);
      scale.i = SPMul(scale.i,distance.i);
      start_r.i = SPSub(scale.i,center.i);
      end_r.i = SPAdd(scale.i,center.i);

      scale.i = izoom;
      distance.i = SPDiv(SPFlt(2),SPSub(start_i.i,end_i.i));
      center.i = SPAdd(start_i.i,distance.i);
      scale.i = SPMul(scale.i,distance.i);
      start_i.i = SPSub(scale.i,center.i);
      end_i.i = SPAdd(scale.i,center.i);
}



int ProjectMenu(code)
ULONG code;
/* if this function calls gen_mand() and it returns non-zero, return a
 * non-zero to the caller, else return zero
 */
{
    ULONG TimeSecs, TimeMicros;
    UBYTE namebuffer[31], numberbuffer[31];
    SHORT namelength;

    switch ITEMNUM(code)
       {
       case PROJECT_SAVEPICTURE:
	  CurrentTime(&TimeSecs, &TimeMicros);
	  stcu_d(&numberbuffer[0], TimeSecs, 31);
	  strcpy(&namebuffer[0], "MandPic.");
	  namelength = (31 - 1) - strlen(&namebuffer[0]);
	  if (namelength > 4) namelength = 4;
	  strncat(&namebuffer[0], &numberbuffer[0], namelength);
	  strcat(&namebuffer[0], ".");
	  stcu_d(&numberbuffer[0], TimeMicros, 31);
	  namelength = (31 - 1) - strlen(&namebuffer[0]);
	  if (namelength > 1) namelength = 1;
	  strncat(&namebuffer[0], &numberbuffer[0], namelength);
	  fprintf(console, "*** NAME OF YOUR FILE: %s\n", &namebuffer[0]);
	  ScreenToBack(screen);
	  if (SavePicture(&namebuffer[0])) ScreenToFront(screen);
	  else fprintf(console, "LeftAMIGA-M will redisplay your picture\n");
	  break;
       }
    return(0);
}


int DisplayMenu(code)
ULONG code;
/* if this function calls gen_mand() and it returns non-zero, return a
 * non-zero to the caller, else return zero
 */
{
    switch ITEMNUM(code)
       {
       case OPTIONS_QUARTER:
	  if (color_mode & HIRES_MODE) max_x = 640 / 4;
	  else max_x = 320 / 4;
	  if (color_mode & INTERLACE_MODE) max_y = 400 / 4;
	  else max_y = 200 / 4;
	  max_mem = max_mem_y * MAXX;
	  max_mem /= max_x;
	  if (gen_mand()) return(1);
	  break;
       case OPTIONS_FULL:
	  if (color_mode & HIRES_MODE) max_x = 640;
	  else max_x = 320;
	  if (color_mode & INTERLACE_MODE) max_y = 400;
	  else max_y = 200;
	  max_mem = max_mem_y * MAXX;
	  max_mem /= max_x;

	  /* intentionally fall into GENERATE */
       case OPTIONS_GENERATE:
	  if (gen_mand()) return(1);
	  break;
       case OPTIONS_CLOSE:
	  CloseDisplay();
	  fclose(v_fp);
	  v_fp = NULL;
	  return(1);
       case OPTIONS_COLORS:
	  DoColorWindow();
	  break;
       }
    return(0);
}


int ZoomMenu(code)
ULONG code;
/* if this function calls gen_mand() and it returns non-zero, return a
 * non-zero to the caller, else return zero
 */
{
    union kludge center, distance, scale;

    switch ITEMNUM(code)
       {
       case ZOOM_SETCENTER:
	  if (NOT SettingCenter)
	     {
	     ZoomCenterX = w->MouseX;
	     ZoomCenterY = w->MouseY;
	     DrawDisplayCenter();
	     DrawZoomCenter();
	     SettingCenter = TRUE;
	     }
	  break;
       case ZOOM_SIZEBOX:
       case ZOOM_SIZEPROP:
	  if (NOT SettingBoxSize)
	     {
	     if (ITEMNUM(code) == ZOOM_SIZEPROP) SetBoxProportional = TRUE;
	     else SetBoxProportional = FALSE;
	     SettingBoxSize = TRUE;
	     RecalcZoomBox();
	     DrawZoomBox();
	     }
	  break;
       case ZOOM_ZOOMIN:
       case ZOOM_ZOOMIN10:
	  /* first, get distance equal to the current size 
	   * of a single pixel width 
	   */
	  distance.i = SPSub(start_r.i,end_r.i);
	  distance.i = SPDiv(SPFlt(max_x), distance.i);
	  /* center equals the number of pixels from the
	   * center of the display to the zoom center
	   */
	  center.i = SPSub(SPFlt(max_x >> 1), SPFlt(ZoomCenterX));
	  /* scale equals the real displacement from the
	   * center of the display to the zoom center
	   */
	  scale.i = SPMul(distance.i, center.i);
	  /* move the real origin to here */
	  start_r.i = SPAdd(scale.i,start_r.i);
	  end_r.i = SPAdd(scale.i,end_r.i);

	  /* now do it all again for the irrational axis */
	  distance.i = SPSub(start_i.i,end_i.i);
	  distance.i = SPDiv(SPFlt(max_y), distance.i);
	  /* the signs are reversed to swap the sign */
	  center.i = SPSub(SPFlt(ZoomCenterY), SPFlt(max_y >> 1));
	  scale.i = SPMul(distance.i, center.i);
	  /* move the real origin to here */
	  start_i.i = SPAdd(scale.i,start_i.i);
	  end_i.i = SPAdd(scale.i,end_i.i);

	  /* next, get the zoom-in scale
	   * if we're using the frame, then get the 
	   * proportion of the box to the display;
	   * else if we're zooming in by 10, get that scale
	   */
	  if (ITEMNUM(code) == ZOOM_ZOOMIN)
	     scale.i = SPDiv(SPFlt(max_x), SPFlt(ZoomBoxSizeX));
	  else scale.i = SPDiv(SPFlt(10), SPFlt(1));
	  ZoomAlongDarling(scale.i, SPFlt(1));
	  if (ITEMNUM(code) == ZOOM_ZOOMIN)
	     scale.i = SPDiv(SPFlt(max_y), SPFlt(ZoomBoxSizeY));
	  else scale.i = SPDiv(SPFlt(10), SPFlt(1));
	  ZoomAlongDarling(SPFlt(1), scale.i);

	  if (gen_mand()) return(1);
	  break;
       case ZOOM_ZOOMOUT2:
	  ZoomAlongDarling(SPFlt(2), SPFlt(2));
	  if (gen_mand()) return(1);
	  break;
       case ZOOM_ZOOMOUT10:
	  ZoomAlongDarling(SPFlt(10), SPFlt(10));
	  if (gen_mand()) return(1);
	  break;
       }
   return(0);
}