javacpp-FFmpeg系列之3： 像素图像数据转换
- 2019 年 11 月 1 日
- 筆記
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/eguid_1/article/details/82767187
前言
ffmpeg获取的数据一般为yuv,argb,rgb,bgr,abgr等图像像素数据，我们可能需要转换为java的图像，来方便我们显示他们，当然不需要转换也可以达到我们的目的。
在线演示demo：https://blog.csdn.net/eguid_1/article/details/82842904
项目维护地址：https://github.com/eguid/easyCV
一、那么先来个RGB像素使用的小demo压压惊
（密集恐惧症预警）
通过这个RGB像素的小demo，更容易理解RGB像素格式
public static int getRGB(int[] rgbarr) {//RGB24数组转整型RGB24 int rgb = ((int) rgbarr[0]) & 0xff | (((int) rgbarr[1]) & 0xff) << 8 | (((int) rgbarr[2]) & 0xff) << 16 | 0xff000000; return rgb; } private static int createRandomRgb() {//随机生成RGB24 int[] rgbarr = new int[3]; rgbarr[0] = (int) (Math.random() * 255); rgbarr[1] = (int) (Math.random() * 255); rgbarr[2] = (int) (Math.random() * 255); return getRGB(rgbarr); } //eguid原创文章，转载请注明出处和作者名(blog.csdn.net/eguid_1) public static void main(String[] args){ int width = 800, height = 600; //使用整型RGB24 BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB); for (int i = 0; i < width; i++) { for (int j = 0; j < height; j++) { int rgb = createRandomRgb(); image.setRGB(i, j, rgb); } } JLabel label = new JLabel(); label.setSize(width, height); label.setIcon(new ImageIcon(image)); JFrame frame = new JFrame(); frame.setSize(width, height); frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); frame.add(label); frame.setVisible(true); Timer timer=new Timer("定时刷新", true); timer.scheduleAtFixedRate(new TimerTask() { @Override public void run() { for (int i = 0; i < width; i++) { for (int j = 0; j < height; j++) { int rgb =createRandomRgb(); image.setRGB(i, j, rgb); } } label.repaint(); } }, 100, 1000/25); }
运行结果：
RGB像素格式demo截图-eguid
二、像素图像数据转换为java图像
上一章我们已经通过ffmpeg获取到了AVFrame，要是能预览这帧视频图像那岂不是更好？
要完成转换为java的图像数据这个转换我们分为两步转换流程。
1、像素图像转换为ByteBuffer
不管是yuv，还是rgb或是bgr，都可以使用该函数转换为ByteBuffer，不同的是ByteBuffer转换为BufferedImage的异同。
public ByteBuffer saveFrame(AVFrame pFrame, int width, int height){ BytePointer data = pFrame.data(0); int size = width * height * 3; ByteBuffer buf = data.position(0).limit(size).asBuffer(); return buf; }
2、ByteBuffer转换为BufferImage
由于不同像素数据转换方法都不相同，以RGB和BGR为例：
（1）BGR24转BufferedImage
/** * 24位BGR转BufferedImage * @param src -源数据 * @param width -宽度 * @param height-高度 * @return */ public static BufferedImage BGR2BufferedImage(ByteBuffer src,int width,int height) { BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_3BYTE_BGR); Raster ra = image.getRaster(); DataBuffer out = ra.getDataBuffer(); DataBufferByte db=(DataBufferByte)out; ByteBuffer.wrap(db.getData()).put(src); return image; }
/** * 24位整型BGR转BufferedImage * @param src -源数据 * @param width -宽度 * @param height-高度 * @return */ public static BufferedImage BGR2BufferedImage(IntBuffer src,int width,int height) { BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_BGR); Raster ra = image.getRaster(); DataBuffer out = ra.getDataBuffer(); DataBufferInt db=(DataBufferInt)out; IntBuffer.wrap(db.getData()).put(src); return image; }
（2）RGB24转BufferedImage
/** * 24位整型RGB转BufferedImage * @param src -源数据 * @param width -宽度 * @param height-高度 * @return */ public static BufferedImage RGB2BufferedImage(IntBuffer src,int width,int height) { BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB); Raster ra = image.getRaster(); DataBuffer out = ra.getDataBuffer(); DataBufferInt db=(DataBufferInt)out; IntBuffer.wrap(db.getData()).put(src); return image; }
三、一个方便预览BufferImage的小函数
/** * 使用窗口显示BufferedImage图片 * @param image -BufferedImage */ public static void viewImage(BufferedImage image) { int width=image.getWidth(),height=image.getHeight(); JLabel label = new JLabel(); label.setSize(width, height); label.setIcon(new ImageIcon(image)); JFrame frame = new JFrame(); frame.setSize(width, height); frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); frame.add(label); frame.setVisible(true); }
四、补充
1、BufferImage转Base64编码，方便网页显示
/** * bufferedImage转base64 * @param format -格式（jpg,png,bmp,gif,jpeg等等） * @return * @throws IOException */ public static String bufferedImage2Base64(BufferedImage image, String format) throws IOException { Encoder encoder = Base64.getEncoder(); ByteArrayOutputStream baos = new ByteArrayOutputStream();// 字节流 ImageIO.write(image, format, baos);// 写出到字节流 byte[] bytes=baos.toByteArray(); // 编码成base64 String jpg_base64 = encoder.encodeToString(bytes); return jpg_base64; }
2、保存成图片文件
public static void saveImage(BufferedImage image,String format,String file) throws IOException { ImageIO.write(image, format, new File(file)); }
