1 /*
2 * jcapimin.c
3 *
4 * This file was part of the Independent JPEG Group's software:
5 * Copyright (C) 1994-1998, Thomas G. Lane.
6 * Modified 2003-2010 by Guido Vollbeding.
7 * libjpeg-turbo Modifications:
8 * Copyright (C) 2014, D. R. Commander.
9 * For conditions of distribution and use, see the accompanying README file.
10 *
11 * This file contains application interface code for the compression half
12 * of the JPEG library. These are the "minimum" API routines that may be
13 * needed in either the normal full-compression case or the transcoding-only
14 * case.
15 *
16 * Most of the routines intended to be called directly by an application
17 * are in this file or in jcapistd.c. But also see jcparam.c for
18 * parameter-setup helper routines, jcomapi.c for routines shared by
19 * compression and decompression, and jctrans.c for the transcoding case.
20 */
21
22 #define JPEG_INTERNALS
23 #include "jinclude.h"
24 #include "jpeglib.h"
25 #include "jmemsys.h"
26 #include "jcmaster.h"
27
28
29 /*
30 * Initialization of a JPEG compression object.
31 * The error manager must already be set up (in case memory manager fails).
32 */
33
34 GLOBAL(void)
jpeg_CreateCompress(j_compress_ptr cinfo,int version,size_t structsize)35 jpeg_CreateCompress(j_compress_ptr cinfo, int version, size_t structsize)
36 {
37 int i;
38
39 /* Guard against version mismatches between library and caller. */
40 cinfo->mem = NULL; /* so jpeg_destroy knows mem mgr not called */
41 if (version != JPEG_LIB_VERSION)
42 ERREXIT2(cinfo, JERR_BAD_LIB_VERSION, JPEG_LIB_VERSION, version);
43 if (structsize != sizeof(struct jpeg_compress_struct))
44 ERREXIT2(cinfo, JERR_BAD_STRUCT_SIZE,
45 (int)sizeof(struct jpeg_compress_struct), (int)structsize);
46
47 /* For debugging purposes, we zero the whole master structure.
48 * But the application has already set the err pointer, and may have set
49 * client_data, so we have to save and restore those fields.
50 * Note: if application hasn't set client_data, tools like Purify may
51 * complain here.
52 */
53 {
54 struct jpeg_error_mgr *err = cinfo->err;
55 void *client_data = cinfo->client_data; /* ignore Purify complaint here */
56 MEMZERO(cinfo, sizeof(struct jpeg_compress_struct));
57 cinfo->err = err;
58 cinfo->client_data = client_data;
59 }
60 cinfo->is_decompressor = FALSE;
61
62 /* Initialize a memory manager instance for this object */
63 jinit_memory_mgr((j_common_ptr)cinfo);
64
65 /* Zero out pointers to permanent structures. */
66 cinfo->progress = NULL;
67 cinfo->dest = NULL;
68
69 cinfo->comp_info = NULL;
70
71 for (i = 0; i < NUM_QUANT_TBLS; i++) {
72 cinfo->quant_tbl_ptrs[i] = NULL;
73 #if JPEG_LIB_VERSION >= 70
74 cinfo->q_scale_factor[i] = 100;
75 #endif
76 }
77
78 for (i = 0; i < NUM_HUFF_TBLS; i++) {
79 cinfo->dc_huff_tbl_ptrs[i] = NULL;
80 cinfo->ac_huff_tbl_ptrs[i] = NULL;
81 }
82
83 #if JPEG_LIB_VERSION >= 80
84 /* Must do it here for emit_dqt in case jpeg_write_tables is used */
85 cinfo->block_size = DCTSIZE;
86 cinfo->natural_order = jpeg_natural_order;
87 cinfo->lim_Se = DCTSIZE2 - 1;
88 #endif
89
90 cinfo->script_space = NULL;
91
92 cinfo->input_gamma = 1.0; /* in case application forgets */
93
94 /* OK, I'm ready */
95 cinfo->global_state = CSTATE_START;
96
97 /* The master struct is used to store extension parameters, so we allocate it
98 * here. It is later reallocated by jinit_c_master_control().
99 */
100 cinfo->master = (struct jpeg_comp_master *)
101 (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_PERMANENT,
102 sizeof(my_comp_master));
103 MEMZERO(cinfo->master, sizeof(my_comp_master));
104
105 cinfo->master->compress_profile = JCP_MAX_COMPRESSION;
106 }
107
108
109 /*
110 * Destruction of a JPEG compression object
111 */
112
113 GLOBAL(void)
jpeg_destroy_compress(j_compress_ptr cinfo)114 jpeg_destroy_compress(j_compress_ptr cinfo)
115 {
116 jpeg_destroy((j_common_ptr)cinfo); /* use common routine */
117 }
118
119
120 /*
121 * Abort processing of a JPEG compression operation,
122 * but don't destroy the object itself.
123 */
124
125 GLOBAL(void)
jpeg_abort_compress(j_compress_ptr cinfo)126 jpeg_abort_compress(j_compress_ptr cinfo)
127 {
128 jpeg_abort((j_common_ptr)cinfo); /* use common routine */
129 }
130
131
132 /*
133 * Forcibly suppress or un-suppress all quantization and Huffman tables.
134 * Marks all currently defined tables as already written (if suppress)
135 * or not written (if !suppress). This will control whether they get emitted
136 * by a subsequent jpeg_start_compress call.
137 *
138 * This routine is exported for use by applications that want to produce
139 * abbreviated JPEG datastreams. It logically belongs in jcparam.c, but
140 * since it is called by jpeg_start_compress, we put it here --- otherwise
141 * jcparam.o would be linked whether the application used it or not.
142 */
143
144 GLOBAL(void)
jpeg_suppress_tables(j_compress_ptr cinfo,boolean suppress)145 jpeg_suppress_tables(j_compress_ptr cinfo, boolean suppress)
146 {
147 int i;
148 JQUANT_TBL *qtbl;
149 JHUFF_TBL *htbl;
150
151 for (i = 0; i < NUM_QUANT_TBLS; i++) {
152 if ((qtbl = cinfo->quant_tbl_ptrs[i]) != NULL)
153 qtbl->sent_table = suppress;
154 }
155
156 for (i = 0; i < NUM_HUFF_TBLS; i++) {
157 if ((htbl = cinfo->dc_huff_tbl_ptrs[i]) != NULL)
158 htbl->sent_table = suppress;
159 if ((htbl = cinfo->ac_huff_tbl_ptrs[i]) != NULL)
160 htbl->sent_table = suppress;
161 }
162 }
163
164
165 /*
166 * Finish JPEG compression.
167 *
168 * If a multipass operating mode was selected, this may do a great deal of
169 * work including most of the actual output.
170 */
171
172 GLOBAL(void)
jpeg_finish_compress(j_compress_ptr cinfo)173 jpeg_finish_compress(j_compress_ptr cinfo)
174 {
175 JDIMENSION iMCU_row;
176
177 if (cinfo->global_state == CSTATE_SCANNING ||
178 cinfo->global_state == CSTATE_RAW_OK) {
179 /* Terminate first pass */
180 if (cinfo->next_scanline < cinfo->image_height)
181 ERREXIT(cinfo, JERR_TOO_LITTLE_DATA);
182 (*cinfo->master->finish_pass) (cinfo);
183 } else if (cinfo->global_state != CSTATE_WRCOEFS)
184 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
185 /* Perform any remaining passes */
186 while (!cinfo->master->is_last_pass) {
187 (*cinfo->master->prepare_for_pass) (cinfo);
188 for (iMCU_row = 0; iMCU_row < cinfo->total_iMCU_rows; iMCU_row++) {
189 if (cinfo->progress != NULL) {
190 cinfo->progress->pass_counter = (long)iMCU_row;
191 cinfo->progress->pass_limit = (long)cinfo->total_iMCU_rows;
192 (*cinfo->progress->progress_monitor) ((j_common_ptr)cinfo);
193 }
194 /* We bypass the main controller and invoke coef controller directly;
195 * all work is being done from the coefficient buffer.
196 */
197 if (!(*cinfo->coef->compress_data) (cinfo, (JSAMPIMAGE)NULL))
198 ERREXIT(cinfo, JERR_CANT_SUSPEND);
199 }
200 (*cinfo->master->finish_pass) (cinfo);
201 }
202 /* Write EOI, do final cleanup */
203 (*cinfo->marker->write_file_trailer) (cinfo);
204 (*cinfo->dest->term_destination) (cinfo);
205 /* We can use jpeg_abort to release memory and reset global_state */
206 jpeg_abort((j_common_ptr)cinfo);
207 }
208
209
210 /*
211 * Write a special marker.
212 * This is only recommended for writing COM or APPn markers.
213 * Must be called after jpeg_start_compress() and before
214 * first call to jpeg_write_scanlines() or jpeg_write_raw_data().
215 */
216
217 GLOBAL(void)
jpeg_write_marker(j_compress_ptr cinfo,int marker,const JOCTET * dataptr,unsigned int datalen)218 jpeg_write_marker(j_compress_ptr cinfo, int marker, const JOCTET *dataptr,
219 unsigned int datalen)
220 {
221 void (*write_marker_byte) (j_compress_ptr info, int val);
222
223 if (cinfo->next_scanline != 0 ||
224 (cinfo->global_state != CSTATE_SCANNING &&
225 cinfo->global_state != CSTATE_RAW_OK &&
226 cinfo->global_state != CSTATE_WRCOEFS))
227 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
228
229 (*cinfo->marker->write_marker_header) (cinfo, marker, datalen);
230 write_marker_byte = cinfo->marker->write_marker_byte; /* copy for speed */
231 while (datalen--) {
232 (*write_marker_byte) (cinfo, *dataptr);
233 dataptr++;
234 }
235 }
236
237 /* Same, but piecemeal. */
238
239 GLOBAL(void)
jpeg_write_m_header(j_compress_ptr cinfo,int marker,unsigned int datalen)240 jpeg_write_m_header(j_compress_ptr cinfo, int marker, unsigned int datalen)
241 {
242 if (cinfo->next_scanline != 0 ||
243 (cinfo->global_state != CSTATE_SCANNING &&
244 cinfo->global_state != CSTATE_RAW_OK &&
245 cinfo->global_state != CSTATE_WRCOEFS))
246 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
247
248 (*cinfo->marker->write_marker_header) (cinfo, marker, datalen);
249 }
250
251 GLOBAL(void)
jpeg_write_m_byte(j_compress_ptr cinfo,int val)252 jpeg_write_m_byte(j_compress_ptr cinfo, int val)
253 {
254 (*cinfo->marker->write_marker_byte) (cinfo, val);
255 }
256
257
258 /*
259 * Alternate compression function: just write an abbreviated table file.
260 * Before calling this, all parameters and a data destination must be set up.
261 *
262 * To produce a pair of files containing abbreviated tables and abbreviated
263 * image data, one would proceed as follows:
264 *
265 * initialize JPEG object
266 * set JPEG parameters
267 * set destination to table file
268 * jpeg_write_tables(cinfo);
269 * set destination to image file
270 * jpeg_start_compress(cinfo, FALSE);
271 * write data...
272 * jpeg_finish_compress(cinfo);
273 *
274 * jpeg_write_tables has the side effect of marking all tables written
275 * (same as jpeg_suppress_tables(..., TRUE)). Thus a subsequent start_compress
276 * will not re-emit the tables unless it is passed write_all_tables=TRUE.
277 */
278
279 GLOBAL(void)
jpeg_write_tables(j_compress_ptr cinfo)280 jpeg_write_tables(j_compress_ptr cinfo)
281 {
282 if (cinfo->global_state != CSTATE_START)
283 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
284
285 /* (Re)initialize error mgr and destination modules */
286 (*cinfo->err->reset_error_mgr) ((j_common_ptr)cinfo);
287 (*cinfo->dest->init_destination) (cinfo);
288 /* Initialize the marker writer ... bit of a crock to do it here. */
289 jinit_marker_writer(cinfo);
290 /* Write them tables! */
291 (*cinfo->marker->write_tables_only) (cinfo);
292 /* And clean up. */
293 (*cinfo->dest->term_destination) (cinfo);
294 /*
295 * In library releases up through v6a, we called jpeg_abort() here to free
296 * any working memory allocated by the destination manager and marker
297 * writer. Some applications had a problem with that: they allocated space
298 * of their own from the library memory manager, and didn't want it to go
299 * away during write_tables. So now we do nothing. This will cause a
300 * memory leak if an app calls write_tables repeatedly without doing a full
301 * compression cycle or otherwise resetting the JPEG object. However, that
302 * seems less bad than unexpectedly freeing memory in the normal case.
303 * An app that prefers the old behavior can call jpeg_abort for itself after
304 * each call to jpeg_write_tables().
305 */
306 }
307