// plot_toro.java  plot toro ortho with height colors

import java.io.*;
import java.*;
import java.awt.*;
import java.awt.event.*;

public class plot_toro extends Frame
{
  // mouse and display variables
  int xm, ym, bm; // mouse position 
  int height = 700;
  int width = 700;
  double xmin = -1.0; // typically determined from data
  double xmax = 1.0;
  double ymin = -1.0;
  double ymax = 1.0;
  double zmin = -1.0;
  double zmax = 1.0;
  double tmin = 0.0;
  double tmax = 6.28318;
  double pmin = 0.0;
  double pmax = 6.28318;
  double qmin = 0.0;
  double qmax = 6.28318;
  double amin = 0.0;
  double amax = 6.28318;
  double R = 0.5;
  double r = 0.2;
  double t = tmin;
  double p = pmin;
  double q = qmin; // rotation X Z
  double a = amin; // rotation Y Z
  int nx = 10;
  int ny = 10;
  int nz = 10;
  int nt = 16;
  int np = 16;
  int nq = 16;
  int na = 16;
  double hx = (xmax-xmin)/nx;
  double hy = (ymax-ymin)/ny;
  double hz = (zmax-zmin)/nz;
  double ht = (tmax-tmin)/nt;
  double hp = (pmax-pmin)/np;
  double hq = (qmax-qmin)/nq;
  double ha = (amax-amin)/na;
  double ya = 0.5;     //  orthographic typ 0.1 to 0.5 
  double xs = (0.666*(width-100))/(xmax-xmin);
  double ys = (0.666*(height-100))/(ymax-ymin);
  int xp, yp; // computed by toro

  public plot_toro()
  {
    System.out.println("plot_toro.java running");
    setTitle("plot_toro");
    setSize(width,height);
    setBackground(Color.white);
    setForeground(Color.black);
    addWindowListener(new WindowAdapter()
    {
      public void windowClosing(WindowEvent e)
      {
        System.exit(0);
      }
    });
    setVisible(true);
    this.addMouseListener (new mousePressHandler());

    do_plot();
  } // plot_toro constructor

  void do_plot()
  {
    // set up here

    // plot
    repaint(); // wait for mouse events
  } // end do_program


  class mousePressHandler extends MouseAdapter
  {
    public void mousePressed (MouseEvent e)
    {
      xm = e.getX();
      ym = e.getY();
      bm = e.getButton();
      // System.out.println("down xm="+xm+", ym="+ym+", bm="+bm); // debug

      if(in_rect(width-70, height-20, 30, 20))
      {
	// do something, run
	requestFocus();
        MyThread do_run = new MyThread(); // allocate do_run thread
        do_run.start(); // terminates
      }
      else if(in_rect(width-125, height-20, 40, 20))
      {
	// do something, nextq
	q = q + hq;
	requestFocus();
        repaint();
      }
      else if(in_rect(width-165, height-20, 40, 20))
      {
	// do something, nexta
        a = a + ha;
	requestFocus();
        repaint();
      }
      if(in_rect(width-35, height-20, 30, 20)) System.exit(0);
    } // end mousePressed 
  } // end mousePressHandler

  boolean in_rect(int x, int y, int w, int h)
  {
    return xm>x && xm<x+w && ym>y && ym<y+h;
  }
 
