Selenium2学习-033-WebUI自动化实战实例-031-页面快照截图应用之二 -- 区域截图

 1     /**
 2      * Get basic snapshot for expected area of display screen area
 3      * 
 4      * @author Aaron.ffp
 5      * @version V1.0.0: autoSeleniumDemo main.aaron.sele.core SeleniumCore.java snapshotPartial, 2015-7-28 01:41:12 Exp $
 6      * 
 7      * @param filename : store png file name
 8      * @param left     : left distance
 9      * @param top      : top distance
10      * @param width    : width distance
11      * @param height   : height distance
12      *  
13      * @return boolean
14      */
15     public boolean snapshotPartial(String filename, int left, int top, int width, int height){
16         boolean success = false;
17         
18         try {
19             // Get byte data of full screen capture
20             byte[] byte_screen_capture = ((TakesScreenshot) this.webdriver).getScreenshotAs(OutputType.BYTES);
21             
22             // create full screen capture
23             BufferedImage img_screen_catpture = ImageIO.read(new ByteArrayInputStream(byte_screen_capture));
24             
25             // get partial image by location and size
26             BufferedImage partial_screen_capture = img_screen_catpture.getSubimage(left, top, width, height);
27             
28             File f = new File(filename);
29             
30             if (f.isFile() && f.exists()) {
31                 f.delete();
32             }
33             
34             // store partial image
35             ImageIO.write(partial_screen_capture, "png", f);
36             
37             success = true;
38         } catch (IOException ioe_sci) {
39             ioe_sci.printStackTrace();
40         } catch (RasterFormatException rfe) {
41             rfe.printStackTrace();
42         }
43         
44         return success;
45     }