生成二维码图片

将指定字符串生成二维码图片

package com.yike.util;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
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.WriterException;
import com.google.zxing.common.BitMatrix;

/**
 * 生成二维码工具
 * @author Devon
 *
 */
public class MatrixToImageWriter {
	private static final int BLACK = 0xFF000000;      
	private static final int WHITE = 0xFFFFFFFF;           
	
	private MatrixToImageWriter() {
	}     
	
	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) ? BLACK : WHITE);
			}
		}
		return image;
	}
	
	private static BitMatrix toBufferedImage(BitMatrix matrix, int margin){
		int tempM = margin*2;
		int[] rec = matrix.getEnclosingRectangle();
		int resWidth = rec[2] + tempM;
		int resHeight = rec[3] + tempM;
		BitMatrix resMatrix = new BitMatrix(resWidth, resHeight);
		resMatrix.clear();
		for(int i= margin; i < resWidth- margin; i++){
			for(int j=margin; j < resHeight-margin; j++){
				if(matrix.get(i-margin + rec[0], j-margin + rec[1])){
					resMatrix.set(i,j);
				}
			}
		}
		return resMatrix;
	}
	
	public static void writeToFile(BitMatrix matrix, String format, File file)  throws IOException {        
		BufferedImage image = toBufferedImage(matrix);
		if (!ImageIO.write(image, format, file)) {
			throw new IOException("Could not write an image of format " + format + " to " + file);        
		}
	}
	
	public static void writeToStream(BitMatrix matrix, String format, OutputStream stream)  throws IOException {        
		BufferedImage image = toBufferedImage(matrix);        
		if (!ImageIO.write(image, format, stream)) {          
			throw new IOException("Could not write an image of format " + format);        
			}   
	}

	//自定义白边
	public static void writeToStream(BitMatrix matrix, String format, OutputStream stream,int margin)  throws IOException {
		matrix = toBufferedImage(matrix, margin);
		BufferedImage image = toBufferedImage(matrix);        
		if (!ImageIO.write(image, format, stream)) {          
			throw new IOException("Could not write an image of format " + format);        
			}   
	}
	/**
	 * 二维码参数
	 */
	protected static final int QRWIDTH=300;//二维码宽度
	protected static final int QRHEIGHT=300;//二维码高度
	protected static final String FORMAT="gif";//二维码格式
	/**
	 * 生成默认二维码方法
	 * @param text  内容
	 * @param stream
	 * @throws WriterException 
	 * @throws IOException 
	 */
	public static void createQRImage(String text,OutputStream stream) throws WriterException, IOException{
		Hashtable hints = new Hashtable();  
		hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
		BitMatrix bitMatrix = null;
		bitMatrix = new MultiFormatWriter().encode(text,BarcodeFormat.QR_CODE, QRWIDTH, QRHEIGHT, hints);
		MatrixToImageWriter.writeToStream(bitMatrix, FORMAT, stream);
	}

	/**
	 * 生成自定义二维码方法
	 * @param text  内容
	 * @param stream
	 * @param width  宽度
	 * @param height 高度
	 * @throws WriterException 
	 * @throws IOException 
	 */
	public static void createQRImage(String text,OutputStream stream,int width,int height) throws WriterException, IOException{
		Hashtable hints = new Hashtable();  
		hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
		BitMatrix bitMatrix = null;
		bitMatrix = new MultiFormatWriter().encode(text,BarcodeFormat.QR_CODE, width, height, hints);
		MatrixToImageWriter.writeToStream(bitMatrix, FORMAT, stream);
	}

	/**
	 * 生成自定义二维码方法
	 * @param text  内容
	 * @param stream
	 * @param width  宽度
	 * @param height 高度
	 * @param margin 白边
	 * @throws WriterException 
	 * @throws IOException 
	 */
	public static void createQRImage(String text,OutputStream stream,int width,int height,int margin) throws WriterException, IOException{
		Hashtable hints = new Hashtable();  
		hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
		BitMatrix bitMatrix = null;
		bitMatrix = new MultiFormatWriter().encode(text,BarcodeFormat.QR_CODE, width, height, hints);
		MatrixToImageWriter.writeToStream(bitMatrix, FORMAT, stream,margin);
	}
}