Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 1 | /* |
| 2 | * jcmainct.c |
| 3 | * |
DRC | 5033f3e | 2014-05-18 18:33:44 +0000 | [diff] [blame] | 4 | * This file was part of the Independent JPEG Group's software: |
Thomas G. Lane | 489583f | 1996-02-07 00:00:00 +0000 | [diff] [blame] | 5 | * Copyright (C) 1994-1996, Thomas G. Lane. |
DRC | 5033f3e | 2014-05-18 18:33:44 +0000 | [diff] [blame] | 6 | * It was modified by The libjpeg-turbo Project to include only code relevant |
| 7 | * to libjpeg-turbo. |
Alex Naidis | 6eb7d37 | 2016-10-16 23:10:08 +0200 | [diff] [blame] | 8 | * For conditions of distribution and use, see the accompanying README.ijg |
| 9 | * file. |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 10 | * |
| 11 | * This file contains the main buffer controller for compression. |
| 12 | * The main buffer lies between the pre-processor and the JPEG |
| 13 | * compressor proper; it holds downsampled data in the JPEG colorspace. |
| 14 | */ |
| 15 | |
| 16 | #define JPEG_INTERNALS |
| 17 | #include "jinclude.h" |
| 18 | #include "jpeglib.h" |
| 19 | |
| 20 | |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 21 | /* Private buffer controller object */ |
| 22 | |
| 23 | typedef struct { |
| 24 | struct jpeg_c_main_controller pub; /* public fields */ |
| 25 | |
DRC | e5eaf37 | 2014-05-09 18:00:32 +0000 | [diff] [blame] | 26 | JDIMENSION cur_iMCU_row; /* number of current iMCU row */ |
| 27 | JDIMENSION rowgroup_ctr; /* counts row groups received in iMCU row */ |
| 28 | boolean suspended; /* remember if we suspended output */ |
| 29 | J_BUF_MODE pass_mode; /* current operating mode */ |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 30 | |
| 31 | /* If using just a strip buffer, this points to the entire set of buffers |
| 32 | * (we allocate one for each component). In the full-image case, this |
| 33 | * points to the currently accessible strips of the virtual arrays. |
| 34 | */ |
| 35 | JSAMPARRAY buffer[MAX_COMPONENTS]; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 36 | } my_main_controller; |
| 37 | |
Alex Naidis | 6eb7d37 | 2016-10-16 23:10:08 +0200 | [diff] [blame] | 38 | typedef my_main_controller *my_main_ptr; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 39 | |
| 40 | |
| 41 | /* Forward declarations */ |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 42 | METHODDEF(void) process_data_simple_main(j_compress_ptr cinfo, |
| 43 | JSAMPARRAY input_buf, |
| 44 | JDIMENSION *in_row_ctr, |
| 45 | JDIMENSION in_rows_avail); |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 46 | |
| 47 | |
| 48 | /* |
| 49 | * Initialize for a processing pass. |
| 50 | */ |
| 51 | |
Thomas G. Lane | 489583f | 1996-02-07 00:00:00 +0000 | [diff] [blame] | 52 | METHODDEF(void) |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 53 | start_pass_main(j_compress_ptr cinfo, J_BUF_MODE pass_mode) |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 54 | { |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 55 | my_main_ptr main_ptr = (my_main_ptr)cinfo->main; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 56 | |
| 57 | /* Do nothing in raw-data mode. */ |
| 58 | if (cinfo->raw_data_in) |
| 59 | return; |
| 60 | |
DRC | a29d6b4 | 2014-12-01 20:13:12 +0000 | [diff] [blame] | 61 | if (pass_mode != JBUF_PASS_THRU) |
| 62 | ERREXIT(cinfo, JERR_BAD_BUFFER_MODE); |
| 63 | |
DRC | e5eaf37 | 2014-05-09 18:00:32 +0000 | [diff] [blame] | 64 | main_ptr->cur_iMCU_row = 0; /* initialize counters */ |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 65 | main_ptr->rowgroup_ctr = 0; |
| 66 | main_ptr->suspended = FALSE; |
DRC | e5eaf37 | 2014-05-09 18:00:32 +0000 | [diff] [blame] | 67 | main_ptr->pass_mode = pass_mode; /* save mode for use by process_data */ |
DRC | a29d6b4 | 2014-12-01 20:13:12 +0000 | [diff] [blame] | 68 | main_ptr->pub.process_data = process_data_simple_main; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 69 | } |
| 70 | |
| 71 | |
| 72 | /* |
| 73 | * Process some data. |
| 74 | * This routine handles the simple pass-through mode, |
| 75 | * where we have only a strip buffer. |
| 76 | */ |
| 77 | |
Thomas G. Lane | 489583f | 1996-02-07 00:00:00 +0000 | [diff] [blame] | 78 | METHODDEF(void) |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 79 | process_data_simple_main(j_compress_ptr cinfo, JSAMPARRAY input_buf, |
| 80 | JDIMENSION *in_row_ctr, JDIMENSION in_rows_avail) |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 81 | { |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 82 | my_main_ptr main_ptr = (my_main_ptr)cinfo->main; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 83 | |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 84 | while (main_ptr->cur_iMCU_row < cinfo->total_iMCU_rows) { |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 85 | /* Read input data if we haven't filled the main buffer yet */ |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 86 | if (main_ptr->rowgroup_ctr < DCTSIZE) |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 87 | (*cinfo->prep->pre_process_data) (cinfo, input_buf, in_row_ctr, |
| 88 | in_rows_avail, main_ptr->buffer, |
| 89 | &main_ptr->rowgroup_ctr, |
| 90 | (JDIMENSION)DCTSIZE); |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 91 | |
| 92 | /* If we don't have a full iMCU row buffered, return to application for |
| 93 | * more data. Note that preprocessor will always pad to fill the iMCU row |
| 94 | * at the bottom of the image. |
| 95 | */ |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 96 | if (main_ptr->rowgroup_ctr != DCTSIZE) |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 97 | return; |
| 98 | |
| 99 | /* Send the completed row to the compressor */ |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 100 | if (!(*cinfo->coef->compress_data) (cinfo, main_ptr->buffer)) { |
Thomas G. Lane | a8b67c4 | 1995-03-15 00:00:00 +0000 | [diff] [blame] | 101 | /* If compressor did not consume the whole row, then we must need to |
| 102 | * suspend processing and return to the application. In this situation |
| 103 | * we pretend we didn't yet consume the last input row; otherwise, if |
| 104 | * it happened to be the last row of the image, the application would |
| 105 | * think we were done. |
| 106 | */ |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 107 | if (!main_ptr->suspended) { |
DRC | e5eaf37 | 2014-05-09 18:00:32 +0000 | [diff] [blame] | 108 | (*in_row_ctr)--; |
| 109 | main_ptr->suspended = TRUE; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 110 | } |
| 111 | return; |
| 112 | } |
| 113 | /* We did finish the row. Undo our little suspension hack if a previous |
| 114 | * call suspended; then mark the main buffer empty. |
| 115 | */ |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 116 | if (main_ptr->suspended) { |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 117 | (*in_row_ctr)++; |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 118 | main_ptr->suspended = FALSE; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 119 | } |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 120 | main_ptr->rowgroup_ctr = 0; |
| 121 | main_ptr->cur_iMCU_row++; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 122 | } |
| 123 | } |
| 124 | |
| 125 | |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 126 | /* |
| 127 | * Initialize main buffer controller. |
| 128 | */ |
| 129 | |
Thomas G. Lane | 489583f | 1996-02-07 00:00:00 +0000 | [diff] [blame] | 130 | GLOBAL(void) |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 131 | jinit_c_main_controller(j_compress_ptr cinfo, boolean need_full_buffer) |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 132 | { |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 133 | my_main_ptr main_ptr; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 134 | int ci; |
| 135 | jpeg_component_info *compptr; |
| 136 | |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 137 | main_ptr = (my_main_ptr) |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 138 | (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE, |
DRC | 5de454b | 2014-05-18 19:04:03 +0000 | [diff] [blame] | 139 | sizeof(my_main_controller)); |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 140 | cinfo->main = (struct jpeg_c_main_controller *)main_ptr; |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 141 | main_ptr->pub.start_pass = start_pass_main; |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 142 | |
| 143 | /* We don't need to create a buffer in raw-data mode. */ |
| 144 | if (cinfo->raw_data_in) |
| 145 | return; |
| 146 | |
| 147 | /* Create the buffer. It holds downsampled data, so each component |
| 148 | * may be of a different size. |
| 149 | */ |
| 150 | if (need_full_buffer) { |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 151 | ERREXIT(cinfo, JERR_BAD_BUFFER_MODE); |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 152 | } else { |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 153 | /* Allocate a strip buffer for each component */ |
| 154 | for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components; |
DRC | e5eaf37 | 2014-05-09 18:00:32 +0000 | [diff] [blame] | 155 | ci++, compptr++) { |
DRC | 7ed7b57 | 2012-02-07 22:51:49 +0000 | [diff] [blame] | 156 | main_ptr->buffer[ci] = (*cinfo->mem->alloc_sarray) |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 157 | ((j_common_ptr)cinfo, JPOOL_IMAGE, |
DRC | e5eaf37 | 2014-05-09 18:00:32 +0000 | [diff] [blame] | 158 | compptr->width_in_blocks * DCTSIZE, |
Leon Scroggins III | 3993b37 | 2018-07-16 10:43:45 -0400 | [diff] [blame] | 159 | (JDIMENSION)(compptr->v_samp_factor * DCTSIZE)); |
Thomas G. Lane | 36a4ccc | 1994-09-24 00:00:00 +0000 | [diff] [blame] | 160 | } |
| 161 | } |
| 162 | } |