Subversion Repositories Kolibri OS

Rev

Blame | Last modification | View Log | RSS feed

  1. /*
  2.  * Copyright (c) 2002-2007, Communications and Remote Sensing Laboratory, Universite catholique de Louvain (UCL), Belgium
  3.  * Copyright (c) 2002-2007, Professor Benoit Macq
  4.  * Copyright (c) 2001-2003, David Janssens
  5.  * Copyright (c) 2002-2003, Yannick Verschueren
  6.  * Copyright (c) 2003-2007, Francois-Olivier Devaux and Antonin Descampe
  7.  * Copyright (c) 2005, Herve Drolon, FreeImage Team
  8.  * All rights reserved.
  9.  *
  10.  * Redistribution and use in source and binary forms, with or without
  11.  * modification, are permitted provided that the following conditions
  12.  * are met:
  13.  * 1. Redistributions of source code must retain the above copyright
  14.  *    notice, this list of conditions and the following disclaimer.
  15.  * 2. Redistributions in binary form must reproduce the above copyright
  16.  *    notice, this list of conditions and the following disclaimer in the
  17.  *    documentation and/or other materials provided with the distribution.
  18.  *
  19.  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
  20.  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  23.  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24.  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25.  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26.  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27.  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28.  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29.  * POSSIBILITY OF SUCH DAMAGE.
  30.  */
  31.  
  32. #include "opj_includes.h"
  33.  
  34. /** @defgroup T2 T2 - Implementation of a tier-2 coding */
  35. /*@{*/
  36.  
  37. /** @name Local static functions */
  38. /*@{*/
  39.  
  40. static void t2_putcommacode(opj_bio_t *bio, int n);
  41. static int t2_getcommacode(opj_bio_t *bio);
  42. /**
  43. Variable length code for signalling delta Zil (truncation point)
  44. @param bio Bit Input/Output component
  45. @param n delta Zil
  46. */
  47. static void t2_putnumpasses(opj_bio_t *bio, int n);
  48. static int t2_getnumpasses(opj_bio_t *bio);
  49. /**
  50. Encode a packet of a tile to a destination buffer
  51. @param tile Tile for which to write the packets
  52. @param tcp Tile coding parameters
  53. @param pi Packet identity
  54. @param dest Destination buffer
  55. @param len Length of the destination buffer
  56. @param cstr_info Codestream information structure
  57. @param tileno Number of the tile encoded
  58. @return
  59. */
  60. static int t2_encode_packet(opj_tcd_tile_t *tile, opj_tcp_t *tcp, opj_pi_iterator_t *pi, unsigned char *dest, int len, opj_codestream_info_t *cstr_info, int tileno);
  61. /**
  62. @param cblk
  63. @param index
  64. @param cblksty
  65. @param first
  66. */
  67. static void t2_init_seg(opj_tcd_cblk_dec_t* cblk, int index, int cblksty, int first);
  68. /**
  69. Decode a packet of a tile from a source buffer
  70. @param t2 T2 handle
  71. @param src Source buffer
  72. @param len Length of the source buffer
  73. @param tile Tile for which to write the packets
  74. @param tcp Tile coding parameters
  75. @param pi Packet identity
  76. @param pack_info Packet information
  77. @return
  78. */
  79. static int t2_decode_packet(opj_t2_t* t2, unsigned char *src, int len, opj_tcd_tile_t *tile,
  80.                                                                                                                 opj_tcp_t *tcp, opj_pi_iterator_t *pi, opj_packet_info_t *pack_info);
  81.  
  82. /*@}*/
  83.  
  84. /*@}*/
  85.  
  86. /* ----------------------------------------------------------------------- */
  87.  
  88. /* #define RESTART 0x04 */
  89.  
  90. static void t2_putcommacode(opj_bio_t *bio, int n) {
  91.         while (--n >= 0) {
  92.                 bio_write(bio, 1, 1);
  93.         }
  94.         bio_write(bio, 0, 1);
  95. }
  96.  
  97. static int t2_getcommacode(opj_bio_t *bio) {
  98.         int n;
  99.         for (n = 0; bio_read(bio, 1); n++) {
  100.                 ;
  101.         }
  102.         return n;
  103. }
  104.  
  105. static void t2_putnumpasses(opj_bio_t *bio, int n) {
  106.         if (n == 1) {
  107.                 bio_write(bio, 0, 1);
  108.         } else if (n == 2) {
  109.                 bio_write(bio, 2, 2);
  110.         } else if (n <= 5) {
  111.                 bio_write(bio, 0xc | (n - 3), 4);
  112.         } else if (n <= 36) {
  113.                 bio_write(bio, 0x1e0 | (n - 6), 9);
  114.         } else if (n <= 164) {
  115.                 bio_write(bio, 0xff80 | (n - 37), 16);
  116.         }
  117. }
  118.  
  119. static int t2_getnumpasses(opj_bio_t *bio) {
  120.         int n;
  121.         if (!bio_read(bio, 1))
  122.                 return 1;
  123.         if (!bio_read(bio, 1))
  124.                 return 2;
  125.         if ((n = bio_read(bio, 2)) != 3)
  126.                 return (3 + n);
  127.         if ((n = bio_read(bio, 5)) != 31)
  128.                 return (6 + n);
  129.         return (37 + bio_read(bio, 7));
  130. }
  131.  
  132. static int t2_encode_packet(opj_tcd_tile_t * tile, opj_tcp_t * tcp, opj_pi_iterator_t *pi, unsigned char *dest, int length, opj_codestream_info_t *cstr_info, int tileno) {
  133.         int bandno, cblkno;
  134.         unsigned char *c = dest;
  135.  
  136.         int compno = pi->compno;        /* component value */
  137.         int resno  = pi->resno;         /* resolution level value */
  138.         int precno = pi->precno;        /* precinct value */
  139.         int layno  = pi->layno;         /* quality layer value */
  140.  
  141.         opj_tcd_tilecomp_t *tilec = &tile->comps[compno];
  142.         opj_tcd_resolution_t *res = &tilec->resolutions[resno];
  143.        
  144.         opj_bio_t *bio = NULL;  /* BIO component */
  145.        
  146.         /* <SOP 0xff91> */
  147.         if (tcp->csty & J2K_CP_CSTY_SOP) {
  148.                 c[0] = 255;
  149.                 c[1] = 145;
  150.                 c[2] = 0;
  151.                 c[3] = 4;
  152.                 c[4] = (tile->packno % 65536) / 256;
  153.                 c[5] = (tile->packno % 65536) % 256;
  154.                 c += 6;
  155.         }
  156.         /* </SOP> */
  157.        
  158.         if (!layno) {
  159.                 for (bandno = 0; bandno < res->numbands; bandno++) {
  160.                         opj_tcd_band_t *band = &res->bands[bandno];
  161.                         opj_tcd_precinct_t *prc = &band->precincts[precno];
  162.                         tgt_reset(prc->incltree);
  163.                         tgt_reset(prc->imsbtree);
  164.                         for (cblkno = 0; cblkno < prc->cw * prc->ch; cblkno++) {
  165.                                 opj_tcd_cblk_enc_t* cblk = &prc->cblks.enc[cblkno];
  166.                                 cblk->numpasses = 0;
  167.                                 tgt_setvalue(prc->imsbtree, cblkno, band->numbps - cblk->numbps);
  168.                         }
  169.                 }
  170.         }
  171.        
  172.         bio = bio_create();
  173.         bio_init_enc(bio, c, length);
  174.         bio_write(bio, 1, 1);           /* Empty header bit */
  175.        
  176.         /* Writing Packet header */
  177.         for (bandno = 0; bandno < res->numbands; bandno++) {
  178.                 opj_tcd_band_t *band = &res->bands[bandno];
  179.                 opj_tcd_precinct_t *prc = &band->precincts[precno];
  180.                 for (cblkno = 0; cblkno < prc->cw * prc->ch; cblkno++) {
  181.                         opj_tcd_cblk_enc_t* cblk = &prc->cblks.enc[cblkno];
  182.                         opj_tcd_layer_t *layer = &cblk->layers[layno];
  183.                         if (!cblk->numpasses && layer->numpasses) {
  184.                                 tgt_setvalue(prc->incltree, cblkno, layno);
  185.                         }
  186.                 }
  187.                 for (cblkno = 0; cblkno < prc->cw * prc->ch; cblkno++) {
  188.                         opj_tcd_cblk_enc_t* cblk = &prc->cblks.enc[cblkno];
  189.                         opj_tcd_layer_t *layer = &cblk->layers[layno];
  190.                         int increment = 0;
  191.                         int nump = 0;
  192.                         int len = 0, passno;
  193.                         /* cblk inclusion bits */
  194.                         if (!cblk->numpasses) {
  195.                                 tgt_encode(bio, prc->incltree, cblkno, layno + 1);
  196.                         } else {
  197.                                 bio_write(bio, layer->numpasses != 0, 1);
  198.                         }
  199.                         /* if cblk not included, go to the next cblk  */
  200.                         if (!layer->numpasses) {
  201.                                 continue;
  202.                         }
  203.                         /* if first instance of cblk --> zero bit-planes information */
  204.                         if (!cblk->numpasses) {
  205.                                 cblk->numlenbits = 3;
  206.                                 tgt_encode(bio, prc->imsbtree, cblkno, 999);
  207.                         }
  208.                         /* number of coding passes included */
  209.                         t2_putnumpasses(bio, layer->numpasses);
  210.                        
  211.                         /* computation of the increase of the length indicator and insertion in the header     */
  212.                         for (passno = cblk->numpasses; passno < cblk->numpasses + layer->numpasses; passno++) {
  213.                                 opj_tcd_pass_t *pass = &cblk->passes[passno];
  214.                                 nump++;
  215.                                 len += pass->len;
  216.                                 if (pass->term || passno == (cblk->numpasses + layer->numpasses) - 1) {
  217.                                         increment = int_max(increment, int_floorlog2(len) + 1 - (cblk->numlenbits + int_floorlog2(nump)));
  218.                                         len = 0;
  219.                                         nump = 0;
  220.                                 }
  221.                         }
  222.                         t2_putcommacode(bio, increment);
  223.  
  224.                         /* computation of the new Length indicator */
  225.                         cblk->numlenbits += increment;
  226.  
  227.                         /* insertion of the codeword segment length */
  228.                         for (passno = cblk->numpasses; passno < cblk->numpasses + layer->numpasses; passno++) {
  229.                                 opj_tcd_pass_t *pass = &cblk->passes[passno];
  230.                                 nump++;
  231.                                 len += pass->len;
  232.                                 if (pass->term || passno == (cblk->numpasses + layer->numpasses) - 1) {
  233.                                         bio_write(bio, len, cblk->numlenbits + int_floorlog2(nump));
  234.                                         len = 0;
  235.                                         nump = 0;
  236.                                 }
  237.                         }
  238.                 }
  239.         }
  240.  
  241.         if (bio_flush(bio)) {
  242.                 bio_destroy(bio);
  243.                 return -999;            /* modified to eliminate longjmp !! */
  244.         }
  245.  
  246.         c += bio_numbytes(bio);
  247.         bio_destroy(bio);
  248.        
  249.         /* <EPH 0xff92> */
  250.         if (tcp->csty & J2K_CP_CSTY_EPH) {
  251.                 c[0] = 255;
  252.                 c[1] = 146;
  253.                 c += 2;
  254.         }
  255.         /* </EPH> */
  256.  
  257.         /* << INDEX */
  258.         // End of packet header position. Currently only represents the distance to start of packet
  259.         // Will be updated later by incrementing with packet start value
  260.         if(cstr_info && cstr_info->index_write) {
  261.                 opj_packet_info_t *info_PK = &cstr_info->tile[tileno].packet[cstr_info->packno];
  262.                 info_PK->end_ph_pos = (int)(c - dest);
  263.         }
  264.         /* INDEX >> */
  265.        
  266.         /* Writing the packet body */
  267.        
  268.         for (bandno = 0; bandno < res->numbands; bandno++) {
  269.                 opj_tcd_band_t *band = &res->bands[bandno];
  270.                 opj_tcd_precinct_t *prc = &band->precincts[precno];
  271.                 for (cblkno = 0; cblkno < prc->cw * prc->ch; cblkno++) {
  272.                         opj_tcd_cblk_enc_t* cblk = &prc->cblks.enc[cblkno];
  273.                         opj_tcd_layer_t *layer = &cblk->layers[layno];
  274.                         if (!layer->numpasses) {
  275.                                 continue;
  276.                         }
  277.                         if (c + layer->len > dest + length) {
  278.                                 return -999;
  279.                         }
  280.                        
  281.                         memcpy(c, layer->data, layer->len);
  282.                         cblk->numpasses += layer->numpasses;
  283.                         c += layer->len;
  284.                         /* << INDEX */
  285.                         if(cstr_info && cstr_info->index_write) {
  286.                                 opj_packet_info_t *info_PK = &cstr_info->tile[tileno].packet[cstr_info->packno];
  287.                                 info_PK->disto += layer->disto;
  288.                                 if (cstr_info->D_max < info_PK->disto) {
  289.                                         cstr_info->D_max = info_PK->disto;
  290.                                 }
  291.                         }
  292.                         /* INDEX >> */
  293.                 }
  294.         }
  295.        
  296.         return (c - dest);
  297. }
  298.  
  299. static void t2_init_seg(opj_tcd_cblk_dec_t* cblk, int index, int cblksty, int first) {
  300.         opj_tcd_seg_t* seg;
  301.         cblk->segs = (opj_tcd_seg_t*) opj_realloc(cblk->segs, (index + 1) * sizeof(opj_tcd_seg_t));
  302.         seg = &cblk->segs[index];
  303.         seg->data = NULL;
  304.         seg->dataindex = 0;
  305.         seg->numpasses = 0;
  306.         seg->len = 0;
  307.         if (cblksty & J2K_CCP_CBLKSTY_TERMALL) {
  308.                 seg->maxpasses = 1;
  309.         }
  310.         else if (cblksty & J2K_CCP_CBLKSTY_LAZY) {
  311.                 if (first) {
  312.                         seg->maxpasses = 10;
  313.                 } else {
  314.                         seg->maxpasses = (((seg - 1)->maxpasses == 1) || ((seg - 1)->maxpasses == 10)) ? 2 : 1;
  315.                 }
  316.         } else {
  317.                 seg->maxpasses = 109;
  318.         }
  319. }
  320.  
  321. static int t2_decode_packet(opj_t2_t* t2, unsigned char *src, int len, opj_tcd_tile_t *tile,
  322.                                                                                                                 opj_tcp_t *tcp, opj_pi_iterator_t *pi, opj_packet_info_t *pack_info) {
  323.         int bandno, cblkno;
  324.         unsigned char *c = src;
  325.  
  326.         opj_cp_t *cp = t2->cp;
  327.  
  328.         int compno = pi->compno;        /* component value */
  329.         int resno  = pi->resno;         /* resolution level value */
  330.         int precno = pi->precno;        /* precinct value */
  331.         int layno  = pi->layno;         /* quality layer value */
  332.  
  333.         opj_tcd_resolution_t* res = &tile->comps[compno].resolutions[resno];
  334.  
  335.         unsigned char *hd = NULL;
  336.         int present;
  337.        
  338.         opj_bio_t *bio = NULL;  /* BIO component */
  339.        
  340.         if (layno == 0) {
  341.                 for (bandno = 0; bandno < res->numbands; bandno++) {
  342.                         opj_tcd_band_t *band = &res->bands[bandno];
  343.                         opj_tcd_precinct_t *prc = &band->precincts[precno];
  344.                        
  345.                         if ((band->x1-band->x0 == 0)||(band->y1-band->y0 == 0)) continue;
  346.                        
  347.                         tgt_reset(prc->incltree);
  348.                         tgt_reset(prc->imsbtree);
  349.                         for (cblkno = 0; cblkno < prc->cw * prc->ch; cblkno++) {
  350.                                 opj_tcd_cblk_dec_t* cblk = &prc->cblks.dec[cblkno];
  351.                                 cblk->numsegs = 0;
  352.                         }
  353.                 }
  354.         }
  355.        
  356.         /* SOP markers */
  357.        
  358.         if (tcp->csty & J2K_CP_CSTY_SOP) {
  359.                 if ((*c) != 0xff || (*(c + 1) != 0x91)) {
  360.                         opj_event_msg(t2->cinfo, EVT_WARNING, "Expected SOP marker\n");
  361.                 } else {
  362.                         c += 6;
  363.                 }
  364.                
  365.                 /** TODO : check the Nsop value */
  366.         }
  367.        
  368.         /*
  369.         When the marker PPT/PPM is used the packet header are store in PPT/PPM marker
  370.         This part deal with this caracteristic
  371.         step 1: Read packet header in the saved structure
  372.         step 2: Return to codestream for decoding
  373.         */
  374.  
  375.         bio = bio_create();
  376.        
  377.         if (cp->ppm == 1) {             /* PPM */
  378.                 hd = cp->ppm_data;
  379.                 bio_init_dec(bio, hd, cp->ppm_len);
  380.         } else if (tcp->ppt == 1) {     /* PPT */
  381.                 hd = tcp->ppt_data;
  382.                 bio_init_dec(bio, hd, tcp->ppt_len);
  383.         } else {                        /* Normal Case */
  384.                 hd = c;
  385.                 bio_init_dec(bio, hd, src+len-hd);
  386.         }
  387.        
  388.         present = bio_read(bio, 1);
  389.        
  390.         if (!present) {
  391.                 bio_inalign(bio);
  392.                 hd += bio_numbytes(bio);
  393.                 bio_destroy(bio);
  394.                
  395.                 /* EPH markers */
  396.                
  397.                 if (tcp->csty & J2K_CP_CSTY_EPH) {
  398.                         if ((*hd) != 0xff || (*(hd + 1) != 0x92)) {
  399.                                 printf("Error : expected EPH marker\n");
  400.                         } else {
  401.                                 hd += 2;
  402.                         }
  403.                 }
  404.  
  405.                 /* << INDEX */
  406.                 // End of packet header position. Currently only represents the distance to start of packet
  407.                 // Will be updated later by incrementing with packet start value
  408.                 if(pack_info) {
  409.                         pack_info->end_ph_pos = (int)(c - src);
  410.                 }
  411.                 /* INDEX >> */
  412.                
  413.                 if (cp->ppm == 1) {             /* PPM case */
  414.                         cp->ppm_len += cp->ppm_data-hd;
  415.                         cp->ppm_data = hd;
  416.                         return (c - src);
  417.                 }
  418.                 if (tcp->ppt == 1) {    /* PPT case */
  419.                         tcp->ppt_len+=tcp->ppt_data-hd;
  420.                         tcp->ppt_data = hd;
  421.                         return (c - src);
  422.                 }
  423.                
  424.                 return (hd - src);
  425.         }
  426.        
  427.         for (bandno = 0; bandno < res->numbands; bandno++) {
  428.                 opj_tcd_band_t *band = &res->bands[bandno];
  429.                 opj_tcd_precinct_t *prc = &band->precincts[precno];
  430.                
  431.                 if ((band->x1-band->x0 == 0)||(band->y1-band->y0 == 0)) continue;
  432.                
  433.                 for (cblkno = 0; cblkno < prc->cw * prc->ch; cblkno++) {
  434.                         int included, increment, n, segno;
  435.                         opj_tcd_cblk_dec_t* cblk = &prc->cblks.dec[cblkno];
  436.                         /* if cblk not yet included before --> inclusion tagtree */
  437.                         if (!cblk->numsegs) {
  438.                                 included = tgt_decode(bio, prc->incltree, cblkno, layno + 1);
  439.                                 /* else one bit */
  440.                         } else {
  441.                                 included = bio_read(bio, 1);
  442.                         }
  443.                         /* if cblk not included */
  444.                         if (!included) {
  445.                                 cblk->numnewpasses = 0;
  446.                                 continue;
  447.                         }
  448.                         /* if cblk not yet included --> zero-bitplane tagtree */
  449.                         if (!cblk->numsegs) {
  450.                                 int i, numimsbs;
  451.                                 for (i = 0; !tgt_decode(bio, prc->imsbtree, cblkno, i); i++) {
  452.                                         ;
  453.                                 }
  454.                                 numimsbs = i - 1;
  455.                                 cblk->numbps = band->numbps - numimsbs;
  456.                                 cblk->numlenbits = 3;
  457.                         }
  458.                         /* number of coding passes */
  459.                         cblk->numnewpasses = t2_getnumpasses(bio);
  460.                         increment = t2_getcommacode(bio);
  461.                         /* length indicator increment */
  462.                         cblk->numlenbits += increment;
  463.                         segno = 0;
  464.                         if (!cblk->numsegs) {
  465.                                 t2_init_seg(cblk, segno, tcp->tccps[compno].cblksty, 1);
  466.                         } else {
  467.                                 segno = cblk->numsegs - 1;
  468.                                 if (cblk->segs[segno].numpasses == cblk->segs[segno].maxpasses) {
  469.                                         ++segno;
  470.                                         t2_init_seg(cblk, segno, tcp->tccps[compno].cblksty, 0);
  471.                                 }
  472.                         }
  473.                         n = cblk->numnewpasses;
  474.                        
  475.                         do {
  476.                                 cblk->segs[segno].numnewpasses = int_min(cblk->segs[segno].maxpasses - cblk->segs[segno].numpasses, n);
  477.                                 cblk->segs[segno].newlen = bio_read(bio, cblk->numlenbits + int_floorlog2(cblk->segs[segno].numnewpasses));
  478.                                 n -= cblk->segs[segno].numnewpasses;
  479.                                 if (n > 0) {
  480.                                         ++segno;
  481.                                         t2_init_seg(cblk, segno, tcp->tccps[compno].cblksty, 0);
  482.                                 }
  483.                         } while (n > 0);
  484.                 }
  485.         }
  486.        
  487.         if (bio_inalign(bio)) {
  488.                 bio_destroy(bio);
  489.                 return -999;
  490.         }
  491.        
  492.         hd += bio_numbytes(bio);
  493.         bio_destroy(bio);
  494.        
  495.         /* EPH markers */
  496.         if (tcp->csty & J2K_CP_CSTY_EPH) {
  497.                 if ((*hd) != 0xff || (*(hd + 1) != 0x92)) {
  498.                         opj_event_msg(t2->cinfo, EVT_ERROR, "Expected EPH marker\n");
  499.                         return -999;
  500.                 } else {
  501.                         hd += 2;
  502.                 }
  503.         }
  504.  
  505.         /* << INDEX */
  506.         // End of packet header position. Currently only represents the distance to start of packet
  507.         // Will be updated later by incrementing with packet start value
  508.         if(pack_info) {
  509.                 pack_info->end_ph_pos = (int)(hd - src);
  510.         }
  511.         /* INDEX >> */
  512.        
  513.         if (cp->ppm==1) {
  514.                 cp->ppm_len+=cp->ppm_data-hd;
  515.                 cp->ppm_data = hd;
  516.         } else if (tcp->ppt == 1) {
  517.                 tcp->ppt_len+=tcp->ppt_data-hd;
  518.                 tcp->ppt_data = hd;
  519.         } else {
  520.                 c=hd;
  521.         }
  522.        
  523.         for (bandno = 0; bandno < res->numbands; bandno++) {
  524.                 opj_tcd_band_t *band = &res->bands[bandno];
  525.                 opj_tcd_precinct_t *prc = &band->precincts[precno];
  526.                
  527.                 if ((band->x1-band->x0 == 0)||(band->y1-band->y0 == 0)) continue;
  528.                
  529.                 for (cblkno = 0; cblkno < prc->cw * prc->ch; cblkno++) {
  530.                         opj_tcd_cblk_dec_t* cblk = &prc->cblks.dec[cblkno];
  531.                         opj_tcd_seg_t *seg = NULL;
  532.                         if (!cblk->numnewpasses)
  533.                                 continue;
  534.                         if (!cblk->numsegs) {
  535.                                 seg = &cblk->segs[0];
  536.                                 cblk->numsegs++;
  537.                                 cblk->len = 0;
  538.                         } else {
  539.                                 seg = &cblk->segs[cblk->numsegs - 1];
  540.                                 if (seg->numpasses == seg->maxpasses) {
  541.                                         seg++;
  542.                                         cblk->numsegs++;
  543.                                 }
  544.                         }
  545.                        
  546.                         do {
  547.                                 if (c + seg->newlen > src + len) {
  548.                                         return -999;
  549.                                 }
  550.  
  551. #ifdef USE_JPWL
  552.                         /* we need here a j2k handle to verify if making a check to
  553.                         the validity of cblocks parameters is selected from user (-W) */
  554.  
  555.                                 /* let's check that we are not exceeding */
  556.                                 if ((cblk->len + seg->newlen) > 8192) {
  557.                                         opj_event_msg(t2->cinfo, EVT_WARNING,
  558.                                                 "JPWL: segment too long (%d) for codeblock %d (p=%d, b=%d, r=%d, c=%d)\n",
  559.                                                 seg->newlen, cblkno, precno, bandno, resno, compno);
  560.                                         if (!JPWL_ASSUME) {
  561.                                                 opj_event_msg(t2->cinfo, EVT_ERROR, "JPWL: giving up\n");
  562.                                                 return -999;
  563.                                         }
  564.                                         seg->newlen = 8192 - cblk->len;
  565.                                         opj_event_msg(t2->cinfo, EVT_WARNING, "      - truncating segment to %d\n", seg->newlen);
  566.                                         break;
  567.                                 };
  568.  
  569. #endif /* USE_JPWL */
  570.                                
  571.                                 cblk->data = (unsigned char*) opj_realloc(cblk->data, (cblk->len + seg->newlen) * sizeof(unsigned char*));
  572.                                 memcpy(cblk->data + cblk->len, c, seg->newlen);
  573.                                 if (seg->numpasses == 0) {
  574.                                         seg->data = &cblk->data;
  575.                                         seg->dataindex = cblk->len;
  576.                                 }
  577.                                 c += seg->newlen;
  578.                                 cblk->len += seg->newlen;
  579.                                 seg->len += seg->newlen;
  580.                                 seg->numpasses += seg->numnewpasses;
  581.                                 cblk->numnewpasses -= seg->numnewpasses;
  582.                                 if (cblk->numnewpasses > 0) {
  583.                                         seg++;
  584.                                         cblk->numsegs++;
  585.                                 }
  586.                         } while (cblk->numnewpasses > 0);
  587.                 }
  588.         }
  589.        
  590.         return (c - src);
  591. }
  592.  
  593. /* ----------------------------------------------------------------------- */
  594.  
  595. int t2_encode_packets(opj_t2_t* t2,int tileno, opj_tcd_tile_t *tile, int maxlayers, unsigned char *dest, int len, opj_codestream_info_t *cstr_info,int tpnum, int tppos,int pino, J2K_T2_MODE t2_mode, int cur_totnum_tp){
  596.         unsigned char *c = dest;
  597.         int e = 0;
  598.         int compno;
  599.         opj_pi_iterator_t *pi = NULL;
  600.         int poc;
  601.         opj_image_t *image = t2->image;
  602.         opj_cp_t *cp = t2->cp;
  603.         opj_tcp_t *tcp = &cp->tcps[tileno];
  604.         int pocno = cp->cinema == CINEMA4K_24? 2: 1;
  605.         int maxcomp = cp->max_comp_size > 0 ? image->numcomps : 1;
  606.        
  607.         pi = pi_initialise_encode(image, cp, tileno, t2_mode);
  608.         if(!pi) {
  609.                 /* TODO: throw an error */
  610.                 return -999;
  611.         }
  612.        
  613.         if(t2_mode == THRESH_CALC ){ /* Calculating threshold */
  614.                 for(compno = 0; compno < maxcomp; compno++ ){
  615.                         for(poc = 0; poc < pocno ; poc++){
  616.                                 int comp_len = 0;
  617.                                 int tpnum = compno;
  618.                                 if (pi_create_encode(pi, cp,tileno,poc,tpnum,tppos,t2_mode,cur_totnum_tp)) {
  619.                                         opj_event_msg(t2->cinfo, EVT_ERROR, "Error initializing Packet Iterator\n");
  620.                                         pi_destroy(pi, cp, tileno);
  621.                                         return -999;
  622.                                 }
  623.                                 while (pi_next(&pi[poc])) {
  624.                                         if (pi[poc].layno < maxlayers) {
  625.                                                 e = t2_encode_packet(tile, &cp->tcps[tileno], &pi[poc], c, dest + len - c, cstr_info, tileno);
  626.                                                 comp_len = comp_len + e;
  627.                                                 if (e == -999) {
  628.                                                         break;
  629.                                                 } else {
  630.                                                         c += e;
  631.                                                 }
  632.                                         }
  633.                                 }
  634.                                 if (e == -999) break;
  635.                                 if (cp->max_comp_size){
  636.                                         if (comp_len > cp->max_comp_size){
  637.                                                 e = -999;
  638.                                                 break;
  639.                                         }
  640.                                 }
  641.                         }
  642.                         if (e == -999)  break;
  643.                 }
  644.         }else{  /* t2_mode == FINAL_PASS  */
  645.                 pi_create_encode(pi, cp,tileno,pino,tpnum,tppos,t2_mode,cur_totnum_tp);
  646.                 while (pi_next(&pi[pino])) {
  647.                         if (pi[pino].layno < maxlayers) {
  648.                                 e = t2_encode_packet(tile, &cp->tcps[tileno], &pi[pino], c, dest + len - c, cstr_info, tileno);
  649.                                 if (e == -999) {
  650.                                         break;
  651.                                 } else {
  652.                                         c += e;
  653.                                 }
  654.                                 /* INDEX >> */
  655.                                 if(cstr_info) {
  656.                                         if(cstr_info->index_write) {
  657.                                                 opj_tile_info_t *info_TL = &cstr_info->tile[tileno];
  658.                                                 opj_packet_info_t *info_PK = &info_TL->packet[cstr_info->packno];
  659.                                                 if (!cstr_info->packno) {
  660.                                                         info_PK->start_pos = info_TL->end_header + 1;
  661.                                                 } else {
  662.                                                         info_PK->start_pos = ((cp->tp_on | tcp->POC)&& info_PK->start_pos) ? info_PK->start_pos : info_TL->packet[cstr_info->packno - 1].end_pos + 1;
  663.                                                 }
  664.                                                 info_PK->end_pos = info_PK->start_pos + e - 1;
  665.                                                 info_PK->end_ph_pos += info_PK->start_pos - 1;  // End of packet header which now only represents the distance
  666.                                                                                                                                                                                                                                                 // to start of packet is incremented by value of start of packet
  667.                                         }
  668.                                        
  669.                                         cstr_info->packno++;
  670.                                 }
  671.                                 /* << INDEX */
  672.                                 tile->packno++;
  673.                         }
  674.                 }
  675.         }
  676.        
  677.         pi_destroy(pi, cp, tileno);
  678.        
  679.         if (e == -999) {
  680.                 return e;
  681.         }
  682.        
  683.   return (c - dest);
  684. }
  685.  
  686. int t2_decode_packets(opj_t2_t *t2, unsigned char *src, int len, int tileno, opj_tcd_tile_t *tile, opj_codestream_info_t *cstr_info) {
  687.         unsigned char *c = src;
  688.         opj_pi_iterator_t *pi;
  689.         int pino, e = 0;
  690.         int n = 0, curtp = 0;
  691.         int tp_start_packno;
  692.  
  693.         opj_image_t *image = t2->image;
  694.         opj_cp_t *cp = t2->cp;
  695.        
  696.         /* create a packet iterator */
  697.         pi = pi_create_decode(image, cp, tileno);
  698.         if(!pi) {
  699.                 /* TODO: throw an error */
  700.                 return -999;
  701.         }
  702.  
  703.         tp_start_packno = 0;
  704.        
  705.         for (pino = 0; pino <= cp->tcps[tileno].numpocs; pino++) {
  706.                 while (pi_next(&pi[pino])) {
  707.                         if ((cp->layer==0) || (cp->layer>=((pi[pino].layno)+1))) {
  708.                                 opj_packet_info_t *pack_info;
  709.                                 if (cstr_info)
  710.                                         pack_info = &cstr_info->tile[tileno].packet[cstr_info->packno];
  711.                                 else
  712.                                         pack_info = NULL;
  713.                                 e = t2_decode_packet(t2, c, src + len - c, tile, &cp->tcps[tileno], &pi[pino], pack_info);
  714.                         } else {
  715.                                 e = 0;
  716.                         }
  717.                         if(e == -999) return -999;
  718.                         /* progression in resolution */
  719.                         image->comps[pi[pino].compno].resno_decoded =  
  720.                                 (e > 0) ?
  721.                                 int_max(pi[pino].resno, image->comps[pi[pino].compno].resno_decoded)
  722.                                 : image->comps[pi[pino].compno].resno_decoded;
  723.                         n++;
  724.  
  725.                         /* INDEX >> */
  726.                         if(cstr_info) {
  727.                                 opj_tile_info_t *info_TL = &cstr_info->tile[tileno];
  728.                                 opj_packet_info_t *info_PK = &info_TL->packet[cstr_info->packno];
  729.                                 if (!cstr_info->packno) {
  730.                                         info_PK->start_pos = info_TL->end_header + 1;
  731.                                 } else if (info_TL->packet[cstr_info->packno-1].end_pos >= (int)cstr_info->tile[tileno].tp[curtp].tp_end_pos){ // New tile part
  732.                                         info_TL->tp[curtp].tp_numpacks = cstr_info->packno - tp_start_packno; // Number of packets in previous tile-part
  733.                                         tp_start_packno = cstr_info->packno;
  734.                                         curtp++;
  735.                                         info_PK->start_pos = cstr_info->tile[tileno].tp[curtp].tp_end_header+1;
  736.                                 } else {
  737.                                         info_PK->start_pos = (cp->tp_on && info_PK->start_pos) ? info_PK->start_pos : info_TL->packet[cstr_info->packno - 1].end_pos + 1;
  738.                                 }
  739.                                 info_PK->end_pos = info_PK->start_pos + e - 1;
  740.                                 info_PK->end_ph_pos += info_PK->start_pos - 1;  // End of packet header which now only represents the distance
  741.                                                                                                                                                                                                                                 // to start of packet is incremented by value of start of packet
  742.                                 cstr_info->packno++;
  743.                         }
  744.                         /* << INDEX */
  745.                        
  746.                         if (e == -999) {                /* ADD */
  747.                                 break;
  748.                         } else {
  749.                                 c += e;
  750.                         }                      
  751.                 }
  752.         }
  753.         /* INDEX >> */
  754.         if(cstr_info) {
  755.                 cstr_info->tile[tileno].tp[curtp].tp_numpacks = cstr_info->packno - tp_start_packno; // Number of packets in last tile-part
  756.         }
  757.         /* << INDEX */
  758.  
  759.         /* don't forget to release pi */
  760.         pi_destroy(pi, cp, tileno);
  761.        
  762.         if (e == -999) {
  763.                 return e;
  764.         }
  765.        
  766.         return (c - src);
  767. }
  768.  
  769. /* ----------------------------------------------------------------------- */
  770.  
  771. opj_t2_t* t2_create(opj_common_ptr cinfo, opj_image_t *image, opj_cp_t *cp) {
  772.         /* create the tcd structure */
  773.         opj_t2_t *t2 = (opj_t2_t*)opj_malloc(sizeof(opj_t2_t));
  774.         if(!t2) return NULL;
  775.         t2->cinfo = cinfo;
  776.         t2->image = image;
  777.         t2->cp = cp;
  778.  
  779.         return t2;
  780. }
  781.  
  782. void t2_destroy(opj_t2_t *t2) {
  783.         if(t2) {
  784.                 opj_free(t2);
  785.         }
  786. }
  787.  
  788.  
  789.  
  790.  
  791.  
  792.