GeometryByReferenceTest.java Source code

Java tutorial

Introduction

Here is the source code for GeometryByReferenceTest.java

Source

/*
 * @(#)GeometryByReferenceTest.java 1.12 02/10/21 13:41:27
 * 
 * Copyright (c) 1996-2002 Sun Microsystems, Inc. All Rights Reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *  - Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *  - Redistribution in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * 
 * Neither the name of Sun Microsystems, Inc. or the names of contributors may
 * be used to endorse or promote products derived from this software without
 * specific prior written permission.
 * 
 * This software is provided "AS IS," without a warranty of any kind. ALL
 * EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND WARRANTIES, INCLUDING ANY
 * IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR
 * NON-INFRINGEMENT, ARE HEREBY EXCLUDED. SUN AND ITS LICENSORS SHALL NOT BE
 * LIABLE FOR ANY DAMAGES SUFFERED BY LICENSEE AS A RESULT OF USING, MODIFYING
 * OR DISTRIBUTING THE SOFTWARE OR ITS DERIVATIVES. IN NO EVENT WILL SUN OR ITS
 * LICENSORS BE LIABLE FOR ANY LOST REVENUE, PROFIT OR DATA, OR FOR DIRECT,
 * INDIRECT, SPECIAL, CONSEQUENTIAL, INCIDENTAL OR PUNITIVE DAMAGES, HOWEVER
 * CAUSED AND REGARDLESS OF THE THEORY OF LIABILITY, ARISING OUT OF THE USE OF
 * OR INABILITY TO USE SOFTWARE, EVEN IF SUN HAS BEEN ADVISED OF THE POSSIBILITY
 * OF SUCH DAMAGES.
 * 
 * You acknowledge that Software is not designed,licensed or intended for use in
 * the design, construction, operation or maintenance of any nuclear facility.
 */

import java.awt.Container;
import java.awt.Frame;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;

import javax.media.j3d.AmbientLight;
import javax.media.j3d.Appearance;
import javax.media.j3d.BoundingSphere;
import javax.media.j3d.BranchGroup;
import javax.media.j3d.Canvas3D;
import javax.media.j3d.ColoringAttributes;
import javax.media.j3d.DirectionalLight;
import javax.media.j3d.Geometry;
import javax.media.j3d.GeometryArray;
import javax.media.j3d.GeometryUpdater;
import javax.media.j3d.IndexedGeometryArray;
import javax.media.j3d.IndexedTriangleArray;
import javax.media.j3d.IndexedTriangleStripArray;
import javax.media.j3d.Material;
import javax.media.j3d.RenderingAttributes;
import javax.media.j3d.Shape3D;
import javax.media.j3d.Transform3D;
import javax.media.j3d.TransformGroup;
import javax.media.j3d.TransparencyAttributes;
import javax.media.j3d.TriangleArray;
import javax.media.j3d.TriangleStripArray;
import javax.swing.BoxLayout;
import javax.swing.JApplet;
import javax.swing.JCheckBox;
import javax.swing.JComboBox;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.border.TitledBorder;
import javax.vecmath.Color3f;
import javax.vecmath.Point3d;
import javax.vecmath.Point3f;
import javax.vecmath.Vector3f;

import com.sun.j3d.utils.applet.MainFrame;
import com.sun.j3d.utils.behaviors.vp.OrbitBehavior;
import com.sun.j3d.utils.universe.SimpleUniverse;
import com.sun.j3d.utils.universe.ViewingPlatform;

public class GeometryByReferenceTest extends JApplet implements ActionListener, GeometryUpdater {

    RenderingAttributes ra;

    ColoringAttributes ca;

    Material mat;

    Appearance app;

    JComboBox geomType;

    JComboBox vertexType;

    JComboBox colorType;

    JCheckBox transparency;

    JComboBox updates;

    Shape3D shape;

    TransparencyAttributes transp;

    int updateIndex = 0;

    int colorCount = 0, vertexCount = 0;

