source: ether_megapoli/trunk/service/implementation/com/ether/JPanelToImageManager.java @ 130

package com.ether;

import gov.noaa.pmel.sgt.JPane;
import gov.noaa.pmel.sgt.swing.JPlotLayout;
import org.jetbrains.annotations.NotNull;

import javax.swing.*;
import java.awt.*;
import java.awt.image.BufferedImage;

/**
 * @Author : E.Mo..
 * This class convert a JPanel into a BufferedImage
 */
public class JPanelToImageManager
{
    public JPanelToImageManager( @NotNull final JPane jPanel )
    {
        _target = jPanel;
    }

    /**
     * @return the image corresponding to the JPanel
     */
    public BufferedImage saveAsImage()
    {
        return paintComponent( _target );
    }

    /**
     * @param component the component to convert
     * @return the buffer of the converted image
     */
    private BufferedImage paintComponent( @NotNull final Component component )
    {
        if( 0 >= component.getWidth() || 0 >= component.getHeight() )
            return null;

        // Set it to it's preferred size. (optional)
        component.setSize( component.getPreferredSize() );
        layoutComponent( component );

        final BufferedImage bufferedImage = new BufferedImage( component.getWidth(), component.getHeight(), BufferedImage.TYPE_INT_RGB );

        if( component instanceof JPlotLayout )
        {
            final Graphics2D graphics2D = bufferedImage.createGraphics();
            graphics2D.setBackground( Color.red );
            ( (JPlotLayout) component ).draw( graphics2D );
        }
        else
        {
            final CellRendererPane crp = new CellRendererPane();
            crp.add( component );
            crp.paintComponent( bufferedImage.createGraphics(), component, crp, component.getBounds() );
        }
        return bufferedImage;
    }

    private void layoutComponent( final Component component )
    {
        synchronized( component.getTreeLock() )
        {
            component.doLayout();
            if( component instanceof Container )
            {
                for( final Component child : ( (Container) component ).getComponents() )
                    layoutComponent( child );
            }
        }
    }

    // Panel to convert into an image
    @NotNull
    private final JPane _target;
}