JCMAINCT.C
上传用户:wep9318
上传日期:2007-01-07
资源大小:893k
文件大小:9k
源码类别:

图片显示

开发平台:

Visual C++

  1. /*
  2.  * jcmainct.c
  3.  *
  4.  * Copyright (C) 1994, Thomas G. Lane.
  5.  * This file is part of the Independent JPEG Group's software.
  6.  * For conditions of distribution and use, see the accompanying README file.
  7.  *
  8.  * This file contains the main buffer controller for compression.
  9.  * The main buffer lies between the pre-processor and the JPEG
  10.  * compressor proper; it holds downsampled data in the JPEG colorspace.
  11.  */
  12. #define JPEG_INTERNALS
  13. #include "jinclude.h"
  14. #include "jpeglib.h"
  15. /* Note: currently, there is no operating mode in which a full-image buffer
  16.  * is needed at this step.  If there were, that mode could not be used with
  17.  * "raw data" input, since this module is bypassed in that case.  However,
  18.  * we've left the code here for possible use in special applications.
  19.  */
  20. #undef FULL_MAIN_BUFFER_SUPPORTED
  21. /* Private buffer controller object */
  22. typedef struct {
  23.   struct jpeg_c_main_controller pub; /* public fields */
  24.   JDIMENSION cur_mcu_row; /* number of current iMCU row */
  25.   JDIMENSION rowgroup_ctr; /* counts row groups received in iMCU row */
  26.   JDIMENSION mcu_ctr; /* counts MCUs output from current row */
  27.   boolean suspended; /* remember if we suspended output */
  28.   J_BUF_MODE pass_mode; /* current operating mode */
  29.   /* If using just a strip buffer, this points to the entire set of buffers
  30.    * (we allocate one for each component).  In the full-image case, this
  31.    * points to the currently accessible strips of the virtual arrays.
  32.    */
  33.   JSAMPARRAY buffer[MAX_COMPONENTS];
  34. #ifdef FULL_MAIN_BUFFER_SUPPORTED
  35.   /* If using full-image storage, this array holds pointers to virtual-array
  36.    * control blocks for each component.  Unused if not full-image storage.
  37.    */
  38.   jvirt_sarray_ptr whole_image[MAX_COMPONENTS];
  39. #endif
  40. } my_main_controller;
  41. typedef my_main_controller * my_main_ptr;
  42. /* Forward declarations */
  43. METHODDEF void process_data_simple_main
  44. JPP((j_compress_ptr cinfo, JSAMPARRAY input_buf,
  45.      JDIMENSION *in_row_ctr, JDIMENSION in_rows_avail));
  46. #ifdef FULL_MAIN_BUFFER_SUPPORTED
  47. METHODDEF void process_data_buffer_main
  48. JPP((j_compress_ptr cinfo, JSAMPARRAY input_buf,
  49.      JDIMENSION *in_row_ctr, JDIMENSION in_rows_avail));
  50. #endif
  51. /*
  52.  * Initialize for a processing pass.
  53.  */
  54. METHODDEF void
  55. start_pass_main (j_compress_ptr cinfo, J_BUF_MODE pass_mode)
  56. {
  57.   my_main_ptr main = (my_main_ptr) cinfo->main;
  58.   /* Do nothing in raw-data mode. */
  59.   if (cinfo->raw_data_in)
  60.     return;
  61.   main->cur_mcu_row = 0; /* initialize counters */
  62.   main->rowgroup_ctr = 0;
  63.   main->mcu_ctr = 0;
  64.   main->suspended = FALSE;
  65.   main->pass_mode = pass_mode; /* save mode for use by process_data */
  66.   switch (pass_mode) {
  67.   case JBUF_PASS_THRU:
  68. #ifdef FULL_MAIN_BUFFER_SUPPORTED
  69.     if (main->whole_image[0] != NULL)
  70.       ERREXIT(cinfo, JERR_BAD_BUFFER_MODE);
  71. #endif
  72.     main->pub.process_data = process_data_simple_main;
  73.     break;
  74. #ifdef FULL_MAIN_BUFFER_SUPPORTED
  75.   case JBUF_SAVE_SOURCE:
  76.   case JBUF_CRANK_DEST:
  77.   case JBUF_SAVE_AND_PASS:
  78.     if (main->whole_image[0] == NULL)
  79.       ERREXIT(cinfo, JERR_BAD_BUFFER_MODE);
  80.     main->pub.process_data = process_data_buffer_main;
  81.     break;
  82. #endif
  83.   default:
  84.     ERREXIT(cinfo, JERR_BAD_BUFFER_MODE);
  85.     break;
  86.   }
  87. }
  88. /*
  89.  * Process some data.
  90.  * This routine handles the simple pass-through mode,
  91.  * where we have only a strip buffer.
  92.  */
  93. METHODDEF void
  94. process_data_simple_main (j_compress_ptr cinfo,
  95.   JSAMPARRAY input_buf, JDIMENSION *in_row_ctr,
  96.   JDIMENSION in_rows_avail)
  97. {
  98.   my_main_ptr main = (my_main_ptr) cinfo->main;
  99.   while (main->cur_mcu_row < cinfo->total_iMCU_rows) {
  100.     /* Read input data if we haven't filled the main buffer yet */
  101.     if (main->rowgroup_ctr < DCTSIZE)
  102.       (*cinfo->prep->pre_process_data) (cinfo,
  103. input_buf, in_row_ctr, in_rows_avail,
  104. main->buffer, &main->rowgroup_ctr,
  105. (JDIMENSION) DCTSIZE);
  106.     /* If we don't have a full iMCU row buffered, return to application for
  107.      * more data.  Note that preprocessor will always pad to fill the iMCU row
  108.      * at the bottom of the image.
  109.      */
  110.     if (main->rowgroup_ctr != DCTSIZE)
  111.       return;
  112.     /* Send the completed row to the compressor */
  113.     (*cinfo->coef->compress_data) (cinfo, main->buffer, &main->mcu_ctr);
  114.     /* If compressor did not consume the whole row, then we must need to
  115.      * suspend processing and return to the application.  In this situation
  116.      * we pretend we didn't yet consume the last input row; otherwise, if
  117.      * it happened to be the last row of the image, the application would
  118.      * think we were done.
  119.      */
  120.     if (main->mcu_ctr < cinfo->MCUs_per_row) {
  121.       if (! main->suspended) {
  122. (*in_row_ctr)--;
  123. main->suspended = TRUE;
  124.       }
  125.       return;
  126.     }
  127.     /* We did finish the row.  Undo our little suspension hack if a previous
  128.      * call suspended; then mark the main buffer empty.
  129.      */
  130.     if (main->suspended) {
  131.       (*in_row_ctr)++;
  132.       main->suspended = FALSE;
  133.     }
  134.     main->mcu_ctr = 0;
  135.     main->rowgroup_ctr = 0;
  136.     main->cur_mcu_row++;
  137.   }
  138. }
  139. #ifdef FULL_MAIN_BUFFER_SUPPORTED
  140. /*
  141.  * Process some data.
  142.  * This routine handles all of the modes that use a full-size buffer.
  143.  */
  144. METHODDEF void
  145. process_data_buffer_main (j_compress_ptr cinfo,
  146.   JSAMPARRAY input_buf, JDIMENSION *in_row_ctr,
  147.   JDIMENSION in_rows_avail)
  148. {
  149.   my_main_ptr main = (my_main_ptr) cinfo->main;
  150.   int ci;
  151.   jpeg_component_info *compptr;
  152.   boolean writing = (main->pass_mode != JBUF_CRANK_DEST);
  153.   while (main->cur_mcu_row < cinfo->total_iMCU_rows) {
  154.     /* Realign the virtual buffers if at the start of an iMCU row. */
  155.     if (main->rowgroup_ctr == 0) {
  156.       for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
  157.    ci++, compptr++) {
  158. main->buffer[ci] = (*cinfo->mem->access_virt_sarray)
  159.   ((j_common_ptr) cinfo, main->whole_image[ci],
  160.    main->cur_mcu_row * (compptr->v_samp_factor * DCTSIZE), writing);
  161.       }
  162.       /* In a read pass, pretend we just read some source data. */
  163.       if (! writing) {
  164. *in_row_ctr += cinfo->max_v_samp_factor * DCTSIZE;
  165. main->rowgroup_ctr = DCTSIZE;
  166.       }
  167.     }
  168.     /* If a write pass, read input data until the current iMCU row is full. */
  169.     /* Note: preprocessor will pad if necessary to fill the last iMCU row. */
  170.     if (writing) {
  171.       (*cinfo->prep->pre_process_data) (cinfo,
  172. input_buf, in_row_ctr, in_rows_avail,
  173. main->buffer, &main->rowgroup_ctr,
  174. (JDIMENSION) DCTSIZE);
  175.       /* Return to application if we need more data to fill the iMCU row. */
  176.       if (main->rowgroup_ctr < DCTSIZE)
  177. return;
  178.     }
  179.     /* Emit data, unless this is a sink-only pass. */
  180.     if (main->pass_mode != JBUF_SAVE_SOURCE) {
  181.       (*cinfo->coef->compress_data) (cinfo, main->buffer, &main->mcu_ctr);
  182.       /* If compressor did not consume the whole row, then we must need to
  183.        * suspend processing and return to the application.  In this situation
  184.        * we pretend we didn't yet consume the last input row; otherwise, if
  185.        * it happened to be the last row of the image, the application would
  186.        * think we were done.
  187.        */
  188.       if (main->mcu_ctr < cinfo->MCUs_per_row) {
  189. if (! main->suspended) {
  190.   (*in_row_ctr)--;
  191.   main->suspended = TRUE;
  192. }
  193. return;
  194.       }
  195.       /* We did finish the row.  Undo our little suspension hack if a previous
  196.        * call suspended; then mark the main buffer empty.
  197.        */
  198.       if (main->suspended) {
  199. (*in_row_ctr)++;
  200. main->suspended = FALSE;
  201.       }
  202.     }
  203.     /* If get here, we are done with this iMCU row.  Mark buffer empty. */
  204.     main->mcu_ctr = 0;
  205.     main->rowgroup_ctr = 0;
  206.     main->cur_mcu_row++;
  207.   }
  208. }
  209. #endif /* FULL_MAIN_BUFFER_SUPPORTED */
  210. /*
  211.  * Initialize main buffer controller.
  212.  */
  213. GLOBAL void
  214. jinit_c_main_controller (j_compress_ptr cinfo, boolean need_full_buffer)
  215. {
  216.   my_main_ptr main;
  217.   int ci;
  218.   jpeg_component_info *compptr;
  219.   main = (my_main_ptr)
  220.     (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
  221. SIZEOF(my_main_controller));
  222.   cinfo->main = (struct jpeg_c_main_controller *) main;
  223.   main->pub.start_pass = start_pass_main;
  224.   /* We don't need to create a buffer in raw-data mode. */
  225.   if (cinfo->raw_data_in)
  226.     return;
  227.   /* Create the buffer.  It holds downsampled data, so each component
  228.    * may be of a different size.
  229.    */
  230.   if (need_full_buffer) {
  231. #ifdef FULL_MAIN_BUFFER_SUPPORTED
  232.     /* Allocate a full-image virtual array for each component */
  233.     /* Note we implicitly pad the bottom to a multiple of the iMCU height */
  234.     for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
  235.  ci++, compptr++) {
  236.       main->whole_image[ci] = (*cinfo->mem->request_virt_sarray)
  237. ((j_common_ptr) cinfo, JPOOL_IMAGE,
  238.  compptr->width_in_blocks * DCTSIZE,
  239.  compptr->height_in_blocks * DCTSIZE,
  240.  (JDIMENSION) (compptr->v_samp_factor * DCTSIZE));
  241.     }
  242. #else
  243.     ERREXIT(cinfo, JERR_BAD_BUFFER_MODE);
  244. #endif
  245.   } else {
  246. #ifdef FULL_MAIN_BUFFER_SUPPORTED
  247.     main->whole_image[0] = NULL; /* flag for no virtual arrays */
  248. #endif
  249.     /* Allocate a strip buffer for each component */
  250.     for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
  251.  ci++, compptr++) {
  252.       main->buffer[ci] = (*cinfo->mem->alloc_sarray)
  253. ((j_common_ptr) cinfo, JPOOL_IMAGE,
  254.  compptr->width_in_blocks * DCTSIZE,
  255.  (JDIMENSION) (compptr->v_samp_factor * DCTSIZE));
  256.     }
  257.   }
  258. }