import javax.swing.*;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.FlowLayout;
import java.awt.Dimension;
import edu.jsu.leathrum.mathlets.shared.*;

public class Grapher3dWires extends AMathlet {

private class GraphPanel extends AGraphPanel3dWires {

private class GraphCanvas extends AGraphCanvas3dWires {

  GraphModelWires model = new GraphModelWires();

  public GraphCanvas(int initialWidth, int initialHeight) {
    super(initialWidth, initialHeight);
  }

  private Dimension size;
  private double xrez, yrez;

  private void drawCurve(double a, double b, double cc,
    			 boolean xdir) {
    double x, y, z, q;
    boolean connected=false;
    double orient = (cc<b)?-1.0:1.0;
    double step = 2*orient*((xdir)?xrez:yrez);
    
    x=y=z=0.0;
    connected=false;
    for (q=b; orient*q<orient*cc; q+=step) {
      x=(xdir)?q:a; y=(xdir)?a:q;
      z=evaluatef(x,y);
      if (Double.isNaN(z) || Double.isInfinite(z) ||
    	  (z > c.zmax) || (z < c.zmin)) connected=false;
      else {
    	if (connected) model.add(x, y, z);
    	else {
    	  connected=true;
              model.newpath();
              model.add(x, y, z);
        }
      }
    }
  }
    
  protected void drawNew() {
    double x, y;
    
    size = this.getPreferredSize();
    xrez=(c.xmax-c.xmin)/((double) size.width);
    yrez=(c.ymax-c.ymin)/((double) size.width);
    model = new GraphModelWires();
    for (x=c.xmin; x<c.xmax+xrez; x+=xgrid) {
      drawCurve(x, c.ymin, c.ymax, false);
    }
    for (y=c.ymin; y<c.ymax+yrez; y+=ygrid) {
      drawCurve(y, c.xmin, c.xmax, true);
    }
  }
    
   protected void paintModel(Graphics2D g) {
     g.setPaint(Color.blue);
     model.paint(g);
   }
}

  GraphCanvas canvas;

  public GraphPanel () {
    super();
    canvas = new GraphCanvas(HEIGHT,WIDTH);
    initPanel(canvas);
  }
}

private class ParserInputSet extends AParserInputSet {

  public ParserInputSet(String namestr, int size) {
    super(namestr, size);
  }
  
  public ParserInputSet(String namestr, int size, boolean b, Color c) {
    super(namestr, size, b, c);
  }
  
  double x, y;

  protected void setVars(double[] v) {
    x = v[0]; y = v[1];
  }

  protected double getVar(int id) {
    switch (id) {
    case Parser.VARX:
      return x;
    case Parser.VARY:
      return y;
    default:
      return Double.NaN;
    }
  }
}

  GraphPanel graph = new GraphPanel();
  ParserInputSet field = new ParserInputSet("z=f(x,y)=", 25);
  JButton compButton = new JButton("graph");
  JButton clearButton = new JButton("clear");
  Box b1, b2;

  public Grapher3dWires() {
    this.getContentPane().setBackground(Color.white);
    this.getContentPane().setLayout(new FlowLayout());
    this.getContentPane().add(graph);

    b1 = new Box(BoxLayout.Y_AXIS);
    b1.add(field);

    b2 = new Box(BoxLayout.X_AXIS);
    b2.add(Box.createHorizontalGlue());
    b2.add(compButton);
    compButton.setOpaque(false);
    b2.add(clearButton);
    clearButton.setOpaque(false);
    b2.add(Box.createHorizontalGlue());
    b1.add(b2);

    this.getContentPane().add(b1);
  }

  public void localinit() {
    field.setName(AMathlet.getProperty("zeq")+AMathlet.getProperty("fxyeq"));
    registerInputSet(field);
    setHandler("f",field);

    registerGraphButton(compButton);
    registerClearButton(clearButton);

    graph.localinit();
  }

  public void draw() {
    field.setDirty();
    graph.reapply();
  }

  public double evaluatef(double x, double y) {
    return field.evaluatef(x,y);
  }
}