source: java/main/src/main/java/com/framsticks/visualization/Viewer.java @ 78

package com.framsticks.visualization;

import com.framsticks.model.World;
import com.sun.j3d.loaders.IncorrectFormatException;
import com.sun.j3d.loaders.ParsingErrorException;
import com.sun.j3d.loaders.Scene;
import com.sun.j3d.loaders.objectfile.ObjectFile;
import com.sun.j3d.utils.behaviors.mouse.MouseRotate;
import com.sun.j3d.utils.universe.PlatformGeometry;
import com.sun.j3d.utils.universe.SimpleUniverse;
import com.sun.j3d.utils.universe.ViewingPlatform;

import javax.media.j3d.*;
import javax.swing.*;
import javax.swing.text.StyledDocument;
import javax.vecmath.Color3f;
import javax.vecmath.Point3d;
import javax.vecmath.Vector3f;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.FileNotFoundException;

import org.apache.log4j.Logger;

public class Viewer {

    private static Logger LOGGER = Logger.getLogger(Viewer.class);

    Canvas3D canvas3d;
    SimpleUniverse universe;

    World world;
    private JMenuBar menuBar;

        private JScrollPane scrollPane;
    private boolean scrollLock = false;

        private JCheckBoxMenuItem loggingItem;
    private JCheckBoxMenuItem autorefreshItem;

    public Viewer(JMenuBar menuBar) {
        super();
        this.menuBar = menuBar;
        loggingItem = new JCheckBoxMenuItem("Logging");
        loggingItem.setSelected(true);

        init();
    }

        public BranchGroup createSceneGraph() {
        // Create the root of the branch graph
        BranchGroup objRoot = new BranchGroup();

        // Create a Transform group to scale all objects so they
        // appear in the scene.
        TransformGroup objScale = new TransformGroup();
        Transform3D t3d = new Transform3D();
        t3d.setScale(0.7);
        objScale.setTransform(t3d);
        objRoot.addChild(objScale);

        TransformGroup objTrans = new TransformGroup();
        objTrans.setCapability(TransformGroup.ALLOW_TRANSFORM_WRITE);
        objTrans.setCapability(TransformGroup.ALLOW_TRANSFORM_READ);
        objScale.addChild(objTrans);

        String filename = "/static/shared/res/obj/cylinder.obj";
        int flags = ObjectFile.RESIZE;
        flags |= ObjectFile.TRIANGULATE;
        flags |= ObjectFile.STRIPIFY;
                double creaseAngle = 60.0;
                ObjectFile f = new ObjectFile(flags,
                (float) (creaseAngle * Math.PI / 180.0));
        Scene s = null;
        try {
            s = f.load(this.getClass().getResource(filename).getFile());
        } catch (FileNotFoundException e) {
            System.err.println(e);
            System.exit(1);
        } catch (ParsingErrorException e) {
            System.err.println(e);
            System.exit(1);
        } catch (IncorrectFormatException e) {
            System.err.println(e);
            System.exit(1);
        }

        objTrans.addChild(s.getSceneGroup());

        BoundingSphere bounds = new BoundingSphere(new Point3d(0.0, 0.0, 0.0),
                100.0);

        Color3f bgColor = new Color3f(0.05f, 0.05f, 0.5f);
        Background bgNode = new Background(bgColor);
        bgNode.setApplicationBounds(bounds);
        objRoot.addChild(bgNode);

        return objRoot;
    }

