package screenshotmaker;

import java.awt.Dimension;
import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.File;
import javax.imageio.ImageIO;

/**
 * Simple class to capture the full screen or a rectangle on the screen.
 * @author  Jean-Francois Denise
 */
public class ScreenCapturer {
    /**
     * Capture a rectangle
     * @param rect The rectangle
     * @return The img file
     * @throws java.lang.Exception In case capture fails
     */
     public static File capture(Rectangle rect) throws Exception {
        Robot robot = new Robot();
        BufferedImage img = robot.createScreenCapture(rect);
        File file =  File.createTempFile("jfx_screen_capture_retouched",".jpg");
        ImageIO.write(img, "jpg", file);
        return file;
    }

    /**
     * Capture the full screen
     * @return The img file
     * @throws java.lang.Exception In case capture fails
     */
    public static File capture() throws Exception {
        Toolkit toolkit = Toolkit.getDefaultToolkit();
        Dimension screenSize = toolkit.getScreenSize();
        Rectangle screenRect = new Rectangle(screenSize);
        Robot robot = new Robot();
        BufferedImage image = robot.createScreenCapture(screenRect);
        File file =  File.createTempFile("jfx_screen_capture",".jpg");
        ImageIO.write(image, "jpg", file);
        return file;
    }
    /**
     * The current screen resolution
     * @return  The current screen resolution
     * @throws java.lang.Exception In case any exception occurs
     */
    public static Dimension getScreenDimension() throws Exception {
        Toolkit toolkit = Toolkit.getDefaultToolkit();
        return toolkit.getScreenSize();
    }
}