package com.bonus.core.qrcode;

import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.Hashtable;

import javax.imageio.ImageIO;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.client.j2se.MatrixToImageConfig;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

public class QRCodeUtil {
	private static int width = 280; // 二维码宽度
	private static int height = 280; // 二维码高度
	private static int onColor = 0xFF000000; // 前景色
	private static int offColor = 0xFFFFFFFF; // 背景色
	private static int margin = 0; // 白边大小，取值范围0~4
	private static ErrorCorrectionLevel level = ErrorCorrectionLevel.L; // 二维码容错率

	/**
	 * 生成二维码
	 * 
	 * @param params
	 *            QRCodeParams属性：txt、fileName、filePath不得为空；
	 * @throws QRParamsException
	 */
	public static void generateQRImage(QRCodeParams params) throws QRParamsException {
		if (params == null || params.getFileName() == null || params.getFilePath() == null || params.getTxt() == null) {

			throw new QRParamsException("参数错误");
		}
		try {
			initData(params);

			String imgPath = params.getFilePath();
			String imgName = params.getFileName();
			String txt = params.getTxt();

			if (params.getLogoPath() != null && !"".equals(params.getLogoPath().trim())) {
				generateQRImage(txt, params.getLogoPath(), imgPath, imgName, params.getSuffixName());
			} else {
				generateQRImage(txt, imgPath, imgName, params.getSuffixName());
			}
		} catch (Exception e) {
			e.printStackTrace();
		}

	}

	/**
	 * 生成二维码
	 * 
	 * @param txt
	 *            //二维码内容
	 * @param imgPath
	 *            //二维码保存物理路径
	 * @param imgName
	 *            //二维码文件名称
	 * @param suffix
	 *            //图片后缀名
	 */
	public static void generateQRImage(String txt, String imgPath, String imgName, String suffix) {
		
		File filePath = new File(imgPath);
		if (!filePath.exists()) {
			filePath.mkdirs();
		}

		File imageFile = new File(imgPath, imgName);
		Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
		// 指定纠错等级
		hints.put(EncodeHintType.ERROR_CORRECTION, level);
		// 指定编码格式
		hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
		hints.put(EncodeHintType.MARGIN, margin); // 设置白边
		try {
			MatrixToImageConfig config = new MatrixToImageConfig(onColor, offColor);
			BitMatrix bitMatrix = new MultiFormatWriter().encode(txt, BarcodeFormat.QR_CODE, width, height, hints);
			// bitMatrix = deleteWhite(bitMatrix);
			MatrixToImageWriter.writeToPath(bitMatrix, suffix, imageFile.toPath(), config);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 生成带logo的二维码图片
	 * 
	 * @param txt
	 *            //二维码内容
	 * @param logoPath
	 *            //logo绝对物理路径
	 * @param imgPath
	 *            //二维码保存绝对物理路径
	 * @param imgName
	 *            //二维码文件名称
	 * @param suffix
	 *            //图片后缀名
	 * @throws Exception
	 */
	public static void generateQRImage(String txt, String logoPath, String imgPath, String imgName, String suffix)
			throws Exception {
		File filePath = new File(imgPath);
		if (!filePath.exists()) {
			filePath.mkdirs();
		}

		if (imgPath.endsWith("/")) {
			imgPath += imgName;
		} else {
			imgPath += "/" + imgName;
		}

		Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
		hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
		hints.put(EncodeHintType.ERROR_CORRECTION, level);
		hints.put(EncodeHintType.MARGIN, margin); // 设置白边
		BitMatrix bitMatrix = new MultiFormatWriter().encode(txt, BarcodeFormat.QR_CODE, width, height, hints);
		File qrcodeFile = new File(imgPath);
		writeToFile(bitMatrix, suffix, qrcodeFile, logoPath);
	}

	/**
	 * 
	 * @param matrix
	 *            二维码矩阵相关
	 * @param format
	 *            二维码图片格式
	 * @param file
	 *            二维码图片文件
	 * @param logoPath
	 *            logo路径
	 * @throws IOException
	 */
	public static void writeToFile(BitMatrix matrix, String format, File file, String logoPath) throws IOException {
		BufferedImage image = toBufferedImage(matrix);
		Graphics2D gs = image.createGraphics();

		int ratioWidth = image.getWidth() * 2 / 10;
		int ratioHeight = image.getHeight() * 2 / 10;
		// 载入logo
		Image img = ImageIO.read(new File(logoPath));
		int logoWidth = img.getWidth(null) > ratioWidth ? ratioWidth : img.getWidth(null);
		int logoHeight = img.getHeight(null) > ratioHeight ? ratioHeight : img.getHeight(null);

		int x = (image.getWidth() - logoWidth) / 2;
		int y = (image.getHeight() - logoHeight) / 2;

		gs.drawImage(img, x, y, logoWidth, logoHeight, null);
		gs.setColor(Color.black);
		gs.setBackground(Color.WHITE);
		gs.dispose();
		img.flush();
		if (!ImageIO.write(image, format, file)) {
			throw new IOException("Could not write an image of format " + format + " to " + file);
		}
	}

	public static BufferedImage toBufferedImage(BitMatrix matrix) {
		int width = matrix.getWidth();
		int height = matrix.getHeight();
		BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);

		for (int x = 0; x < width; x++) {
			for (int y = 0; y < height; y++) {
				image.setRGB(x, y, matrix.get(x, y) ? onColor : offColor);
			}
		}
		return image;
	}

	public static BitMatrix deleteWhite(BitMatrix matrix) {
		int[] rec = matrix.getEnclosingRectangle();
		int resWidth = rec[2] + 1;
		int resHeight = rec[3] + 1;

		BitMatrix resMatrix = new BitMatrix(resWidth, resHeight);
		resMatrix.clear();
		for (int i = 0; i < resWidth; i++) {
			for (int j = 0; j < resHeight; j++) {
				if (matrix.get(i + rec[0], j + rec[1]))
					resMatrix.set(i, j);
			}
		}
		return resMatrix;
	}

	private static void initData(QRCodeParams params) {
		if (params.getWidth() != null) {
			width = params.getWidth();
		}
		if (params.getHeight() != null) {
			height = params.getHeight();
		}
		if (params.getOnColor() != null) {
			onColor = params.getOnColor();
		}
		if (params.getOffColor() != null) {
			offColor = params.getOffColor();
		}
		if (params.getLevel() != null) {
			level = params.getLevel();
		}

	}
}