GeometryByReferenceNIOBuffer.java Source code

Java tutorial

Introduction

Here is the source code for GeometryByReferenceNIOBuffer.java

Source

/*
 * @(#)GeometryByReferenceNIOBuffer.java 1.7 02/10/21 13:42:07
 * 
 * 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 java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.nio.FloatBuffer;

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.J3DBuffer;
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 GeometryByReferenceNIOBuffer 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 int[] indices = { 0, 1, 3, 0, 3, 2, 1, 2, 3, 0, 2, 1 };

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

    private SimpleUniverse u;

    private J3DBuffer floatBufferCoord;

    private J3DBuffer floatBufferColor;

    private J3DBuffer indexedFloatBufferCoord;

    private J3DBuffer indexedFloatBufferColor;

    void createJ3DBuffers() {
        int i;
        ByteOrder order = ByteOrder.nativeOrder();

        FloatBuffer coord = ByteBuffer.allocateDirect(36 * 4).order(order).asFloatBuffer();
        coord.put(floatVerts, 0, 36);
        floatBufferCoord = new J3DBuffer(coord);

        FloatBuffer color = ByteBuffer.allocateDirect(36 * 4).order(order).asFloatBuffer();
        color.put(floatClrs, 0, 36);
        floatBufferColor = new J3DBuffer(color);

        FloatBuffer indexedCoord = ByteBuffer.allocateDirect(12 * 4).order(order).asFloatBuffer();
        indexedCoord.put(indexedFloatVerts, 0, 12);
        indexedFloatBufferCoord = new J3DBuffer(indexedCoord);

        FloatBuffer indexedColor = ByteBuffer.allocateDirect(12 * 4).order(order).asFloatBuffer();
        indexedColor.put(indexedFloatClrs, 0, 12);
        indexedFloatBufferColor = new J3DBuffer(indexedColor);
    }

    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);

        //create the direct nio buffer
        createJ3DBuffers();

        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);

        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 GeometryByReferenceNIOBuffer() {
    }

    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.setColorRefBuffer(null);
            geo.setCoordRefBuffer(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);
            }

        }

        if (setVertex) {
            geo = (GeometryArray) shape.getGeometry();
            if (geo instanceof IndexedGeometryArray)
                geo.setCoordRefBuffer(indexedFloatBufferCoord);
            else
                geo.setCoordRefBuffer(floatBufferCoord);

        }
        if (setColor) {
            geo = (GeometryArray) shape.getGeometry();
            if (geo instanceof IndexedGeometryArray)
                geo.setColorRefBuffer(indexedFloatBufferColor);
            else
                geo.setColorRefBuffer(floatBufferColor);
        }
    }

    public static void main(String[] args) {
        Frame frame = new MainFrame(new GeometryByReferenceNIOBuffer(), 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 | TriangleArray.USE_NIO_BUFFER);

            tetra.setCoordRefBuffer(floatBufferCoord);
            tetra.setColorRefBuffer(floatBufferColor);

        } else if (type == 2) {
            tetra = new TriangleStripArray(
                    12, TriangleStripArray.COORDINATES | TriangleStripArray.COLOR_3
                            | TriangleStripArray.BY_REFERENCE | TriangleStripArray.USE_NIO_BUFFER,
                    stripVertexCounts);
            tetra.setCoordRefBuffer(floatBufferCoord);
            tetra.setColorRefBuffer(floatBufferColor);

        } else if (type == 3) { // Indexed Geometry
            tetra = new IndexedTriangleArray(4,
                    IndexedTriangleArray.COORDINATES | IndexedTriangleArray.COLOR_3
                            | IndexedTriangleArray.BY_REFERENCE | IndexedTriangleArray.USE_NIO_BUFFER,
                    //IndexedTriangleStripArray.USE_COORD_INDEX_ONLY,
                    12);
            tetra.setCoordRefBuffer(indexedFloatBufferCoord);
            tetra.setColorRefBuffer(indexedFloatBufferColor);
            ((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 | IndexedTriangleStripArray.USE_NIO_BUFFER
                            | IndexedTriangleStripArray.USE_COORD_INDEX_ONLY,
                    12, stripVertexCounts);
            tetra.setCoordRefBuffer(indexedFloatBufferCoord);
            tetra.setColorRefBuffer(indexedFloatBufferColor);
            ((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;
        float val1;
        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;

            FloatBuffer indexedCoord = (FloatBuffer) indexedFloatBufferCoord.getBuffer();
            indexedCoord.rewind();
            FloatBuffer coord = (FloatBuffer) floatBufferCoord.getBuffer();
            coord.rewind();

            if (vertexIndex == 0) {
                // Do Indexed geometry
                for (i = 0; i < indexedCoord.limit(); i += 3) {
                    val1 = indexedCoord.get(i);
                    indexedCoord.put(i, val1 + val);
                }
                // Do non-indexed float geometry
                for (i = 0; i < coord.limit(); i += 3) {
                    val1 = coord.get(i);
                    coord.put(i, val1 + val);
                }
            }
        } else if (updateIndex == 2) { // colors
            colorCount++;
            if ((colorCount & 1) == 1)
                val = 0.4f;
            else
                val = -0.4f;

            FloatBuffer indexedColors = (FloatBuffer) indexedFloatBufferColor.getBuffer();
            indexedColors.rewind();
            FloatBuffer colors = (FloatBuffer) floatBufferColor.getBuffer();
            colors.rewind();

            if (colorIndex == 0) {
                // Do Indexed geometry
                for (i = 0; i < indexedColors.limit(); i += 3) {
                    indexedColors.put(i, indexedColors.get(i) + val);
                }
                // Do non-indexed float geometry
                for (i = 0; i < colors.limit(); i += 3) {
                    colors.put(i, colors.get(i) + val);
                }
            }

        }

    }
}