OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * jdphuff.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 Huffman entropy decoding routines for progressive JPEG. |
| 9 * |
| 10 * Much of the complexity here has to do with supporting input suspension. |
| 11 * If the data source module demands suspension, we want to be able to back |
| 12 * up to the start of the current MCU. To do this, we copy state variables |
| 13 * into local working storage, and update them back to the permanent |
| 14 * storage only upon successful completion of an MCU. |
| 15 */ |
| 16 |
| 17 #define JPEG_INTERNALS |
| 18 #include "jinclude.h" |
| 19 #include "jpeglib.h" |
| 20 #include "jdhuff.h" /* Declarations shared with jdhuff.c */ |
| 21 |
| 22 |
| 23 #ifdef D_PROGRESSIVE_SUPPORTED |
| 24 |
| 25 /* |
| 26 * Expanded entropy decoder object for progressive Huffman decoding. |
| 27 * |
| 28 * The savable_state subrecord contains fields that change within an MCU, |
| 29 * but must not be updated permanently until we complete the MCU. |
| 30 */ |
| 31 |
| 32 typedef struct { |
| 33 unsigned int EOBRUN; /* remaining EOBs in EOBRUN */ |
| 34 int last_dc_val[MAX_COMPS_IN_SCAN]; /* last DC coef for each component */ |
| 35 } savable_state; |
| 36 |
| 37 /* This macro is to work around compilers with missing or broken |
| 38 * structure assignment. You'll need to fix this code if you have |
| 39 * such a compiler and you change MAX_COMPS_IN_SCAN. |
| 40 */ |
| 41 |
| 42 #ifndef NO_STRUCT_ASSIGN |
| 43 #define ASSIGN_STATE(dest,src) ((dest) = (src)) |
| 44 #else |
| 45 #if MAX_COMPS_IN_SCAN == 4 |
| 46 #define ASSIGN_STATE(dest,src) \ |
| 47 ((dest).EOBRUN = (src).EOBRUN, \ |
| 48 (dest).last_dc_val[0] = (src).last_dc_val[0], \ |
| 49 (dest).last_dc_val[1] = (src).last_dc_val[1], \ |
| 50 (dest).last_dc_val[2] = (src).last_dc_val[2], \ |
| 51 (dest).last_dc_val[3] = (src).last_dc_val[3]) |
| 52 #endif |
| 53 #endif |
| 54 |
| 55 |
| 56 typedef struct { |
| 57 struct jpeg_entropy_decoder pub; /* public fields */ |
| 58 |
| 59 /* These fields are loaded into local variables at start of each MCU. |
| 60 * In case of suspension, we exit WITHOUT updating them. |
| 61 */ |
| 62 bitread_perm_state bitstate; /* Bit buffer at start of MCU */ |
| 63 savable_state saved; /* Other state at start of MCU */ |
| 64 |
| 65 /* These fields are NOT loaded into local working state. */ |
| 66 unsigned int restarts_to_go; /* MCUs left in this restart interval */ |
| 67 |
| 68 /* Pointers to derived tables (these workspaces have image lifespan) */ |
| 69 d_derived_tbl * derived_tbls[NUM_HUFF_TBLS]; |
| 70 |
| 71 d_derived_tbl * ac_derived_tbl; /* active table during an AC scan */ |
| 72 } phuff_entropy_decoder; |
| 73 |
| 74 typedef phuff_entropy_decoder * phuff_entropy_ptr; |
| 75 |
| 76 /* Forward declarations */ |
| 77 METHODDEF(boolean) decode_mcu_DC_first JPP((j_decompress_ptr cinfo, |
| 78 JBLOCKROW *MCU_data)); |
| 79 METHODDEF(boolean) decode_mcu_AC_first JPP((j_decompress_ptr cinfo, |
| 80 JBLOCKROW *MCU_data)); |
| 81 METHODDEF(boolean) decode_mcu_DC_refine JPP((j_decompress_ptr cinfo, |
| 82 JBLOCKROW *MCU_data)); |
| 83 METHODDEF(boolean) decode_mcu_AC_refine JPP((j_decompress_ptr cinfo, |
| 84 JBLOCKROW *MCU_data)); |
| 85 |
| 86 |
| 87 /* |
| 88 * Initialize for a Huffman-compressed scan. |
| 89 */ |
| 90 |
| 91 METHODDEF(void) |
| 92 start_pass_phuff_decoder (j_decompress_ptr cinfo) |
| 93 { |
| 94 phuff_entropy_ptr entropy = (phuff_entropy_ptr) cinfo->entropy; |
| 95 boolean is_DC_band, bad; |
| 96 int ci, coefi, tbl; |
| 97 int *coef_bit_ptr; |
| 98 jpeg_component_info * compptr; |
| 99 |
| 100 is_DC_band = (cinfo->Ss == 0); |
| 101 |
| 102 /* Validate scan parameters */ |
| 103 bad = FALSE; |
| 104 if (is_DC_band) { |
| 105 if (cinfo->Se != 0) |
| 106 bad = TRUE; |
| 107 } else { |
| 108 /* need not check Ss/Se < 0 since they came from unsigned bytes */ |
| 109 if (cinfo->Ss > cinfo->Se || cinfo->Se >= DCTSIZE2) |
| 110 bad = TRUE; |
| 111 /* AC scans may have only one component */ |
| 112 if (cinfo->comps_in_scan != 1) |
| 113 bad = TRUE; |
| 114 } |
| 115 if (cinfo->Ah != 0) { |
| 116 /* Successive approximation refinement scan: must have Al = Ah-1. */ |
| 117 if (cinfo->Al != cinfo->Ah-1) |
| 118 bad = TRUE; |
| 119 } |
| 120 if (cinfo->Al > 13) /* need not check for < 0 */ |
| 121 bad = TRUE; |
| 122 /* Arguably the maximum Al value should be less than 13 for 8-bit precision, |
| 123 * but the spec doesn't say so, and we try to be liberal about what we |
| 124 * accept. Note: large Al values could result in out-of-range DC |
| 125 * coefficients during early scans, leading to bizarre displays due to |
| 126 * overflows in the IDCT math. But we won't crash. |
| 127 */ |
| 128 if (bad) |
| 129 ERREXIT4(cinfo, JERR_BAD_PROGRESSION, |
| 130 cinfo->Ss, cinfo->Se, cinfo->Ah, cinfo->Al); |
| 131 /* Update progression status, and verify that scan order is legal. |
| 132 * Note that inter-scan inconsistencies are treated as warnings |
| 133 * not fatal errors ... not clear if this is right way to behave. |
| 134 */ |
| 135 for (ci = 0; ci < cinfo->comps_in_scan; ci++) { |
| 136 int cindex = cinfo->cur_comp_info[ci]->component_index; |
| 137 coef_bit_ptr = & cinfo->coef_bits[cindex][0]; |
| 138 if (!is_DC_band && coef_bit_ptr[0] < 0) /* AC without prior DC scan */ |
| 139 WARNMS2(cinfo, JWRN_BOGUS_PROGRESSION, cindex, 0); |
| 140 for (coefi = cinfo->Ss; coefi <= cinfo->Se; coefi++) { |
| 141 int expected = (coef_bit_ptr[coefi] < 0) ? 0 : coef_bit_ptr[coefi]; |
| 142 if (cinfo->Ah != expected) |
| 143 WARNMS2(cinfo, JWRN_BOGUS_PROGRESSION, cindex, coefi); |
| 144 coef_bit_ptr[coefi] = cinfo->Al; |
| 145 } |
| 146 } |
| 147 |
| 148 /* Select MCU decoding routine */ |
| 149 if (cinfo->Ah == 0) { |
| 150 if (is_DC_band) |
| 151 entropy->pub.decode_mcu = decode_mcu_DC_first; |
| 152 else |
| 153 entropy->pub.decode_mcu = decode_mcu_AC_first; |
| 154 } else { |
| 155 if (is_DC_band) |
| 156 entropy->pub.decode_mcu = decode_mcu_DC_refine; |
| 157 else |
| 158 entropy->pub.decode_mcu = decode_mcu_AC_refine; |
| 159 } |
| 160 |
| 161 for (ci = 0; ci < cinfo->comps_in_scan; ci++) { |
| 162 compptr = cinfo->cur_comp_info[ci]; |
| 163 /* Make sure requested tables are present, and compute derived tables. |
| 164 * We may build same derived table more than once, but it's not expensive. |
| 165 */ |
| 166 if (is_DC_band) { |
| 167 if (cinfo->Ah == 0) { /* DC refinement needs no table */ |
| 168 tbl = compptr->dc_tbl_no; |
| 169 jpeg_make_d_derived_tbl(cinfo, TRUE, tbl, |
| 170 & entropy->derived_tbls[tbl]); |
| 171 } |
| 172 } else { |
| 173 tbl = compptr->ac_tbl_no; |
| 174 jpeg_make_d_derived_tbl(cinfo, FALSE, tbl, |
| 175 & entropy->derived_tbls[tbl]); |
| 176 /* remember the single active table */ |
| 177 entropy->ac_derived_tbl = entropy->derived_tbls[tbl]; |
| 178 } |
| 179 /* Initialize DC predictions to 0 */ |
| 180 entropy->saved.last_dc_val[ci] = 0; |
| 181 } |
| 182 |
| 183 /* Initialize bitread state variables */ |
| 184 entropy->bitstate.bits_left = 0; |
| 185 entropy->bitstate.get_buffer = 0; /* unnecessary, but keeps Purify quiet */ |
| 186 entropy->pub.insufficient_data = FALSE; |
| 187 |
| 188 /* Initialize private state variables */ |
| 189 entropy->saved.EOBRUN = 0; |
| 190 |
| 191 /* Initialize restart counter */ |
| 192 entropy->restarts_to_go = cinfo->restart_interval; |
| 193 } |
| 194 |
| 195 |
| 196 /* |
| 197 * Figure F.12: extend sign bit. |
| 198 * On some machines, a shift and add will be faster than a table lookup. |
| 199 */ |
| 200 |
| 201 #ifdef AVOID_TABLES |
| 202 |
| 203 #define HUFF_EXTEND(x,s) ((x) < (1<<((s)-1)) ? (x) + (((-1)<<(s)) + 1) : (x)) |
| 204 |
| 205 #else |
| 206 |
| 207 #define HUFF_EXTEND(x,s) ((x) < extend_test[s] ? (x) + extend_offset[s] : (x)) |
| 208 |
| 209 static const int extend_test[16] = /* entry n is 2**(n-1) */ |
| 210 { 0, 0x0001, 0x0002, 0x0004, 0x0008, 0x0010, 0x0020, 0x0040, 0x0080, |
| 211 0x0100, 0x0200, 0x0400, 0x0800, 0x1000, 0x2000, 0x4000 }; |
| 212 |
| 213 static const int extend_offset[16] = /* entry n is (-1 << n) + 1 */ |
| 214 { 0, ((-1)<<1) + 1, ((-1)<<2) + 1, ((-1)<<3) + 1, ((-1)<<4) + 1, |
| 215 ((-1)<<5) + 1, ((-1)<<6) + 1, ((-1)<<7) + 1, ((-1)<<8) + 1, |
| 216 ((-1)<<9) + 1, ((-1)<<10) + 1, ((-1)<<11) + 1, ((-1)<<12) + 1, |
| 217 ((-1)<<13) + 1, ((-1)<<14) + 1, ((-1)<<15) + 1 }; |
| 218 |
| 219 #endif /* AVOID_TABLES */ |
| 220 |
| 221 |
| 222 /* |
| 223 * Check for a restart marker & resynchronize decoder. |
| 224 * Returns FALSE if must suspend. |
| 225 */ |
| 226 |
| 227 LOCAL(boolean) |
| 228 process_restart (j_decompress_ptr cinfo) |
| 229 { |
| 230 phuff_entropy_ptr entropy = (phuff_entropy_ptr) cinfo->entropy; |
| 231 int ci; |
| 232 |
| 233 /* Throw away any unused bits remaining in bit buffer; */ |
| 234 /* include any full bytes in next_marker's count of discarded bytes */ |
| 235 cinfo->marker->discarded_bytes += entropy->bitstate.bits_left / 8; |
| 236 entropy->bitstate.bits_left = 0; |
| 237 |
| 238 /* Advance past the RSTn marker */ |
| 239 if (! (*cinfo->marker->read_restart_marker) (cinfo)) |
| 240 return FALSE; |
| 241 |
| 242 /* Re-initialize DC predictions to 0 */ |
| 243 for (ci = 0; ci < cinfo->comps_in_scan; ci++) |
| 244 entropy->saved.last_dc_val[ci] = 0; |
| 245 /* Re-init EOB run count, too */ |
| 246 entropy->saved.EOBRUN = 0; |
| 247 |
| 248 /* Reset restart counter */ |
| 249 entropy->restarts_to_go = cinfo->restart_interval; |
| 250 |
| 251 /* Reset out-of-data flag, unless read_restart_marker left us smack up |
| 252 * against a marker. In that case we will end up treating the next data |
| 253 * segment as empty, and we can avoid producing bogus output pixels by |
| 254 * leaving the flag set. |
| 255 */ |
| 256 if (cinfo->unread_marker == 0) |
| 257 entropy->pub.insufficient_data = FALSE; |
| 258 |
| 259 return TRUE; |
| 260 } |
| 261 |
| 262 |
| 263 /* |
| 264 * Huffman MCU decoding. |
| 265 * Each of these routines decodes and returns one MCU's worth of |
| 266 * Huffman-compressed coefficients. |
| 267 * The coefficients are reordered from zigzag order into natural array order, |
| 268 * but are not dequantized. |
| 269 * |
| 270 * The i'th block of the MCU is stored into the block pointed to by |
| 271 * MCU_data[i]. WE ASSUME THIS AREA IS INITIALLY ZEROED BY THE CALLER. |
| 272 * |
| 273 * We return FALSE if data source requested suspension. In that case no |
| 274 * changes have been made to permanent state. (Exception: some output |
| 275 * coefficients may already have been assigned. This is harmless for |
| 276 * spectral selection, since we'll just re-assign them on the next call. |
| 277 * Successive approximation AC refinement has to be more careful, however.) |
| 278 */ |
| 279 |
| 280 /* |
| 281 * MCU decoding for DC initial scan (either spectral selection, |
| 282 * or first pass of successive approximation). |
| 283 */ |
| 284 |
| 285 METHODDEF(boolean) |
| 286 decode_mcu_DC_first (j_decompress_ptr cinfo, JBLOCKROW *MCU_data) |
| 287 { |
| 288 phuff_entropy_ptr entropy = (phuff_entropy_ptr) cinfo->entropy; |
| 289 int Al = cinfo->Al; |
| 290 register int s, r; |
| 291 int blkn, ci; |
| 292 JBLOCKROW block; |
| 293 BITREAD_STATE_VARS; |
| 294 savable_state state; |
| 295 d_derived_tbl * tbl; |
| 296 jpeg_component_info * compptr; |
| 297 |
| 298 /* Process restart marker if needed; may have to suspend */ |
| 299 if (cinfo->restart_interval) { |
| 300 if (entropy->restarts_to_go == 0) |
| 301 if (! process_restart(cinfo)) |
| 302 return FALSE; |
| 303 } |
| 304 |
| 305 /* If we've run out of data, just leave the MCU set to zeroes. |
| 306 * This way, we return uniform gray for the remainder of the segment. |
| 307 */ |
| 308 if (! entropy->pub.insufficient_data) { |
| 309 |
| 310 /* Load up working state */ |
| 311 BITREAD_LOAD_STATE(cinfo,entropy->bitstate); |
| 312 ASSIGN_STATE(state, entropy->saved); |
| 313 |
| 314 /* Outer loop handles each block in the MCU */ |
| 315 |
| 316 for (blkn = 0; blkn < cinfo->blocks_in_MCU; blkn++) { |
| 317 block = MCU_data[blkn]; |
| 318 ci = cinfo->MCU_membership[blkn]; |
| 319 compptr = cinfo->cur_comp_info[ci]; |
| 320 tbl = entropy->derived_tbls[compptr->dc_tbl_no]; |
| 321 |
| 322 /* Decode a single block's worth of coefficients */ |
| 323 |
| 324 /* Section F.2.2.1: decode the DC coefficient difference */ |
| 325 HUFF_DECODE(s, br_state, tbl, return FALSE, label1); |
| 326 if (s) { |
| 327 CHECK_BIT_BUFFER(br_state, s, return FALSE); |
| 328 r = GET_BITS(s); |
| 329 s = HUFF_EXTEND(r, s); |
| 330 } |
| 331 |
| 332 /* Convert DC difference to actual value, update last_dc_val */ |
| 333 s += state.last_dc_val[ci]; |
| 334 state.last_dc_val[ci] = s; |
| 335 /* Scale and output the coefficient (assumes jpeg_natural_order[0]=0) */ |
| 336 (*block)[0] = (JCOEF) (s << Al); |
| 337 } |
| 338 |
| 339 /* Completed MCU, so update state */ |
| 340 BITREAD_SAVE_STATE(cinfo,entropy->bitstate); |
| 341 ASSIGN_STATE(entropy->saved, state); |
| 342 } |
| 343 |
| 344 /* Account for restart interval (no-op if not using restarts) */ |
| 345 entropy->restarts_to_go--; |
| 346 |
| 347 return TRUE; |
| 348 } |
| 349 |
| 350 |
| 351 /* |
| 352 * MCU decoding for AC initial scan (either spectral selection, |
| 353 * or first pass of successive approximation). |
| 354 */ |
| 355 |
| 356 METHODDEF(boolean) |
| 357 decode_mcu_AC_first (j_decompress_ptr cinfo, JBLOCKROW *MCU_data) |
| 358 { |
| 359 phuff_entropy_ptr entropy = (phuff_entropy_ptr) cinfo->entropy; |
| 360 int Se = cinfo->Se; |
| 361 int Al = cinfo->Al; |
| 362 register int s, k, r; |
| 363 unsigned int EOBRUN; |
| 364 JBLOCKROW block; |
| 365 BITREAD_STATE_VARS; |
| 366 d_derived_tbl * tbl; |
| 367 |
| 368 /* Process restart marker if needed; may have to suspend */ |
| 369 if (cinfo->restart_interval) { |
| 370 if (entropy->restarts_to_go == 0) |
| 371 if (! process_restart(cinfo)) |
| 372 return FALSE; |
| 373 } |
| 374 |
| 375 /* If we've run out of data, just leave the MCU set to zeroes. |
| 376 * This way, we return uniform gray for the remainder of the segment. |
| 377 */ |
| 378 if (! entropy->pub.insufficient_data) { |
| 379 |
| 380 /* Load up working state. |
| 381 * We can avoid loading/saving bitread state if in an EOB run. |
| 382 */ |
| 383 EOBRUN = entropy->saved.EOBRUN; /* only part of saved state we need */ |
| 384 |
| 385 /* There is always only one block per MCU */ |
| 386 |
| 387 if (EOBRUN > 0) /* if it's a band of zeroes... */ |
| 388 EOBRUN--; /* ...process it now (we do nothing) */ |
| 389 else { |
| 390 BITREAD_LOAD_STATE(cinfo,entropy->bitstate); |
| 391 block = MCU_data[0]; |
| 392 tbl = entropy->ac_derived_tbl; |
| 393 |
| 394 for (k = cinfo->Ss; k <= Se; k++) { |
| 395 HUFF_DECODE(s, br_state, tbl, return FALSE, label2); |
| 396 r = s >> 4; |
| 397 s &= 15; |
| 398 if (s) { |
| 399 k += r; |
| 400 CHECK_BIT_BUFFER(br_state, s, return FALSE); |
| 401 r = GET_BITS(s); |
| 402 s = HUFF_EXTEND(r, s); |
| 403 /* Scale and output coefficient in natural (dezigzagged) order */ |
| 404 (*block)[jpeg_natural_order[k]] = (JCOEF) (s << Al); |
| 405 } else { |
| 406 if (r == 15) { /* ZRL */ |
| 407 k += 15; /* skip 15 zeroes in band */ |
| 408 } else { /* EOBr, run length is 2^r + appended bits */ |
| 409 EOBRUN = 1 << r; |
| 410 if (r) { /* EOBr, r > 0 */ |
| 411 CHECK_BIT_BUFFER(br_state, r, return FALSE); |
| 412 r = GET_BITS(r); |
| 413 EOBRUN += r; |
| 414 } |
| 415 EOBRUN--; /* this band is processed at this moment */ |
| 416 break; /* force end-of-band */ |
| 417 } |
| 418 } |
| 419 } |
| 420 |
| 421 BITREAD_SAVE_STATE(cinfo,entropy->bitstate); |
| 422 } |
| 423 |
| 424 /* Completed MCU, so update state */ |
| 425 entropy->saved.EOBRUN = EOBRUN; /* only part of saved state we need */ |
| 426 } |
| 427 |
| 428 /* Account for restart interval (no-op if not using restarts) */ |
| 429 entropy->restarts_to_go--; |
| 430 |
| 431 return TRUE; |
| 432 } |
| 433 |
| 434 |
| 435 /* |
| 436 * MCU decoding for DC successive approximation refinement scan. |
| 437 * Note: we assume such scans can be multi-component, although the spec |
| 438 * is not very clear on the point. |
| 439 */ |
| 440 |
| 441 METHODDEF(boolean) |
| 442 decode_mcu_DC_refine (j_decompress_ptr cinfo, JBLOCKROW *MCU_data) |
| 443 { |
| 444 phuff_entropy_ptr entropy = (phuff_entropy_ptr) cinfo->entropy; |
| 445 int p1 = 1 << cinfo->Al; /* 1 in the bit position being coded */ |
| 446 int blkn; |
| 447 JBLOCKROW block; |
| 448 BITREAD_STATE_VARS; |
| 449 |
| 450 /* Process restart marker if needed; may have to suspend */ |
| 451 if (cinfo->restart_interval) { |
| 452 if (entropy->restarts_to_go == 0) |
| 453 if (! process_restart(cinfo)) |
| 454 return FALSE; |
| 455 } |
| 456 |
| 457 /* Not worth the cycles to check insufficient_data here, |
| 458 * since we will not change the data anyway if we read zeroes. |
| 459 */ |
| 460 |
| 461 /* Load up working state */ |
| 462 BITREAD_LOAD_STATE(cinfo,entropy->bitstate); |
| 463 |
| 464 /* Outer loop handles each block in the MCU */ |
| 465 |
| 466 for (blkn = 0; blkn < cinfo->blocks_in_MCU; blkn++) { |
| 467 block = MCU_data[blkn]; |
| 468 |
| 469 /* Encoded data is simply the next bit of the two's-complement DC value */ |
| 470 CHECK_BIT_BUFFER(br_state, 1, return FALSE); |
| 471 if (GET_BITS(1)) |
| 472 (*block)[0] |= p1; |
| 473 /* Note: since we use |=, repeating the assignment later is safe */ |
| 474 } |
| 475 |
| 476 /* Completed MCU, so update state */ |
| 477 BITREAD_SAVE_STATE(cinfo,entropy->bitstate); |
| 478 |
| 479 /* Account for restart interval (no-op if not using restarts) */ |
| 480 entropy->restarts_to_go--; |
| 481 |
| 482 return TRUE; |
| 483 } |
| 484 |
| 485 |
| 486 /* |
| 487 * MCU decoding for AC successive approximation refinement scan. |
| 488 */ |
| 489 |
| 490 METHODDEF(boolean) |
| 491 decode_mcu_AC_refine (j_decompress_ptr cinfo, JBLOCKROW *MCU_data) |
| 492 { |
| 493 phuff_entropy_ptr entropy = (phuff_entropy_ptr) cinfo->entropy; |
| 494 int Se = cinfo->Se; |
| 495 int p1 = 1 << cinfo->Al; /* 1 in the bit position being coded */ |
| 496 int m1 = (-1) << cinfo->Al; /* -1 in the bit position being coded */ |
| 497 register int s, k, r; |
| 498 unsigned int EOBRUN; |
| 499 JBLOCKROW block; |
| 500 JCOEFPTR thiscoef; |
| 501 BITREAD_STATE_VARS; |
| 502 d_derived_tbl * tbl; |
| 503 int num_newnz; |
| 504 int newnz_pos[DCTSIZE2]; |
| 505 |
| 506 /* Process restart marker if needed; may have to suspend */ |
| 507 if (cinfo->restart_interval) { |
| 508 if (entropy->restarts_to_go == 0) |
| 509 if (! process_restart(cinfo)) |
| 510 return FALSE; |
| 511 } |
| 512 |
| 513 /* If we've run out of data, don't modify the MCU. |
| 514 */ |
| 515 if (! entropy->pub.insufficient_data) { |
| 516 |
| 517 /* Load up working state */ |
| 518 BITREAD_LOAD_STATE(cinfo,entropy->bitstate); |
| 519 EOBRUN = entropy->saved.EOBRUN; /* only part of saved state we need */ |
| 520 |
| 521 /* There is always only one block per MCU */ |
| 522 block = MCU_data[0]; |
| 523 tbl = entropy->ac_derived_tbl; |
| 524 |
| 525 /* If we are forced to suspend, we must undo the assignments to any newly |
| 526 * nonzero coefficients in the block, because otherwise we'd get confused |
| 527 * next time about which coefficients were already nonzero. |
| 528 * But we need not undo addition of bits to already-nonzero coefficients; |
| 529 * instead, we can test the current bit to see if we already did it. |
| 530 */ |
| 531 num_newnz = 0; |
| 532 |
| 533 /* initialize coefficient loop counter to start of band */ |
| 534 k = cinfo->Ss; |
| 535 |
| 536 if (EOBRUN == 0) { |
| 537 for (; k <= Se; k++) { |
| 538 HUFF_DECODE(s, br_state, tbl, goto undoit, label3); |
| 539 r = s >> 4; |
| 540 s &= 15; |
| 541 if (s) { |
| 542 if (s != 1) /* size of new coef should always be 1 */ |
| 543 WARNMS(cinfo, JWRN_HUFF_BAD_CODE); |
| 544 CHECK_BIT_BUFFER(br_state, 1, goto undoit); |
| 545 if (GET_BITS(1)) |
| 546 s = p1; /* newly nonzero coef is positive */ |
| 547 else |
| 548 s = m1; /* newly nonzero coef is negative */ |
| 549 } else { |
| 550 if (r != 15) { |
| 551 EOBRUN = 1 << r; /* EOBr, run length is 2^r + appended bits */ |
| 552 if (r) { |
| 553 CHECK_BIT_BUFFER(br_state, r, goto undoit); |
| 554 r = GET_BITS(r); |
| 555 EOBRUN += r; |
| 556 } |
| 557 break; /* rest of block is handled by EOB logic */ |
| 558 } |
| 559 /* note s = 0 for processing ZRL */ |
| 560 } |
| 561 /* Advance over already-nonzero coefs and r still-zero coefs, |
| 562 * appending correction bits to the nonzeroes. A correction bit is 1 |
| 563 * if the absolute value of the coefficient must be increased. |
| 564 */ |
| 565 do { |
| 566 thiscoef = *block + jpeg_natural_order[k]; |
| 567 if (*thiscoef != 0) { |
| 568 CHECK_BIT_BUFFER(br_state, 1, goto undoit); |
| 569 if (GET_BITS(1)) { |
| 570 if ((*thiscoef & p1) == 0) { /* do nothing if already set it */ |
| 571 if (*thiscoef >= 0) |
| 572 *thiscoef += p1; |
| 573 else |
| 574 *thiscoef += m1; |
| 575 } |
| 576 } |
| 577 } else { |
| 578 if (--r < 0) |
| 579 break; /* reached target zero coefficient */ |
| 580 } |
| 581 k++; |
| 582 } while (k <= Se); |
| 583 if (s) { |
| 584 int pos = jpeg_natural_order[k]; |
| 585 /* Output newly nonzero coefficient */ |
| 586 (*block)[pos] = (JCOEF) s; |
| 587 /* Remember its position in case we have to suspend */ |
| 588 newnz_pos[num_newnz++] = pos; |
| 589 } |
| 590 } |
| 591 } |
| 592 |
| 593 if (EOBRUN > 0) { |
| 594 /* Scan any remaining coefficient positions after the end-of-band |
| 595 * (the last newly nonzero coefficient, if any). Append a correction |
| 596 * bit to each already-nonzero coefficient. A correction bit is 1 |
| 597 * if the absolute value of the coefficient must be increased. |
| 598 */ |
| 599 for (; k <= Se; k++) { |
| 600 thiscoef = *block + jpeg_natural_order[k]; |
| 601 if (*thiscoef != 0) { |
| 602 CHECK_BIT_BUFFER(br_state, 1, goto undoit); |
| 603 if (GET_BITS(1)) { |
| 604 if ((*thiscoef & p1) == 0) { /* do nothing if already changed it */ |
| 605 if (*thiscoef >= 0) |
| 606 *thiscoef += p1; |
| 607 else |
| 608 *thiscoef += m1; |
| 609 } |
| 610 } |
| 611 } |
| 612 } |
| 613 /* Count one block completed in EOB run */ |
| 614 EOBRUN--; |
| 615 } |
| 616 |
| 617 /* Completed MCU, so update state */ |
| 618 BITREAD_SAVE_STATE(cinfo,entropy->bitstate); |
| 619 entropy->saved.EOBRUN = EOBRUN; /* only part of saved state we need */ |
| 620 } |
| 621 |
| 622 /* Account for restart interval (no-op if not using restarts) */ |
| 623 entropy->restarts_to_go--; |
| 624 |
| 625 return TRUE; |
| 626 |
| 627 undoit: |
| 628 /* Re-zero any output coefficients that we made newly nonzero */ |
| 629 while (num_newnz > 0) |
| 630 (*block)[newnz_pos[--num_newnz]] = 0; |
| 631 |
| 632 return FALSE; |
| 633 } |
| 634 |
| 635 |
| 636 /* |
| 637 * Module initialization routine for progressive Huffman entropy decoding. |
| 638 */ |
| 639 |
| 640 GLOBAL(void) |
| 641 jinit_phuff_decoder (j_decompress_ptr cinfo) |
| 642 { |
| 643 phuff_entropy_ptr entropy; |
| 644 int *coef_bit_ptr; |
| 645 int ci, i; |
| 646 |
| 647 entropy = (phuff_entropy_ptr) |
| 648 (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE, |
| 649 SIZEOF(phuff_entropy_decoder)); |
| 650 cinfo->entropy = (struct jpeg_entropy_decoder *) entropy; |
| 651 entropy->pub.start_pass = start_pass_phuff_decoder; |
| 652 |
| 653 /* Mark derived tables unallocated */ |
| 654 for (i = 0; i < NUM_HUFF_TBLS; i++) { |
| 655 entropy->derived_tbls[i] = NULL; |
| 656 } |
| 657 |
| 658 /* Create progression status table */ |
| 659 cinfo->coef_bits = (int (*)[DCTSIZE2]) |
| 660 (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE, |
| 661 cinfo->num_components*DCTSIZE2*SIZEOF(int)); |
| 662 coef_bit_ptr = & cinfo->coef_bits[0][0]; |
| 663 for (ci = 0; ci < cinfo->num_components; ci++) |
| 664 for (i = 0; i < DCTSIZE2; i++) |
| 665 *coef_bit_ptr++ = -1; |
| 666 } |
| 667 |
| 668 #endif /* D_PROGRESSIVE_SUPPORTED */ |
OLD | NEW |