source: main/trunk/gli/src/org/greenstone/gatherer/feedback/ScreenShot.java@ 24915

package org.greenstone.gatherer.feedback;

import java.awt.*;
import java.applet.Applet;
import java.awt.image.*;
import java.awt.event.WindowEvent;
import java.awt.event.WindowListener;
import java.awt.event.WindowAdapter;
import java.awt.event.*;
import java.awt.image.BufferedImage;
import java.awt.image.DataBuffer;
import java.awt.geom.GeneralPath;
import java.io.*;
import javax.imageio.*;
import javax.swing.*; 
import com.sun.image.codec.jpeg.*;
import java.awt.geom.*;

/**
 * This class can take screen shot of the whole screen size or screen shot of a specified
 * rectangle only.
 * @author Veronica Liesaputra
 */
public class ScreenShot
{
    /**
     * This is the buffered image that hold the screen shot.
     */
    private BufferedImage img;
    
    /**
     * The dimension of the whole screen size.
     */
    private final static Dimension dim = Toolkit.getDefaultToolkit().getScreenSize();
    
    /**
     * Rectangle with a dimension of a whole screen size.
     */
    private Rectangle mRect = new Rectangle(dim);
    
    /**
     * The desired width of the image.
     */
    private int iw;

    /**
     * The desired height of the image.
     */
    private int ih;

    /**
     * BufferedImage that hold the screen shot image with the desired height and width.
     */
    private BufferedImage bi;
    
    /**
     * This is the robot that allows to take screen shot.
     */
    private Robot mRobot;
  
    /**
     * This constructor will setup the robot to alow taking a screen shot.
     */
    public ScreenShot () 
    {
    try
        {
        mRobot = new Robot();
        iw = (int) (dim.getWidth() - 60);
        ih = (int) (dim.getHeight() * 0.75) - 50;
        }
    catch (AWTException exp) {}
    }

    /**
     * This method will only take screen shot of the particular rectangle.
     * @param rect is the specified rectangle that we want to take the screen shot of.
     * @return     the image taken.
     */
    //public byte[] getImageIcon (Rectangle rect)
    public BufferedImage getImageIcon (Rectangle rect)
    {
    BufferedImage image;
    image = null;
    
    //try
    //  {
        img = mRobot.createScreenCapture(rect);
        
        int w,h;
        boolean rescale;

        rescale = false;
        w = rect.width;
        
        if (w >= iw)
            {
            w = iw - 50;
            rescale = true;
            }
        
        h = rect.height;
        
        if ( h >= ih)
            {
            h = ih  - 50;
            rescale = true;
            }
        
        Graphics2D big;
        big = null;
        
        if (rescale == true)
            {
            image = new BufferedImage(w,h, BufferedImage.TYPE_INT_RGB);
            big = image.createGraphics();
            big.drawImage(img.getScaledInstance(w,h,Image.SCALE_SMOOTH),0,0,null);
            }
        else
            {
            image = img;
            }

        
        /*ByteArrayOutputStream stream;
        stream = new ByteArrayOutputStream();
        ImageIO.write(image,"jpeg",stream);
        byte[] bytes;
        bytes = stream.toByteArray();*/
        
        if (big != null)
            big.dispose();
        //image = null;
        //img = null;
        ActionRecorderDialog.setSavefinish(true);
        return image;
        //return bytes;
        //  }
        //catch(IOException ex) {ex.printStackTrace();}
    
    //return null;
    }

    /**
     * This method will only take screen shot of the particular rectangle and encode
     * the image to its string representation.
     * @param rect is the specified rectangle that we want to take the screen shot of.
     * @return     the string representation of the image taken.
     */
    public String getImage (Rectangle rect)
    {
    String txt;
    txt = null;

    BufferedImage image;
    image = null;
    
    try
        {
        img = mRobot.createScreenCapture(rect);
        
        int w,h;
        boolean rescale;

        rescale = false;
        w = rect.width;
        
        if (w >= iw)
            {
            w = iw - 50;
            rescale = true;
            }
        
        h = rect.height;
        
        if ( h >= ih)
            {
            h = ih  - 50;
            rescale = true;
            }
    
        Graphics2D big;
        big = null;

        if (rescale == true)
            {
            image = new BufferedImage(w,h, BufferedImage.TYPE_INT_RGB);
            big = image.createGraphics();
            big.drawImage(img.getScaledInstance(w,h,Image.SCALE_SMOOTH),0,0,null);
            }
        else
            {
            image = img;
            }

        ByteArrayOutputStream stream;
        stream = new ByteArrayOutputStream();
        ImageIO.write(image,"gif",stream);
        byte[] bytes;
        bytes = stream.toByteArray();
        txt = Base64.encodeBytes(bytes);
        
        bytes = null;
        if (big != null)
            big.dispose();
        image = null;
        img = null;
        }
    catch(IOException ex) {ex.printStackTrace();}
    
    ActionRecorderDialog.setSavefinish(true);
    
    return txt;
    }

    /**
     * This method will return the buffered image of the whole screen size screen shot
     * with the desired width and height.
     * @return the buffered image with the desired height and width.
     */
    public BufferedImage getBuffImage()
    {
    return bi;
    }

    /**
     * This method will take screen shot of the whole screen and resize it to the
     * desired width and height.
     */
    public BufferedImage captureScreen()
    {
    img = mRobot.createScreenCapture(mRect);
    
    Graphics2D big;
    bi = new BufferedImage(iw,ih, BufferedImage.TYPE_INT_RGB);
    big = bi.createGraphics();
    big.drawImage(img.getScaledInstance(iw,ih,Image.SCALE_SMOOTH),0,0,null);
    big.dispose();
    big = null;

    return bi;
    }

    /**
     * This method will return the desired width.
     * @return the desired width.
     */
    public int getWidth()
    {
    return iw;
    }

    /**
     * This method will return the desired height.
     * @return the desired height.
     */
    public int getHeight()
    {
    return ih;
    }
}