Форум программистов [Powered by Invision Power Board]

Код


public class CanvasSupplier extends Applet
{
    private BufferedImage bufferedImage;
    
    public void createImage()
    {
        GraphicsConfiguration gc = SimpleUniverse.getPreferredConfiguration();
        Canvas3D cv = new Canvas3D(gc);
        setLayout(new BorderLayout());
        add(cv, BorderLayout.CENTER);
        BranchGroup bg = createSceneGraph();
        bg.compile();
        SimpleUniverse su = new SimpleUniverse(cv);
        Locale locale = new Locale(su);
        su.getViewingPlatform().setNominalViewingTransform();
        su.addBranchGraph(bg);

        bufferedImage = takeScreenshot(cv);
    }

    public BufferedImage takeScreenshot(Canvas3D canvas)
    {
        Screen3D on = canvas.getScreen3D();

        Canvas3D shot = new Canvas3D(canvas.getGraphicsConfiguration(), true);
        canvas.getView().stopView();
        canvas.getView().addCanvas3D(shot);
        canvas.getView().startView();


        Screen3D off = shot.getScreen3D();
        off.setSize(on.getSize());
        off.setPhysicalScreenHeight(on.getPhysicalScreenHeight());
        off.setPhysicalScreenWidth(on.getPhysicalScreenWidth());
        shot.setOffScreenLocation(canvas.getLocationOnScreen());

        BufferedImage bi = new BufferedImage(500, 500, BufferedImage.TYPE_INT_ARGB);
        ImageComponent2D buffer = new ImageComponent2D(ImageComponent.FORMAT_RGBA, bi);
        shot.setOffScreenBuffer(buffer);
        shot.renderOffScreenBuffer();
        shot.waitForOffScreenRendering();
        BufferedImage res = shot.getOffScreenBuffer().getImage();
        canvas.getView().removeCanvas3D(shot);
        return res;

    }

    public BranchGroup createSceneGraph()
    {
        BranchGroup objRoot = new BranchGroup();
        TransformGroup objSpin = new TransformGroup();
        objSpin.setCapability(TransformGroup.ALLOW_TRANSFORM_WRITE);
        objRoot.addChild(objSpin);
        objSpin.addChild(new ColorCube(0.4));
          return objRoot;
    }

    public BufferedImage getBufferedImage()
    {
        return bufferedImage;
    }
}


public class Test3dBean extends BaseBean
{
    public BufferedImage getBufferedImage()
    {
       try
        {
         CanvasSupplier canvasSupplier = new CanvasSupplier();
         canvasSupplier.createImage();
         bufferedImage = canvasSupplier.getBufferedImage();

            ImageIO.write(bufferedImage, "png", new File("C:\\image123.png"));
        }
        catch (IOException e)
        {
            System.out.println(e.getMessage());
        }

        return bufferedImage;
    }
   
}