encode YUV420 to jpeg, using libjpeg

最新推荐文章于 2024-05-16 15:58:09 发布

carlsonlee_freec

最新推荐文章于 2024-05-16 15:58:09 发布

阅读量2.1k

点赞数

文章标签： buffer null struct components include file

本文链接：https://blog.csdn.net/carlsonlee_freec/article/details/6233793

版权

#include <stdlib.h>
#include <stdio.h>
#include "jpeglib.h"
#include <setjmp.h>
#include <glib.h>

gboolean jpeg_enc_yv12(guchar* buffer, gint width, gint height, gint quality, char* filename)
{
   struct jpeg_compress_struct cinfo;
   struct jpeg_error_mgr jerr;
   FILE *outfile = NULL;
   gboolean ret = TRUE;
   if(buffer == NULL || width <=0 || height <=0|| filename == NULL)
       return FALSE;
   if ((outfile = fopen(filename, "wb")) == NULL)
   {
       return FALSE;
   }
   cinfo.err = jpeg_std_error(&jerr);
   jpeg_create_compress(&cinfo);
   jpeg_stdio_dest(&cinfo, outfile);

   cinfo.image_width = width;
   cinfo.image_height = height;
   cinfo.input_components = 3;
   cinfo.in_color_space = JCS_YCbCr;
   jpeg_set_defaults(&cinfo);
   jpeg_set_quality(&cinfo, quality, TRUE);
   cinfo.raw_data_in = TRUE;

   {
       JSAMPARRAY pp[3];
       JSAMPROW *rpY = (JSAMPROW*)malloc(sizeof(JSAMPROW) * height);
       JSAMPROW *rpU = (JSAMPROW*)malloc(sizeof(JSAMPROW) * height);
       JSAMPROW *rpV = (JSAMPROW*)malloc(sizeof(JSAMPROW) * height);
       int k;
       if(rpY == NULL && rpU == NULL && rpV == NULL)
       {
           ret = FALSE;
           goto exit;
       }
       cinfo.comp_info[0].h_samp_factor =
       cinfo.comp_info[0].v_samp_factor = 2;
       cinfo.comp_info[1].h_samp_factor =
       cinfo.comp_info[1].v_samp_factor =
       cinfo.comp_info[2].h_samp_factor =
       cinfo.comp_info[2].v_samp_factor = 1;
       jpeg_start_compress(&cinfo, TRUE);

       for (k = 0; k < height; k+=2)
       {
           rpY[k]   = buffer + k*width;
           rpY[k+1] = buffer + (k+1)*width;
           rpU[k/2] = buffer+width*height + (k/2)*width/2;
           rpV[k/2] = buffer+width*height*5/4 + (k/2)*width/2;
       }
       for (k = 0; k < height; k+=2*DCTSIZE)
       {
           pp[0] = &rpY[k];
           pp[1] = &rpU[k/2];
           pp[2] = &rpV[k/2];
           jpeg_write_raw_data(&cinfo, pp, 2*DCTSIZE);
       }
       jpeg_finish_compress(&cinfo);
       free(rpY);
       free(rpU);
       free(rpV);
   }
exit:
   fclose(outfile);
   jpeg_destroy_compress(&cinfo);
   return ret;
}

int main()
{
   guchar *buffer = malloc(640*480*3/2);
   jpeg_enc_yv12(buffer, 640, 480, 90, "test.jpg");
   free(buffer);
}