    int vertexIndex = 0, colorIndex = 0;

    GeometryArray tetraRegular, tetraStrip, tetraIndexed, tetraIndexedStrip;

    GeometryArray[] geoArrays = new GeometryArray[4];

    private static final float sqrt3 = (float) Math.sqrt(3.0);

    private static final float sqrt3_3 = sqrt3 / 3.0f;

    private static final float sqrt24_3 = (float) Math.sqrt(24.0) / 3.0f;

    private static final float ycenter = 0.5f * sqrt24_3;

    private static final float zcenter = -sqrt3_3;

    private static final Point3f p1 = new Point3f(-1.0f, -ycenter, -zcenter);

    private static final Point3f p2 = new Point3f(1.0f, -ycenter, -zcenter);

    private static final Point3f p3 = new Point3f(0.0f, -ycenter, -sqrt3 - zcenter);

    private static final Point3f p4 = new Point3f(0.0f, sqrt24_3 - ycenter, 0.0f);

    private static final float[] floatVerts = { p1.x, p1.y, p1.z, // front face
            p2.x, p2.y, p2.z, p4.x, p4.y, p4.z,

            p1.x, p1.y, p1.z, // left, back face
            p4.x, p4.y, p4.z, p3.x, p3.y, p3.z,

            p2.x, p2.y, p2.z, // right, back face
            p3.x, p3.y, p3.z, p4.x, p4.y, p4.z,

            p1.x, p1.y, p1.z, // bottom face
            p3.x, p3.y, p3.z, p2.x, p2.y, p2.z, };

    private static final Color3f c1 = new Color3f(0.6f, 0.0f, 0.0f);

    private static final Color3f c2 = new Color3f(0.0f, 0.6f, 0.0f);

    private static final Color3f c3 = new Color3f(0.0f, 0.6f, 0.6f);

    private static final Color3f c4 = new Color3f(0.6f, 0.6f, 0.0f);

    private static final float[] floatClrs = { c1.x, c1.y, c1.z, // front face
            c2.x, c2.y, c2.z, c4.x, c4.y, c4.z,

            c1.x, c1.y, c1.z, // left, back face
            c4.x, c4.y, c4.z, c3.x, c3.y, c3.z,

            c2.x, c2.y, c2.z, // right, back face
            c3.x, c3.y, c3.z, c4.x, c4.y, c4.z,

            c1.x, c1.y, c1.z, // bottom face
            c3.x, c3.y, c3.z, c2.x, c2.y, c2.z, };

    private static final float[] indexedFloatVerts = { p1.x, p1.y, p1.z, p2.x, p2.y, p2.z, p3.x, p3.y, p3.z, p4.x,
            p4.y, p4.z,

    };

    private static final float[] indexedFloatClrs = { c1.x, c1.y, c1.z, c2.x, c2.y, c2.z, c3.x, c3.y, c3.z, c4.x,
            c4.y, c4.z, };

    private static final Point3f[] p3fVerts = { p1, p2, p4, p1, p4, p3, p2, p3, p4, p1, p3, p2 };

    private static final Point3f[] indexedP3fVerts = { p1, p2, p3, p4 };

    private static final Color3f[] c3fClrs = { c1, c2, c4, c1, c4, c3, c2, c3, c4, c1, c3, c2 };

    private static final Color3f[] indexedC3fClrs = { c1, c2, c3, c4 };

    private static final int[] indices = { 0, 1, 3, 0, 3, 2, 1, 2, 3, 0, 2, 1 };

    private int[] stripVertexCounts = { 3, 3, 3, 3 };

    private SimpleUniverse u;

