import com.google.zxing.*;
import com.google.zxing.client.j2se.BufferedImageLuminanceSource;
import com.google.zxing.client.j2se.MatrixToImageConfig;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.common.HybridBinarizer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;


public class ZXingUtil {  
    private ZXingUtil(){}  
      
    /** 
     * 為二維碼圖片增加logo頭像 
     * @param imagePath 二維碼圖片存放路徑(含文件名)
     * @param logoPath  logo頭像存放路徑(含文件名) 
     */  
    private static void overlapImage(String imagePath, String logoPath) throws IOException {
        BufferedImage image = ImageIO.read(new File(imagePath));
        int logoWidth = image.getWidth()/5;   //設置logo圖片寬度為二維碼圖片的五分之一  
        int logoHeight = image.getHeight()/5; //設置logo圖片高度為二維碼圖片的五分之一  
        int logoX = (image.getWidth()-logoWidth)/2;   //設置logo圖片的位置,這里令其居中  
        int logoY = (image.getHeight()-logoHeight)/2; //設置logo圖片的位置,這里令其居中  
        Graphics2D graphics = image.createGraphics();
        graphics.drawImage(ImageIO.read(new File(logoPath)), logoX, logoY, logoWidth, logoHeight, null);
        graphics.dispose();  
        ImageIO.write(image, imagePath.substring(imagePath.lastIndexOf(".") + 1), new File(imagePath));  
    }  
  
      
    /** 
     * 生成二維碼 
     * @param content   二維碼內容 
     * @param charset   編碼二維碼內容時采用的字符集(傳null時默認采用UTF-8編碼) 
     * @param imagePath 二維碼圖片存放路徑(含文件名) 
     * @param width     生成的二維碼圖片寬度 
     * @param height    生成的二維碼圖片高度 
     * @param logoPath  logo頭像存放路徑(含文件名,若不加logo則傳null即可) 
     * @return 生成二維碼結果(true or false) 
     */  
    public static boolean encodeQRCodeImage(String content, String charset, String imagePath, int width, int height, String logoPath) {  
        Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>();
        //指定編碼格式  
        //hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");  
        //指定糾錯級別(L--7%,M--15%,Q--25%,H--30%)  
        hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
        //編碼內容,編碼類型(這里指定為二維碼),生成圖片寬度,生成圖片高度,設置參數  
        BitMatrix bitMatrix = null;
        try {  
            bitMatrix = new MultiFormatWriter().encode(new String(content.getBytes(charset==null?"UTF-8":charset), "ISO-8859-1"), BarcodeFormat.QR_CODE, width, height, hints);
        } catch (Exception e) {  
            System.out.println("編碼待生成二維碼圖片的文本時發生異常,堆棧軌跡如下");  
            e.printStackTrace();  
            return false;  
        }  
        //生成的二維碼圖片默認背景為白色,前景為黑色,但是在加入logo圖像后會導致logo也變為黑白色,至于是什么原因還沒有仔細去讀它的源碼  
        //所以這里對其第一個參數黑色將ZXing默認的前景色0xFF000000稍微改了一下0xFF000001,最終效果也是白色背景黑色前景的二維碼,且logo顏色保持原有不變  
        MatrixToImageConfig config = new MatrixToImageConfig(0xFF000001, 0xFFFFFFFF);
        //這里要顯式指定MatrixToImageConfig,否則還會按照默認處理將logo圖像也變為黑白色(如果打算加logo的話,反之則不須傳MatrixToImageConfig參數)  
        try {  
            MatrixToImageWriter.writeToFile(bitMatrix, imagePath.substring(imagePath.lastIndexOf(".") + 1), new File(imagePath), config);
        } catch (IOException e) {  
            System.out.println("生成二維碼圖片[" + imagePath + "]時遇到異常,堆棧軌跡如下");  
            e.printStackTrace();  
            return false;  
        }  
        //此時二維碼圖片已經生成了,只不過沒有logo頭像,所以接下來根據傳入的logoPath參數來決定是否加logo頭像  
        if(null == logoPath){  
            return true;  
        }else{  
            //如果此時最終生成的二維碼不是我們想要的,那么可以擴展MatrixToImageConfig類(反正ZXing提供了源碼)  
            //擴展時可以重寫其writeToFile方法,令其返回toBufferedImage()方法所生成的BufferedImage對象(盡管這種做法未必能解決為題,故需根據實際情景測試)  
            //然后替換這里overlapImage()里面的第一行BufferedImage image = ImageIO.read(new File(imagePath));  
            //即private static void overlapImage(BufferedImage image, String imagePath, String logoPath)  
            try {  
                //這里不需要判斷logoPath是否指向了一個具體的文件,因為這種情景下overlapImage會拋IO異常  
                overlapImage(imagePath, logoPath);  
                return true;  
            } catch (IOException e) {  
                System.out.println("為二維碼圖片[" + imagePath + "]添加logo頭像[" + logoPath + "]時遇到異常,堆棧軌跡如下");  
                e.printStackTrace();  
                return false;  
            }  
        }  
    }  
      
      
    /** 
     * 解析二維碼 
     * @param imagePath 二維碼圖片存放路徑(含文件名) 
     * @param charset   解碼二維碼內容時采用的字符集(傳null時默認采用UTF-8編碼) 
     * @return 解析成功后返回二維碼文本,否則返回空字符串 
     */  
    public static String decodeQRCodeImage(String imagePath, String charset) {  
        BufferedImage image = null;  
        try {  
            image = ImageIO.read(new File(imagePath));  
        } catch (IOException e) {  
            e.printStackTrace();  
            return "";  
        }  
        if(null == image){  
            System.out.println("Could not decode QRCodeImage");  
            return "";  
        }  
        BinaryBitmap bitmap = new BinaryBitmap(new HybridBinarizer(new BufferedImageLuminanceSource(image)));
        Map<DecodeHintType, String> hints = new HashMap<DecodeHintType, String>();
        hints.put(DecodeHintType.CHARACTER_SET, charset==null ? "UTF-8" : charset);  
        Result result = null;  
        try {  
            result = new MultiFormatReader().decode(bitmap, hints);  
            return result.getText();  
        } catch (NotFoundException e) {  
            System.out.println("二維碼圖片[" + imagePath + "]解析失敗,堆棧軌跡如下");  
            e.printStackTrace();  
            return "";  
        }  
    }  
}