package com.sunda.spmsweb.util;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.WriterException;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.oned.Code128Writer;
import org.apache.commons.lang.ObjectUtils;
import org.apache.commons.lang.StringUtils;
import org.krysalis.barcode4j.HumanReadablePlacement;
import org.krysalis.barcode4j.impl.code128.Code128Bean;
import org.krysalis.barcode4j.output.bitmap.BitmapCanvasProvider;

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

/**
 * @program: spms
 * @description: BarCodeUtil
 * @author: Wayne Wu
 * @create: 2021-06-28 09:52
 **/

public class BarCodeOverseaWhsInUtil {

    /**
     * 生成code128条形码
     *
     * @param height        条形码的高度
     * @param width         条形码的宽度
     * @param message       要生成的文本
     * @param withQuietZone 是否两边留白
     * @param hideText      隐藏可读文本
     * @return 图片对应的字节码
     */
    public static byte[] generateBarCode128(String message, Double height, Double width, boolean withQuietZone, boolean hideText) {
        Code128Bean bean = new Code128Bean();
        // 分辨率
        int dpi = 512;
        // 设置两侧是否留白
        bean.doQuietZone(withQuietZone);

        // 设置条形码高度和宽度
        bean.setBarHeight((double) ObjectUtils.defaultIfNull(height, 9.0D));
        if (width != null) {
            bean.setModuleWidth(width);
        }
        // 设置文本位置（包括是否显示）
        if (hideText) {
            bean.setMsgPosition(HumanReadablePlacement.HRP_NONE);
        }
        // 设置图片类型
        String format = "image/png";
        ByteArrayOutputStream ous = new ByteArrayOutputStream();
        BitmapCanvasProvider canvas = new BitmapCanvasProvider(ous, format, dpi,
                BufferedImage.TYPE_BYTE_BINARY, false, 0);

        // 生产条形码
        bean.generateBarcode(canvas, message);
        try {
            canvas.finish();
        } catch (IOException e) {
        }

        return ous.toByteArray();
    }


    /** 条形码宽度 */
    //private static final int WIDTH = 325;
    private static final int WIDTH = 600;

    /** 条形码高度 */
    //private static final int HEIGHT = 50;
    private static final int HEIGHT = 140;

    /** 加文字 条形码 */
    private static final int WORDHEIGHT = 400;

    /**
     * 设置 条形码参数
     */
    private static Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>() {
        private static final long serialVersionUID = 1L;
        {
            // 设置编码方式
            put(EncodeHintType.CHARACTER_SET, "utf-8");
        }
    };

