1 /* 2 * Copyright (c) 1999-2000 Image Power, Inc. and the University of 3 * British Columbia. 4 * Copyright (c) 2001-2002 Michael David Adams. 5 * All rights reserved. 6 */ 7 8 /* __START_OF_JASPER_LICENSE__ 9 * 10 * JasPer License Version 2.0 11 * 12 * Copyright (c) 2001-2006 Michael David Adams 13 * Copyright (c) 1999-2000 Image Power, Inc. 14 * Copyright (c) 1999-2000 The University of British Columbia 15 * 16 * All rights reserved. 17 * 18 * Permission is hereby granted, free of charge, to any person (the 19 * "User") obtaining a copy of this software and associated documentation 20 * files (the "Software"), to deal in the Software without restriction, 21 * including without limitation the rights to use, copy, modify, merge, 22 * publish, distribute, and/or sell copies of the Software, and to permit 23 * persons to whom the Software is furnished to do so, subject to the 24 * following conditions: 25 * 26 * 1. The above copyright notices and this permission notice (which 27 * includes the disclaimer below) shall be included in all copies or 28 * substantial portions of the Software. 29 * 30 * 2. The name of a copyright holder shall not be used to endorse or 31 * promote products derived from the Software without specific prior 32 * written permission. 33 * 34 * THIS DISCLAIMER OF WARRANTY CONSTITUTES AN ESSENTIAL PART OF THIS 35 * LICENSE. NO USE OF THE SOFTWARE IS AUTHORIZED HEREUNDER EXCEPT UNDER 36 * THIS DISCLAIMER. THE SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 37 * "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING 38 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A 39 * PARTICULAR PURPOSE AND NONINFRINGEMENT OF THIRD PARTY RIGHTS. IN NO 40 * EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, OR ANY SPECIAL 41 * INDIRECT OR CONSEQUENTIAL DAMAGES, OR ANY DAMAGES WHATSOEVER RESULTING 42 * FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, 43 * NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION 44 * WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. NO ASSURANCES ARE 45 * PROVIDED BY THE COPYRIGHT HOLDERS THAT THE SOFTWARE DOES NOT INFRINGE 46 * THE PATENT OR OTHER INTELLECTUAL PROPERTY RIGHTS OF ANY OTHER ENTITY. 47 * EACH COPYRIGHT HOLDER DISCLAIMS ANY LIABILITY TO THE USER FOR CLAIMS 48 * BROUGHT BY ANY OTHER ENTITY BASED ON INFRINGEMENT OF INTELLECTUAL 49 * PROPERTY RIGHTS OR OTHERWISE. AS A CONDITION TO EXERCISING THE RIGHTS 50 * GRANTED HEREUNDER, EACH USER HEREBY ASSUMES SOLE RESPONSIBILITY TO SECURE 51 * ANY OTHER INTELLECTUAL PROPERTY RIGHTS NEEDED, IF ANY. THE SOFTWARE 52 * IS NOT FAULT-TOLERANT AND IS NOT INTENDED FOR USE IN MISSION-CRITICAL 53 * SYSTEMS, SUCH AS THOSE USED IN THE OPERATION OF NUCLEAR FACILITIES, 54 * AIRCRAFT NAVIGATION OR COMMUNICATION SYSTEMS, AIR TRAFFIC CONTROL 55 * SYSTEMS, DIRECT LIFE SUPPORT MACHINES, OR WEAPONS SYSTEMS, IN WHICH 56 * THE FAILURE OF THE SOFTWARE OR SYSTEM COULD LEAD DIRECTLY TO DEATH, 57 * PERSONAL INJURY, OR SEVERE PHYSICAL OR ENVIRONMENTAL DAMAGE ("HIGH 58 * RISK ACTIVITIES"). THE COPYRIGHT HOLDERS SPECIFICALLY DISCLAIM ANY 59 * EXPRESS OR IMPLIED WARRANTY OF FITNESS FOR HIGH RISK ACTIVITIES. 60 * 61 * __END_OF_JASPER_LICENSE__ 62 */ 63 64 /* 65 * Tier 1 Encoder 66 * 67 * $Id: jpc_t1enc.c,v 1.2 2008-05-26 09:40:52 vp153 Exp $ 68 */ 69 70 /******************************************************************************\ 71 * Includes. 72 \******************************************************************************/ 73 74 #include <stdio.h> 75 #include <stdlib.h> 76 #include <assert.h> 77 78 #include "jasper/jas_fix.h" 79 #include "jasper/jas_malloc.h" 80 #include "jasper/jas_math.h" 81 82 #include "jpc_t1enc.h" 83 #include "jpc_t1cod.h" 84 #include "jpc_enc.h" 85 #include "jpc_cod.h" 86 #include "jpc_math.h" 87 88 static int jpc_encsigpass(jpc_mqenc_t *mqenc, int bitpos, int orient, int, 89 jas_matrix_t *flags, jas_matrix_t *data, int term, long *nmsedec); 90 91 static int jpc_encrefpass(jpc_mqenc_t *mqenc, int bitpos, int, jas_matrix_t *flags, 92 jas_matrix_t *data, int term, long *nmsedec); 93 94 static int jpc_encclnpass(jpc_mqenc_t *mqenc, int bitpos, int orient, int, 95 int, jas_matrix_t *flags, jas_matrix_t *data, int term, long *nmsedec); 96 97 static int jpc_encrawsigpass(jpc_bitstream_t *out, int bitpos, int, 98 jas_matrix_t *flags, jas_matrix_t *data, int term, long *nmsedec); 99 100 static int jpc_encrawrefpass(jpc_bitstream_t *out, int bitpos, int, 101 jas_matrix_t *flags, jas_matrix_t *data, int term, long *nmsedec); 102 103 /******************************************************************************\ 104 * Code for encoding code blocks. 105 \******************************************************************************/ 106 107 /* Encode all of the code blocks associated with the current tile. */ 108 int jpc_enc_enccblks(jpc_enc_t *enc) 109 { 110 jpc_enc_tcmpt_t *tcmpt; 111 jpc_enc_tcmpt_t *endcomps; 112 jpc_enc_rlvl_t *lvl; 113 jpc_enc_rlvl_t *endlvls; 114 jpc_enc_band_t *band; 115 jpc_enc_band_t *endbands; 116 jpc_enc_cblk_t *cblk; 117 jpc_enc_cblk_t *endcblks; 118 int i; 119 int j; 120 int mx; 121 int bmx; 122 int v; 123 jpc_enc_tile_t *tile; 124 uint_fast32_t prcno; 125 jpc_enc_prc_t *prc; 126 127 tile = enc->curtile; 128 129 endcomps = &tile->tcmpts[tile->numtcmpts]; 130 for (tcmpt = tile->tcmpts; tcmpt != endcomps; ++tcmpt) { 131 endlvls = &tcmpt->rlvls[tcmpt->numrlvls]; 132 for (lvl = tcmpt->rlvls; lvl != endlvls; ++lvl) { 133 if (!lvl->bands) { 134 continue; 135 } 136 endbands = &lvl->bands[lvl->numbands]; 137 for (band = lvl->bands; band != endbands; ++band) { 138 if (!band->data) { 139 continue; 140 } 141 for (prcno = 0, prc = band->prcs; prcno < lvl->numprcs; ++prcno, ++prc) { 142 if (!prc->cblks) { 143 continue; 144 } 145 bmx = 0; 146 endcblks = &prc->cblks[prc->numcblks]; 147 for (cblk = prc->cblks; cblk != endcblks; ++cblk) { 148 mx = 0; 149 for (i = 0; i < jas_matrix_numrows(cblk->data); ++i) { 150 for (j = 0; j < jas_matrix_numcols(cblk->data); ++j) { 151 v = abs(jas_matrix_get(cblk->data, i, j)); 152 if (v > mx) { 153 mx = v; 154 } 155 } 156 } 157 if (mx > bmx) { 158 bmx = mx; 159 } 160 cblk->numbps = JAS_MAX(jpc_firstone(mx) + 1 - JPC_NUMEXTRABITS, 0); 161 } 162 163 for (cblk = prc->cblks; cblk != endcblks; ++cblk) { 164 cblk->numimsbs = band->numbps - cblk->numbps; 165 assert(cblk->numimsbs >= 0); 166 } 167 168 for (cblk = prc->cblks; cblk != endcblks; ++cblk) { 169 if (jpc_enc_enccblk(enc, cblk->stream, tcmpt, band, cblk)) { 170 return -1; 171 } 172 } 173 } 174 } 175 } 176 } 177 return 0; 178 } 179 180 int getthebyte(jas_stream_t *in, long off) 181 { 182 int c; 183 long oldpos; 184 oldpos = jas_stream_tell(in); 185 assert(oldpos >= 0); 186 jas_stream_seek(in, off, SEEK_SET); 187 c = jas_stream_peekc(in); 188 jas_stream_seek(in, oldpos, SEEK_SET); 189 return c; 190 } 191 192 /* Encode a single code block. */ 193 int jpc_enc_enccblk(jpc_enc_t *enc, jas_stream_t *out, jpc_enc_tcmpt_t *tcmpt, jpc_enc_band_t *band, jpc_enc_cblk_t *cblk) 194 { 195 jpc_enc_pass_t *pass; 196 jpc_enc_pass_t *endpasses; 197 int bitpos; 198 int n; 199 int adjust; 200 int ret; 201 int passtype; 202 int t; 203 jpc_bitstream_t *bout; 204 jpc_enc_pass_t *termpass; 205 jpc_enc_rlvl_t *rlvl; 206 int vcausal; 207 int segsym; 208 int termmode; 209 int c; 210 211 bout = 0; 212 rlvl = band->rlvl; 213 214 cblk->stream = jas_stream_memopen(0, 0); 215 assert(cblk->stream); 216 cblk->mqenc = jpc_mqenc_create(JPC_NUMCTXS, cblk->stream); 217 assert(cblk->mqenc); 218 jpc_mqenc_setctxs(cblk->mqenc, JPC_NUMCTXS, jpc_mqctxs); 219 220 cblk->numpasses = (cblk->numbps > 0) ? (3 * cblk->numbps - 2) : 0; 221 if (cblk->numpasses > 0) { 222 cblk->passes = jas_alloc2(cblk->numpasses, sizeof(jpc_enc_pass_t)); 223 assert(cblk->passes); 224 } else { 225 cblk->passes = 0; 226 } 227 endpasses = &cblk->passes[cblk->numpasses]; 228 for (pass = cblk->passes; pass != endpasses; ++pass) { 229 pass->start = 0; 230 pass->end = 0; 231 pass->term = JPC_ISTERMINATED(pass - cblk->passes, 0, cblk->numpasses, (tcmpt->cblksty & JPC_COX_TERMALL) != 0, (tcmpt->cblksty & JPC_COX_LAZY) != 0); 232 pass->type = JPC_SEGTYPE(pass - cblk->passes, 0, (tcmpt->cblksty & JPC_COX_LAZY) != 0); 233 pass->lyrno = -1; 234 if (pass == endpasses - 1) { 235 assert(pass->term == 1); 236 pass->term = 1; 237 } 238 } 239 240 cblk->flags = jas_matrix_create(jas_matrix_numrows(cblk->data) + 2, 241 jas_matrix_numcols(cblk->data) + 2); 242 assert(cblk->flags); 243 244 245 bitpos = cblk->numbps - 1; 246 pass = cblk->passes; 247 n = cblk->numpasses; 248 while (--n >= 0) { 249 250 if (pass->type == JPC_SEG_MQ) { 251 /* NOP */ 252 } else { 253 assert(pass->type == JPC_SEG_RAW); 254 if (!bout) { 255 bout = jpc_bitstream_sopen(cblk->stream, "w"); 256 assert(bout); 257 } 258 } 259 260 #if 1 261 passtype = (pass - cblk->passes + 2) % 3; 262 #else 263 passtype = JPC_PASSTYPE(pass - cblk->passes + 2); 264 #endif 265 pass->start = jas_stream_tell(cblk->stream); 266 #if 0 267 assert(jas_stream_tell(cblk->stream) == jas_stream_getrwcount(cblk->stream)); 268 #endif 269 assert(bitpos >= 0); 270 vcausal = (tcmpt->cblksty & JPC_COX_VSC) != 0; 271 segsym = (tcmpt->cblksty & JPC_COX_SEGSYM) != 0; 272 if (pass->term) { 273 termmode = ((tcmpt->cblksty & JPC_COX_PTERM) ? 274 JPC_MQENC_PTERM : JPC_MQENC_DEFTERM) + 1; 275 } else { 276 termmode = 0; 277 } 278 switch (passtype) { 279 case JPC_SIGPASS: 280 ret = (pass->type == JPC_SEG_MQ) ? jpc_encsigpass(cblk->mqenc, 281 bitpos, band->orient, vcausal, cblk->flags, 282 cblk->data, termmode, &pass->nmsedec) : 283 jpc_encrawsigpass(bout, bitpos, vcausal, cblk->flags, 284 cblk->data, termmode, &pass->nmsedec); 285 break; 286 case JPC_REFPASS: 287 ret = (pass->type == JPC_SEG_MQ) ? jpc_encrefpass(cblk->mqenc, 288 bitpos, vcausal, cblk->flags, cblk->data, termmode, 289 &pass->nmsedec) : jpc_encrawrefpass(bout, bitpos, 290 vcausal, cblk->flags, cblk->data, termmode, 291 &pass->nmsedec); 292 break; 293 case JPC_CLNPASS: 294 assert(pass->type == JPC_SEG_MQ); 295 ret = jpc_encclnpass(cblk->mqenc, bitpos, band->orient, 296 vcausal, segsym, cblk->flags, cblk->data, termmode, 297 &pass->nmsedec); 298 break; 299 default: 300 assert(0); 301 break; 302 } 303 304 if (pass->type == JPC_SEG_MQ) { 305 if (pass->term) { 306 jpc_mqenc_init(cblk->mqenc); 307 } 308 jpc_mqenc_getstate(cblk->mqenc, &pass->mqencstate); 309 pass->end = jas_stream_tell(cblk->stream); 310 if (tcmpt->cblksty & JPC_COX_RESET) { 311 jpc_mqenc_setctxs(cblk->mqenc, JPC_NUMCTXS, jpc_mqctxs); 312 } 313 } else { 314 if (pass->term) { 315 if (jpc_bitstream_pending(bout)) { 316 jpc_bitstream_outalign(bout, 0x2a); 317 } 318 jpc_bitstream_close(bout); 319 bout = 0; 320 pass->end = jas_stream_tell(cblk->stream); 321 } else { 322 pass->end = jas_stream_tell(cblk->stream) + 323 jpc_bitstream_pending(bout); 324 /* NOTE - This will not work. need to adjust by # of pending output bytes */ 325 } 326 } 327 #if 0 328 /* XXX - This assertion fails sometimes when various coding modes are used. 329 This seems to be harmless, but why does it happen at all? */ 330 assert(jas_stream_tell(cblk->stream) == jas_stream_getrwcount(cblk->stream)); 331 #endif 332 333 pass->wmsedec = jpc_fixtodbl(band->rlvl->tcmpt->synweight) * 334 jpc_fixtodbl(band->rlvl->tcmpt->synweight) * 335 jpc_fixtodbl(band->synweight) * 336 jpc_fixtodbl(band->synweight) * 337 jpc_fixtodbl(band->absstepsize) * jpc_fixtodbl(band->absstepsize) * 338 ((double) (1 << bitpos)) * ((double)(1 << bitpos)) * 339 jpc_fixtodbl(pass->nmsedec); 340 pass->cumwmsedec = pass->wmsedec; 341 if (pass != cblk->passes) { 342 pass->cumwmsedec += pass[-1].cumwmsedec; 343 } 344 if (passtype == JPC_CLNPASS) { 345 --bitpos; 346 } 347 ++pass; 348 } 349 350 #if 0 351 dump_passes(cblk->passes, cblk->numpasses, cblk); 352 #endif 353 354 n = 0; 355 endpasses = &cblk->passes[cblk->numpasses]; 356 for (pass = cblk->passes; pass != endpasses; ++pass) { 357 if (pass->start < n) { 358 pass->start = n; 359 } 360 if (pass->end < n) { 361 pass->end = n; 362 } 363 if (!pass->term) { 364 termpass = pass; 365 while (termpass - pass < cblk->numpasses && 366 !termpass->term) { 367 ++termpass; 368 } 369 if (pass->type == JPC_SEG_MQ) { 370 t = (pass->mqencstate.lastbyte == 0xff) ? 1 : 0; 371 if (pass->mqencstate.ctreg >= 5) { 372 adjust = 4 + t; 373 } else { 374 adjust = 5 + t; 375 } 376 pass->end += adjust; 377 } 378 if (pass->end > termpass->end) { 379 pass->end = termpass->end; 380 } 381 if ((c = getthebyte(cblk->stream, pass->end - 1)) == EOF) { 382 abort(); 383 } 384 if (c == 0xff) { 385 ++pass->end; 386 } 387 n = JAS_MAX(n, pass->end); 388 } else { 389 n = JAS_MAX(n, pass->end); 390 } 391 } 392 393 #if 0 394 dump_passes(cblk->passes, cblk->numpasses, cblk); 395 #endif 396 397 if (bout) { 398 jpc_bitstream_close(bout); 399 } 400 401 return 0; 402 } 403 404 /******************************************************************************\ 405 * Code for significance pass. 406 \******************************************************************************/ 407 408 #define sigpass_step(fp, frowstep, dp, bitpos, one, nmsedec, orient, mqenc, vcausalflag) \ 409 { \ 410 int f; \ 411 int v; \ 412 f = *(fp); \ 413 if ((f & JPC_OTHSIGMSK) && !(f & (JPC_SIG | JPC_VISIT))) { \ 414 v = (abs(*(dp)) & (one)) ? 1 : 0; \ 415 jpc_mqenc_setcurctx(mqenc, JPC_GETZCCTXNO(f, (orient))); \ 416 jpc_mqenc_putbit(mqenc, v); \ 417 if (v) { \ 418 *(nmsedec) += JPC_GETSIGNMSEDEC(abs(*(dp)), (bitpos) + JPC_NUMEXTRABITS); \ 419 v = ((*(dp) < 0) ? 1 : 0); \ 420 jpc_mqenc_setcurctx(mqenc, JPC_GETSCCTXNO(f)); \ 421 jpc_mqenc_putbit(mqenc, v ^ JPC_GETSPB(f)); \ 422 JPC_UPDATEFLAGS4(fp, frowstep, v, vcausalflag); \ 423 *(fp) |= JPC_SIG; \ 424 } \ 425 *(fp) |= JPC_VISIT; \ 426 } \ 427 } 428 429 static int jpc_encsigpass(jpc_mqenc_t *mqenc, int bitpos, int orient, int vcausalflag, 430 jas_matrix_t *flags, jas_matrix_t *data, int term, long *nmsedec) 431 { 432 int i; 433 int j; 434 int one; 435 int vscanlen; 436 int width; 437 int height; 438 int frowstep; 439 int drowstep; 440 int fstripestep; 441 int dstripestep; 442 jpc_fix_t *fstripestart; 443 jpc_fix_t *dstripestart; 444 jpc_fix_t *fp; 445 jpc_fix_t *dp; 446 jpc_fix_t *fvscanstart; 447 jpc_fix_t *dvscanstart; 448 int k; 449 450 *nmsedec = 0; 451 width = jas_matrix_numcols(data); 452 height = jas_matrix_numrows(data); 453 frowstep = jas_matrix_rowstep(flags); 454 drowstep = jas_matrix_rowstep(data); 455 fstripestep = frowstep << 2; 456 dstripestep = drowstep << 2; 457 458 one = 1 << (bitpos + JPC_NUMEXTRABITS); 459 460 fstripestart = jas_matrix_getref(flags, 1, 1); 461 dstripestart = jas_matrix_getref(data, 0, 0); 462 for (i = height; i > 0; i -= 4, fstripestart += fstripestep, 463 dstripestart += dstripestep) { 464 fvscanstart = fstripestart; 465 dvscanstart = dstripestart; 466 vscanlen = JAS_MIN(i, 4); 467 for (j = width; j > 0; --j, ++fvscanstart, ++dvscanstart) { 468 fp = fvscanstart; 469 dp = dvscanstart; 470 k = vscanlen; 471 472 sigpass_step(fp, frowstep, dp, bitpos, one, 473 nmsedec, orient, mqenc, vcausalflag); 474 if (--k <= 0) { 475 continue; 476 } 477 fp += frowstep; 478 dp += drowstep; 479 sigpass_step(fp, frowstep, dp, bitpos, one, 480 nmsedec, orient, mqenc, 0); 481 if (--k <= 0) { 482 continue; 483 } 484 fp += frowstep; 485 dp += drowstep; 486 sigpass_step(fp, frowstep, dp, bitpos, one, 487 nmsedec, orient, mqenc, 0); 488 if (--k <= 0) { 489 continue; 490 } 491 fp += frowstep; 492 dp += drowstep; 493 sigpass_step(fp, frowstep, dp, bitpos, one, 494 nmsedec, orient, mqenc, 0); 495 496 } 497 } 498 499 if (term) { 500 jpc_mqenc_flush(mqenc, term - 1); 501 } 502 503 return jpc_mqenc_error(mqenc) ? (-1) : 0; 504 } 505 506 #define rawsigpass_step(fp, frowstep, dp, bitpos, one, nmsedec, out, vcausalflag) \ 507 { \ 508 jpc_fix_t f = *(fp); \ 509 jpc_fix_t v; \ 510 if ((f & JPC_OTHSIGMSK) && !(f & (JPC_SIG | JPC_VISIT))) { \ 511 v = (abs(*(dp)) & (one)) ? 1 : 0; \ 512 if ((jpc_bitstream_putbit((out), v)) == EOF) { \ 513 return -1; \ 514 } \ 515 if (v) { \ 516 *(nmsedec) += JPC_GETSIGNMSEDEC(abs(*(dp)), (bitpos) + JPC_NUMEXTRABITS); \ 517 v = ((*(dp) < 0) ? 1 : 0); \ 518 if (jpc_bitstream_putbit(out, v) == EOF) { \ 519 return -1; \ 520 } \ 521 JPC_UPDATEFLAGS4(fp, frowstep, v, vcausalflag); \ 522 *(fp) |= JPC_SIG; \ 523 } \ 524 *(fp) |= JPC_VISIT; \ 525 } \ 526 } 527 528 static int jpc_encrawsigpass(jpc_bitstream_t *out, int bitpos, int vcausalflag, jas_matrix_t *flags, 529 jas_matrix_t *data, int term, long *nmsedec) 530 { 531 int i; 532 int j; 533 int k; 534 int one; 535 int vscanlen; 536 int width; 537 int height; 538 int frowstep; 539 int drowstep; 540 int fstripestep; 541 int dstripestep; 542 jpc_fix_t *fstripestart; 543 jpc_fix_t *dstripestart; 544 jpc_fix_t *fp; 545 jpc_fix_t *dp; 546 jpc_fix_t *fvscanstart; 547 jpc_fix_t *dvscanstart; 548 549 *nmsedec = 0; 550 width = jas_matrix_numcols(data); 551 height = jas_matrix_numrows(data); 552 frowstep = jas_matrix_rowstep(flags); 553 drowstep = jas_matrix_rowstep(data); 554 fstripestep = frowstep << 2; 555 dstripestep = drowstep << 2; 556 557 one = 1 << (bitpos + JPC_NUMEXTRABITS); 558 559 fstripestart = jas_matrix_getref(flags, 1, 1); 560 dstripestart = jas_matrix_getref(data, 0, 0); 561 for (i = height; i > 0; i -= 4, fstripestart += fstripestep, 562 dstripestart += dstripestep) { 563 fvscanstart = fstripestart; 564 dvscanstart = dstripestart; 565 vscanlen = JAS_MIN(i, 4); 566 for (j = width; j > 0; --j, ++fvscanstart, ++dvscanstart) { 567 fp = fvscanstart; 568 dp = dvscanstart; 569 k = vscanlen; 570 571 rawsigpass_step(fp, frowstep, dp, bitpos, one, 572 nmsedec, out, vcausalflag); 573 if (--k <= 0) { 574 continue; 575 } 576 fp += frowstep; 577 dp += drowstep; 578 579 rawsigpass_step(fp, frowstep, dp, bitpos, one, 580 nmsedec, out, 0); 581 if (--k <= 0) { 582 continue; 583 } 584 fp += frowstep; 585 dp += drowstep; 586 587 rawsigpass_step(fp, frowstep, dp, bitpos, one, 588 nmsedec, out, 0); 589 if (--k <= 0) { 590 continue; 591 } 592 fp += frowstep; 593 dp += drowstep; 594 595 rawsigpass_step(fp, frowstep, dp, bitpos, one, 596 nmsedec, out, 0); 597 if (--k <= 0) { 598 continue; 599 } 600 fp += frowstep; 601 dp += drowstep; 602 603 } 604 } 605 606 if (term) { 607 jpc_bitstream_outalign(out, 0x2a); 608 } 609 610 return 0; 611 } 612 613 /******************************************************************************\ 614 * Code for refinement pass. 615 \******************************************************************************/ 616 617 #define refpass_step(fp, dp, bitpos, one, nmsedec, mqenc, vcausalflag) \ 618 { \ 619 int v; \ 620 if (((*(fp)) & (JPC_SIG | JPC_VISIT)) == JPC_SIG) { \ 621 (d) = *(dp); \ 622 *(nmsedec) += JPC_GETREFNMSEDEC(abs(d), (bitpos) + JPC_NUMEXTRABITS); \ 623 jpc_mqenc_setcurctx((mqenc), JPC_GETMAGCTXNO(*(fp))); \ 624 v = (abs(d) & (one)) ? 1 : 0; \ 625 jpc_mqenc_putbit((mqenc), v); \ 626 *(fp) |= JPC_REFINE; \ 627 } \ 628 } 629 630 static int jpc_encrefpass(jpc_mqenc_t *mqenc, int bitpos, int vcausalflag, jas_matrix_t *flags, jas_matrix_t *data, 631 int term, long *nmsedec) 632 { 633 int i; 634 int j; 635 int one; 636 int vscanlen; 637 int d; 638 int width; 639 int height; 640 int frowstep; 641 int drowstep; 642 int fstripestep; 643 int dstripestep; 644 jpc_fix_t *fstripestart; 645 jpc_fix_t *dstripestart; 646 jpc_fix_t *fvscanstart; 647 jpc_fix_t *dvscanstart; 648 jpc_fix_t *dp; 649 jpc_fix_t *fp; 650 int k; 651 652 *nmsedec = 0; 653 width = jas_matrix_numcols(data); 654 height = jas_matrix_numrows(data); 655 frowstep = jas_matrix_rowstep(flags); 656 drowstep = jas_matrix_rowstep(data); 657 fstripestep = frowstep << 2; 658 dstripestep = drowstep << 2; 659 660 one = 1 << (bitpos + JPC_NUMEXTRABITS); 661 662 fstripestart = jas_matrix_getref(flags, 1, 1); 663 dstripestart = jas_matrix_getref(data, 0, 0); 664 for (i = height; i > 0; i -= 4, fstripestart += fstripestep, 665 dstripestart += dstripestep) { 666 fvscanstart = fstripestart; 667 dvscanstart = dstripestart; 668 vscanlen = JAS_MIN(i, 4); 669 for (j = width; j > 0; --j, ++fvscanstart, ++dvscanstart) { 670 fp = fvscanstart; 671 dp = dvscanstart; 672 k = vscanlen; 673 674 refpass_step(fp, dp, bitpos, one, nmsedec, 675 mqenc, vcausalflag); 676 if (--k <= 0) { 677 continue; 678 } 679 fp += frowstep; 680 dp += drowstep; 681 refpass_step(fp, dp, bitpos, one, nmsedec, 682 mqenc, 0); 683 if (--k <= 0) { 684 continue; 685 } 686 fp += frowstep; 687 dp += drowstep; 688 refpass_step(fp, dp, bitpos, one, nmsedec, 689 mqenc, 0); 690 if (--k <= 0) { 691 continue; 692 } 693 fp += frowstep; 694 dp += drowstep; 695 refpass_step(fp, dp, bitpos, one, nmsedec, 696 mqenc, 0); 697 698 } 699 } 700 701 if (term) { 702 jpc_mqenc_flush(mqenc, term - 1); 703 } 704 705 return jpc_mqenc_error(mqenc) ? (-1) : 0; 706 } 707 708 #define rawrefpass_step(fp, dp, bitpos, one, nmsedec, out, vcausalflag) \ 709 { \ 710 jpc_fix_t d; \ 711 jpc_fix_t v; \ 712 if (((*(fp)) & (JPC_SIG | JPC_VISIT)) == JPC_SIG) { \ 713 d = *(dp); \ 714 *(nmsedec) += JPC_GETREFNMSEDEC(abs(d), (bitpos) + JPC_NUMEXTRABITS); \ 715 v = (abs(d) & (one)) ? 1 : 0; \ 716 if (jpc_bitstream_putbit((out), v) == EOF) { \ 717 return -1; \ 718 } \ 719 *(fp) |= JPC_REFINE; \ 720 } \ 721 } 722 723 static int jpc_encrawrefpass(jpc_bitstream_t *out, int bitpos, int vcausalflag, jas_matrix_t *flags, 724 jas_matrix_t *data, int term, long *nmsedec) 725 { 726 int i; 727 int j; 728 int k; 729 int one; 730 int vscanlen; 731 int width; 732 int height; 733 int frowstep; 734 int drowstep; 735 int fstripestep; 736 int dstripestep; 737 jpc_fix_t *fstripestart; 738 jpc_fix_t *dstripestart; 739 jpc_fix_t *fvscanstart; 740 jpc_fix_t *dvscanstart; 741 jpc_fix_t *dp; 742 jpc_fix_t *fp; 743 744 *nmsedec = 0; 745 width = jas_matrix_numcols(data); 746 height = jas_matrix_numrows(data); 747 frowstep = jas_matrix_rowstep(flags); 748 drowstep = jas_matrix_rowstep(data); 749 fstripestep = frowstep << 2; 750 dstripestep = drowstep << 2; 751 752 one = 1 << (bitpos + JPC_NUMEXTRABITS); 753 754 fstripestart = jas_matrix_getref(flags, 1, 1); 755 dstripestart = jas_matrix_getref(data, 0, 0); 756 for (i = height; i > 0; i -= 4, fstripestart += fstripestep, 757 dstripestart += dstripestep) { 758 fvscanstart = fstripestart; 759 dvscanstart = dstripestart; 760 vscanlen = JAS_MIN(i, 4); 761 for (j = width; j > 0; --j, ++fvscanstart, ++dvscanstart) { 762 fp = fvscanstart; 763 dp = dvscanstart; 764 k = vscanlen; 765 766 rawrefpass_step(fp, dp, bitpos, one, nmsedec, 767 out, vcausalflag); 768 if (--k <= 0) { 769 continue; 770 } 771 fp += frowstep; 772 dp += drowstep; 773 rawrefpass_step(fp, dp, bitpos, one, nmsedec, 774 out, vcausalflag); 775 if (--k <= 0) { 776 continue; 777 } 778 fp += frowstep; 779 dp += drowstep; 780 rawrefpass_step(fp, dp, bitpos, one, nmsedec, 781 out, vcausalflag); 782 if (--k <= 0) { 783 continue; 784 } 785 fp += frowstep; 786 dp += drowstep; 787 rawrefpass_step(fp, dp, bitpos, one, nmsedec, 788 out, vcausalflag); 789 790 } 791 } 792 793 if (term) { 794 jpc_bitstream_outalign(out, 0x2a); 795 } 796 797 return 0; 798 } 799 800 /******************************************************************************\ 801 * Code for cleanup pass. 802 \******************************************************************************/ 803 804 #define clnpass_step(fp, frowstep, dp, bitpos, one, orient, nmsedec, mqenc, label1, label2, vcausalflag) \ 805 { \ 806 int f; \ 807 int v; \ 808 label1 \ 809 f = *(fp); \ 810 if (!(f & (JPC_SIG | JPC_VISIT))) { \ 811 jpc_mqenc_setcurctx(mqenc, JPC_GETZCCTXNO(f, (orient))); \ 812 v = (abs(*(dp)) & (one)) ? 1 : 0; \ 813 jpc_mqenc_putbit((mqenc), v); \ 814 if (v) { \ 815 label2 \ 816 f = *(fp); \ 817 /* Coefficient is significant. */ \ 818 *(nmsedec) += JPC_GETSIGNMSEDEC(abs(*(dp)), (bitpos) + JPC_NUMEXTRABITS); \ 819 jpc_mqenc_setcurctx((mqenc), JPC_GETSCCTXNO(f)); \ 820 v = ((*(dp) < 0) ? 1 : 0); \ 821 jpc_mqenc_putbit((mqenc), v ^ JPC_GETSPB(f)); \ 822 JPC_UPDATEFLAGS4((fp), (frowstep), v, vcausalflag); \ 823 *(fp) |= JPC_SIG; \ 824 } \ 825 } \ 826 *(fp) &= ~JPC_VISIT; \ 827 } 828 829 static int jpc_encclnpass(jpc_mqenc_t *mqenc, int bitpos, int orient, int vcausalflag, int segsymflag, jas_matrix_t *flags, 830 jas_matrix_t *data, int term, long *nmsedec) 831 { 832 int i; 833 int j; 834 int k; 835 int vscanlen; 836 int v; 837 int runlen; 838 jpc_fix_t *fp; 839 int width; 840 int height; 841 jpc_fix_t *dp; 842 int one; 843 int frowstep; 844 int drowstep; 845 int fstripestep; 846 int dstripestep; 847 jpc_fix_t *fstripestart; 848 jpc_fix_t *dstripestart; 849 jpc_fix_t *fvscanstart; 850 jpc_fix_t *dvscanstart; 851 852 *nmsedec = 0; 853 width = jas_matrix_numcols(data); 854 height = jas_matrix_numrows(data); 855 frowstep = jas_matrix_rowstep(flags); 856 drowstep = jas_matrix_rowstep(data); 857 fstripestep = frowstep << 2; 858 dstripestep = drowstep << 2; 859 860 one = 1 << (bitpos + JPC_NUMEXTRABITS); 861 862 fstripestart = jas_matrix_getref(flags, 1, 1); 863 dstripestart = jas_matrix_getref(data, 0, 0); 864 for (i = height; i > 0; i -= 4, fstripestart += fstripestep, 865 dstripestart += dstripestep) { 866 fvscanstart = fstripestart; 867 dvscanstart = dstripestart; 868 vscanlen = JAS_MIN(i, 4); 869 for (j = width; j > 0; --j, ++fvscanstart, ++dvscanstart) { 870 871 fp = fvscanstart; 872 if (vscanlen >= 4 && !((*fp) & (JPC_SIG | JPC_VISIT | 873 JPC_OTHSIGMSK)) && (fp += frowstep, !((*fp) & (JPC_SIG | 874 JPC_VISIT | JPC_OTHSIGMSK))) && (fp += frowstep, !((*fp) & 875 (JPC_SIG | JPC_VISIT | JPC_OTHSIGMSK))) && (fp += frowstep, 876 !((*fp) & (JPC_SIG | JPC_VISIT | JPC_OTHSIGMSK)))) { 877 dp = dvscanstart; 878 for (k = 0; k < vscanlen; ++k) { 879 v = (abs(*dp) & one) ? 1 : 0; 880 if (v) { 881 break; 882 } 883 dp += drowstep; 884 } 885 runlen = k; 886 if (runlen >= 4) { 887 jpc_mqenc_setcurctx(mqenc, JPC_AGGCTXNO); 888 jpc_mqenc_putbit(mqenc, 0); 889 continue; 890 } 891 jpc_mqenc_setcurctx(mqenc, JPC_AGGCTXNO); 892 jpc_mqenc_putbit(mqenc, 1); 893 jpc_mqenc_setcurctx(mqenc, JPC_UCTXNO); 894 jpc_mqenc_putbit(mqenc, runlen >> 1); 895 jpc_mqenc_putbit(mqenc, runlen & 1); 896 fp = fvscanstart + frowstep * runlen; 897 dp = dvscanstart + drowstep * runlen; 898 k = vscanlen - runlen; 899 switch (runlen) { 900 case 0: 901 goto clnpass_partial0; 902 break; 903 case 1: 904 goto clnpass_partial1; 905 break; 906 case 2: 907 goto clnpass_partial2; 908 break; 909 case 3: 910 goto clnpass_partial3; 911 break; 912 } 913 } else { 914 runlen = 0; 915 fp = fvscanstart; 916 dp = dvscanstart; 917 k = vscanlen; 918 goto clnpass_full0; 919 } 920 clnpass_step(fp, frowstep, dp, bitpos, one, 921 orient, nmsedec, mqenc, clnpass_full0:, clnpass_partial0:, vcausalflag); 922 if (--k <= 0) { 923 continue; 924 } 925 fp += frowstep; 926 dp += drowstep; 927 clnpass_step(fp, frowstep, dp, bitpos, one, 928 orient, nmsedec, mqenc, ;, clnpass_partial1:, 0); 929 if (--k <= 0) { 930 continue; 931 } 932 fp += frowstep; 933 dp += drowstep; 934 clnpass_step(fp, frowstep, dp, bitpos, one, 935 orient, nmsedec, mqenc, ;, clnpass_partial2:, 0); 936 if (--k <= 0) { 937 continue; 938 } 939 fp += frowstep; 940 dp += drowstep; 941 clnpass_step(fp, frowstep, dp, bitpos, one, 942 orient, nmsedec, mqenc, ;, clnpass_partial3:, 0); 943 } 944 } 945 946 if (segsymflag) { 947 jpc_mqenc_setcurctx(mqenc, JPC_UCTXNO); 948 jpc_mqenc_putbit(mqenc, 1); 949 jpc_mqenc_putbit(mqenc, 0); 950 jpc_mqenc_putbit(mqenc, 1); 951 jpc_mqenc_putbit(mqenc, 0); 952 } 953 954 if (term) { 955 jpc_mqenc_flush(mqenc, term - 1); 956 } 957 958 return jpc_mqenc_error(mqenc) ? (-1) : 0; 959 } 960