    BranchGroup createSceneGraph() {
        BranchGroup objRoot = new BranchGroup();

        // Set up attributes to render lines
        app = new Appearance();

        transp = new TransparencyAttributes();
        transp.setTransparency(0.5f);
        transp.setCapability(TransparencyAttributes.ALLOW_MODE_WRITE);
        transp.setTransparencyMode(TransparencyAttributes.NONE);
        app.setTransparencyAttributes(transp);

        tetraRegular = createGeometry(1);
        tetraStrip = createGeometry(2);
        tetraIndexed = createGeometry(3);
        tetraIndexedStrip = createGeometry(4);

        geoArrays[0] = tetraRegular;
        geoArrays[1] = tetraStrip;
        geoArrays[2] = tetraIndexed;
        geoArrays[3] = tetraIndexedStrip;

        shape = new Shape3D(tetraRegular, app);
        shape.setCapability(Shape3D.ALLOW_GEOMETRY_WRITE);
        shape.setCapability(Shape3D.ALLOW_GEOMETRY_READ);

        Transform3D t = new Transform3D();
        // move the object upwards
        t.set(new Vector3f(0.0f, 0.3f, 0.0f));

        // rotate the shape
        Transform3D temp = new Transform3D();
        temp.rotX(Math.PI / 4.0d);
        t.mul(temp);
        temp.rotY(Math.PI / 4.0d);
        t.mul(temp);

        // Shrink the object
        t.setScale(0.6);

        TransformGroup trans = new TransformGroup(t);
        trans.setCapability(TransformGroup.ALLOW_TRANSFORM_WRITE);
        trans.setCapability(TransformGroup.ALLOW_TRANSFORM_READ);

        objRoot.addChild(trans);
        trans.addChild(shape);

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

        // Set up the global lights
        Color3f lColor1 = new Color3f(0.7f, 0.7f, 0.7f);
        Vector3f lDir1 = new Vector3f(-1.0f, -1.0f, -1.0f);
        Color3f alColor = new Color3f(0.2f, 0.2f, 0.2f);

        AmbientLight aLgt = new AmbientLight(alColor);
        aLgt.setInfluencingBounds(bounds);
        DirectionalLight lgt1 = new DirectionalLight(lColor1, lDir1);
        lgt1.setInfluencingBounds(bounds);
        objRoot.addChild(aLgt);
        objRoot.addChild(lgt1);

        // Let Java 3D perform optimizations on this scene graph.
        objRoot.compile();

        return objRoot;
    }

    JPanel createGeometryByReferencePanel() {
        JPanel panel = new JPanel();
        panel.setBorder(new TitledBorder("Geometry Type"));

        String values[] = { "Array", "Strip", "Indexed", "IndexedStrip" };
        geomType = new JComboBox(values);
        geomType.setLightWeightPopupEnabled(false);
        geomType.addActionListener(this);
        geomType.setSelectedIndex(0);
        panel.add(new JLabel("Geometry Type"));
        panel.add(geomType);

        String vertex_types[] = { "Float", "P3F" };

        vertexType = new JComboBox(vertex_types);
        vertexType.setLightWeightPopupEnabled(false);
        vertexType.addActionListener(this);
        vertexType.setSelectedIndex(0);
        panel.add(new JLabel("VertexType"));
        panel.add(vertexType);

        String color_types[] = { "Float", "C3F" };

        colorType = new JComboBox(color_types);
        colorType.setLightWeightPopupEnabled(false);
        colorType.addActionListener(this);
        colorType.setSelectedIndex(0);
        panel.add(new JLabel("ColorType"));
        panel.add(colorType);

        return panel;
    }

    JPanel createUpdatePanel() {

        JPanel panel = new JPanel();
        panel.setBorder(new TitledBorder("Other Attributes"));

        String updateComp[] = { "None", "Geometry", "Color" };

        transparency = new JCheckBox("EnableTransparency", false);
        transparency.addActionListener(this);
        panel.add(transparency);

        updates = new JComboBox(updateComp);
        updates.setLightWeightPopupEnabled(false);
        updates.addActionListener(this);
        updates.setSelectedIndex(0);
        panel.add(new JLabel("UpdateData"));
        panel.add(updates);

        return panel;
    }

    public GeometryByReferenceTest() {
    }

