Balls3DWithUI.java

/*
  Copyright 2006 by Sean Luke and George Mason University
  Licensed under the Academic Free License version 3.0
  See the file "LICENSE" for more information
*/

package sim.app.balls3d;
import sim.portrayal3d.network.*;
import sim.portrayal3d.continuous.*;
import sim.display3d.*;
import sim.engine.*;
import sim.display.*;
import javax.swing.*;
import sim.portrayal3d.simple.*;

public class Balls3DWithUI extends GUIState
    {
    public Display3D display;
    public JFrame displayFrame;

    NetworkPortrayal3D edgePortrayal = new NetworkPortrayal3D();
    ContinuousPortrayal3D nodePortrayal = new ContinuousPortrayal3D();

    public static void main(String[] args)
        {
        new Balls3DWithUI().createController();
        }

    public Balls3DWithUI() { super(new Balls3D( System.currentTimeMillis())); }

    public Balls3DWithUI(SimState state) 
        { 
        super(state); 
        }

    public static String getName() { return "3D Balls and Bands"; }
    
    public Object getSimulationInspectedObject() { return state; }

    public void start()
        {
        super.start();
        setupPortrayals();
        }

    public void load(SimState state)
        {
        super.load(state);
        setupPortrayals();
        }

    public void setupPortrayals()
        {
        // display.destroySceneGraph();
        
        Balls3D tut = (Balls3D) state;
        
        final java.text.NumberFormat strengthFormat = java.text.NumberFormat.getInstance();
        strengthFormat.setMinimumIntegerDigits(1);
        strengthFormat.setMaximumFractionDigits(2);
        
        // tell the portrayals what to portray and how to portray them
        edgePortrayal.setField( new SpatialNetwork3D( tut.balls, tut.bands ) );
        SimpleEdgePortrayal3D portrayal = new CylinderEdgePortrayal3D()
            {
            public String getLabel(sim.field.network.Edge e)
                {
                return strengthFormat.format(e.getWeight());
                }
            }; 
        // portrayal.setShowLabels(true);  // deprecated.  The default constructor for CylinderEdgePortrayal shows labels by default anyway
        portrayal.setLabelScale(1.0);
        
        edgePortrayal.setPortrayalForAll( portrayal );
        nodePortrayal.setField( tut.balls );
        try
            {
            nodePortrayal.setPortrayalForAll(new CircledPortrayal3D(
                    new BallPortrayal(5.0f),
                    20f, true));

            // Or to see some teapots....
            //            nodePortrayal.setPortrayalForAll(new CircledPortrayal3D(
            //                    new BranchGroupPortrayal3D(BranchGroupPortrayal3D.getBranchGroupForResource(BranchGroupPortrayal3D.class, "shapes/teapot.obj")),
            //                    20f, true));

            // Or some colorful cubes....
            //            nodePortrayal.setPortrayalForAll(new CircledPortrayal3D(
            //                    new Shape3DPortrayal3D(new com.sun.j3d.utils.geometry.ColorCube()),
            //                    20f, true));

            }
        catch (Exception e) { throw new RuntimeException("yo", e); }

        display.createSceneGraph(); 
        display.reset();
        }

    public void init(Controller c)
        {
        super.init(c);

        Balls3D tut = (Balls3D) state;

        // make the displayer
        display = new Display3D(600,600,this);                        
                        
        display.attach( edgePortrayal, "Bands" );
        display.attach( nodePortrayal, "Balls" );

        display.translate(-tut.gridWidth/2,
            -tut.gridHeight/2,
            -tut.gridLength/2);
        
        display.scale(1.0/tut.gridWidth);

        displayFrame = display.createFrame();
        displayFrame.setTitle("Balls and Bands");
        c.registerFrame(displayFrame);   // register the frame so it appears in the "Display" list
        displayFrame.setVisible(true);

        display.getSelectionBehavior().setTolerance(10.0f);
        }

    public void quit()
        {
        super.quit();

        if (displayFrame!=null) displayFrame.dispose();
        displayFrame = null;
        display = null;
        }

    }