    /**
     * 生成 图片缓冲
     * @author fxbin
     * @param vaNumber  VA 码
     * @return 返回BufferedImage
     */
    public static BufferedImage getBarCode(String vaNumber){
        try {
            Code128Writer writer = new Code128Writer();
            // 编码内容, 编码类型, 宽度, 高度, 设置参数
            BitMatrix bitMatrix = writer.encode(vaNumber, BarcodeFormat.CODE_128, WIDTH, HEIGHT, hints);
            return MatrixToImageWriter.toBufferedImage(bitMatrix);
        } catch (WriterException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 把带logo的条形码下面加上文字
     */
    public static BufferedImage insertWords(BufferedImage image, String materialNo, String materialZhDesc, String materialEnDesc, String boxNote, String createDate){
        //SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
        // 新的图片，把带logo的二维码下面加上文字
        if (StringUtils.isNotEmpty(materialNo)) {

            BufferedImage outImage = new BufferedImage(WIDTH, WORDHEIGHT, BufferedImage.TYPE_INT_RGB);

            Graphics2D g2d = outImage.createGraphics();

            // 抗锯齿
            setGraphics2D(g2d);
            // 设置白色
            setColorWhite(g2d);

            // 画条形码到新的面板
            g2d.drawImage(image, 0, 50, image.getWidth(), image.getHeight(), null);
            // 画文字到新的面板
            Color color=new Color(0, 0, 0);
            g2d.setColor(color);
            // 字体、字型、字号
            g2d.setFont(new Font("Arial", Font.PLAIN, 40));
            //文字长度
            int strWidth = g2d.getFontMetrics().stringWidth(materialNo);
            //总长度减去文字长度的一半  （居中显示）
            int wordStartX=(WIDTH - strWidth) / 2;
            //height + (outImage.getHeight() - height) / 2 + 12
            //int wordStartY=HEIGHT+50;
            int wordStartY = 300;

            // 画文字
            if (StringUtils.isNotEmpty(boxNote)){
                g2d.setFont(new Font("Arial", Font.PLAIN, 42));
                g2d.drawString(materialNo, wordStartX, 240);

                g2d.setFont(new Font("Arial", Font.PLAIN, 20));
                //g2d.drawString(materialZhDesc, (WIDTH - g2d.getFontMetrics().stringWidth(materialZhDesc)) / 2, 270);
                g2d.drawString(materialZhDesc, 0, 270);

                g2d.setFont(new Font("Arial", Font.PLAIN, 20));
                //g2d.drawString(materialEnDesc, (WIDTH - g2d.getFontMetrics().stringWidth(materialEnDesc)) / 2, 310);
                g2d.drawString(materialEnDesc, 0, 310);

                g2d.setFont(new Font("Arial", Font.PLAIN, 24));
                g2d.drawString(boxNote, (WIDTH - g2d.getFontMetrics().stringWidth(boxNote)) / 2, 350);

                if (StringUtils.isNotEmpty(createDate)){
                    g2d.drawString(createDate, (WIDTH - g2d.getFontMetrics().stringWidth(createDate)) / 2, 380);
                } else {
                    g2d.drawString(sdf.format(new Date()), (WIDTH - g2d.getFontMetrics().stringWidth(sdf.format(new Date()))) / 2, 380);
                }

                g2d.dispose();
                outImage.flush();
                return outImage;
            } else {
                g2d.setFont(new Font("Arial", Font.PLAIN, 42));
                g2d.drawString(materialNo, wordStartX, 240);

                g2d.setFont(new Font("Arial", Font.PLAIN, 20));
                //g2d.drawString(materialZhDesc, (WIDTH - g2d.getFontMetrics().stringWidth(materialZhDesc)) / 2, 280);
                g2d.drawString(materialZhDesc, 0, 280);

                g2d.setFont(new Font("Arial", Font.PLAIN, 20));
                //g2d.drawString(materialEnDesc, (WIDTH - g2d.getFontMetrics().stringWidth(materialEnDesc)) / 2, 330);
                g2d.drawString(materialEnDesc, 0, 330);

                g2d.setFont(new Font("Arial", Font.PLAIN, 24));
                if (StringUtils.isNotEmpty(createDate)){
                    g2d.drawString(createDate, (WIDTH - g2d.getFontMetrics().stringWidth(createDate)) / 2, 380);
                } else {
                    g2d.drawString(sdf.format(new Date()), (WIDTH - g2d.getFontMetrics().stringWidth(sdf.format(new Date()))) / 2, 380);
                }

                g2d.dispose();
                outImage.flush();
                return outImage;
            }

        }
        return null;
    }

    /**
     * 设置 Graphics2D 属性  （抗锯齿）
     * @param g2d  Graphics2D提供对几何形状、坐标转换、颜色管理和文本布局更为复杂的控制
     */
    private static void setGraphics2D(Graphics2D g2d){
        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
        g2d.setRenderingHint(RenderingHints.KEY_STROKE_CONTROL, RenderingHints.VALUE_STROKE_DEFAULT);
        Stroke s = new BasicStroke(1, BasicStroke.CAP_ROUND, BasicStroke.JOIN_MITER);
        g2d.setStroke(s);
    }

    /**
     * 设置背景为白色
     * @param g2d Graphics2D提供对几何形状、坐标转换、颜色管理和文本布局更为复杂的控制
     */
    private static void setColorWhite(Graphics2D g2d){
        g2d.setColor(Color.WHITE);
        //填充整个屏幕
        g2d.fillRect(0,0,600,600);
        //设置笔刷
        g2d.setColor(Color.BLACK);
    }

    public static void main(String[] args) throws IOException {
        String materialNo = "210054327";
        String materialZhDesc = "快速无缝输送带接头套_B800/普通型_快速无缝输送带接头套_B800/普通型";
        String materialEnDesc = "快速无缝输送带接头套_B800/普通型_快速无缝输送带接头套_B800/普通型";
        String boxNote = "BJ-11801149752-194200";
        String createDate = "2022-01-14";
        BufferedImage image = insertWords(getBarCode(materialNo), materialNo, materialZhDesc, materialEnDesc, boxNote, createDate);
        ImageIO.write(image, "jpg", new File("BarCode/" + boxNote + "-" + materialNo + ".jpg"));

        String materialNo2 = "210046549";
        String materialZhDesc2 = "混合器配件_含拖轮轴/图号HZ-03+轴套+拖轮+锁帽+拖轮挡板+拖轮内套+齿轮";
        String materialEnDesc2 = "混合器配件_含拖轮轴/图号HZ-03+轴套+拖轮+锁帽+拖轮挡板+拖轮内套+齿轮";
        BufferedImage image2 = insertWords(getBarCode(materialNo2), materialNo2, materialZhDesc2, materialEnDesc2, "", "");
        ImageIO.write(image2, "jpg", new File("BarCode/" + materialNo2 + ".jpg"));
    }

    public static void generateBarCodeWhsIn (String materialNo, String materialZhDesc, String materialEnDesc, String boxNote, String createDate) throws IOException {
        BufferedImage image = insertWords(getBarCode(materialNo), materialNo, materialZhDesc, materialEnDesc, boxNote, createDate);
        if (StringUtils.isNotEmpty(boxNote)){
            ImageIO.write(image, "jpg", new File("BarCode/" + boxNote + "-" + materialNo + ".jpg"));
        } else {
            ImageIO.write(image, "jpg", new File("BarCode/" + materialNo + ".jpg"));
        }
    }

}