Print Canvas3D
/*
* @(#)PrintCanvas3D.java 1.5 02/04/01 15:04:11
*
* 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.BorderLayout;
import java.awt.Color;
import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.GraphicsConfiguration;
import java.awt.Image;
import java.awt.Point;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.awt.image.ImageObserver;
import java.awt.print.PageFormat;
import java.awt.print.Printable;
import java.awt.print.PrinterException;
import java.awt.print.PrinterJob;
import java.io.FileNotFoundException;
import javax.media.j3d.Alpha;
import javax.media.j3d.AmbientLight;
import javax.media.j3d.Background;
import javax.media.j3d.BoundingSphere;
import javax.media.j3d.BranchGroup;
import javax.media.j3d.Canvas3D;
import javax.media.j3d.DirectionalLight;
import javax.media.j3d.ImageComponent;
import javax.media.j3d.ImageComponent2D;
import javax.media.j3d.RotationInterpolator;
import javax.media.j3d.Screen3D;
import javax.media.j3d.Transform3D;
import javax.media.j3d.TransformGroup;
import javax.swing.JFrame;
import javax.swing.JMenu;
import javax.swing.JMenuBar;
import javax.swing.JMenuItem;
import javax.swing.JPanel;
import javax.swing.JPopupMenu;
import javax.swing.JScrollPane;
import javax.swing.JSeparator;
import javax.vecmath.Color3f;
import javax.vecmath.Point3d;
import javax.vecmath.Vector3f;
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.behaviors.mouse.MouseTranslate;
import com.sun.j3d.utils.behaviors.mouse.MouseZoom;
import com.sun.j3d.utils.universe.SimpleUniverse;
public class PrintCanvas3D extends JFrame implements ActionListener {
private static final boolean spin = false;
private static final boolean noTriangulate = false;
private static final boolean noStripify = false;
private static final double creaseAngle = 60.0;
private JMenuItem snapshotItem;
private JMenuItem printItem;
private JMenuItem quitItem;
private SimpleUniverse u;
private Canvas3D canvas3D;
private OffScreenCanvas3D offScreenCanvas3D;
private static final int OFF_SCREEN_SCALE = 3;
private class AppPanel extends JPanel {
private String filename = null;
public BranchGroup createSceneGraph(String args[]) {
// Create the root of the branch graph
BranchGroup objRoot = new BranchGroup();
// Create a Transformgroup 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);
// Create the transform group node and initialize it to the
// identity. Enable the TRANSFORM_WRITE capability so that
// our behavior code can modify it at runtime. Add it to the
// root of the subgraph.
TransformGroup objTrans = new TransformGroup();
objTrans.setCapability(TransformGroup.ALLOW_TRANSFORM_WRITE);
objTrans.setCapability(TransformGroup.ALLOW_TRANSFORM_READ);
objScale.addChild(objTrans);
int flags = ObjectFile.RESIZE;
if (!noTriangulate)
flags |= ObjectFile.TRIANGULATE;
if (!noStripify)
flags |= ObjectFile.STRIPIFY;
ObjectFile f = new ObjectFile(flags,
(float) (creaseAngle * Math.PI / 180.0));
Scene s = null;
try {
s = f.load(filename);
} 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);
if (spin) {
Transform3D yAxis = new Transform3D();
Alpha rotationAlpha = new Alpha(-1, Alpha.INCREASING_ENABLE, 0,
0, 4000, 0, 0, 0, 0, 0);
RotationInterpolator rotator = new RotationInterpolator(
rotationAlpha, objTrans, yAxis, 0.0f,
(float) Math.PI * 2.0f);
rotator.setSchedulingBounds(bounds);
objTrans.addChild(rotator);
} else {
// Create the rotate behavior node
MouseRotate behavior = new MouseRotate();
behavior.setTransformGroup(objTrans);
objTrans.addChild(behavior);
behavior.setSchedulingBounds(bounds);
// Create the zoom behavior node
MouseZoom behavior2 = new MouseZoom();
behavior2.setTransformGroup(objTrans);
objTrans.addChild(behavior2);
behavior2.setSchedulingBounds(bounds);
// Create the translate behavior node
MouseTranslate behavior3 = new MouseTranslate();
behavior3.setTransformGroup(objTrans);
objTrans.addChild(behavior3);
behavior3.setSchedulingBounds(bounds);
}
// Set up the background
Color3f bgColor = new Color3f(0.05f, 0.05f, 0.5f);
Background bgNode = new Background(bgColor);
bgNode.setApplicationBounds(bounds);
objRoot.addChild(bgNode);
// Set up the ambient light
Color3f ambientColor = new Color3f(0.1f, 0.1f, 0.1f);
AmbientLight ambientLightNode = new AmbientLight(ambientColor);
ambientLightNode.setInfluencingBounds(bounds);
objRoot.addChild(ambientLightNode);
// Set up the directional lights
Color3f light1Color = new Color3f(1.0f, 1.0f, 0.9f);
Vector3f light1Direction = new Vector3f(4.0f, -7.0f, -12.0f);
Color3f light2Color = new Color3f(0.3f, 0.3f, 0.4f);
Vector3f light2Direction = new Vector3f(-6.0f, -2.0f, -1.0f);
DirectionalLight light1 = new DirectionalLight(light1Color,
light1Direction);
light1.setInfluencingBounds(bounds);
objRoot.addChild(light1);
DirectionalLight light2 = new DirectionalLight(light2Color,
light2Direction);
light2.setInfluencingBounds(bounds);
objRoot.addChild(light2);
return objRoot;
}
private void usage() {
System.out.println("Usage: java PrintCanvas3D <.obj file>");
System.exit(0);
} // End of usage
// Create the Canvas3D (both on-screen and off-screen)
private void createCanvas3D(String[] args) {
if (args.length == 0) {
usage();
} else {
for (int i = 0; i < args.length; i++) {
if (args[i].startsWith("-")) {
System.err.println("Argument '" + args[i]
+ "' ignored.");
} else {
filename = args[i];
}
}
}
if (filename == null) {
usage();
}
// Create Canvas3D
GraphicsConfiguration config = SimpleUniverse
.getPreferredConfiguration();
canvas3D = new Canvas3D(config);
canvas3D.setSize(600, 450);
// Create a simple scene and attach it to the virtual universe
BranchGroup scene = createSceneGraph(args);
u = new SimpleUniverse(canvas3D);
// This will move the ViewPlatform back a bit so the
// objects in the scene can be viewed.
u.getViewingPlatform().setNominalViewingTransform();
u.addBranchGraph(scene);
// Create the off-screen Canvas3D object
offScreenCanvas3D = new OffScreenCanvas3D(config, true);
// Set the off-screen size based on a scale factor times the
// on-screen size
Screen3D sOn = canvas3D.getScreen3D();
Screen3D sOff = offScreenCanvas3D.getScreen3D();
Dimension dim = sOn.getSize();
dim.width *= OFF_SCREEN_SCALE;
dim.height *= OFF_SCREEN_SCALE;
sOff.setSize(dim);
sOff.setPhysicalScreenWidth(sOn.getPhysicalScreenWidth()
* OFF_SCREEN_SCALE);
sOff.setPhysicalScreenHeight(sOn.getPhysicalScreenHeight()
* OFF_SCREEN_SCALE);
// attach the offscreen canvas to the view
u.getViewer().getView().addCanvas3D(offScreenCanvas3D);
}
private AppPanel(String args[]) {
setLayout(new BorderLayout());
// Create Canvas3D and scene graph
createCanvas3D(args);
add("Center", canvas3D);
}
}
public void actionPerformed(ActionEvent event) {
Object target = event.getSource();
if ((target == snapshotItem) || (target == printItem)) {
Point loc = canvas3D.getLocationOnScreen();
offScreenCanvas3D.setOffScreenLocation(loc);
Dimension dim = canvas3D.getSize();
dim.width *= OFF_SCREEN_SCALE;
dim.height *= OFF_SCREEN_SCALE;
BufferedImage bImage = offScreenCanvas3D.doRender(dim.width,
dim.height);
if (target == snapshotItem) {
new ImageDisplayer(bImage);
} else { // (target == printItem)
new ImagePrinter(bImage).print();
}
} else if (target == quitItem) {
u.removeAllLocales();
System.exit(0);
}
}
private JMenuBar createMenuBar() {
JMenuBar menuBar = new JMenuBar();
JMenu fileMenu = new JMenu("File");
snapshotItem = new JMenuItem("Snapshot");
snapshotItem.addActionListener(this);
printItem = new JMenuItem("Print...");
printItem.addActionListener(this);
quitItem = new JMenuItem("Quit");
quitItem.addActionListener(this);
fileMenu.add(snapshotItem);
fileMenu.add(printItem);
fileMenu.add(new JSeparator());
fileMenu.add(quitItem);
menuBar.add(fileMenu);
return menuBar;
}
private PrintCanvas3D(String args[]) {
this.setTitle("Canvas3D Print Test");
// Create and initialize menu bar
JPopupMenu.setDefaultLightWeightPopupEnabled(false);
this.setJMenuBar(createMenuBar());
// Handle the close event
this.addWindowListener(new WindowAdapter() {
public void windowClosing(WindowEvent winEvent) {
System.exit(0);
}
});
// Add main panel to top-level frame and make it visible
this.getContentPane().add(new AppPanel(args));
this.pack();
this.setVisible(true);
}
public static void main(String[] args) {
new PrintCanvas3D(args);
}
}
class OffScreenCanvas3D extends Canvas3D {
OffScreenCanvas3D(GraphicsConfiguration graphicsConfiguration,
boolean offScreen) {
super(graphicsConfiguration, offScreen);
}
BufferedImage doRender(int width, int height) {
BufferedImage bImage = new BufferedImage(width, height,
BufferedImage.TYPE_INT_ARGB);
ImageComponent2D buffer = new ImageComponent2D(
ImageComponent.FORMAT_RGBA, bImage);
setOffScreenBuffer(buffer);
renderOffScreenBuffer();
waitForOffScreenRendering();
bImage = getOffScreenBuffer().getImage();
return bImage;
}
public void postSwap() {
// No-op since we always wait for off-screen rendering to complete
}
}
class ImageDisplayer extends JFrame implements ActionListener {
BufferedImage bImage;
private class ImagePanel extends JPanel {
public void paint(Graphics g) {
g.setColor(Color.black);
g.fillRect(0, 0, getSize().width, getSize().height);
g.drawImage(bImage, 0, 0, this);
}
private ImagePanel() {
setPreferredSize(new Dimension(bImage.getWidth(), bImage
.getHeight()));
}
}
private JMenuItem printItem;
private JMenuItem closeItem;
public void actionPerformed(ActionEvent event) {
Object target = event.getSource();
if (target == printItem) {
new ImagePrinter(bImage).print();
} else if (target == closeItem) {
this.removeAll();
this.setVisible(false);
bImage = null;
}
}
private JMenuBar createMenuBar() {
JMenuBar menuBar = new JMenuBar();
JMenu fileMenu = new JMenu("File");
printItem = new JMenuItem("Print...");
printItem.addActionListener(this);
closeItem = new JMenuItem("Close");
closeItem.addActionListener(this);
fileMenu.add(printItem);
fileMenu.add(new JSeparator());
fileMenu.add(closeItem);
menuBar.add(fileMenu);
return menuBar;
}
ImageDisplayer(BufferedImage bImage) {
this.bImage = bImage;
this.setTitle("Off-screen Canvas3D Snapshot");
// Create and initialize menu bar
this.setJMenuBar(createMenuBar());
// Create scroll pane, and embedded image panel
ImagePanel imagePanel = new ImagePanel();
JScrollPane scrollPane = new JScrollPane(imagePanel);
scrollPane.getViewport().setPreferredSize(new Dimension(700, 700));
// Add scroll pane to the frame and make it visible
this.getContentPane().add(scrollPane);
this.pack();
this.setVisible(true);
}
}
class ImagePrinter implements Printable, ImageObserver {
BufferedImage bImage;
public int print(Graphics g, PageFormat pf, int pi) throws PrinterException {
if (pi >= 1) {
return Printable.NO_SUCH_PAGE;
}
Graphics2D g2d = (Graphics2D) g;
//g2d.translate(pf.getImageableX(), pf.getImageableY());
AffineTransform t2d = new AffineTransform();
t2d.translate(pf.getImageableX(), pf.getImageableY());
double xscale = pf.getImageableWidth() / (double) bImage.getWidth();
double yscale = pf.getImageableHeight() / (double) bImage.getHeight();
double scale = Math.min(xscale, yscale);
t2d.scale(scale, scale);
try {
g2d.drawImage(bImage, t2d, this);
} catch (Exception ex) {
ex.printStackTrace();
return Printable.NO_SUCH_PAGE;
}
return Printable.PAGE_EXISTS;
}
void print() {
PrinterJob printJob = PrinterJob.getPrinterJob();
PageFormat pageFormat = printJob.defaultPage();
pageFormat.setOrientation(PageFormat.LANDSCAPE);
pageFormat = printJob.validatePage(pageFormat);
printJob.setPrintable(this, pageFormat);
if (printJob.printDialog()) {
try {
printJob.print();
} catch (PrinterException ex) {
ex.printStackTrace();
}
}
}
public boolean imageUpdate(Image img, int infoflags, int x, int y,
int width, int height) {
return false;
}
ImagePrinter(BufferedImage bImage) {
this.bImage = bImage;
}
}
Related examples in the same category
