package com.xn;
 
import javax.imageio.ImageIO; 
import java.awt.*; 
import java.awt.image.BufferedImage; 
import java.io.FileOutputStream; 
import java.io.IOException; 
import java.io.OutputStream; 
import java.util.Date; 
import java.util.Random; 
 
/** 
 * 驗證碼生成工具類
 * 
 * @leo 
 */ 
public class ValidateCode { 
 // 圖片的寬度。 
 private int width = 160; 
 // 圖片的高度。 
 private int height = 40; 
 // 驗證碼字符個數 
 private int codeCount = 5; 
 // 驗證碼干擾線數 
 private int lineCount = 150; 
 // 驗證碼 
 private String code = null; 
 // 驗證碼圖片Buffer 
 private BufferedImage buffImg = null; 
 
 // 驗證碼范圍,去掉0(數字)和O(拼音)容易混淆的(小寫的1和L也可以去掉,大寫不用了) 
 private char[] codeSequence = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 
   'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 
   'X', 'Y', 'Z', '1', '2', '3', '4', '5', '6', '7', '8', '9'}; 
 
 /** 
  * 默認構造函數,設置默認參數 
  */ 
 public ValidateCode() { 
  this.createCode(); 
 } 
 
 /** 
  * @param width 圖片寬 
  * @param height 圖片高 
  */ 
 public ValidateCode(int width, int height) { 
  this.width = width; 
  this.height = height; 
  this.createCode(); 
 } 
 
 /** 
  * @param width  圖片寬 
  * @param height 圖片高 
  * @param codeCount 字符個數 
  * @param lineCount 干擾線條數 
  */ 
 public ValidateCode(int width, int height, int codeCount, int lineCount) { 
  this.width = width; 
  this.height = height; 
  this.codeCount = codeCount; 
  this.lineCount = lineCount; 
  this.createCode(); 
 } 
 
 public void createCode() { 
  int x = 0, fontHeight = 0, codeY = 0; 
  int red = 0, green = 0, blue = 0; 
 
  x = width / (codeCount + 2);//每個字符的寬度(左右各空出一個字符) 
  fontHeight = height - 2;//字體的高度 
  codeY = height - 4; 
 
  // 圖像buffer 
  buffImg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB); 
  Graphics2D g = buffImg.createGraphics();  
   
  /*// 將圖像背景填充為白色 
  g.setColor(Color.WHITE); 
  g.fillRect(0, 0, width, height);*/
  // 增加下面代碼使得背景透明 
  buffImg = g.getDeviceConfiguration().createCompatibleImage(width, height, Transparency.TRANSLUCENT); 
  g.dispose(); 
  g = buffImg.createGraphics(); 
  // 背景透明代碼結束 
   
  // 畫圖BasicStroke是JDK中提供的一個基本的畫筆類,我們對他設置畫筆的粗細，就可以在drawPanel上任意畫出自己想要的圖形了。 
  g.setColor(new Color(255, 0, 0)); 
  g.setStroke(new BasicStroke(1f)); 
  g.fillRect(128, 128, width, height);
  
  // 生成隨機數 
  Random random = new Random(); 
      //設置字體類型、字體大小、字體樣式　
    Font font = new Font("微軟雅黑",Font.PLAIN, fontHeight);
 
  g.setFont(font); 
 
  for (int i = 0; i < lineCount; i++) { 
   // 設置隨機開始和結束坐標 
   int xs = random.nextInt(width);//x坐標開始 
   int ys = random.nextInt(height);//y坐標開始 
   int xe = xs + random.nextInt(width / 8);//x坐標結束 
   int ye = ys + random.nextInt(height / 8);//y坐標結束 
 
   // 產生隨機的顏色值，讓輸出的每個干擾線的顏色值都將不同。 
   red = random.nextInt(255); 
   green = random.nextInt(255); 
   blue = random.nextInt(255); 
   g.setColor(new Color(red, green, blue)); 
   g.drawLine(xs, ys, xe, ye); 
  } 
 
  // randomCode記錄隨機產生的驗證碼 
  StringBuffer randomCode = new StringBuffer(); 
  // 隨機產生codeCount個字符的驗證碼。 
  for (int i = 0; i < codeCount; i++) { 
   String strRand = String.valueOf(codeSequence[random.nextInt(codeSequence.length)]); 
   // 產生隨機的顏色值，讓輸出的每個字符的顏色值都將不同。 
   red = random.nextInt(255); 
   green = random.nextInt(255); 
   blue = random.nextInt(255); 
   //指定某種顏色
   //g.setColor(new Color(252, 145, 83));
   g.setColor(new Color(red, green, blue)); 
   g.drawString(strRand, (i + 1) * x, codeY); 
   // 將產生的四個隨機數組合在一起。 
   randomCode.append(strRand); 
  } 
  // 將四位數字的驗證碼保存到Session中。 
  code = randomCode.toString(); 
 } 
 
 public void write(String path) throws IOException { 
  OutputStream sos = new FileOutputStream(path); 
  this.write(sos); 
 } 
 
 public void write(OutputStream sos) throws IOException { 
  ImageIO.write(buffImg, "png", sos); 
  sos.close(); 
 } 
 
 public BufferedImage getBuffImg() { 
  return buffImg; 
 } 
 
 public String getCode() { 
  return code; 
 } 
 
 /** 
  * 測試函數,默認生成到d盤 
  * @param args 
  */ 
 public static void main(String[] args) { 
  ValidateCode vCode = new ValidateCode(160,40,5,150); 
  try { 
   String path="D:/"+new Date().getTime()+".png"; 
   System.out.println(vCode.getCode()+" >"+path); 
   vCode.write(path); 
  } catch (IOException e) { 
   e.printStackTrace(); 
  } 
 } 
}