PrintCanvas3D.java Source code

Java tutorial

Introduction

Here is the source code for PrintCanvas3D.java

Source

/*
 * @(#)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;
    }
}