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