3rdparty/libjpeg/jdtrans.c

/*
 * jdtrans.c
 *
 * Copyright (C) 1995-1997, Thomas G. Lane.
 * This file is part of the Independent JPEG Group's software.
 * For conditions of distribution and use, see the accompanying README file.
 *
 * This file contains library routines for transcoding decompression,
 * that is, reading raw DCT coefficient arrays from an input JPEG file.
 * The routines in jdapimin.c will also be needed by a transcoder.
 */

#define JPEG_INTERNALS
#include "jinclude.h"
#include "jpeglib.h"


/* Forward declarations */
LOCAL(void) transdecode_master_selection JPP((j_decompress_ptr cinfo));


/*
 * Read the coefficient arrays from a JPEG file.
 * jpeg_read_header must be completed before calling this.
 *
 * The entire image is read into a set of virtual coefficient-block arrays,
 * one per component.  The return value is a pointer to the array of
 * virtual-array descriptors.  These can be manipulated directly via the
 * JPEG memory manager, or handed off to jpeg_write_coefficients().
 * To release the memory occupied by the virtual arrays, call
 * jpeg_finish_decompress() when done with the data.
 *
 * An alternative usage is to simply obtain access to the coefficient arrays
 * during a buffered-image-mode decompression operation.  This is allowed
 * after any jpeg_finish_output() call.  The arrays can be accessed until
 * jpeg_finish_decompress() is called.  (Note that any call to the library
 * may reposition the arrays, so don't rely on access_virt_barray() results
 * to stay valid across library calls.)
 *
 * Returns NULL if suspended.  This case need be checked only if
 * a suspending data source is used.
 */

GLOBAL(jvirt_barray_ptr *)
jpeg_read_coefficients (j_decompress_ptr cinfo)
{
  if (cinfo->global_state == DSTATE_READY) {
    /* First call: initialize active modules */
    transdecode_master_selection(cinfo);
    cinfo->global_state = DSTATE_RDCOEFS;
  }
  if (cinfo->global_state == DSTATE_RDCOEFS) {
    /* Absorb whole file into the coef buffer */
    for (;;) {
      int retcode;
      /* Call progress monitor hook if present */
      if (cinfo->progress != NULL)
        (*cinfo->progress->progress_monitor) ((j_common_ptr) cinfo);
      /* Absorb some more input */
      retcode = (*cinfo->inputctl->consume_input) (cinfo);
      if (retcode == JPEG_SUSPENDED)
        return NULL;
      if (retcode == JPEG_REACHED_EOI)
        break;
      /* Advance progress counter if appropriate */
      if (cinfo->progress != NULL &&
          (retcode == JPEG_ROW_COMPLETED || retcode == JPEG_REACHED_SOS)) {
        if (++cinfo->progress->pass_counter >= cinfo->progress->pass_limit) {
          /* startup underestimated number of scans; ratchet up one scan */
          cinfo->progress->pass_limit += (long) cinfo->total_iMCU_rows;
        }
      }
    }
    /* Set state so that jpeg_finish_decompress does the right thing */
    cinfo->global_state = DSTATE_STOPPING;
  }
  /* At this point we should be in state DSTATE_STOPPING if being used
   * standalone, or in state DSTATE_BUFIMAGE if being invoked to get access
   * to the coefficients during a full buffered-image-mode decompression.
   */
  if ((cinfo->global_state == DSTATE_STOPPING ||
       cinfo->global_state == DSTATE_BUFIMAGE) && cinfo->buffered_image) {
    return cinfo->coef->coef_arrays;
  }
  /* Oops, improper usage */
  ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
  return NULL;                  /* keep compiler happy */
}


/*
 * Master selection of decompression modules for transcoding.
 * This substitutes for jdmaster.c's initialization of the full decompressor.
 */

