selenium中java代码截图

public void getScreenShot(){
        try {
            System.setProperty("webdriver.firefox.bin", "E:/火狐浏览器/Mozilla Firefox/firefox.exe");
            WebDriver driver = new FirefoxDriver();
            driver.get("http://www.baidu.com/");
            SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss");      //转换时间格式
            String time = dateFormat.format(Calendar.getInstance().getTime());      //获取当前时间
            Thread.sleep(2000);
            File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);  //执行屏幕截取
            FileUtils.copyFile(srcFile, new File("F:\\资料图片", time + ".jpg"));
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } 
        //利用FileUtils工具类的copyFile()方法保存getScreenshotAs()返回的文件;"屏幕截图"即时保存截图的文件夹
    }