Selenium Webdriver实现截图功能

前几天在研究中自动化的时候突发奇想，想着能不能来截个图，以便之后查看，实现的方法其实也不难，毕竟selenium webdriver已经提供了截图额功能，TakesScreenshot接口函数（英文意思就是获取屏幕截图takes-screenshot）。废话不多说了，直接上代码

package com.wch;

import java.io.File;
import java.io.IOException;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.support.ui.WebDriverWait;

import com.sun.jna.platform.FileUtils;

public class TestTakesScreenshot {
	public static void main(String[] args) {
		System.setProperty("webdriver.firefox.bin", "D:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe");
		WebDriver driver = new FirefoxDriver();
		driver.get("http://www.baidu.com");
		File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); //讲截取的图片以文件的形式返回
		try {
			org.apache.commons.io.FileUtils.copyFile(srcFile, new File("d:\\screenshot.png"));  //使用copyFile()方法保存获取到的截图文件
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		driver.quit();
	}
}

还有其他方法的话希望各位也能提供下，互相学习