    private void init() {

            JTextPane textPane = new JTextPane();
        textPane.setEditable(false);

            StyledDocument styledDocument = textPane.getStyledDocument();

        scrollPane = new JScrollPane(textPane);
        //scrollPane.
        //scrollPane
        //              .setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_ALWAYS);

        // paneScrollPane.setMinimumSize(new Dimension(400, 300));

        createMenuBar();

        GraphicsConfiguration config = SimpleUniverse
                .getPreferredConfiguration();

        // Create a Canvas3D using the preferred configuration
        canvas3d = new Canvas3D(config);

        // Create simple universe with view branch
        universe = new SimpleUniverse(canvas3d);
        BoundingSphere bounds = new BoundingSphere(new Point3d(0.0, 0.0, 0.0),
                100.0);

        // add mouse behaviours to the ViewingPlatform
        ViewingPlatform viewingPlatform = universe.getViewingPlatform();

        PlatformGeometry platformGeometry = new PlatformGeometry();

        // Set up the ambient light
        Color3f ambientColor = new Color3f(0.1f, 0.1f, 0.1f);
        AmbientLight ambientLightNode = new AmbientLight(ambientColor);
        ambientLightNode.setInfluencingBounds(bounds);
        platformGeometry.addChild(ambientLightNode);

        // Set up the directional lights
        Color3f light1Color = new Color3f(1.0f, 1.0f, 0.9f);
        Vector3f light1Direction = new Vector3f(1.0f, 1.0f, 1.0f);
        Color3f light2Color = new Color3f(1.0f, 1.0f, 1.0f);
        Vector3f light2Direction = new Vector3f(-1.0f, -1.0f, -1.0f);

        DirectionalLight light1 = new DirectionalLight(light1Color,
                light1Direction);
        light1.setInfluencingBounds(bounds);
        platformGeometry.addChild(light1);

        DirectionalLight light2 = new DirectionalLight(light2Color,
                light2Direction);
        light2.setInfluencingBounds(bounds);
        platformGeometry.addChild(light2);

        viewingPlatform.setPlatformGeometry(platformGeometry);

        viewingPlatform.setNominalViewingTransform();

        // Ensure at least 5 msec per frame (i.e., < 200Hz)
        universe.getViewer().getView().setMinimumFrameCycleTime(5);

        BranchGroup scene = new BranchGroup();
        BranchGroup content = createSceneGraph();

        // canvas3d.addMouseListener(behaviour);
        TransformGroup transformGroup = new TransformGroup(new Transform3D());
        transformGroup.setCapability(TransformGroup.ALLOW_TRANSFORM_READ);
        transformGroup.setCapability(TransformGroup.ALLOW_TRANSFORM_WRITE);
        MouseRotate behaviour = new MouseRotate();
        behaviour.setTransformGroup(transformGroup);
        transformGroup.addChild(behaviour);
        // behaviour.addListener(canvas3d);
        // behaviour.initialize();
        behaviour.setSchedulingBounds(new BoundingSphere(new Point3d(0.0, 0.0,
                0.0), 100.0));
        transformGroup.addChild(content);
        // transformGroup.addChild(behaviour);
        scene.addChild(transformGroup);

        universe.addBranchGraph(scene);


    }

    private void createMenuBar() {

        JMenu serverItem = new JMenu("Manager");
        menuBar.add(serverItem);

        JMenuItem testModeItem = new JMenuItem("Test mode");
        serverItem.add(testModeItem);

            JMenuItem connectItem = new JMenuItem("Connect");
        serverItem.add(connectItem);

        JMenu simulationItem = new JMenu("Simulation");
        menuBar.add(simulationItem);

            //TODO: move init to main window
        JMenuItem initItem = new JMenuItem("Init");
        simulationItem.add(initItem);
        initItem.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
            }
        });

        JMenuItem refreshCreaturesItem = new JMenuItem("Refresh creatures");
        simulationItem.add(refreshCreaturesItem);
        refreshCreaturesItem.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {

            }
        });

        JMenuItem refreshWorldItem = new JMenuItem("Refresh world");
        simulationItem.add(refreshWorldItem);
        refreshWorldItem.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                refreshWorld();
            }
        });

            JMenu styleMenu = new JMenu("Style");
        menuBar.add(styleMenu);

            JMenu viewMenu = new JMenu("View");
        menuBar.add(viewMenu);
        rebuildViewMenu(null);

        JMenu optionsMenu = new JMenu("Options");
        menuBar.add(optionsMenu);

        autorefreshItem = new JCheckBoxMenuItem("Autorefresh");
        autorefreshItem.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                if (autorefreshItem.isSelected()) {

                }
            }
        });
        optionsMenu.add(autorefreshItem);

        optionsMenu.add(loggingItem);

    }

    public void refreshWorld() {


    }

    public void rebuildViewMenu(Object creatures) {

    }

    public Component getViewComponent() {
        return canvas3d;
    }

    public Component getControlComponent() {
        return scrollPane;
    }

}