Coverage Report

Created: 2024-05-18 12:36

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