    public void init() {
        Container contentPane = getContentPane();

        Canvas3D c = new Canvas3D(SimpleUniverse.getPreferredConfiguration());
        contentPane.add("Center", c);

        BranchGroup scene = createSceneGraph();
        // SimpleUniverse is a Convenience Utility class
        u = new SimpleUniverse(c);

        // add mouse behaviors to the viewingPlatform
        ViewingPlatform viewingPlatform = u.getViewingPlatform();

        // This will move the ViewPlatform back a bit so the
        // objects in the scene can be viewed.
        viewingPlatform.setNominalViewingTransform();
        u.addBranchGraph(scene);

        // add Orbit behavior to the ViewingPlatform
        OrbitBehavior orbit = new OrbitBehavior(c, OrbitBehavior.REVERSE_ALL);
        BoundingSphere bounds = new BoundingSphere(new Point3d(0.0, 0.0, 0.0), 100.0);
        orbit.setSchedulingBounds(bounds);
        viewingPlatform.setViewPlatformBehavior(orbit);

        // Create GUI
        JPanel p = new JPanel();
        BoxLayout boxlayout = new BoxLayout(p, BoxLayout.Y_AXIS);
        p.add(createGeometryByReferencePanel());
        p.add(createUpdatePanel());
        p.setLayout(boxlayout);

        contentPane.add("South", p);
    }

    public void destroy() {
        u.cleanup();
    }

    public void actionPerformed(ActionEvent e) {
        Object target = e.getSource();
        GeometryArray geo;
        boolean setColor = false, setVertex = false;
        if (target == geomType) {
            geo = geoArrays[geomType.getSelectedIndex()];
            // Set everything to null, and set it later ..
            geo.setColorRefFloat(null);
            geo.setColorRef3f(null);
            geo.setCoordRefFloat(null);
            geo.setCoordRef3f(null);
            shape.setGeometry(geoArrays[geomType.getSelectedIndex()]);

            setColor = true;
            setVertex = true;

        } else if (target == transparency) {
            if (transparency.isSelected()) {
                transp.setTransparencyMode(TransparencyAttributes.BLENDED);
            } else {
                transp.setTransparencyMode(TransparencyAttributes.NONE);
            }

        } else if (target == updates) {
            updateIndex = updates.getSelectedIndex();
            if (updateIndex == 1) {
                System.out.println("Doing coordinate update");
                ((GeometryArray) (shape.getGeometry())).updateData(this);
            } else if (updateIndex == 2) {
                System.out.println("Doing color update");
                ((GeometryArray) (shape.getGeometry())).updateData(this);
            }

        } else if (target == vertexType) {
            geo = ((GeometryArray) shape.getGeometry());
            if (vertexIndex == 0) {
                geo.setCoordRefFloat(null);
            } else if (vertexIndex == 1) {
                geo.setCoordRef3f(null);
            }
            vertexIndex = vertexType.getSelectedIndex();
            setVertex = true;
        } else if (target == colorType) {
            geo = (GeometryArray) shape.getGeometry();
            if (colorIndex == 0) {
                geo.setColorRefFloat(null);
            } else if (colorIndex == 1) {
                geo.setColorRef3f(null);
            }
            colorIndex = colorType.getSelectedIndex();
            setColor = true;
        }

        if (setVertex) {
            geo = (GeometryArray) shape.getGeometry();
            if (vertexIndex == 0) {
                if (geo instanceof IndexedGeometryArray)
                    geo.setCoordRefFloat(indexedFloatVerts);
                else
                    geo.setCoordRefFloat(floatVerts);
            } else if (vertexIndex == 1) {
                if (geo instanceof IndexedGeometryArray)
                    geo.setCoordRef3f(indexedP3fVerts);
                else
                    geo.setCoordRef3f(p3fVerts);
            }

        }
        if (setColor) {
            geo = (GeometryArray) shape.getGeometry();
            if (colorIndex == 0) {
                if (geo instanceof IndexedGeometryArray)
                    geo.setColorRefFloat(indexedFloatClrs);
                else
                    geo.setColorRefFloat(floatClrs);
            } else if (colorIndex == 1) {
                if (geo instanceof IndexedGeometryArray)
                    geo.setColorRef3f(indexedC3fClrs);
                else
                    geo.setColorRef3f(c3fClrs);
            }
        }

    }