LOCAL(void)
transdecode_master_selection (j_decompress_ptr cinfo)
{
  /* This is effectively a buffered-image operation. */
  cinfo->buffered_image = TRUE;

  /* Entropy decoding: either Huffman or arithmetic coding. */
  if (cinfo->arith_code)
    jinit_arith_decoder(cinfo);
  else {
    jinit_huff_decoder(cinfo);
  }

  /* Always get a full-image coefficient buffer. */
  jinit_d_coef_controller(cinfo, TRUE);

  /* We can now tell the memory manager to allocate virtual arrays. */
  (*cinfo->mem->realize_virt_arrays) ((j_common_ptr) cinfo);

  /* Initialize input side of decompressor to consume first scan. */
  (*cinfo->inputctl->start_input_pass) (cinfo);

  /* Initialize progress monitoring. */
  if (cinfo->progress != NULL) {
    int nscans;
    /* Estimate number of scans to set pass_limit. */
    if (cinfo->progressive_mode) {
      /* Arbitrarily estimate 2 interleaved DC scans + 3 AC scans/component. */
      nscans = 2 + 3 * cinfo->num_components;
    } else if (cinfo->inputctl->has_multiple_scans) {
      /* For a nonprogressive multiscan file, estimate 1 scan per component. */
      nscans = cinfo->num_components;
    } else {
      nscans = 1;
    }
    cinfo->progress->pass_counter = 0L;
    cinfo->progress->pass_limit = (long) cinfo->total_iMCU_rows * nscans;
    cinfo->progress->completed_passes = 0;
    cinfo->progress->total_passes = 1;
  }
}
1	/*
2	* jdtrans.c
3	*
4	* Copyright (C) 1995-1997, 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 library routines for transcoding decompression,
9	* that is, reading raw DCT coefficient arrays from an input JPEG file.
10	* The routines in jdapimin.c will also be needed by a transcoder.
11	*/
12
13	#define JPEG_INTERNALS
14	#include "jinclude.h"
15	#include "jpeglib.h"
16
17
18	/* Forward declarations */
19	LOCAL(void) transdecode_master_selection JPP((j_decompress_ptr cinfo));
20
21
22	/*
23	* Read the coefficient arrays from a JPEG file.
24	* jpeg_read_header must be completed before calling this.
25	*
26	* The entire image is read into a set of virtual coefficient-block arrays,
27	* one per component. The return value is a pointer to the array of
28	* virtual-array descriptors. These can be manipulated directly via the
29	* JPEG memory manager, or handed off to jpeg_write_coefficients().
30	* To release the memory occupied by the virtual arrays, call
31	* jpeg_finish_decompress() when done with the data.
32	*
33	* An alternative usage is to simply obtain access to the coefficient arrays
34	* during a buffered-image-mode decompression operation. This is allowed
35	* after any jpeg_finish_output() call. The arrays can be accessed until
36	* jpeg_finish_decompress() is called. (Note that any call to the library
37	* may reposition the arrays, so don't rely on access_virt_barray() results
38	* to stay valid across library calls.)
39	*
40	* Returns NULL if suspended. This case need be checked only if
41	* a suspending data source is used.
42	*/
43
44	GLOBAL(jvirt_barray_ptr *)
45	jpeg_read_coefficients (j_decompress_ptr cinfo)
46	{
47	if (cinfo->global_state == DSTATE_READY) {
48	/* First call: initialize active modules */
49	transdecode_master_selection(cinfo);
50	cinfo->global_state = DSTATE_RDCOEFS;
51	}
52	if (cinfo->global_state == DSTATE_RDCOEFS) {
53	/* Absorb whole file into the coef buffer */
54	for (;;) {
55	int retcode;
56	/* Call progress monitor hook if present */
57	if (cinfo->progress != NULL)
58	(*cinfo->progress->progress_monitor) ((j_common_ptr) cinfo);
59	/* Absorb some more input */
60	retcode = (*cinfo->inputctl->consume_input) (cinfo);
61	if (retcode == JPEG_SUSPENDED)
62	return NULL;
63	if (retcode == JPEG_REACHED_EOI)
64	break;
65	/* Advance progress counter if appropriate */
66	if (cinfo->progress != NULL &&
67	(retcode == JPEG_ROW_COMPLETED \|\| retcode == JPEG_REACHED_SOS)) {
68	if (++cinfo->progress->pass_counter >= cinfo->progress->pass_limit) {
69	/* startup underestimated number of scans; ratchet up one scan */
70	cinfo->progress->pass_limit += (long) cinfo->total_iMCU_rows;
71	}
72	}
73	}
74	/* Set state so that jpeg_finish_decompress does the right thing */
75	cinfo->global_state = DSTATE_STOPPING;
76	}
77	/* At this point we should be in state DSTATE_STOPPING if being used
78	* standalone, or in state DSTATE_BUFIMAGE if being invoked to get access
79	* to the coefficients during a full buffered-image-mode decompression.
80	*/
81	if ((cinfo->global_state == DSTATE_STOPPING \|\|
82	cinfo->global_state == DSTATE_BUFIMAGE) && cinfo->buffered_image) {
83	return cinfo->coef->coef_arrays;
84	}
85	/* Oops, improper usage */
86	ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
87	return NULL; /* keep compiler happy */
88	}
89
90
91	/*
92	* Master selection of decompression modules for transcoding.
93	* This substitutes for jdmaster.c's initialization of the full decompressor.
94	*/
95
96	LOCAL(void)
97	transdecode_master_selection (j_decompress_ptr cinfo)
98	{
99	/* This is effectively a buffered-image operation. */
100	cinfo->buffered_image = TRUE;
101
102	/* Entropy decoding: either Huffman or arithmetic coding. */
103	if (cinfo->arith_code)
104	jinit_arith_decoder(cinfo);
105	else {
106	jinit_huff_decoder(cinfo);
107	}
108
109	/* Always get a full-image coefficient buffer. */
110	jinit_d_coef_controller(cinfo, TRUE);
111
112	/* We can now tell the memory manager to allocate virtual arrays. */
113	(*cinfo->mem->realize_virt_arrays) ((j_common_ptr) cinfo);
114
115	/* Initialize input side of decompressor to consume first scan. */
116	(*cinfo->inputctl->start_input_pass) (cinfo);
117
118	/* Initialize progress monitoring. */
119	if (cinfo->progress != NULL) {
120	int nscans;
121	/* Estimate number of scans to set pass_limit. */
122	if (cinfo->progressive_mode) {
123	/* Arbitrarily estimate 2 interleaved DC scans + 3 AC scans/component. */
124	nscans = 2 + 3 * cinfo->num_components;
125	} else if (cinfo->inputctl->has_multiple_scans) {
126	/* For a nonprogressive multiscan file, estimate 1 scan per component. */
127	nscans = cinfo->num_components;
128	} else {
129	nscans = 1;
130	}
131	cinfo->progress->pass_counter = 0L;
132	cinfo->progress->pass_limit = (long) cinfo->total_iMCU_rows * nscans;
133	cinfo->progress->completed_passes = 0;
134	cinfo->progress->total_passes = 1;
135	}
136	}