使用libjpeg 压缩yuv420到jpg (内存方式)

#include <Windows.h>#include <stdio.h>  extern  "C" {#include <jpeglib.h> }  #define WIDTH 352#define HEIGHT 288#define QUALITY 80#define BUFFER_SZIE (WIDTH*HEIGHT*2)  /* The following declarations and 5 functions are jpeg related  * functions used by put_jpeg_grey_memory and put_jpeg_yuv420p_memory */typedef struct {    struct jpeg_destination_mgr pub;    JOCTET *buf;    size_t bufsize;    size_t jpegsize;} mem_destination_mgr;  typedef mem_destination_mgr *mem_dest_ptr;    METHODDEF(void) init_destination(j_compress_ptr cinfo){    mem_dest_ptr dest = (mem_dest_ptr) cinfo->dest;    dest->pub.next_output_byte = dest->buf;    dest->pub.free_in_buffer = dest->bufsize;    dest->jpegsize = 0;}  METHODDEF(boolean) empty_output_buffer(j_compress_ptr cinfo){    mem_dest_ptr dest = (mem_dest_ptr) cinfo->dest;    dest->pub.next_output_byte = dest->buf;    dest->pub.free_in_buffer = dest->bufsize;      return FALSE;}  METHODDEF(void) term_destination(j_compress_ptr cinfo){    mem_dest_ptr dest = (mem_dest_ptr) cinfo->dest;    dest->jpegsize = dest->bufsize - dest->pub.free_in_buffer;}  static GLOBAL(void) jpeg_mem_dest(j_compress_ptr cinfo, JOCTET* buf, size_t bufsize){    mem_dest_ptr dest;      if (cinfo->dest == NULL) {        cinfo->dest = (struct jpeg_destination_mgr *)            (*cinfo->mem->alloc_small)((j_common_ptr)cinfo, JPOOL_PERMANENT,            sizeof(mem_destination_mgr));    }      dest = (mem_dest_ptr) cinfo->dest;      dest->pub.init_destination    = init_destination;    dest->pub.empty_output_buffer = empty_output_buffer;    dest->pub.term_destination    = term_destination;      dest->buf      = buf;    dest->bufsize  = bufsize;    dest->jpegsize = 0;}  static GLOBAL(int) jpeg_mem_size(j_compress_ptr cinfo){    mem_dest_ptr dest = (mem_dest_ptr) cinfo->dest;    return dest->jpegsize;}      /* put_jpeg_yuv420p_memory converts an input image in the YUV420P format into a jpeg image and puts * it in a memory buffer. * Inputs: * - input_image is the image in YUV420P format. * - width and height are the dimensions of the image * Output: * - dest_image is a pointer to the jpeg image buffer * Returns buffer size of jpeg image      */static int put_jpeg_yuv420p_memory(unsigned char *dest_image,                                   unsigned char *input_image, int width, int height){    int i, j, jpeg_image_size;      JSAMPROW y[16],cb[16],cr[16]; // y[2][5] = color sample of row 2 and pixel column 5; (one plane)    JSAMPARRAY data[3]; // t[0][2][5] = color sample 0 of row 2 and column 5      struct jpeg_compress_struct cinfo;    struct jpeg_error_mgr jerr;      data[0] = y;    data[1] = cb;    data[2] = cr;      cinfo.err = jpeg_std_error(&jerr);  // errors get written to stderr           jpeg_create_compress(&cinfo);    cinfo.image_width = width;    cinfo.image_height = height;    cinfo.input_components = 3;    jpeg_set_defaults (&cinfo);      jpeg_set_colorspace(&cinfo, JCS_YCbCr);      cinfo.raw_data_in = TRUE;                  // supply downsampled data    cinfo.do_fancy_downsampling = FALSE;       // fix segfaulst with v7    cinfo.comp_info[0].h_samp_factor = 2;    cinfo.comp_info[0].v_samp_factor = 2;    cinfo.comp_info[1].h_samp_factor = 1;    cinfo.comp_info[1].v_samp_factor = 1;    cinfo.comp_info[2].h_samp_factor = 1;    cinfo.comp_info[2].v_samp_factor = 1;      jpeg_set_quality(&cinfo, QUALITY, TRUE);    cinfo.dct_method = JDCT_FASTEST;      jpeg_mem_dest(&cinfo, dest_image, BUFFER_SZIE);    // data written to mem          jpeg_start_compress (&cinfo, TRUE);      for (j = 0; j < height; j += 16) {        for (i = 0; i < 16; i++) {            y[i] = input_image + width * (i + j);            if (i%2 == 0) {                cb[i/2] = input_image + width * height + width / 2 * ((i + j) / 2);                cr[i/2] = input_image + width * height + width * height / 4 + width / 2 * ((i + j) / 2);            }        }        jpeg_write_raw_data(&cinfo, data, 16);    }      jpeg_finish_compress(&cinfo);    jpeg_image_size = jpeg_mem_size(&cinfo);    jpeg_destroy_compress(&cinfo);          return jpeg_image_size;}    int main( int argc, TCHAR * argv[], TCHAR * envp[] ){    HANDLE fyuv,fyuvjpg;    BYTE *pSrc ,*pDst;    LONG  lSize = 0;    DWORD  readsize;    DWORD  writesize;      pSrc = new BYTE[BUFFER_SZIE];    fyuv = CreateFile(L"cif.yuv", GENERIC_ALL, 0, NULL, OPEN_EXISTING, 0, NULL);    ReadFile(fyuv , pSrc ,BUFFER_SZIE ,&readsize,NULL);      pDst = new BYTE[BUFFER_SZIE];    lSize = put_jpeg_yuv420p_memory(pDst,pSrc , WIDTH ,HEIGHT);    fyuvjpg = CreateFile(L"cif_yuv.jpg", GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);    WriteFile(fyuvjpg, pDst, lSize, &writesize, NULL);      CloseHandle(fyuv);    CloseHandle(fyuvjpg);    return 0;}