net.gdface.image.ImageUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-image-android Show documentation
There is a newer version: 2.10.3
package net.gdface.image;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import android.graphics.Bitmap;
import android.graphics.Bitmap.CompressFormat;
import android.graphics.ImageFormat;
import android.graphics.Rect;
import android.graphics.YuvImage;
import gu.jimgutil.MatrixUtils;
import net.gdface.utils.Assert;

/**
 * 图像工具类
 * @author guyadong
 *
 */
public class ImageUtil {
	/**
	 * 如果{@code bitmap}不是RGBA({@link Bitmap.Config#ARGB_8888})格式则转为RGBA格式
	 * @param bitmap
	 * @return RGBA格式的{@link Bitmap}
	 */
	private static Bitmap rgbaCopyOf(Bitmap bitmap){
		if(null==bitmap){
			throw new NullPointerException("bitmap is null");
		}
		if( !Bitmap.Config.ARGB_8888.equals(bitmap.getConfig()) ){		
			bitmap = bitmap.copy(Bitmap.Config.ARGB_8888, false);
			if(null==bitmap){
				throw new NullPointerException("FAIL TO Bitmap.copy");
			}
		}
		return bitmap;
	}
	/**
	 * 对图像解码返回RGBA格式矩阵数据
	 * @param bitmap
	 * @return RGBA格式矩阵数据
	 */
	public static byte[] getMatrixRGBA(Bitmap bitmap){
		Bitmap rgba = rgbaCopyOf(bitmap);
		ByteBuffer buffer = ByteBuffer.allocate(rgba.getByteCount());
		rgba.copyPixelsToBuffer(buffer);
		return buffer.array();
	}
	/**
	 * 对图像解码返回BGR格式矩阵数据
	 * @param bitmap
	 * @return BGR格式矩阵数据
	 */
	public static byte[] getMatrixBGR(Bitmap bitmap){
		Bitmap rgba = rgbaCopyOf(bitmap);
		ByteBuffer buffer = ByteBuffer.allocate(rgba.getByteCount());
		rgba.copyPixelsToBuffer(buffer);
		return MatrixUtils.RGBA2BGR(buffer.array(), rgba.getWidth(), rgba.getHeight(),rgba.getRowBytes());
	}
	/**
	 * 对图像解码返回RGB格式矩阵数据
	 * @param bitmap
	 * @return RGB格式矩阵数据
	 */
	public static byte[] getMatrixRGB(Bitmap bitmap){
		Bitmap rgba = rgbaCopyOf(bitmap);
		ByteBuffer buffer = ByteBuffer.allocate(rgba.getByteCount());
		rgba.copyPixelsToBuffer(buffer);
		return MatrixUtils.RGBA2RGB(buffer.array(), rgba.getWidth(), rgba.getHeight(),rgba.getRowBytes());
	}
	/**
	 * 对图像解码返回GRAY格式矩阵数据
	 * @param bitmap
	 * @return GRAY格式矩阵数据
	 */
	public static byte[] getMatrixGRAY(Bitmap bitmap){
		Bitmap rgba = rgbaCopyOf(bitmap);
		ByteBuffer buffer = ByteBuffer.allocate(rgba.getByteCount());
		rgba.copyPixelsToBuffer(buffer);
		return MatrixUtils.RGBA2GRAY(buffer.array(), rgba.getWidth(), rgba.getHeight(),rgba.getRowBytes());
	}
	public static byte[] wirteJPEGBytes(Bitmap source){
		return wirteJPEGBytes(source,null);
	}
	public static byte[] wirteBMPBytes(Bitmap source){
		throw new UnsupportedOperationException();
	}
	public static byte[] wirtePNGBytes(Bitmap source){
		return wirteBytes(source,"PNG");
	}
	public static byte[] wirteGIFBytes(Bitmap source){
		throw new UnsupportedOperationException();
	}
	public static byte[] wirteWEBPBytes(Bitmap source){
		return wirteBytes(source,"WEBP");
	}
	/**
	 * 将原图压缩生成jpeg格式的数据
	 * @param source
	 * @return jpeg格式的二进制数据
	 * @see #wirteBytes(Bitmap, String)
	 */
	public static byte[] wirteJPEGBytes(Bitmap source,Float compressionQuality){
		return wirteBytes(source,"JPEG",compressionQuality);
	}
	public static byte[] wirteBytes(Bitmap source,String formatName){
		return wirteBytes(source,formatName,null);
	}
	
	/**
	 * 将{@link Bitmap}生成formatName指定格式的图像数据
	 * @param source
	 * @param formatName 图像格式名，图像格式名错误则抛出异常,可用的值 'PNG','JPEG','WEBP'
	 * @param compressionQuality 压缩质量(0.0~1.0),超过此范围抛出异常,为null使用默认值
	 * @return 图像字节数组
	 */
	public static byte[] wirteBytes(Bitmap source,String formatName,Float compressionQuality){
		ByteArrayOutputStream output = new ByteArrayOutputStream();
		try {
			wirte(source, formatName, compressionQuality, output);
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
		return output.toByteArray();		
	}
	/**
	 * 将{@link Bitmap}生成formatName指定格式的图像数据
	 * @param source
	 * @param formatName 图像格式名，图像格式名错误则抛出异常,可用的值 'PNG','JPEG','WEBP'
	 * @param compressionQuality 压缩质量(0.0~1.0),超过此范围抛出异常,为null使用默认值
	 * @param output 输出流
	 * @throws IOException
	 */
	public static void wirte(Bitmap source,String formatName,Float compressionQuality,OutputStream output) throws IOException{
		Assert.notNull(source, "source");
		Assert.notEmpty(formatName, "formatName");
		Assert.notNull(output, "output");
		if( ! (compressionQuality==null || (compressionQuality>0f && compressionQuality <= 1.0f)) ){
			throw new IllegalArgumentException("INVALID compressionQuality");
		}
		CompressFormat format = Bitmap.CompressFormat.valueOf(formatName);
		int quality = compressionQuality !=null ? (int)(compressionQuality*100):100;
		source.compress(format, quality, output);		
	}
	
	public static byte[] nv21ToJPEG(byte[] nv21, int width, int height) {
		Assert.notNull(nv21, "nv21");
		YuvImage image = new YuvImage(nv21, ImageFormat.NV21, width, height, null);
		ByteArrayOutputStream stream = new ByteArrayOutputStream();
		image.compressToJpeg(new Rect(0, 0, width, height), 80, stream);
		return stream.toByteArray();
	}
}