Java Tutorial/2D Graphics/Screen Capture

Capture Screen

   <source lang="java">

import java.awt.Dimension; import java.awt.Graphics; import java.awt.Rectangle; import java.awt.Robot; import java.awt.Toolkit; import java.awt.image.BufferedImage; import javax.swing.JFrame; import javax.swing.JPanel; public class Capture {

 public static void main(String[] args) {
   JFrame capture = new JFrame();
   capture.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
   Toolkit kit = Toolkit.getDefaultToolkit();
   final Dimension d = kit.getScreenSize();
   capture.setSize(d);
   
   Rectangle rect = new Rectangle(d);
   try {
     Robot robot = new Robot();
     final BufferedImage image = robot.createScreenCapture(rect);
     image.flush();
     JPanel panel = new JPanel() {
       public void paintComponent(Graphics g) {
         g.drawImage(image, 0, 0, d.width, d.height, this);
       }
     };
     panel.setOpaque(false);
     panel.prepareImage(image, panel);
     panel.repaint();
     capture.getContentPane().add(panel);
   } catch (Exception e) {
     e.printStackTrace();
   }
   capture.setVisible(true);
 }

}</source>

Screen capture with crop operation

   <source lang="java">

import java.awt.Dimension; import java.awt.Graphics; import java.awt.Graphics2D; import java.awt.Image; import java.awt.Point; import java.awt.Rectangle; import java.awt.Robot; import java.awt.Toolkit; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.awt.event.MouseAdapter; import java.awt.event.MouseEvent; import java.awt.event.MouseMotionAdapter; import java.awt.image.BufferedImage; import java.io.File; import java.util.Iterator; import javax.imageio.IIOImage; import javax.imageio.ImageIO; import javax.imageio.ImageWriteParam; import javax.imageio.ImageWriter; import javax.imageio.stream.ImageOutputStream; import javax.swing.JFrame; import javax.swing.JMenu; import javax.swing.JMenuBar; import javax.swing.JMenuItem; import javax.swing.JPanel; import javax.swing.JScrollPane; public class ScreenCapture extends JFrame {

 ImageArea ia = new ImageArea();
 Rectangle rectScreenSize = new Rectangle(Toolkit.getDefaultToolkit().getScreenSize());
 static Robot robot;
 JScrollPane jsp;
 public ScreenCapture(String title) {
   super(title);
   setDefaultCloseOperation(EXIT_ON_CLOSE);
   JMenuBar mb = new JMenuBar();
   JMenu menu = new JMenu("File");
   ActionListener al;
   JMenuItem mi = new JMenuItem("Save");
   al = new ActionListener() {
     public void actionPerformed(ActionEvent e) {
       save();
     }
   };
   mi.addActionListener(al);
   menu.add(mi);
   mb.add(menu);
   menu = new JMenu("Capture");
   mi = new JMenuItem("Capture");
   al = new ActionListener() {
     public void actionPerformed(ActionEvent e) {
       setVisible(false);
       BufferedImage biScreen = robot.createScreenCapture(rectScreenSize);
       setVisible(true);
       ia.setImage(biScreen);
       jsp.getHorizontalScrollBar().setValue(0);
       jsp.getVerticalScrollBar().setValue(0);
     }
   };
   mi.addActionListener(al);
   menu.add(mi);
   mb.add(menu);
   mi = new JMenuItem("Crop");
   al = new ActionListener() {
     public void actionPerformed(ActionEvent e) {
       if (ia.crop()) {
         jsp.getHorizontalScrollBar().setValue(0);
         jsp.getVerticalScrollBar().setValue(0);
       }
     }
   };
   mi.addActionListener(al);
   menu.add(mi);
   mb.add(menu);
   setJMenuBar(mb);
   getContentPane().add(jsp = new JScrollPane(ia));
   setVisible(true);
 }
 private void save() {
   if (ia.getImage() == null) {
     System.out.println("No captured image.");
     return;
   }
   ImageWriter writer = null;
   ImageOutputStream ios = null;
   try {
     Iterator iter = ImageIO.getImageWritersByFormatName("jpeg");
     if (!iter.hasNext()) {
       System.out.println("Unable to save image to jpeg file type.");
       return;
     }
     writer = (ImageWriter) iter.next();
     ios = ImageIO.createImageOutputStream(new File("c:\\a.jpg"));
     writer.setOutput(ios);
     ImageWriteParam iwp = writer.getDefaultWriteParam();
     iwp.setCompressionMode(ImageWriteParam.MODE_EXPLICIT);
     iwp.setCompressionQuality(0.95f);
     writer.write(null,
         new IIOImage((BufferedImage) ia.getImage(), null, null), iwp);
   } catch (Exception e2) {
     e2.printStackTrace();
   }
 }
 public static void main(String[] args) throws Exception {
   robot = new Robot();
   new ScreenCapture("Capture");
 }

} class ImageArea extends JPanel {

 private Image image;
 Point startPoint = new Point(), endPoint = new Point();
 private Rectangle selectedRectangle = new Rectangle();
 public ImageArea() {
   addMouseListener(new MouseAdapter() {
     public void mousePressed(MouseEvent e) {
       if (image == null)
         return;
       startPoint = endPoint = e.getPoint();
       repaint();
     }
   });
   addMouseMotionListener(new MouseMotionAdapter() {
     public void mouseDragged(MouseEvent e) {
       if (image == null)
         return;
       endPoint = e.getPoint();
       repaint();
     }
   });
 }
 public boolean crop() {
   if (startPoint.equals(endPoint))
     return true;
   boolean succeeded = true;
   int x1 = (startPoint.x < endPoint.x) ? startPoint.x : endPoint.x;
   int y1 = (startPoint.y < endPoint.y) ? startPoint.y : endPoint.y;
   int x2 = (startPoint.x > endPoint.x) ? startPoint.x : endPoint.x;
   int y2 = (startPoint.y > endPoint.y) ? startPoint.y : endPoint.y;
   int width = (x2 - x1) + 1;
   int height = (y2 - y1) + 1;
   BufferedImage biCrop = new BufferedImage(width, height,
       BufferedImage.TYPE_INT_RGB);
   Graphics2D g2d = biCrop.createGraphics();
   BufferedImage bi = (BufferedImage) image;
   BufferedImage bi2 = bi.getSubimage(x1, y1, width, height);
   g2d.drawImage(bi2, null, 0, 0);
   g2d.dispose();
   if (succeeded)
     setImage(biCrop);
   else {
     startPoint.x = endPoint.x;
     startPoint.y = endPoint.y;
     repaint();
   }
   return succeeded;
 }
 public Image getImage() {
   return image;
 }
 public void paintComponent(Graphics g) {
   super.paintComponent(g);
   if (image != null)
     g.drawImage(image, 0, 0, this);
   if (startPoint.x != endPoint.x || startPoint.y != endPoint.y) {
     int x1 = (startPoint.x < endPoint.x) ? startPoint.x : endPoint.x;
     int y1 = (startPoint.y < endPoint.y) ? startPoint.y : endPoint.y;
     int x2 = (startPoint.x > endPoint.x) ? startPoint.x : endPoint.x;
     int y2 = (startPoint.y > endPoint.y) ? startPoint.y : endPoint.y;
     selectedRectangle.x = x1;
     selectedRectangle.y = y1;
     selectedRectangle.width = (x2 - x1) + 1;
     selectedRectangle.height = (y2 - y1) + 1;
     Graphics2D g2d = (Graphics2D) g;
     g2d.draw(selectedRectangle);
   }
 }
 public void setImage(Image image) {
   this.image = image;
   setPreferredSize(new Dimension(image.getWidth(this), image.getHeight(this)));
   revalidate();
   startPoint.x = endPoint.x;
   startPoint.y = endPoint.y;
   repaint();
 }

}</source>