1. package com.lms.jdk8.util;
    3. import javax.imageio.ImageIO;
    4. import java.awt.*;
    5. import java.awt.geom.AffineTransform;
    6. import java.awt.image.BufferedImage;
    7. import java.io.File;
    8. import java.io.FileOutputStream;
    9. import java.io.IOException;
    10. import java.io.OutputStream;
    11. import java.util.Arrays;
    12. import java.util.Random;
    14. /**
    15.  *@描述   验证码生成
    16.  */
    17. public class VerifyCodeUtil{
    19.     //使用到Algerian字体，系统里没有的话需要安装字体，字体只显示大写，去掉了1,0,i,o几个容易混淆的字符
    20.     public static final String VERIFY_CODES = "23456789ABCDEFGHJKLMNPQRSTUVWXYZ";
    21.     private static Random random = new Random();
    24.     /**
    25.      * 使用系统默认字符源生成验证码
    26.      * @param verifySize    验证码长度
    27.      */
    28.     public static String generateVerifyCode(int verifySize){
    29.         return generateVerifyCode(verifySize, VERIFY_CODES);
    30.     }
    31.     /**
    32.      * 使用指定源生成验证码
    33.      * @param verifySize    验证码长度
    34.      * @param sources   验证码字符源
    35.      * @return
    36.      */
    37.     public static String generateVerifyCode(int verifySize, String sources){
    38.         if(sources == null || sources.length() == 0){
    39.             sources = VERIFY_CODES;
    40.         }
    41.         int codesLen = sources.length();
    42.         Random rand = new Random(System.currentTimeMillis());
    43.         StringBuilder verifyCode = new StringBuilder(verifySize);
    44.         for(int i = 0; i < verifySize; i++){
    45.             verifyCode.append(sources.charAt(rand.nextInt(codesLen-1)));
    46.         }
    47.         return verifyCode.toString();
    48.     }
    50.     /**
    51.      * 生成随机验证码文件,并返回验证码值
    52.      * @param w
    53.      * @param h
    54.      * @param outputFile
    55.      * @param verifySize
    56.      * @return
    57.      * @throws IOException
    58.      */
    59.     public static String outputVerifyImage(int w, int h, File outputFile, int verifySize) throws IOException{
    60.         String verifyCode = generateVerifyCode(verifySize);
    61.         outputImage(w, h, outputFile, verifyCode);
    62.         return verifyCode;
    63.     }
    65.     /**
    66.      * 输出随机验证码图片流,并返回验证码值
    67.      */
    68.     public static String outputVerifyImage(int w, int h, OutputStream os, int verifySize) throws IOException{
    69.         String verifyCode = generateVerifyCode(verifySize);
    70.         outputImage(w, h, os, verifyCode);
    71.         return verifyCode;
    72.     }
    74.     /**
    75.      * 生成指定验证码图像文件
    76.      */
    77.     public static void outputImage(int w, int h, File outputFile, String code) throws IOException{
    78.         if(outputFile == null){
    79.             return;
    80.         }
    81.         File dir = outputFile.getParentFile();
    82.         if(!dir.exists()){
    83.             dir.mkdirs();
    84.         }
    85.         try{
    86.             outputFile.createNewFile();
    87.             FileOutputStream fos = new FileOutputStream(outputFile);
    88.             outputImage(w, h, fos, code);
    89.             fos.close();
    90.         } catch(IOException e){
    91.             throw e;
    92.         }
    93.     }
    95.     /**
    96.      * 输出指定验证码图片流
    97.      */
    98.     public static void outputImage(int w, int h, OutputStream os, String code) throws IOException{
    99.         int verifySize = code.length();
    100.         BufferedImage image = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB);
    101.         Random rand = new Random();
    102.         Graphics2D g2 = image.createGraphics();
    103.         g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,RenderingHints.VALUE_ANTIALIAS_ON);
    104.         Color[] colors = new Color[5];
    105.         Color[] colorSpaces = new Color[] { Color.WHITE, Color.CYAN,
    106.                 Color.GRAY, Color.LIGHT_GRAY, Color.MAGENTA, Color.ORANGE,
    107.                 Color.PINK, Color.YELLOW };
    108.         float[] fractions = new float[colors.length];
    109.         for(int i = 0; i < colors.length; i++){
    110.             colors[i] = colorSpaces[rand.nextInt(colorSpaces.length)];
    111.             fractions[i] = rand.nextFloat();
    112.         }
    113.         Arrays.sort(fractions);
    115.         g2.setColor(Color.GRAY);// 设置边框色
    116.         g2.fillRect(0, 0, w, h);
    118.         Color c = getRandColor(200, 250);
    119.         g2.setColor(c);// 设置背景色
    120.         g2.fillRect(0, 2, w, h-4);
    122.         //绘制干扰线
    123.         Random random = new Random();
    124.         g2.setColor(getRandColor(160, 200));// 设置线条的颜色
    125.         for (int i = 0; i < 20; i++) {
    126.             int x = random.nextInt(w - 1);
    127.             int y = random.nextInt(h - 1);
    128.             int xl = random.nextInt(6) + 1;
    129.             int yl = random.nextInt(12) + 1;
    130.             g2.drawLine(x, y, x + xl + 40, y + yl + 20);
    131.         }
    133.         // 添加噪点
    134.         float yawpRate = 0.05f;// 噪声率
    135.         int area = (int) (yawpRate * w * h);
    136.         for (int i = 0; i < area; i++) {
    137.             int x = random.nextInt(w);
    138.             int y = random.nextInt(h);
    139.             int rgb = getRandomIntColor();
    140.             image.setRGB(x, y, rgb);
    141.         }
    143.         shear(g2, w, h, c);// 使图片扭曲
    145.         g2.setColor(getRandColor(100, 160));
    146.         int fontSize = h-4;
    147.         Font font = new Font("Algerian", Font.ITALIC, fontSize);
    148.         g2.setFont(font);
    149.         char[] chars = code.toCharArray();
    150.         for(int i = 0; i < verifySize; i++){
    151.             AffineTransform affine = new AffineTransform();
    152.             affine.setToRotation(Math.PI / 4 * rand.nextDouble() * (rand.nextBoolean() ? 1 : -1), (w / verifySize) * i + fontSize/2, h/2);
    153.             g2.setTransform(affine);
    154.             g2.drawChars(chars, i, 1, ((w-10) / verifySize) * i + 5, h/2 + fontSize/2 - 10);
    155.         }
    157.         g2.dispose();
    158.         ImageIO.write(image, "jpg", os);
    159.     }
    162.     private static Color getRandColor(int fc, int bc) {
    163.         if (fc > 255) {
    164.             fc = 255;
    165.         }
    166.         if (bc > 255) {
    167.             bc = 255;
    168.         }
    169.         int r = fc + random.nextInt(bc - fc);
    170.         int g = fc + random.nextInt(bc - fc);
    171.         int b = fc + random.nextInt(bc - fc);
    172.         return new Color(r, g, b);
    173.     }
    175.     private static int getRandomIntColor() {
    176.         int[] rgb = getRandomRgb();
    177.         int color = 0;
    178.         for (int c : rgb) {
    179.             color = color << 8;
    180.             color = color | c;
    181.         }
    182.         return color;
    183.     }
    185.     private static int[] getRandomRgb() {
    186.         int[] rgb = new int[3];
    187.         for (int i = 0; i < 3; i++) {
    188.             rgb[i] = random.nextInt(255);
    189.         }
    190.         return rgb;
    191.     }
    193.     private static void shear(Graphics g, int w1, int h1, Color color) {
    194.         shearX(g, w1, h1, color);
    195.         shearY(g, w1, h1, color);
    196.     }
    198.     private static void shearX(Graphics g, int w1, int h1, Color color) {
    200.         int period = random.nextInt(2);
    202.         boolean borderGap = true;
    203.         int frames = 1;
    204.         int phase = random.nextInt(2);
    206.         for (int i = 0; i < h1; i++) {
    207.             double d = (double) (period >> 1)
    208.                     * Math.sin((double) i / (double) period
    209.                     + (6.2831853071795862D * (double) phase)
    210.                     / (double) frames);
    211.             g.copyArea(0, i, w1, 1, (int) d, 0);
    212.             if (borderGap) {
    213.                 g.setColor(color);
    214.                 g.drawLine((int) d, i, 0, i);
    215.                 g.drawLine((int) d + w1, i, w1, i);
    216.             }
    217.         }
    219.     }
    221.     private static void shearY(Graphics g, int w1, int h1, Color color) {
    223.         int period = random.nextInt(40) + 10; // 50;
    225.         boolean borderGap = true;
    226.         int frames = 20;
    227.         int phase = 7;
    228.         for (int i = 0; i < w1; i++) {
    229.             double d = (double) (period >> 1)
    230.                     * Math.sin((double) i / (double) period
    231.                     + (6.2831853071795862D * (double) phase)
    232.                     / (double) frames);
    233.             g.copyArea(i, 0, 1, h1, 0, (int) d);
    234.             if (borderGap) {
    235.                 g.setColor(color);
    236.                 g.drawLine(i, (int) d, i, 0);
    237.                 g.drawLine(i, (int) d + h1, i, h1);
    238.             }
    240.         }
    242.     }
    243.     public static void main(String[] args) throws IOException {
    244.         //获取验证码
    245.         String s = generateVerifyCode(4);
    246.         //将验证码放入图片中
    247.         outputImage(260,60,new File("D:\\study\\aa.jpg"),s);
    248.         System.out.println(s);
    249.     }
    250. }