package photo;

import java.awt.Color;

import java.awt.Font; import java.awt.Graphics; import java.awt.image.BufferedImage; import java.io.ByteArrayInputStream; import java.io.ByteArrayOutputStream; import java.io.InputStream; import java.util.Random;

import com.sun.image.codec.jpeg.JPEGCodec;

import com.sun.image.codec.jpeg.JPEGImageEncoder;

public class ImageCodeAction  {  private InputStream input;

 private static int WIDTH = 300;

 private static int HEIGHT = 100;

 private static int NUM = 4;

 private static char[] seq = { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I',

   'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V',    'W', 'X', 'Y', 'Z', '0', '1', '2', '3', '4', '5', '6', '7', '8',    '9' };

 public String execute() {

  byte[] image = randomImage();   input = new ByteArrayInputStream(image);   return "success";  }

 /**

  *   * @return   */  private byte[] randomImage() {   Random r = new Random();

  // 图片的内存映像

  BufferedImage image = new BufferedImage(WIDTH, HEIGHT,     BufferedImage.TYPE_INT_RGB);

  // 获得画笔对象

  Graphics g = image.getGraphics();   g.setColor(randomColor(r));   g.fillRect(0, 0, WIDTH, HEIGHT);   g.setColor(new Color(0, 0, 0));

  // 用于存储随机生成的验证码

  StringBuffer number = new StringBuffer();

  // 绘制验证码

  for (int i = 0; i < NUM; i++) {    g.setColor(randomColor(r));    int h = (int) ((HEIGHT * 60 / 100) * r.nextDouble() + (HEIGHT * 30 / 100));    g.setFont(new Font(null, Font.BOLD | Font.ITALIC, h));    String ch = String.valueOf(seq[r.nextInt(seq.length)]);    number.append(ch);    g.drawString(ch, i * WIDTH / NUM * 90 / 100, h);   }

  session.put("code", number.toString());

  System.out.println(number.toString());

  // 绘制干扰线

  for (int i = 0; i <= 12; i++) {    g.setColor(randomColor(r));    g.drawLine(r.nextInt(WIDTH), r.nextInt(HEIGHT), r.nextInt(WIDTH), r      .nextInt(HEIGHT));

  }

  ByteArrayOutputStream os = new ByteArrayOutputStream();

  JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(os);

  // 把BufferedImage对象中的图像信息编码后

  // 向创建该对象(encoder)时指定的输出流输出   try {    encoder.encode(image);    return os.toByteArray();   } catch (Exception e) {    throw new RuntimeException(e);   }  }

 private Color randomColor(Random r) {

  return new Color(r.nextInt(255), r.nextInt(255), r.nextInt(255));  }

 public InputStream getInput() {

  return input;  }

 public void setInput(InputStream input) {

  this.input = input;  }

}