import org.opencv.core.Core;
import org.opencv.core.CvType;
import org.opencv.core.Mat;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.awt.image.DataBufferByte;
import java.awt.image.WritableRaster;
import java.net.URL;
import java.util.List;

/**
 * @program: test-wechatqrcode
 * @description: 微信二维码扫描opencv库方法调用
 * @author: test
 * @create: 221-08-21 17:05
 **/
public class WeChatQRCodeTool {
    private static volatile WeChatQRCodeTool instance;
    private static volatile org.opencv.wechat_qrcode.WeChatQRCode detector;
　　//封装程了工具类，所以做成了单例
    private WeChatQRCodeTool() {
        System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
　　　　　//微信提供的4个模型配置文件放在resource/wechatqrcode文件夹里
        URL detectprototxt = ClassLoader.getSystemResource("wechatqrcode/detect.prototxt");
        URL detectcaffemodel = ClassLoader.getSystemResource("wechatqrcode/detect.caffemodel");
        URL srprototxt = ClassLoader.getSystemResource("wechatqrcode/sr.prototxt");
        URL srcaffemodel = ClassLoader.getSystemResource("wechatqrcode/sr.caffemodel");
　　　　 //实例化微信二维码扫描对象
        detector = new org.opencv.wechat_qrcode.WeChatQRCode(
                detectprototxt.getPath().substring(1),
                //substring1去掉/D:/xx里的/
                detectcaffemodel.getPath().substring(1),
                srprototxt.getPath().substring(1),
                srcaffemodel.getPath().substring(1));

    }

    public static WeChatQRCodeTool getInstance() {
        if (instance == null) {
            synchronized (WeChatQRCodeTool.class) {
                if (instance == null) {
                    instance = new WeChatQRCodeTool();
                }
            }
        }
        return instance;
    }

　　/**
    * 这里提供一个bufferimage的入参。如果是File对象，可以用ImageIO.read()获取一下bufferimage.
　　*/
public String decode(BufferedImage srcImage) {
        Mat image = bufImg2Mat(srcImage, srcImage.getType(), CvType.CV_8UC3);
        //返回解析的字符串，如果图片有多个二维码，则会返回多个。
        List<String> result2 = detector.detectAndDecode(image);
        if (result2 != null && result2.size() > 0) {
            return result2.get(0);
        }
        return null;
    }


    /**
     * BufferedImage转换成Mat
     *
     * @param original 要转换的BufferedImage
     * @param imgType  bufferedImage的类型 如 BufferedImage.TYPE_3BYTE_BGR
     * @param matType  转换成mat的type 如 CvType.CV_8UC3
     */
    public static Mat bufImg2Mat(BufferedImage original, int imgType, int matType) {
        if (original == null) {
            throw new IllegalArgumentException("original == null");
        }

        // Don‘t convert if it already has correct type
        if (original.getType() != imgType) {

            // Create a buffered image
            BufferedImage image = new BufferedImage(original.getWidth(), original.getHeight(), imgType);

            // Draw the image onto the new buffer
            Graphics2D g = image.createGraphics();
            try {
                g.setComposite(AlphaComposite.Src);
                g.drawImage(original, 0, 0, null);
            } finally {
                g.dispose();
            }
        }

        byte[] pixels = ((DataBufferByte) original.getRaster().getDataBuffer()).getData();
        Mat mat = Mat.eye(original.getHeight(), original.getWidth(), matType);
        mat.put(0, 0, pixels);
        return mat;
    }

    /**
     *  Mat转换为BufferedImage
     */
    public static BufferedImage matToBufferedImage(Mat mat) {
        if (mat.height() > 0 && mat.width() > 0) {
            BufferedImage image = new BufferedImage(mat.width(), mat.height(),
                    BufferedImage.TYPE_3BYTE_BGR);
            WritableRaster raster = image.getRaster();
            DataBufferByte dataBuffer = (DataBufferByte) raster.getDataBuffer();
            byte[] data = dataBuffer.getData();
            mat.get(0, 0, data);
            return image;
        }

        return null;
    }

}