org.cattleframework.utils.auxiliary.BarcodeUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cattle-utils Show documentation
Cattle framework utils component pom
There is a newer version: 1.0.1-M3
/*
 * Copyright (C) 2018 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.cattleframework.utils.auxiliary;

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.geom.RoundRectangle2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.Charset;
import java.util.HashMap;
import java.util.Map;

import javax.imageio.ImageIO;

import org.apache.commons.lang3.StringUtils;
import org.cattleframework.exception.CattleException;
import org.cattleframework.exception.ExceptionWrapUtils;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.BinaryBitmap;
import com.google.zxing.DecodeHintType;
import com.google.zxing.EncodeHintType;
import com.google.zxing.LuminanceSource;
import com.google.zxing.MultiFormatReader;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.NotFoundException;
import com.google.zxing.Result;
import com.google.zxing.WriterException;
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.CharacterSetECI;
import com.google.zxing.common.GlobalHistogramBinarizer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

/**
 * 条形码工具
 * 
 * @author orange
 *
 */
public final class BarcodeUtils {

    private static final String IMAGE_DEFAULT_FORMAT = "png";

    private BarcodeUtils() {
    }

    public static void createBarcode(OutputStream outputStream, String content, int width, int height, int margin,
	    Charset charset, ErrorCorrectionLevel errorCorrectionLevel, Color front, Color background, String format) {
	MatrixToImageConfig config = new MatrixToImageConfig(front == null ? Color.BLACK.getRGB() : front.getRGB(),
		background == null ? Color.WHITE.getRGB() : background.getRGB());
	BitMatrix matrix = createBarcode(content, width, height, margin, charset, errorCorrectionLevel);
	try {
	    MatrixToImageWriter.writeToStream(matrix, StringUtils.isBlank(format) ? IMAGE_DEFAULT_FORMAT : format,
		    outputStream, config);
	} catch (IOException e) {
	    throw ExceptionWrapUtils.wrap(e);
	}
    }

    public static void createLogoBarcode(OutputStream outputStream, String content, InputStream logoInputStream,
	    int width, int height, int margin, Charset charset, ErrorCorrectionLevel errorCorrectionLevel, Color front,
	    Color background, String format) {
	MatrixToImageConfig config = new MatrixToImageConfig(front == null ? Color.BLACK.getRGB() : front.getRGB(),
		background == null ? Color.WHITE.getRGB() : background.getRGB());
	BitMatrix matrix = createBarcode(content, width, height, margin, charset, errorCorrectionLevel);
	int qrWidth = matrix.getWidth();
	int qrHeight = matrix.getHeight();
	BufferedImage qrImage = new BufferedImage(qrWidth, qrHeight, BufferedImage.TYPE_INT_RGB);
	for (int x = 0; x < qrWidth; x++) {
	    for (int y = 0; y < qrHeight; y++) {
		qrImage.setRGB(x, y, (matrix.get(x, y) ? config.getPixelOnColor() : config.getPixelOffColor()));
	    }
	}
	try {
	    BufferedImage logoImage = ImageIO.read(logoInputStream);
	    Graphics2D g2 = qrImage.createGraphics();
	    int matrixWidth = qrImage.getWidth();
	    int matrixHeigh = qrImage.getHeight();
	    // 绘制
	    g2.drawImage(logoImage, matrixWidth / 5 * 2, matrixHeigh / 5 * 2, matrixWidth / 5, matrixHeigh / 5, null);
	    BasicStroke stroke = new BasicStroke(5, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND);
	    // 设置笔画对象
	    g2.setStroke(stroke);
	    // 指定弧度的圆角矩形
	    RoundRectangle2D.Float round = new RoundRectangle2D.Float(matrixWidth / 5 * 2, matrixHeigh / 5 * 2,
		    matrixWidth / 5, matrixHeigh / 5, 20, 20);
	    g2.setColor(Color.white);
	    // 绘制圆弧矩形
	    g2.draw(round);
	    // 设置logo 有一道灰色边框
	    BasicStroke stroke2 = new BasicStroke(1, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND);
	    // 设置笔画对象
	    g2.setStroke(stroke2);
	    RoundRectangle2D.Float round2 = new RoundRectangle2D.Float(matrixWidth / 5 * 2 + 2, matrixHeigh / 5 * 2 + 2,
		    matrixWidth / 5 - 4, matrixHeigh / 5 - 4, 20, 20);
	    g2.setColor(new Color(128, 128, 128));
	    // 绘制圆弧矩形
	    g2.draw(round2);
	    g2.dispose();
	    qrImage.flush();
	    if (!ImageIO.write(qrImage, StringUtils.isBlank(format) ? IMAGE_DEFAULT_FORMAT : format, outputStream)) {
		throw new CattleException("不能生成图片格式:" + format);
	    }
	} catch (IOException e) {
	    throw ExceptionWrapUtils.wrap(e);
	}
    }

    private static BitMatrix createBarcode(String content, int width, int height, int margin, Charset charset,
	    ErrorCorrectionLevel errorCorrectionLevel) {
	try {
	    Map hints = new HashMap(3);
	    hints.put(EncodeHintType.CHARACTER_SET, charset == null ? CharacterSetECI.UTF8 : charset);
	    hints.put(EncodeHintType.ERROR_CORRECTION,
		    errorCorrectionLevel == null ? ErrorCorrectionLevel.H : errorCorrectionLevel);
	    hints.put(EncodeHintType.MARGIN, margin);
	    return new MultiFormatWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);
	} catch (WriterException e) {
	    throw ExceptionWrapUtils.wrap(e);
	}
    }

    public static BarcodeResult scan(byte[] imageBytes, Charset charset) {
	try (ByteArrayInputStream bais = new ByteArrayInputStream(imageBytes)) {
	    BufferedImage bi = ImageIO.read(bais);
	    if (bi != null) {
		LuminanceSource source = new BufferedImageLuminanceSource(bi);
		BinaryBitmap bitmap = new BinaryBitmap(new GlobalHistogramBinarizer(source));
		Map hints = new HashMap(2);
		hints.put(DecodeHintType.CHARACTER_SET, charset == null ? CharacterSetECI.UTF8 : charset);
		hints.put(DecodeHintType.TRY_HARDER, Boolean.TRUE);
		Result result = new MultiFormatReader().decode(bitmap, hints);
		return new BarcodeResult(result.getText(), result.getBarcodeFormat().toString());
	    }
	    return null;
	} catch (IOException e) {
	    throw ExceptionWrapUtils.wrap(e);
	} catch (NotFoundException e) {
	    return null;
	}
    }
}