// Java 3D Test Applet
// FogTest.java
//   Copyright (c) 1999 ENDO Yasuyuki
//                      mailto:yasuyuki@javaopen.org
//                      http://www.javaopen.org/j3dbook/index.html

import java.applet.*;
import java.awt.*;
import java.awt.event.*;
import javax.media.j3d.*;
import javax.vecmath.*;
import com.sun.j3d.utils.applet.MainFrame;
import com.sun.j3d.utils.universe.SimpleUniverse;
import com.sun.j3d.utils.geometry.Sphere;

public class FogTest extends Applet {
  private Canvas3D canvas = null;
  private SimpleUniverse universe = null;
  private BranchGroup scene = null;
  private Appearance app = null;
  private Fog fog = null;
  private LinearFog lfog = null;
  private ExponentialFog efog = null;

  public FogTest() {
    this.setLayout(new BorderLayout());

    Panel downpanel = new Panel();
    downpanel.setLayout( new GridLayout(3, 1) );
    this.add(downpanel, BorderLayout.SOUTH);
    
    Panel[] dpanels = new Panel[3];
    for (int i=0; i<3; i++) {
      dpanels[i] = new Panel();
      downpanel.add(dpanels[i]);
    }
    
    dpanels[0].add( new Label("Model Color - R") );
    TextField mrfield = new TextField("1.0");
    mrfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
        float r = 0.0f;
        try {
          r = Float.parseFloat(e.getActionCommand());
          if (r < 0.0f) r = 0.0f;
          if (r > 1.0f) r = 1.0f;
          Color3f color = new Color3f();
          app.getMaterial().getDiffuseColor(color);
          color.x = r;
          app.getMaterial().setDiffuseColor(color);
        } catch (NumberFormatException ex) {}
      }
    });
    dpanels[0].add(mrfield);

    dpanels[0].add( new Label("G") );
    TextField mgfield = new TextField("0.0");
    mgfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
        float g = 0.0f;
        try {
          g = Float.parseFloat(e.getActionCommand());
          if (g < 0.0f) g = 0.0f;
          if (g > 1.0f) g = 1.0f;
          Color3f color = new Color3f();
          app.getMaterial().getDiffuseColor(color);
          color.y = g;
          app.getMaterial().setDiffuseColor(color);
        } catch (NumberFormatException ex) {}
      }
    });
    dpanels[0].add(mgfield);

    dpanels[0].add( new Label("B") );
    TextField mbfield = new TextField("0.0");
    mbfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
        float b = 0.0f;
        try {
          b = Float.parseFloat(e.getActionCommand());
          if (b < 0.0f) b = 0.0f;
          if (b > 1.0f) b = 1.0f;
          Color3f color = new Color3f();
          app.getMaterial().getDiffuseColor(color);
          color.z = b;
          app.getMaterial().setDiffuseColor(color);
        } catch (NumberFormatException ex) {}
      }
    });
    dpanels[0].add(mbfield);

    dpanels[0].add( new Label("BackClip") );
    TextField bcfield = new TextField("10.0");
    bcfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
	double clip = 0.0;
	try {
	  clip = Double.parseDouble(e.getActionCommand());
	  universe.getViewer().getView().setBackClipDistance(clip);
	} catch (NumberFormatException ex) {}
      }
    });
    dpanels[0].add(bcfield);

    Panel fpanel = new Panel();
    fpanel.setLayout( new GridLayout(1, 2) );
    dpanels[1].add(fpanel);
    
    final Panel fdpanel = new Panel();
    final CardLayout clayout = new CardLayout();
    fdpanel.setLayout(clayout);
    dpanels[1].add(fdpanel);

    dpanels[2].add( new Label("Fog Color - R") );
    final TextField rfield = new TextField("0.0");
    rfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
	float r = 0.0f;
	try {
	  r = Float.parseFloat(e.getActionCommand());
	  if (r < 0.0f) r = 0.0f;
	  if (r > 1.0f) r = 1.0f;
	  Color3f color = new Color3f();
	  fog.getColor(color);
	  color.x = r;
	  fog.setColor(color);
	} catch (NumberFormatException ex) {}
      }
    });
    dpanels[2].add(rfield);

    dpanels[2].add( new Label("G") );
    final TextField gfield = new TextField("0.0");
    gfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
	float g = 0.0f;
	if (g < 0.0f) g = 0.0f;
	if (g > 1.0f) g = 1.0f;
	try {
	  g = Float.parseFloat(e.getActionCommand());
	  Color3f color = new Color3f();
	  fog.getColor(color);
	  color.y = g;
	  fog.setColor(color);
	} catch (NumberFormatException ex) {}
      }
    });
    dpanels[2].add(gfield);

    dpanels[2].add( new Label("B") );
    final TextField bfield = new TextField("0.0");
    bfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
	float b = 0.0f;
	if (b < 0.0f) b = 0.0f;
	if (b > 1.0f) b = 1.0f;
	try {
	  b = Float.parseFloat(e.getActionCommand());
	  Color3f color = new Color3f();
	  fog.getColor(color);
	  color.z = b;
	  fog.setColor(color);
	} catch (NumberFormatException ex) {}
      }
    });
    dpanels[2].add(bfield);

    Panel[] fdpanels = new Panel[3];
    for (int i=0; i<3; i++) {
      fdpanels[i] = new Panel();
      fdpanel.add(fdpanels[i], Integer.toString(i));
    }
    clayout.show(fdpanel, "2");

    Panel fgpanel = new Panel();
    fgpanel.setLayout( new GridLayout(1, 3) );
    fpanel.add(fgpanel);

    CheckboxGroup cbg = new CheckboxGroup();
    
    Checkbox ncheck = new Checkbox("None", cbg, false);
    ncheck.addItemListener( new ItemListener() {
      public void itemStateChanged(ItemEvent e) {
	if (e.getStateChange() == ItemEvent.SELECTED) {
	  universe.getLocale().removeBranchGraph(scene);
	  removeChild(scene, fog);
	  clayout.show(fdpanel, "0");
	  universe.addBranchGraph(scene);
	} else if (e.getStateChange() == ItemEvent.DESELECTED) {
	  universe.getLocale().removeBranchGraph(scene);
	  addChild(scene, fog);
	  universe.addBranchGraph(scene);
	}
      }
    });
    fgpanel.add(ncheck);

    Checkbox lcheck = new Checkbox("Linear", cbg, false);
    lcheck.addItemListener( new ItemListener() {
      public void itemStateChanged(ItemEvent e) {
	if (e.getStateChange() == ItemEvent.SELECTED) {
	  universe.getLocale().removeBranchGraph(scene);
	  removeChild(scene, fog);
	  fog = lfog;
	  Color3f color = new Color3f();
	  fog.getColor(color);
	  rfield.setText(Float.toString(color.x));
	  gfield.setText(Float.toString(color.y));
	  bfield.setText(Float.toString(color.z));
	  clayout.show(fdpanel, "1");
	  addChild(scene, fog);
	  universe.addBranchGraph(scene);
	}
      }
    });
    fgpanel.add(lcheck);

    Checkbox echeck = new Checkbox("Exp", cbg, true);
    echeck.addItemListener( new ItemListener() {
      public void itemStateChanged(ItemEvent e) {
	if (e.getStateChange() == ItemEvent.SELECTED) {
	  universe.getLocale().removeBranchGraph(scene);
	  removeChild(scene, fog);
	  fog = efog;
	  Color3f color = new Color3f();
	  fog.getColor(color);
	  rfield.setText(Float.toString(color.x));
	  gfield.setText(Float.toString(color.y));
	  bfield.setText(Float.toString(color.z));
	  clayout.show(fdpanel, "2");
	  addChild(scene, fog);
	  universe.addBranchGraph(scene);
	}
      }
    });
    fgpanel.add(echeck);

    fdpanels[1].add( new Label("FrDistance") );
    TextField frfield = new TextField("0.1");
    frfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
	double d = 0.0;
	try {
	  d= Double.parseDouble(e.getActionCommand());
	  lfog.setFrontDistance(d);
	} catch (NumberFormatException ex) {}	
      }
    });
    fdpanels[1].add(frfield);

    fdpanels[1].add( new Label("BkDistance") );
    TextField brfield = new TextField("1.0");
    brfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
	double d = 0.0;
	try {
	  d = Double.parseDouble(e.getActionCommand());
	  lfog.setBackDistance(d);
	} catch (NumberFormatException ex) {}	
      }
    });
    fdpanels[1].add(brfield);

    fdpanels[2].add( new Label("Dencity") );
    TextField dcfield = new TextField("1.0");
    dcfield.addActionListener( new ActionListener() {
      public void actionPerformed(ActionEvent e) {
	float d = 0.0f;
	try {
	  d = Float.parseFloat(e.getActionCommand());
	  efog.setDensity(d);
	} catch (NumberFormatException ex) {}
      }
    });
    fdpanels[2].add(dcfield);

    GraphicsConfiguration config = SimpleUniverse.getPreferredConfiguration();
    this.canvas = new Canvas3D(config);
    this.add(this.canvas, BorderLayout.CENTER);
    
    universe = new SimpleUniverse(canvas);
    universe.getViewingPlatform().setNominalViewingTransform();
    
    scene = createSceneGraph();
    universe.addBranchGraph(scene);
  }
  
  private BranchGroup createSceneGraph() {
    BranchGroup root = new BranchGroup();
    root.setCapability(BranchGroup.ALLOW_DETACH);

    root.addChild(createLight(0.3f, -0.3f, -0.3f));
    
    BoundingSphere bounds = new BoundingSphere(new Point3d(), 10000.0);
    
    app = createAppearance();
    
    double x = -2.0;
    double y = -2.0;
    double z = 2.0;
    for (int i=0; i<20; i++) {
      root.addChild( createSphere(0.5f, x, y, z, app) );
      x += 0.3 * (double)i;
      y += 0.3 * (double)i;
      z -= (double)i;
    }

    lfog = new LinearFog();
    lfog.setCapability(Fog.ALLOW_COLOR_READ);
    lfog.setCapability(Fog.ALLOW_COLOR_WRITE);
    lfog.setCapability(LinearFog.ALLOW_DISTANCE_WRITE);
    lfog.setInfluencingBounds(bounds);

    efog = new ExponentialFog();
    efog.setCapability(Fog.ALLOW_COLOR_READ);
    efog.setCapability(Fog.ALLOW_COLOR_WRITE);
    efog.setCapability(ExponentialFog.ALLOW_DENSITY_WRITE);
    efog.setInfluencingBounds(bounds);

    fog = efog;
    root.addChild(fog);
    
    return root;
  }
  
  private TransformGroup createSphere( float radius, double x, double y, double z,
                                       Appearance app)
  {
    Transform3D t3d = new Transform3D();
    Vector3d position = new Vector3d(x, y, z);
    t3d.set(position);
    TransformGroup trans = new TransformGroup(t3d);
    trans.addChild( new Sphere(radius, Sphere.GENERATE_NORMALS, app) );
    return trans;
  }
  
  private Appearance createAppearance() {
    Appearance app = new Appearance();
    app.setCapability(Appearance.ALLOW_MATERIAL_READ);
    Material mat = new Material();
    mat.setCapability(Material.ALLOW_COMPONENT_READ);
    mat.setCapability(Material.ALLOW_COMPONENT_WRITE);
    mat.setDiffuseColor(1.0f, 0.0f, 0.0f);
    mat.setSpecularColor(0.1f, 0.1f, 0.1f);
    app.setMaterial(mat);
    return app;
  }
  
  private Light createLight(float x, float y, float z) {
    DirectionalLight light =
      new DirectionalLight( true,
                            new Color3f(1.0f, 1.0f, 1.0f),
                            new Vector3f(x, y, z));
    light.setInfluencingBounds(new BoundingSphere(new Point3d(), 100.0));
    
    return light;
  }

  private void addChild(Group group, Node node) {
    for (int i=0; i<group.numChildren(); i++) if (node == group.getChild(i)) return;
    group.addChild(node);
  }

  private void removeChild(Group group, Node node) {
    for (int i=0; i<group.numChildren(); i++) {
      if (node == group.getChild(i)) group.removeChild(i);
    }
  }

  public static void main(String[] args) {
    FogTest applet = new FogTest();
    Frame frame = new MainFrame(applet, 500, 500);
  }
}