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

图片显示

开发平台:

Visual C++

  1. /*
  2.  * jcmaster.c
  3.  *
  4.  * Copyright (C) 1991-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 master control logic for the JPEG compressor.
  9.  * These routines are concerned with selecting the modules to be executed
  10.  * and with determining the number of passes and the work to be done in each
  11.  * pass.
  12.  */
  13. #define JPEG_INTERNALS
  14. #include "jinclude.h"
  15. #include "jpeglib.h"
  16. /* Private state */
  17. typedef struct {
  18.   struct jpeg_comp_master pub; /* public fields */
  19.   int pass_number; /* eventually need more complex state... */
  20. } my_comp_master;
  21. typedef my_comp_master * my_master_ptr;
  22. /*
  23.  * Support routines that do various essential calculations.
  24.  */
  25. LOCAL void
  26. initial_setup (j_compress_ptr cinfo)
  27. /* Do computations that are needed before master selection phase */
  28. {
  29.   int ci;
  30.   jpeg_component_info *compptr;
  31.   long samplesperrow;
  32.   JDIMENSION jd_samplesperrow;
  33.   /* Sanity check on image dimensions */
  34.   if (cinfo->image_height <= 0 || cinfo->image_width <= 0
  35.       || cinfo->num_components <= 0 || cinfo->input_components <= 0)
  36.     ERREXIT(cinfo, JERR_EMPTY_IMAGE);
  37.   /* Make sure image isn't bigger than I can handle */
  38.   if ((long) cinfo->image_height > (long) JPEG_MAX_DIMENSION ||
  39.       (long) cinfo->image_width > (long) JPEG_MAX_DIMENSION)
  40.     ERREXIT1(cinfo, JERR_IMAGE_TOO_BIG, (unsigned int) JPEG_MAX_DIMENSION);
  41.   /* Width of an input scanline must be representable as JDIMENSION. */
  42.   samplesperrow = (long) cinfo->image_width * (long) cinfo->input_components;
  43.   jd_samplesperrow = (JDIMENSION) samplesperrow;
  44.   if ((long) jd_samplesperrow != samplesperrow)
  45.     ERREXIT(cinfo, JERR_WIDTH_OVERFLOW);
  46.   /* For now, precision must match compiled-in value... */
  47.   if (cinfo->data_precision != BITS_IN_JSAMPLE)
  48.     ERREXIT1(cinfo, JERR_BAD_PRECISION, cinfo->data_precision);
  49.   /* Check that number of components won't exceed internal array sizes */
  50.   if (cinfo->num_components > MAX_COMPONENTS)
  51.     ERREXIT2(cinfo, JERR_COMPONENT_COUNT, cinfo->num_components,
  52.      MAX_COMPONENTS);
  53.   /* Compute maximum sampling factors; check factor validity */
  54.   cinfo->max_h_samp_factor = 1;
  55.   cinfo->max_v_samp_factor = 1;
  56.   for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
  57.        ci++, compptr++) {
  58.     if (compptr->h_samp_factor<=0 || compptr->h_samp_factor>MAX_SAMP_FACTOR ||
  59. compptr->v_samp_factor<=0 || compptr->v_samp_factor>MAX_SAMP_FACTOR)
  60.       ERREXIT(cinfo, JERR_BAD_SAMPLING);
  61.     cinfo->max_h_samp_factor = MAX(cinfo->max_h_samp_factor,
  62.    compptr->h_samp_factor);
  63.     cinfo->max_v_samp_factor = MAX(cinfo->max_v_samp_factor,
  64.    compptr->v_samp_factor);
  65.   }
  66.   /* Compute dimensions of components */
  67.   for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
  68.        ci++, compptr++) {
  69.     /* For compression, we never do DCT scaling. */
  70.     compptr->DCT_scaled_size = DCTSIZE;
  71.     /* Size in DCT blocks */
  72.     compptr->width_in_blocks = (JDIMENSION)
  73.       jdiv_round_up((long) cinfo->image_width * (long) compptr->h_samp_factor,
  74.     (long) (cinfo->max_h_samp_factor * DCTSIZE));
  75.     compptr->height_in_blocks = (JDIMENSION)
  76.       jdiv_round_up((long) cinfo->image_height * (long) compptr->v_samp_factor,
  77.     (long) (cinfo->max_v_samp_factor * DCTSIZE));
  78.     /* Size in samples */
  79.     compptr->downsampled_width = (JDIMENSION)
  80.       jdiv_round_up((long) cinfo->image_width * (long) compptr->h_samp_factor,
  81.     (long) cinfo->max_h_samp_factor);
  82.     compptr->downsampled_height = (JDIMENSION)
  83.       jdiv_round_up((long) cinfo->image_height * (long) compptr->v_samp_factor,
  84.     (long) cinfo->max_v_samp_factor);
  85.     /* Mark component needed (this flag isn't actually used for compression) */
  86.     compptr->component_needed = TRUE;
  87.   }
  88.   /* Compute number of fully interleaved MCU rows (number of times that
  89.    * main controller will call coefficient controller).
  90.    */
  91.   cinfo->total_iMCU_rows = (JDIMENSION)
  92.     jdiv_round_up((long) cinfo->image_height,
  93.   (long) (cinfo->max_v_samp_factor*DCTSIZE));
  94. }
  95. LOCAL void
  96. per_scan_setup (j_compress_ptr cinfo)
  97. /* Do computations that are needed before processing a JPEG scan */
  98. /* cinfo->comps_in_scan and cinfo->cur_comp_info[] are already set */
  99. {
  100.   int ci, mcublks, tmp;
  101.   jpeg_component_info *compptr;
  102.   
  103.   if (cinfo->comps_in_scan == 1) {
  104.     
  105.     /* Noninterleaved (single-component) scan */
  106.     compptr = cinfo->cur_comp_info[0];
  107.     
  108.     /* Overall image size in MCUs */
  109.     cinfo->MCUs_per_row = compptr->width_in_blocks;
  110.     cinfo->MCU_rows_in_scan = compptr->height_in_blocks;
  111.     
  112.     /* For noninterleaved scan, always one block per MCU */
  113.     compptr->MCU_width = 1;
  114.     compptr->MCU_height = 1;
  115.     compptr->MCU_blocks = 1;
  116.     compptr->MCU_sample_width = DCTSIZE;
  117.     compptr->last_col_width = 1;
  118.     compptr->last_row_height = 1;
  119.     
  120.     /* Prepare array describing MCU composition */
  121.     cinfo->blocks_in_MCU = 1;
  122.     cinfo->MCU_membership[0] = 0;
  123.     
  124.   } else {
  125.     
  126.     /* Interleaved (multi-component) scan */
  127.     if (cinfo->comps_in_scan <= 0 || cinfo->comps_in_scan > MAX_COMPS_IN_SCAN)
  128.       ERREXIT2(cinfo, JERR_COMPONENT_COUNT, cinfo->comps_in_scan,
  129.        MAX_COMPS_IN_SCAN);
  130.     
  131.     /* Overall image size in MCUs */
  132.     cinfo->MCUs_per_row = (JDIMENSION)
  133.       jdiv_round_up((long) cinfo->image_width,
  134.     (long) (cinfo->max_h_samp_factor*DCTSIZE));
  135.     cinfo->MCU_rows_in_scan = (JDIMENSION)
  136.       jdiv_round_up((long) cinfo->image_height,
  137.     (long) (cinfo->max_v_samp_factor*DCTSIZE));
  138.     
  139.     cinfo->blocks_in_MCU = 0;
  140.     
  141.     for (ci = 0; ci < cinfo->comps_in_scan; ci++) {
  142.       compptr = cinfo->cur_comp_info[ci];
  143.       /* Sampling factors give # of blocks of component in each MCU */
  144.       compptr->MCU_width = compptr->h_samp_factor;
  145.       compptr->MCU_height = compptr->v_samp_factor;
  146.       compptr->MCU_blocks = compptr->MCU_width * compptr->MCU_height;
  147.       compptr->MCU_sample_width = compptr->MCU_width * DCTSIZE;
  148.       /* Figure number of non-dummy blocks in last MCU column & row */
  149.       tmp = (int) (compptr->width_in_blocks % compptr->MCU_width);
  150.       if (tmp == 0) tmp = compptr->MCU_width;
  151.       compptr->last_col_width = tmp;
  152.       tmp = (int) (compptr->height_in_blocks % compptr->MCU_height);
  153.       if (tmp == 0) tmp = compptr->MCU_height;
  154.       compptr->last_row_height = tmp;
  155.       /* Prepare array describing MCU composition */
  156.       mcublks = compptr->MCU_blocks;
  157.       if (cinfo->blocks_in_MCU + mcublks > MAX_BLOCKS_IN_MCU)
  158. ERREXIT(cinfo, JERR_BAD_MCU_SIZE);
  159.       while (mcublks-- > 0) {
  160. cinfo->MCU_membership[cinfo->blocks_in_MCU++] = ci;
  161.       }
  162.     }
  163.     
  164.   }
  165.   /* Convert restart specified in rows to actual MCU count. */
  166.   /* Note that count must fit in 16 bits, so we provide limiting. */
  167.   if (cinfo->restart_in_rows > 0) {
  168.     long nominal = (long) cinfo->restart_in_rows * (long) cinfo->MCUs_per_row;
  169.     cinfo->restart_interval = (unsigned int) MIN(nominal, 65535L);
  170.   }
  171. }
  172. /*
  173.  * Master selection of compression modules.
  174.  * This is done once at the start of processing an image.  We determine
  175.  * which modules will be used and give them appropriate initialization calls.
  176.  */
  177. LOCAL void
  178. master_selection (j_compress_ptr cinfo)
  179. {
  180.   my_master_ptr master = (my_master_ptr) cinfo->master;
  181.   initial_setup(cinfo);
  182.   master->pass_number = 0;
  183.   /* There's not a lot of smarts here right now, but it'll get more
  184.    * complicated when we have multiple implementations available...
  185.    */
  186.   /* Preprocessing */
  187.   if (! cinfo->raw_data_in) {
  188.     jinit_color_converter(cinfo);
  189.     jinit_downsampler(cinfo);
  190.     jinit_c_prep_controller(cinfo, FALSE /* never need full buffer here */);
  191.   }
  192.   /* Forward DCT */
  193.   jinit_forward_dct(cinfo);
  194.   /* Entropy encoding: either Huffman or arithmetic coding. */
  195.   if (cinfo->arith_code) {
  196. #ifdef C_ARITH_CODING_SUPPORTED
  197.     jinit_arith_encoder(cinfo);
  198. #else
  199.     ERREXIT(cinfo, JERR_ARITH_NOTIMPL);
  200. #endif
  201.   } else
  202.     jinit_huff_encoder(cinfo);
  203.   /* For now, a full buffer is needed only for Huffman optimization. */
  204.   jinit_c_coef_controller(cinfo, cinfo->optimize_coding);
  205.   jinit_c_main_controller(cinfo, FALSE /* never need full buffer here */);
  206.   jinit_marker_writer(cinfo);
  207.   /* We can now tell the memory manager to allocate virtual arrays. */
  208.   (*cinfo->mem->realize_virt_arrays) ((j_common_ptr) cinfo);
  209.   /* Write the datastream header (SOI) immediately.
  210.    * Frame and scan headers are postponed till later.
  211.    * This lets application insert special markers after the SOI.
  212.    */
  213.   (*cinfo->marker->write_file_header) (cinfo);
  214. }
  215. /*
  216.  * Per-pass setup.
  217.  * This is called at the beginning of each pass.  We determine which modules
  218.  * will be active during this pass and give them appropriate start_pass calls.
  219.  * We also set is_last_pass to indicate whether any more passes will be
  220.  * required.
  221.  */
  222. METHODDEF void
  223. prepare_for_pass (j_compress_ptr cinfo)
  224. {
  225.   my_master_ptr master = (my_master_ptr) cinfo->master;
  226.   int ci;
  227.   int npasses;
  228.   /* ???? JUST A QUICK CROCK FOR NOW ??? */
  229.   /* For now, handle only single interleaved output scan; */
  230.   /* we support two passes for Huffman optimization. */
  231.   /* Prepare for single scan containing all components */
  232.   if (cinfo->num_components > MAX_COMPS_IN_SCAN)
  233.     ERREXIT2(cinfo, JERR_COMPONENT_COUNT, cinfo->num_components,
  234.      MAX_COMPS_IN_SCAN);
  235.   cinfo->comps_in_scan = cinfo->num_components;
  236.   for (ci = 0; ci < cinfo->num_components; ci++) {
  237.     cinfo->cur_comp_info[ci] = &cinfo->comp_info[ci];
  238.   }
  239.   per_scan_setup(cinfo);
  240.   if (! cinfo->optimize_coding) {
  241.     /* Standard single-pass case */
  242.     npasses = 1;
  243.     master->pub.call_pass_startup = TRUE;
  244.     master->pub.is_last_pass = TRUE;
  245.     if (! cinfo->raw_data_in) {
  246.       (*cinfo->cconvert->start_pass) (cinfo);
  247.       (*cinfo->downsample->start_pass) (cinfo);
  248.       (*cinfo->prep->start_pass) (cinfo, JBUF_PASS_THRU);
  249.     }
  250.     (*cinfo->fdct->start_pass) (cinfo);
  251.     (*cinfo->entropy->start_pass) (cinfo, FALSE);
  252.     (*cinfo->coef->start_pass) (cinfo, JBUF_PASS_THRU);
  253.     (*cinfo->main->start_pass) (cinfo, JBUF_PASS_THRU);
  254.   } else {
  255.     npasses = 2;
  256.     switch (master->pass_number) {
  257.     case 0:
  258.       /* Huffman optimization: run all modules, gather statistics */
  259.       master->pub.call_pass_startup = FALSE;
  260.       master->pub.is_last_pass = FALSE;
  261.       if (! cinfo->raw_data_in) {
  262. (*cinfo->cconvert->start_pass) (cinfo);
  263. (*cinfo->downsample->start_pass) (cinfo);
  264. (*cinfo->prep->start_pass) (cinfo, JBUF_PASS_THRU);
  265.       }
  266.       (*cinfo->fdct->start_pass) (cinfo);
  267.       (*cinfo->entropy->start_pass) (cinfo, TRUE);
  268.       (*cinfo->coef->start_pass) (cinfo, JBUF_SAVE_AND_PASS);
  269.       (*cinfo->main->start_pass) (cinfo, JBUF_PASS_THRU);
  270.       break;
  271.     case 1:
  272.       /* Second pass: reread data from coefficient buffer */
  273.       master->pub.is_last_pass = TRUE;
  274.       (*cinfo->entropy->start_pass) (cinfo, FALSE);
  275.       (*cinfo->coef->start_pass) (cinfo, JBUF_CRANK_DEST);
  276.       /* We emit frame/scan headers now */
  277.       (*cinfo->marker->write_frame_header) (cinfo);
  278.       (*cinfo->marker->write_scan_header) (cinfo);
  279.       break;
  280.     }
  281.   }
  282.   /* Set up progress monitor's pass info if present */
  283.   if (cinfo->progress != NULL) {
  284.     cinfo->progress->completed_passes = master->pass_number;
  285.     cinfo->progress->total_passes = npasses;
  286.   }
  287.   master->pass_number++;
  288. }
  289. /*
  290.  * Special start-of-pass hook.
  291.  * This is called by jpeg_write_scanlines if call_pass_startup is TRUE.
  292.  * In single-pass processing, we need this hook because we don't want to
  293.  * write frame/scan headers during jpeg_start_compress; we want to let the
  294.  * application write COM markers etc. between jpeg_start_compress and the
  295.  * jpeg_write_scanlines loop.
  296.  * In multi-pass processing, this routine is not used.
  297.  */
  298. METHODDEF void
  299. pass_startup (j_compress_ptr cinfo)
  300. {
  301.   cinfo->master->call_pass_startup = FALSE; /* reset flag so call only once */
  302.   (*cinfo->marker->write_frame_header) (cinfo);
  303.   (*cinfo->marker->write_scan_header) (cinfo);
  304. }
  305. /*
  306.  * Finish up at end of pass.
  307.  */
  308. METHODDEF void
  309. finish_pass_master (j_compress_ptr cinfo)
  310. {
  311.   /* More complex logic later ??? */
  312.   /* The entropy coder needs an end-of-pass call, either to analyze
  313.    * statistics or to flush its output buffer.
  314.    */
  315.   (*cinfo->entropy->finish_pass) (cinfo);
  316. }
  317. /*
  318.  * Initialize master compression control.
  319.  * This creates my own subrecord and also performs the master selection phase,
  320.  * which causes other modules to create their subrecords.
  321.  */
  322. GLOBAL void
  323. jinit_master_compress (j_compress_ptr cinfo)
  324. {
  325.   my_master_ptr master;
  326.   master = (my_master_ptr)
  327.       (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
  328.   SIZEOF(my_comp_master));
  329.   cinfo->master = (struct jpeg_comp_master *) master;
  330.   master->pub.prepare_for_pass = prepare_for_pass;
  331.   master->pub.pass_startup = pass_startup;
  332.   master->pub.finish_pass = finish_pass_master;
  333.   master_selection(cinfo);
  334. }