java抓屏程序

package com.tomtop.test;

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;

public class Test2 {

 public static void main(String[] args) throws Exception {
  captureScreen("D:\\a.png");
 }
 
 public static void captureScreen(String fileName) throws Exception {      
  Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();    
  Rectangle screenRectangle = new Rectangle(screenSize);    
  Robot robot = new Robot();    
  BufferedImage image = robot.createScreenCapture(screenRectangle);    
  ImageIO.write(image, "png", new File(fileName));
  System.out.println("wancheng");
 }
}