  public void paint(Graphics g)
  {
    double x, y, z;
    double z1, z2 , xr, yr, zr;
    int xp1, yp1, xp2, yp2; // line segment

    // buttons
    g.setColor(Color.red);
    g.fillRect(width-35,height-20,30,15);
    g.setColor(Color.black);
    g.drawString("quit", width-30, height-10);
    g.setColor(Color.green);
    g.fillRect(width-70,height-20,30,15);
    g.setColor(Color.black);
    g.drawString("run", width-65, height-10);
    g.setColor(Color.yellow);
    g.fillRect(width-125,height-20,40,15);
    g.setColor(Color.black);
    g.drawString("nextq", width-120, height-10);
    g.drawString("q="+Double.toString(q), width-160, height-55);
    g.setColor(Color.orange);
    g.fillRect(width-165,height-20,40,15);
    g.setColor(Color.black);
    g.drawString("nexta", width-160, height-10);
    g.drawString("a="+Double.toString(a), width-160, height-40);

    // axis labels
    g.drawString("xmin", 25, height-15);
    g.drawString("xmax", (int)(0.66*width-70), height-15);
    g.drawString("ymin", (int)(0.66*width-30), height-25);
    g.drawString("ymax", (int)(width-80), (int)(0.66*height+25));
    g.drawString("zmin", (int)(width-75), (int)(0.66*height+5));
    g.drawString("zmax", (int)(width-75), (int)(85));

    // height color scale
    int hgt = 210;
    int ncol = 10;
    for(int i=0; i<ncol; i++)
    {
      set_color(i, g);
      g.fillRect(20,hgt-20*i,50,15);
    }
    g.drawString("z value", 80, 40);
    g.drawString("xmin="+Double.toString(xmin), 250, 40);
    g.drawString("xmax="+Double.toString(xmax), 400, 40);
    g.drawString("R="+Double.toString(R)      , 550, 40);
    g.drawString("ymin="+Double.toString(ymin), 250, 55);
    g.drawString("ymax="+Double.toString(ymax), 400, 55);
    g.drawString("r="+Double.toString(r)      , 550, 55);
    g.drawString("zmin="+Double.toString(zmin), 250, 70);
    g.drawString("zmax="+Double.toString(zmax), 400, 70);

    // draw box, bottom
    set_color(0, g);
    ortho(xmin, ymin, zmin); xp1=xp; yp1=yp;
    ortho(xmax, ymin, zmin); xp2=xp; yp2=yp;
    g.drawLine(xp1, yp1, xp2, yp2);
    ortho(xmax, ymax, zmin); xp1=xp; yp1=yp;
    g.drawLine(xp2, yp2, xp1, yp1);
    ortho(xmin, ymax, zmin); xp2=xp; yp2=yp;
    g.drawLine(xp1, yp1, xp2, yp2);
    ortho(xmin, ymin, zmin); xp1=xp; yp1=yp;
    g.drawLine(xp2, yp2, xp1, yp1);
    set_color(ncol-1, g); // top
    ortho(xmin, ymin, zmax); xp1=xp; yp1=yp;
    ortho(xmax, ymin, zmax); xp2=xp; yp2=yp;
    g.drawLine(xp1, yp1, xp2, yp2);
    ortho(xmax, ymax, zmax); xp1=xp; yp1=yp;
    g.drawLine(xp2, yp2, xp1, yp1);
    ortho(xmin, ymax, zmax); xp2=xp; yp2=yp;
    g.drawLine(xp1, yp1, xp2, yp2);
    ortho(xmin, ymin, zmax); xp1=xp; yp1=yp;
    g.drawLine(xp2, yp2, xp1, yp1);
    // draw box edges with z value colors
    z1 = zmin;
    hz = (zmax-zmin)/(ncol);
    for(int i=0; i<ncol; i++)
    {
      z2 = z1+hz;
      set_color(i,g);
      ortho(xmin, ymin, z1); xp1=xp; yp1=yp;
      ortho(xmin, ymin, z2); xp2=xp; yp2=yp;
      g.drawLine(xp1, yp1, xp2, yp2);
      ortho(xmin, ymax, z1); xp1=xp; yp1=yp;
      ortho(xmin, ymax, z2); xp2=xp; yp2=yp;
      g.drawLine(xp1, yp1, xp2, yp2);
      ortho(xmax, ymax, z1); xp1=xp; yp1=yp;
      ortho(xmax, ymax, z2); xp2=xp; yp2=yp;
      g.drawLine(xp1, yp1, xp2, yp2);
      ortho(xmax, ymin, z1); xp1=xp; yp1=yp;
      ortho(xmax, ymin, z2); xp2=xp; yp2=yp;
      g.drawLine(xp1, yp1, xp2, yp2);
      z1 = z2;
    }

    // plot function or data with z value colors
    for(int i=0; i<nt; i++)
    {
      t = tmin+(double)i*ht;
      for(int j=0; j<np; j++)
      {
	p = pmin+(double)j*hp;
        x  = xt1(t, p);
        y  = yt1(t, p);
        z1 = zt1(t, p);
        xr = x*Math.cos(q)-z1*Math.sin(q); // rotate X Z
        z1 = z1*Math.cos(q)+x*Math.sin(q);
        yr = y*Math.cos(a)-z1*Math.sin(a); // rotate Y Z
        z1 = z1*Math.cos(a)+y*Math.sin(a);
        // System.out.println("t="+t+", p="+p+", x="+x+", y="+y+", z1="+z1);
        ortho(xr, yr, z1); xp1=xp; yp1=yp;
        x  = xt1(t+ht, p);
        y  = yt1(t+ht, p);
        z2 = zt1(t+ht, p);
        xr = x*Math.cos(q)-z2*Math.sin(q); // rotate X Z
        z2 = z2*Math.cos(q)+x*Math.sin(q);
        yr = y*Math.cos(a)-z2*Math.sin(a); // rotate Y Z
        z2 = z2*Math.cos(a)+y*Math.sin(a);
        ortho(xr, yr, z2); xp2=xp; yp2=yp;
        set_color(0, g);
        g.drawLine(xp1, yp1, xp2, yp2);
        x  = xt1(t, p+hp);
        y  = yt1(t, p+hp);
        z2 = zt1(t, p+hp);
        xr = x*Math.cos(q)-z2*Math.sin(q); // rotate X Z
        z2 = z2*Math.cos(q)+x*Math.sin(q);
        yr = y*Math.cos(a)-z2*Math.sin(a); // rotate Y Z
        z2 = z2*Math.cos(a)+y*Math.sin(a);
        ortho(xr, yr, z2); xp2=xp; yp2=yp;
        set_color(2, g);
        g.drawLine(xp1, yp1, xp2, yp2);
      } // j p
    } // i t
  } // end paint