    public static void main(String[] args) {
        Frame frame = new MainFrame(new GeometryByReferenceTest(), 800, 800);
    }

    public GeometryArray createGeometry(int type) {
        GeometryArray tetra = null;
        if (type == 1) {
            tetra = new TriangleArray(12,
                    TriangleArray.COORDINATES | TriangleArray.COLOR_3 | TriangleArray.BY_REFERENCE);

            tetra.setCoordRefFloat(floatVerts);
            tetra.setColorRefFloat(floatClrs);

        } else if (type == 2) {
            tetra = new TriangleStripArray(12,
                    TriangleStripArray.COORDINATES | TriangleStripArray.COLOR_3 | TriangleStripArray.BY_REFERENCE,
                    stripVertexCounts);
            tetra.setCoordRefFloat(floatVerts);
            tetra.setColorRefFloat(floatClrs);

        } else if (type == 3) { // Indexed Geometry
            tetra = new IndexedTriangleArray(4, IndexedTriangleArray.COORDINATES | IndexedTriangleArray.COLOR_3
                    | IndexedTriangleArray.BY_REFERENCE, 12);
            tetra.setCoordRefFloat(indexedFloatVerts);
            tetra.setColorRefFloat(indexedFloatClrs);
            ((IndexedTriangleArray) tetra).setCoordinateIndices(0, indices);
            ((IndexedTriangleArray) tetra).setColorIndices(0, indices);
        } else if (type == 4) { // Indexed strip geometry
            tetra = new IndexedTriangleStripArray(4, IndexedTriangleStripArray.COORDINATES
                    | IndexedTriangleStripArray.COLOR_3 | IndexedTriangleStripArray.BY_REFERENCE, 12,
                    stripVertexCounts);
            tetra.setCoordRefFloat(indexedFloatVerts);
            tetra.setColorRefFloat(indexedFloatClrs);
            ((IndexedTriangleStripArray) tetra).setCoordinateIndices(0, indices);
            ((IndexedTriangleStripArray) tetra).setColorIndices(0, indices);
        }

        if (tetra != null)
            tetra.setCapability(GeometryArray.ALLOW_REF_DATA_WRITE);
        return tetra;
    }

    public void updateData(Geometry geometry) {
        int i;
        float val;

        if (updateIndex == 1) { // geometry
            // Translate the geometry by a small amount in x
            vertexCount++;
            if ((vertexCount & 1) == 1)
                val = 0.2f;
            else
                val = -0.2f;

            if (vertexIndex == 0) {
                // Do Indexed geometry
                for (i = 0; i < indexedFloatVerts.length; i += 3) {
                    indexedFloatVerts[i] += val;
                }
                // Do non-indexed float geometry
                for (i = 0; i < floatVerts.length; i += 3) {
                    floatVerts[i] += val;
                }
            } else {
                // If p3f do each point only once
                for (i = 0; i < indexedP3fVerts.length; i++) {
                    indexedP3fVerts[i].x += val;
                }
            }

        } else if (updateIndex == 2) { // colors
            colorCount++;
            if ((colorCount & 1) == 1)
                val = 0.4f;
            else
                val = -0.4f;
            if (colorIndex == 0) {
                // Do Indexed geometry
                for (i = 0; i < indexedFloatClrs.length; i += 3) {
                    indexedFloatClrs[i] += val;
                }
                // Do non-indexed float geometry
                for (i = 0; i < floatClrs.length; i += 3) {
                    floatClrs[i] += val;
                }
            } else {
                // If c3f do each point only once
                for (i = 0; i < indexedC3fClrs.length; i++) {
                    indexedC3fClrs[i].x += val;
                }
            }

        }

    }
}