3D Graphics Java

/*
 *  @(#)OffScreenTest.java 1.13 02/10/21 13:46:49
 *
 * 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.applet.Applet;
import java.awt.BorderLayout;
import java.awt.GraphicsConfiguration;
import java.awt.image.BufferedImage;
import javax.media.j3d.BranchGroup;
import javax.media.j3d.Canvas3D;
import javax.media.j3d.ImageComponent;
import javax.media.j3d.ImageComponent2D;
import javax.media.j3d.Raster;
import javax.media.j3d.Screen3D;
import javax.media.j3d.Shape3D;
import javax.media.j3d.Transform3D;
import javax.media.j3d.TransformGroup;
import javax.media.j3d.View;
import javax.vecmath.Point3f;
import javax.vecmath.Vector3d;
import com.sun.j3d.utils.applet.MainFrame;
import com.sun.j3d.utils.geometry.ColorCube;
import com.sun.j3d.utils.universe.SimpleUniverse;
/**
 * OffScreenTest issues renderOffScreenBuffer from the postSwap callback of the
 * OnScreen canvas.
 */
public class OffScreenTest extends Applet {
  private SimpleUniverse u = null;
  public BranchGroup createSceneGraph(Raster drawRaster) {
    // Create the root of the branch graph
    BranchGroup objRoot = new BranchGroup();
    // spin object has composited transformation matrix
    Transform3D spin = new Transform3D();
    Transform3D tempspin = new Transform3D();
    spin.rotX(Math.PI / 4.0d);
    tempspin.rotY(Math.PI / 5.0d);
    spin.mul(tempspin);
    spin.setScale(0.7);
    spin.setTranslation(new Vector3d(-0.4, 0.3, 0.0));
    TransformGroup objTrans = new TransformGroup(spin);
    objRoot.addChild(objTrans);
    // Create a simple shape leaf node, add it to the scene graph.
    // ColorCube is a Convenience Utility class
    objTrans.addChild(new ColorCube(0.4));
    //Create a raster
    Shape3D shape = new Shape3D(drawRaster);
    objRoot.addChild(shape);
    // Let Java 3D perform optimizations on this scene graph.
    objRoot.compile();
    return objRoot;
  }
  public OffScreenTest() {
  }
  public void init() {
    setLayout(new BorderLayout());
    GraphicsConfiguration config = SimpleUniverse
        .getPreferredConfiguration();
    BufferedImage bImage = new BufferedImage(200, 200,
        BufferedImage.TYPE_INT_ARGB);
    ImageComponent2D buffer = new ImageComponent2D(
        ImageComponent.FORMAT_RGBA, bImage);
    buffer.setCapability(ImageComponent2D.ALLOW_IMAGE_READ);
    Raster drawRaster = new Raster(new Point3f(0.0f, 0.0f, 0.0f),
        Raster.RASTER_COLOR, 0, 0, 200, 200, buffer, null);
    drawRaster.setCapability(Raster.ALLOW_IMAGE_WRITE);
    // create the main scene graph
    BranchGroup scene = createSceneGraph(drawRaster);
    // create the on-screen canvas
    OnScreenCanvas3D d = new OnScreenCanvas3D(config, false);
    add("Center", d);
    // create a simple universe
    u = new SimpleUniverse(d);
    // This will move the ViewPlatform back a bit so the
    // objects in the scene can be viewed.
    u.getViewingPlatform().setNominalViewingTransform();
    // create an off Screen Canvas
    OffScreenCanvas3D c = new OffScreenCanvas3D(config, true, drawRaster);
    // set the offscreen to match the onscreen
    Screen3D sOn = d.getScreen3D();
    Screen3D sOff = c.getScreen3D();
    sOff.setSize(sOn.getSize());
    sOff.setPhysicalScreenWidth(sOn.getPhysicalScreenWidth());
    sOff.setPhysicalScreenHeight(sOn.getPhysicalScreenHeight());
    // attach the same view to the offscreen canvas
    View v = u.getViewer().getView();
    v.addCanvas3D(c);
    // tell onscreen about the offscreen so it knows to
    // render to the offscreen at postswap
    d.setOffScreenCanvas(c);
    u.addBranchGraph(scene);
    v.stopView();
    // Make sure that image are render completely
    // before grab it in postSwap().
    d.setImageReady();
    v.startView();
  }
  public void destroy() {
    u.cleanup();
  }
  public static void main(String argv[]) {
    new MainFrame(new OffScreenTest(), 500, 500);
  }
}
class OffScreenCanvas3D extends Canvas3D {
  Raster drawRaster;
  boolean printing = false;
  public OffScreenCanvas3D(GraphicsConfiguration gconfig,
      boolean offscreenflag, Raster drawRaster) {
    super(gconfig, offscreenflag);
    this.drawRaster = drawRaster;
  }
  public void print(boolean toWait) {
    if (!toWait)
      printing = true;
    BufferedImage bImage = new BufferedImage(200, 200,
        BufferedImage.TYPE_INT_ARGB);
    ImageComponent2D buffer = new ImageComponent2D(
        ImageComponent.FORMAT_RGBA, bImage);
    buffer.setCapability(ImageComponent2D.ALLOW_IMAGE_READ);
    this.setOffScreenBuffer(buffer);
    this.renderOffScreenBuffer();
    if (toWait) {
      this.waitForOffScreenRendering();
      drawOffScreenBuffer();
    }
  }
  public void postSwap() {
    if (printing) {
      super.postSwap();
      drawOffScreenBuffer();
      printing = false;
    }
  }
  void drawOffScreenBuffer() {
    BufferedImage bImage = this.getOffScreenBuffer().getImage();
    ImageComponent2D newImageComponent = new ImageComponent2D(
        ImageComponent.FORMAT_RGBA, bImage);
    drawRaster.setImage(newImageComponent);
  }
}
class OnScreenCanvas3D extends Canvas3D {
  OffScreenCanvas3D c;
  boolean print = false;
  boolean imageReady = false;
  public OnScreenCanvas3D(GraphicsConfiguration gconfig, boolean offscreenflag) {
    super(gconfig, offscreenflag);
  }
  public void setOffScreenCanvas(OffScreenCanvas3D c) {
    this.c = c;
  }
  public void setImageReady() {
    imageReady = true;
  }
  public void postSwap() {
    if (imageReady && !print) {
      c.print(false);
      print = true;
    }
  }
}