  // orthographic projection, sets xp,yp
  void ortho(double x, double y, double z)
  {
    double x1, y1, z1, x2, y2, x3, y3;
    x1 = x-xmin;   // zero base
    y1 = y-ymin; 
    z1 = z-zmin;
    x2 = x1+y1*ya; // ortho z is up, y is back
    y2 = z1+y1*ya;
    x3 = x2*xs;    // scale 
    y3 = y2*ys;
    x3 = x3+25;
    y3 = height-y3-25; // away from boarder
    xp = (int)x3;
    yp = (int)y3;
  }

  void set_color(int i, Graphics g)
  {
    if(i<0) {g.setColor(Color.black); return;}
    if(i>9) {g.setColor(Color.gray); return; }
    switch(i)
    {
      case 0: g.setColor(Color.black); break;
      case 1: g.setColor(Color.pink); break; // should be brown
      case 2: g.setColor(Color.red); break;
      case 3: g.setColor(Color.orange); break;
      case 4: g.setColor(Color.yellow); break;
      case 5: g.setColor(Color.green); break;
      case 6: g.setColor(Color.blue); break;
      case 7: g.setColor(Color.magenta); break;
      case 8: g.setColor(Color.cyan); break;
      case 9: g.setColor(Color.gray); break;
    }
  } // end set_color
 
  class MyThread extends Thread
  {
    MyThread()
    {
    } // end MyThread constructor
    
    public void run()
    {
      try
      {
        q = qmin;
        a = amin;
	for(int i=0; i<nq; i++)
	{        
          repaint();
          sleep(250);
          q = q + hq;
	}
        q = qmin;
        a = amin;
	for(int i=0; i<na; i++)
	{        
          repaint();
          sleep(250);
          a = a + ha;
	}
        q = qmin;
        a = amin;
      } // end try
      catch(InterruptedException e)
      {
	System.out.println(" InterruptedException");
      }
    } // end Run
  } // end class MyThread


  // functions to plot, 4 cases of  x,y,z (R, r, t, p) all look same
  // area = (2Pi r)*(2Pi R)   volume = (Pi r*r)*(2PimR)
  double xt1(double tt, double pp) { return (R+r*Math.cos(tt))*Math.cos(pp); }
  double yt1(double tt, double pp) { return (R+r*Math.cos(tt))*Math.sin(pp); }
  double zt1(double tt, double pp) { return r*Math.sin(tt); }
  double xt2(double tt, double pp) { return (R+r*Math.cos(tt))*Math.sin(pp); }
  double yt2(double tt, double pp) { return (R+r*Math.cos(tt))*Math.cos(pp); }
  double zt2(double tt, double pp) { return r*Math.sin(tt); }
  double xt3(double tt, double pp) { return (R+r*Math.sin(tt))*Math.cos(pp); }
  double yt3(double tt, double pp) { return (R+r*Math.sin(tt))*Math.sin(pp); }
  double zt3(double tt, double pp) { return r*Math.cos(tt); }
  double xt4(double tt, double pp) { return (R+r*Math.sin(tt))*Math.sin(pp); }
  double yt4(double tt, double pp) { return (R+r*Math.sin(tt))*Math.cos(pp); }
  double zt4(double tt, double pp) { return r*Math.cos(tt); }

  public static void main(String[] args)
  {
    new plot_toro();
  }
} // end class  plot_toro.java