yading@10: /* yading@10: * JPEG2000 image decoder yading@10: * Copyright (c) 2007 Kamil Nowosad yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * JPEG2000 image decoder yading@10: * @file yading@10: * @author Kamil Nowosad yading@10: */ yading@10: yading@10: // #define DEBUG yading@10: yading@10: #include "avcodec.h" yading@10: #include "bytestream.h" yading@10: #include "internal.h" yading@10: #include "j2k.h" yading@10: #include "libavutil/common.h" yading@10: yading@10: #define JP2_SIG_TYPE 0x6A502020 yading@10: #define JP2_SIG_VALUE 0x0D0A870A yading@10: #define JP2_CODESTREAM 0x6A703263 yading@10: yading@10: #define HAD_COC 0x01 yading@10: #define HAD_QCC 0x02 yading@10: yading@10: typedef struct { yading@10: J2kComponent *comp; yading@10: uint8_t properties[4]; yading@10: J2kCodingStyle codsty[4]; yading@10: J2kQuantStyle qntsty[4]; yading@10: } J2kTile; yading@10: yading@10: typedef struct { yading@10: AVCodecContext *avctx; yading@10: AVFrame *picture; yading@10: GetByteContext g; yading@10: yading@10: int width, height; ///< image width and height yading@10: int image_offset_x, image_offset_y; yading@10: int tile_offset_x, tile_offset_y; yading@10: uint8_t cbps[4]; ///< bits per sample in particular components yading@10: uint8_t sgnd[4]; ///< if a component is signed yading@10: uint8_t properties[4]; yading@10: int cdx[4], cdy[4]; yading@10: int precision; yading@10: int ncomponents; yading@10: int tile_width, tile_height; ///< tile size yading@10: int numXtiles, numYtiles; yading@10: int maxtilelen; yading@10: yading@10: J2kCodingStyle codsty[4]; yading@10: J2kQuantStyle qntsty[4]; yading@10: yading@10: int bit_index; yading@10: yading@10: int curtileno; yading@10: yading@10: J2kTile *tile; yading@10: } J2kDecoderContext; yading@10: yading@10: static int get_bits(J2kDecoderContext *s, int n) yading@10: { yading@10: int res = 0; yading@10: yading@10: while (--n >= 0){ yading@10: res <<= 1; yading@10: if (s->bit_index == 0) { yading@10: s->bit_index = 7 + (bytestream2_get_byte(&s->g) != 0xFFu); yading@10: } yading@10: s->bit_index--; yading@10: res |= (bytestream2_peek_byte(&s->g) >> s->bit_index) & 1; yading@10: } yading@10: return res; yading@10: } yading@10: yading@10: static void j2k_flush(J2kDecoderContext *s) yading@10: { yading@10: if (bytestream2_get_byte(&s->g) == 0xff) yading@10: bytestream2_skip(&s->g, 1); yading@10: s->bit_index = 8; yading@10: } yading@10: #if 0 yading@10: void printcomp(J2kComponent *comp) yading@10: { yading@10: int i; yading@10: for (i = 0; i < comp->y1 - comp->y0; i++) yading@10: ff_j2k_printv(comp->data + i * (comp->x1 - comp->x0), comp->x1 - comp->x0); yading@10: } yading@10: yading@10: static void nspaces(FILE *fd, int n) yading@10: { yading@10: while(n--) putc(' ', fd); yading@10: } yading@10: yading@10: static void dump(J2kDecoderContext *s, FILE *fd) yading@10: { yading@10: int tileno, compno, reslevelno, bandno, precno; yading@10: fprintf(fd, "XSiz = %d, YSiz = %d, tile_width = %d, tile_height = %d\n" yading@10: "numXtiles = %d, numYtiles = %d, ncomponents = %d\n" yading@10: "tiles:\n", yading@10: s->width, s->height, s->tile_width, s->tile_height, yading@10: s->numXtiles, s->numYtiles, s->ncomponents); yading@10: for (tileno = 0; tileno < s->numXtiles * s->numYtiles; tileno++){ yading@10: J2kTile *tile = s->tile + tileno; yading@10: nspaces(fd, 2); yading@10: fprintf(fd, "tile %d:\n", tileno); yading@10: for(compno = 0; compno < s->ncomponents; compno++){ yading@10: J2kComponent *comp = tile->comp + compno; yading@10: nspaces(fd, 4); yading@10: fprintf(fd, "component %d:\n", compno); yading@10: nspaces(fd, 4); yading@10: fprintf(fd, "x0 = %d, x1 = %d, y0 = %d, y1 = %d\n", yading@10: comp->x0, comp->x1, comp->y0, comp->y1); yading@10: for(reslevelno = 0; reslevelno < codsty->nreslevels; reslevelno++){ yading@10: J2kResLevel *reslevel = comp->reslevel + reslevelno; yading@10: nspaces(fd, 6); yading@10: fprintf(fd, "reslevel %d:\n", reslevelno); yading@10: nspaces(fd, 6); yading@10: fprintf(fd, "x0 = %d, x1 = %d, y0 = %d, y1 = %d, nbands = %d\n", yading@10: reslevel->x0, reslevel->x1, reslevel->y0, yading@10: reslevel->y1, reslevel->nbands); yading@10: for(bandno = 0; bandno < reslevel->nbands; bandno++){ yading@10: J2kBand *band = reslevel->band + bandno; yading@10: nspaces(fd, 8); yading@10: fprintf(fd, "band %d:\n", bandno); yading@10: nspaces(fd, 8); yading@10: fprintf(fd, "x0 = %d, x1 = %d, y0 = %d, y1 = %d," yading@10: "codeblock_width = %d, codeblock_height = %d cblknx = %d cblkny = %d\n", yading@10: band->x0, band->x1, yading@10: band->y0, band->y1, yading@10: band->codeblock_width, band->codeblock_height, yading@10: band->cblknx, band->cblkny); yading@10: for (precno = 0; precno < reslevel->num_precincts_x * reslevel->num_precincts_y; precno++){ yading@10: J2kPrec *prec = band->prec + precno; yading@10: nspaces(fd, 10); yading@10: fprintf(fd, "prec %d:\n", precno); yading@10: nspaces(fd, 10); yading@10: fprintf(fd, "xi0 = %d, xi1 = %d, yi0 = %d, yi1 = %d\n", yading@10: prec->xi0, prec->xi1, prec->yi0, prec->yi1); yading@10: } yading@10: } yading@10: } yading@10: } yading@10: } yading@10: } yading@10: #endif yading@10: yading@10: /** decode the value stored in node */ yading@10: static int tag_tree_decode(J2kDecoderContext *s, J2kTgtNode *node, int threshold) yading@10: { yading@10: J2kTgtNode *stack[30]; yading@10: int sp = -1, curval = 0; yading@10: yading@10: if(!node) yading@10: return AVERROR(EINVAL); yading@10: yading@10: while(node && !node->vis){ yading@10: stack[++sp] = node; yading@10: node = node->parent; yading@10: } yading@10: yading@10: if (node) yading@10: curval = node->val; yading@10: else yading@10: curval = stack[sp]->val; yading@10: yading@10: while(curval < threshold && sp >= 0){ yading@10: if (curval < stack[sp]->val) yading@10: curval = stack[sp]->val; yading@10: while (curval < threshold){ yading@10: int ret; yading@10: if ((ret = get_bits(s, 1)) > 0){ yading@10: stack[sp]->vis++; yading@10: break; yading@10: } else if (!ret) yading@10: curval++; yading@10: else yading@10: return ret; yading@10: } yading@10: stack[sp]->val = curval; yading@10: sp--; yading@10: } yading@10: return curval; yading@10: } yading@10: yading@10: /* marker segments */ yading@10: /** get sizes and offsets of image, tiles; number of components */ yading@10: static int get_siz(J2kDecoderContext *s) yading@10: { yading@10: int i, ret; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 36) yading@10: return AVERROR(EINVAL); yading@10: yading@10: bytestream2_get_be16u(&s->g); // Rsiz (skipped) yading@10: s->width = bytestream2_get_be32u(&s->g); // width yading@10: s->height = bytestream2_get_be32u(&s->g); // height yading@10: s->image_offset_x = bytestream2_get_be32u(&s->g); // X0Siz yading@10: s->image_offset_y = bytestream2_get_be32u(&s->g); // Y0Siz yading@10: yading@10: s->tile_width = bytestream2_get_be32u(&s->g); // XTSiz yading@10: s->tile_height = bytestream2_get_be32u(&s->g); // YTSiz yading@10: s->tile_offset_x = bytestream2_get_be32u(&s->g); // XT0Siz yading@10: s->tile_offset_y = bytestream2_get_be32u(&s->g); // YT0Siz yading@10: s->ncomponents = bytestream2_get_be16u(&s->g); // CSiz yading@10: yading@10: if(s->ncomponents <= 0 || s->ncomponents > 4) { yading@10: av_log(s->avctx, AV_LOG_ERROR, "unsupported/invalid ncomponents: %d\n", s->ncomponents); yading@10: return AVERROR(EINVAL); yading@10: } yading@10: if(s->tile_width<=0 || s->tile_height<=0) yading@10: return AVERROR(EINVAL); yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 3 * s->ncomponents) yading@10: return AVERROR(EINVAL); yading@10: yading@10: for (i = 0; i < s->ncomponents; i++){ // Ssiz_i XRsiz_i, YRsiz_i yading@10: uint8_t x = bytestream2_get_byteu(&s->g); yading@10: s->cbps[i] = (x & 0x7f) + 1; yading@10: s->precision = FFMAX(s->cbps[i], s->precision); yading@10: s->sgnd[i] = !!(x & 0x80); yading@10: s->cdx[i] = bytestream2_get_byteu(&s->g); yading@10: s->cdy[i] = bytestream2_get_byteu(&s->g); yading@10: } yading@10: yading@10: s->numXtiles = ff_j2k_ceildiv(s->width - s->tile_offset_x, s->tile_width); yading@10: s->numYtiles = ff_j2k_ceildiv(s->height - s->tile_offset_y, s->tile_height); yading@10: yading@10: if(s->numXtiles * (uint64_t)s->numYtiles > INT_MAX/sizeof(J2kTile)) yading@10: return AVERROR(EINVAL); yading@10: yading@10: s->tile = av_mallocz(s->numXtiles * s->numYtiles * sizeof(J2kTile)); yading@10: if (!s->tile) yading@10: return AVERROR(ENOMEM); yading@10: yading@10: for (i = 0; i < s->numXtiles * s->numYtiles; i++){ yading@10: J2kTile *tile = s->tile + i; yading@10: yading@10: tile->comp = av_mallocz(s->ncomponents * sizeof(J2kComponent)); yading@10: if (!tile->comp) yading@10: return AVERROR(ENOMEM); yading@10: } yading@10: yading@10: s->avctx->width = s->width - s->image_offset_x; yading@10: s->avctx->height = s->height - s->image_offset_y; yading@10: yading@10: switch(s->ncomponents){ yading@10: case 1: yading@10: if (s->precision > 8) { yading@10: s->avctx->pix_fmt = AV_PIX_FMT_GRAY16; yading@10: } else { yading@10: s->avctx->pix_fmt = AV_PIX_FMT_GRAY8; yading@10: } yading@10: break; yading@10: case 3: yading@10: if (s->precision > 8) { yading@10: s->avctx->pix_fmt = AV_PIX_FMT_RGB48; yading@10: } else { yading@10: s->avctx->pix_fmt = AV_PIX_FMT_RGB24; yading@10: } yading@10: break; yading@10: case 4: yading@10: s->avctx->pix_fmt = AV_PIX_FMT_RGBA; yading@10: break; yading@10: } yading@10: yading@10: yading@10: if ((ret = ff_get_buffer(s->avctx, s->picture, 0)) < 0) yading@10: return ret; yading@10: yading@10: s->picture->pict_type = AV_PICTURE_TYPE_I; yading@10: s->picture->key_frame = 1; yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: /** get common part for COD and COC segments */ yading@10: static int get_cox(J2kDecoderContext *s, J2kCodingStyle *c) yading@10: { yading@10: if (bytestream2_get_bytes_left(&s->g) < 5) yading@10: return AVERROR(EINVAL); yading@10: c->nreslevels = bytestream2_get_byteu(&s->g) + 1; // num of resolution levels - 1 yading@10: c->log2_cblk_width = bytestream2_get_byteu(&s->g) + 2; // cblk width yading@10: c->log2_cblk_height = bytestream2_get_byteu(&s->g) + 2; // cblk height yading@10: yading@10: c->cblk_style = bytestream2_get_byteu(&s->g); yading@10: if (c->cblk_style != 0){ // cblk style yading@10: av_log(s->avctx, AV_LOG_WARNING, "extra cblk styles %X\n", c->cblk_style); yading@10: } yading@10: c->transform = bytestream2_get_byteu(&s->g); // transformation yading@10: if (c->csty & J2K_CSTY_PREC) { yading@10: int i; yading@10: yading@10: for (i = 0; i < c->nreslevels; i++) yading@10: bytestream2_get_byte(&s->g); yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: /** get coding parameters for a particular tile or whole image*/ yading@10: static int get_cod(J2kDecoderContext *s, J2kCodingStyle *c, uint8_t *properties) yading@10: { yading@10: J2kCodingStyle tmp; yading@10: int compno; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 5) yading@10: return AVERROR(EINVAL); yading@10: yading@10: tmp.log2_prec_width = yading@10: tmp.log2_prec_height = 15; yading@10: yading@10: tmp.csty = bytestream2_get_byteu(&s->g); yading@10: yading@10: if (bytestream2_get_byteu(&s->g)){ // progression level yading@10: av_log(s->avctx, AV_LOG_ERROR, "only LRCP progression supported\n"); yading@10: return -1; yading@10: } yading@10: yading@10: tmp.nlayers = bytestream2_get_be16u(&s->g); yading@10: tmp.mct = bytestream2_get_byteu(&s->g); // multiple component transformation yading@10: yading@10: get_cox(s, &tmp); yading@10: for (compno = 0; compno < s->ncomponents; compno++){ yading@10: if (!(properties[compno] & HAD_COC)) yading@10: memcpy(c + compno, &tmp, sizeof(J2kCodingStyle)); yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: /** get coding parameters for a component in the whole image on a particular tile */ yading@10: static int get_coc(J2kDecoderContext *s, J2kCodingStyle *c, uint8_t *properties) yading@10: { yading@10: int compno; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 2) yading@10: return AVERROR(EINVAL); yading@10: yading@10: compno = bytestream2_get_byteu(&s->g); yading@10: yading@10: c += compno; yading@10: c->csty = bytestream2_get_byte(&s->g); yading@10: get_cox(s, c); yading@10: yading@10: properties[compno] |= HAD_COC; yading@10: return 0; yading@10: } yading@10: yading@10: /** get common part for QCD and QCC segments */ yading@10: static int get_qcx(J2kDecoderContext *s, int n, J2kQuantStyle *q) yading@10: { yading@10: int i, x; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 1) yading@10: return AVERROR(EINVAL); yading@10: yading@10: x = bytestream2_get_byteu(&s->g); // Sqcd yading@10: yading@10: q->nguardbits = x >> 5; yading@10: q->quantsty = x & 0x1f; yading@10: yading@10: if (q->quantsty == J2K_QSTY_NONE){ yading@10: n -= 3; yading@10: if (bytestream2_get_bytes_left(&s->g) < n || 32*3 < n) yading@10: return AVERROR(EINVAL); yading@10: for (i = 0; i < n; i++) yading@10: q->expn[i] = bytestream2_get_byteu(&s->g) >> 3; yading@10: } else if (q->quantsty == J2K_QSTY_SI){ yading@10: if (bytestream2_get_bytes_left(&s->g) < 2) yading@10: return AVERROR(EINVAL); yading@10: x = bytestream2_get_be16u(&s->g); yading@10: q->expn[0] = x >> 11; yading@10: q->mant[0] = x & 0x7ff; yading@10: for (i = 1; i < 32 * 3; i++){ yading@10: int curexpn = FFMAX(0, q->expn[0] - (i-1)/3); yading@10: q->expn[i] = curexpn; yading@10: q->mant[i] = q->mant[0]; yading@10: } yading@10: } else{ yading@10: n = (n - 3) >> 1; yading@10: if (bytestream2_get_bytes_left(&s->g) < 2 * n || 32*3 < n) yading@10: return AVERROR(EINVAL); yading@10: for (i = 0; i < n; i++){ yading@10: x = bytestream2_get_be16u(&s->g); yading@10: q->expn[i] = x >> 11; yading@10: q->mant[i] = x & 0x7ff; yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: /** get quantization parameters for a particular tile or a whole image */ yading@10: static int get_qcd(J2kDecoderContext *s, int n, J2kQuantStyle *q, uint8_t *properties) yading@10: { yading@10: J2kQuantStyle tmp; yading@10: int compno; yading@10: yading@10: if (get_qcx(s, n, &tmp)) yading@10: return -1; yading@10: for (compno = 0; compno < s->ncomponents; compno++) yading@10: if (!(properties[compno] & HAD_QCC)) yading@10: memcpy(q + compno, &tmp, sizeof(J2kQuantStyle)); yading@10: return 0; yading@10: } yading@10: yading@10: /** get quantization parameters for a component in the whole image on in a particular tile */ yading@10: static int get_qcc(J2kDecoderContext *s, int n, J2kQuantStyle *q, uint8_t *properties) yading@10: { yading@10: int compno; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 1) yading@10: return AVERROR(EINVAL); yading@10: yading@10: compno = bytestream2_get_byteu(&s->g); yading@10: properties[compno] |= HAD_QCC; yading@10: return get_qcx(s, n-1, q+compno); yading@10: } yading@10: yading@10: /** get start of tile segment */ yading@10: static uint8_t get_sot(J2kDecoderContext *s) yading@10: { yading@10: if (bytestream2_get_bytes_left(&s->g) < 8) yading@10: return AVERROR(EINVAL); yading@10: yading@10: s->curtileno = bytestream2_get_be16u(&s->g); ///< Isot yading@10: if((unsigned)s->curtileno >= s->numXtiles * s->numYtiles){ yading@10: s->curtileno=0; yading@10: return AVERROR(EINVAL); yading@10: } yading@10: yading@10: bytestream2_skipu(&s->g, 4); ///< Psot (ignored) yading@10: yading@10: if (!bytestream2_get_byteu(&s->g)){ ///< TPsot yading@10: J2kTile *tile = s->tile + s->curtileno; yading@10: yading@10: /* copy defaults */ yading@10: memcpy(tile->codsty, s->codsty, s->ncomponents * sizeof(J2kCodingStyle)); yading@10: memcpy(tile->qntsty, s->qntsty, s->ncomponents * sizeof(J2kQuantStyle)); yading@10: } yading@10: bytestream2_get_byteu(&s->g); ///< TNsot yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static int init_tile(J2kDecoderContext *s, int tileno) yading@10: { yading@10: int compno, yading@10: tilex = tileno % s->numXtiles, yading@10: tiley = tileno / s->numXtiles; yading@10: J2kTile *tile = s->tile + tileno; yading@10: yading@10: if (!tile->comp) yading@10: return AVERROR(ENOMEM); yading@10: for (compno = 0; compno < s->ncomponents; compno++){ yading@10: J2kComponent *comp = tile->comp + compno; yading@10: J2kCodingStyle *codsty = tile->codsty + compno; yading@10: J2kQuantStyle *qntsty = tile->qntsty + compno; yading@10: int ret; // global bandno yading@10: yading@10: comp->coord[0][0] = FFMAX(tilex * s->tile_width + s->tile_offset_x, s->image_offset_x); yading@10: comp->coord[0][1] = FFMIN((tilex+1)*s->tile_width + s->tile_offset_x, s->width); yading@10: comp->coord[1][0] = FFMAX(tiley * s->tile_height + s->tile_offset_y, s->image_offset_y); yading@10: comp->coord[1][1] = FFMIN((tiley+1)*s->tile_height + s->tile_offset_y, s->height); yading@10: yading@10: if (ret = ff_j2k_init_component(comp, codsty, qntsty, s->cbps[compno], s->cdx[compno], s->cdy[compno])) yading@10: return ret; yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: /** read the number of coding passes */ yading@10: static int getnpasses(J2kDecoderContext *s) yading@10: { yading@10: int num; yading@10: if (!get_bits(s, 1)) yading@10: return 1; yading@10: if (!get_bits(s, 1)) yading@10: return 2; yading@10: if ((num = get_bits(s, 2)) != 3) yading@10: return num < 0 ? num : 3 + num; yading@10: if ((num = get_bits(s, 5)) != 31) yading@10: return num < 0 ? num : 6 + num; yading@10: num = get_bits(s, 7); yading@10: return num < 0 ? num : 37 + num; yading@10: } yading@10: yading@10: static int getlblockinc(J2kDecoderContext *s) yading@10: { yading@10: int res = 0, ret; yading@10: while (ret = get_bits(s, 1)){ yading@10: if (ret < 0) yading@10: return ret; yading@10: res++; yading@10: } yading@10: return res; yading@10: } yading@10: yading@10: static int decode_packet(J2kDecoderContext *s, J2kCodingStyle *codsty, J2kResLevel *rlevel, int precno, yading@10: int layno, uint8_t *expn, int numgbits) yading@10: { yading@10: int bandno, cblkny, cblknx, cblkno, ret; yading@10: yading@10: if (!(ret = get_bits(s, 1))){ yading@10: j2k_flush(s); yading@10: return 0; yading@10: } else if (ret < 0) yading@10: return ret; yading@10: yading@10: for (bandno = 0; bandno < rlevel->nbands; bandno++){ yading@10: J2kBand *band = rlevel->band + bandno; yading@10: J2kPrec *prec = band->prec + precno; yading@10: int pos = 0; yading@10: yading@10: if (band->coord[0][0] == band->coord[0][1] yading@10: || band->coord[1][0] == band->coord[1][1]) yading@10: continue; yading@10: yading@10: for (cblkny = prec->yi0; cblkny < prec->yi1; cblkny++) yading@10: for(cblknx = prec->xi0, cblkno = cblkny * band->cblknx + cblknx; cblknx < prec->xi1; cblknx++, cblkno++, pos++){ yading@10: J2kCblk *cblk = band->cblk + cblkno; yading@10: int incl, newpasses, llen; yading@10: yading@10: if (cblk->npasses) yading@10: incl = get_bits(s, 1); yading@10: else yading@10: incl = tag_tree_decode(s, prec->cblkincl + pos, layno+1) == layno; yading@10: if (!incl) yading@10: continue; yading@10: else if (incl < 0) yading@10: return incl; yading@10: yading@10: if (!cblk->npasses) yading@10: cblk->nonzerobits = expn[bandno] + numgbits - 1 - tag_tree_decode(s, prec->zerobits + pos, 100); yading@10: if ((newpasses = getnpasses(s)) < 0) yading@10: return newpasses; yading@10: if ((llen = getlblockinc(s)) < 0) yading@10: return llen; yading@10: cblk->lblock += llen; yading@10: if ((ret = get_bits(s, av_log2(newpasses) + cblk->lblock)) < 0) yading@10: return ret; yading@10: cblk->lengthinc = ret; yading@10: cblk->npasses += newpasses; yading@10: } yading@10: } yading@10: j2k_flush(s); yading@10: yading@10: if (codsty->csty & J2K_CSTY_EPH) { yading@10: if (bytestream2_peek_be16(&s->g) == J2K_EPH) { yading@10: bytestream2_skip(&s->g, 2); yading@10: } else { yading@10: av_log(s->avctx, AV_LOG_ERROR, "EPH marker not found.\n"); yading@10: } yading@10: } yading@10: yading@10: for (bandno = 0; bandno < rlevel->nbands; bandno++){ yading@10: J2kBand *band = rlevel->band + bandno; yading@10: int yi, cblknw = band->prec[precno].xi1 - band->prec[precno].xi0; yading@10: for (yi = band->prec[precno].yi0; yi < band->prec[precno].yi1; yi++){ yading@10: int xi; yading@10: for (xi = band->prec[precno].xi0; xi < band->prec[precno].xi1; xi++){ yading@10: J2kCblk *cblk = band->cblk + yi * cblknw + xi; yading@10: if (bytestream2_get_bytes_left(&s->g) < cblk->lengthinc) yading@10: return AVERROR(EINVAL); yading@10: bytestream2_get_bufferu(&s->g, cblk->data, cblk->lengthinc); yading@10: cblk->length += cblk->lengthinc; yading@10: cblk->lengthinc = 0; yading@10: } yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: static int decode_packets(J2kDecoderContext *s, J2kTile *tile) yading@10: { yading@10: int layno, reslevelno, compno, precno, ok_reslevel; yading@10: s->bit_index = 8; yading@10: for (layno = 0; layno < tile->codsty[0].nlayers; layno++){ yading@10: ok_reslevel = 1; yading@10: for (reslevelno = 0; ok_reslevel; reslevelno++){ yading@10: ok_reslevel = 0; yading@10: for (compno = 0; compno < s->ncomponents; compno++){ yading@10: J2kCodingStyle *codsty = tile->codsty + compno; yading@10: J2kQuantStyle *qntsty = tile->qntsty + compno; yading@10: if (reslevelno < codsty->nreslevels){ yading@10: J2kResLevel *rlevel = tile->comp[compno].reslevel + reslevelno; yading@10: ok_reslevel = 1; yading@10: for (precno = 0; precno < rlevel->num_precincts_x * rlevel->num_precincts_y; precno++){ yading@10: if (decode_packet(s, codsty, rlevel, precno, layno, qntsty->expn + yading@10: (reslevelno ? 3*(reslevelno-1)+1 : 0), qntsty->nguardbits)) yading@10: return -1; yading@10: } yading@10: } yading@10: } yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: /* TIER-1 routines */ yading@10: static void decode_sigpass(J2kT1Context *t1, int width, int height, int bpno, int bandno, int bpass_csty_symbol, yading@10: int vert_causal_ctx_csty_symbol) yading@10: { yading@10: int mask = 3 << (bpno - 1), y0, x, y; yading@10: yading@10: for (y0 = 0; y0 < height; y0 += 4) yading@10: for (x = 0; x < width; x++) yading@10: for (y = y0; y < height && y < y0+4; y++){ yading@10: if ((t1->flags[y+1][x+1] & J2K_T1_SIG_NB) yading@10: && !(t1->flags[y+1][x+1] & (J2K_T1_SIG | J2K_T1_VIS))){ yading@10: int vert_causal_ctx_csty_loc_symbol = vert_causal_ctx_csty_symbol && (x == 3 && y == 3); yading@10: if (ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + ff_j2k_getnbctxno(t1->flags[y+1][x+1], bandno, yading@10: vert_causal_ctx_csty_loc_symbol))){ yading@10: int xorbit, ctxno = ff_j2k_getsgnctxno(t1->flags[y+1][x+1], &xorbit); yading@10: if (bpass_csty_symbol) yading@10: t1->data[y][x] = ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + ctxno) ? -mask : mask; yading@10: else yading@10: t1->data[y][x] = (ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + ctxno) ^ xorbit) ? yading@10: -mask : mask; yading@10: yading@10: ff_j2k_set_significant(t1, x, y, t1->data[y][x] < 0); yading@10: } yading@10: t1->flags[y+1][x+1] |= J2K_T1_VIS; yading@10: } yading@10: } yading@10: } yading@10: yading@10: static void decode_refpass(J2kT1Context *t1, int width, int height, int bpno) yading@10: { yading@10: int phalf, nhalf; yading@10: int y0, x, y; yading@10: yading@10: phalf = 1 << (bpno - 1); yading@10: nhalf = -phalf; yading@10: yading@10: for (y0 = 0; y0 < height; y0 += 4) yading@10: for (x = 0; x < width; x++) yading@10: for (y = y0; y < height && y < y0+4; y++){ yading@10: if ((t1->flags[y+1][x+1] & (J2K_T1_SIG | J2K_T1_VIS)) == J2K_T1_SIG){ yading@10: int ctxno = ff_j2k_getrefctxno(t1->flags[y+1][x+1]); yading@10: int r = ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + ctxno) ? phalf : nhalf; yading@10: t1->data[y][x] += t1->data[y][x] < 0 ? -r : r; yading@10: t1->flags[y+1][x+1] |= J2K_T1_REF; yading@10: } yading@10: } yading@10: } yading@10: yading@10: static void decode_clnpass(J2kDecoderContext *s, J2kT1Context *t1, int width, int height, yading@10: int bpno, int bandno, int seg_symbols) yading@10: { yading@10: int mask = 3 << (bpno - 1), y0, x, y, runlen, dec; yading@10: yading@10: for (y0 = 0; y0 < height; y0 += 4) { yading@10: for (x = 0; x < width; x++){ yading@10: if (y0 + 3 < height && !( yading@10: (t1->flags[y0+1][x+1] & (J2K_T1_SIG_NB | J2K_T1_VIS | J2K_T1_SIG)) || yading@10: (t1->flags[y0+2][x+1] & (J2K_T1_SIG_NB | J2K_T1_VIS | J2K_T1_SIG)) || yading@10: (t1->flags[y0+3][x+1] & (J2K_T1_SIG_NB | J2K_T1_VIS | J2K_T1_SIG)) || yading@10: (t1->flags[y0+4][x+1] & (J2K_T1_SIG_NB | J2K_T1_VIS | J2K_T1_SIG)))){ yading@10: if (!ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + MQC_CX_RL)) yading@10: continue; yading@10: runlen = ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + MQC_CX_UNI); yading@10: runlen = (runlen << 1) | ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + MQC_CX_UNI); yading@10: dec = 1; yading@10: } else{ yading@10: runlen = 0; yading@10: dec = 0; yading@10: } yading@10: yading@10: for (y = y0 + runlen; y < y0 + 4 && y < height; y++){ yading@10: if (!dec){ yading@10: if (!(t1->flags[y+1][x+1] & (J2K_T1_SIG | J2K_T1_VIS))) yading@10: dec = ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + ff_j2k_getnbctxno(t1->flags[y+1][x+1], yading@10: bandno, 0)); yading@10: } yading@10: if (dec){ yading@10: int xorbit, ctxno = ff_j2k_getsgnctxno(t1->flags[y+1][x+1], &xorbit); yading@10: t1->data[y][x] = (ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + ctxno) ^ xorbit) ? -mask : mask; yading@10: ff_j2k_set_significant(t1, x, y, t1->data[y][x] < 0); yading@10: } yading@10: dec = 0; yading@10: t1->flags[y+1][x+1] &= ~J2K_T1_VIS; yading@10: } yading@10: } yading@10: } yading@10: if (seg_symbols) { yading@10: int val; yading@10: val = ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + MQC_CX_UNI); yading@10: val = (val << 1) + ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + MQC_CX_UNI); yading@10: val = (val << 1) + ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + MQC_CX_UNI); yading@10: val = (val << 1) + ff_mqc_decode(&t1->mqc, t1->mqc.cx_states + MQC_CX_UNI); yading@10: if (val != 0xa) { yading@10: av_log(s->avctx, AV_LOG_ERROR,"Segmentation symbol value incorrect\n"); yading@10: } yading@10: } yading@10: } yading@10: yading@10: static int decode_cblk(J2kDecoderContext *s, J2kCodingStyle *codsty, J2kT1Context *t1, J2kCblk *cblk, yading@10: int width, int height, int bandpos) yading@10: { yading@10: int passno = cblk->npasses, pass_t = 2, bpno = cblk->nonzerobits - 1, y, clnpass_cnt = 0; yading@10: int bpass_csty_symbol = J2K_CBLK_BYPASS & codsty->cblk_style; yading@10: int vert_causal_ctx_csty_symbol = J2K_CBLK_VSC & codsty->cblk_style; yading@10: yading@10: for (y = 0; y < height+2; y++) yading@10: memset(t1->flags[y], 0, (width+2)*sizeof(int)); yading@10: yading@10: for (y = 0; y < height; y++) yading@10: memset(t1->data[y], 0, width*sizeof(int)); yading@10: yading@10: cblk->data[cblk->length] = 0xff; yading@10: cblk->data[cblk->length+1] = 0xff; yading@10: ff_mqc_initdec(&t1->mqc, cblk->data); yading@10: yading@10: while(passno--){ yading@10: switch(pass_t){ yading@10: case 0: decode_sigpass(t1, width, height, bpno+1, bandpos, yading@10: bpass_csty_symbol && (clnpass_cnt >= 4), vert_causal_ctx_csty_symbol); yading@10: break; yading@10: case 1: decode_refpass(t1, width, height, bpno+1); yading@10: if (bpass_csty_symbol && clnpass_cnt >= 4) yading@10: ff_mqc_initdec(&t1->mqc, cblk->data); yading@10: break; yading@10: case 2: decode_clnpass(s, t1, width, height, bpno+1, bandpos, yading@10: codsty->cblk_style & J2K_CBLK_SEGSYM); yading@10: clnpass_cnt = clnpass_cnt + 1; yading@10: if (bpass_csty_symbol && clnpass_cnt >= 4) yading@10: ff_mqc_initdec(&t1->mqc, cblk->data); yading@10: break; yading@10: } yading@10: yading@10: pass_t++; yading@10: if (pass_t == 3){ yading@10: bpno--; yading@10: pass_t = 0; yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: static void mct_decode(J2kDecoderContext *s, J2kTile *tile) yading@10: { yading@10: int i, *src[3], i0, i1, i2, csize = 1; yading@10: yading@10: for (i = 0; i < 3; i++) yading@10: src[i] = tile->comp[i].data; yading@10: yading@10: for (i = 0; i < 2; i++) yading@10: csize *= tile->comp[0].coord[i][1] - tile->comp[0].coord[i][0]; yading@10: yading@10: if (tile->codsty[0].transform == FF_DWT97){ yading@10: for (i = 0; i < csize; i++){ yading@10: i0 = *src[0] + (*src[2] * 46802 >> 16); yading@10: i1 = *src[0] - (*src[1] * 22553 + *src[2] * 46802 >> 16); yading@10: i2 = *src[0] + (116130 * *src[1] >> 16); yading@10: *src[0]++ = i0; yading@10: *src[1]++ = i1; yading@10: *src[2]++ = i2; yading@10: } yading@10: } else{ yading@10: for (i = 0; i < csize; i++){ yading@10: i1 = *src[0] - (*src[2] + *src[1] >> 2); yading@10: i0 = i1 + *src[2]; yading@10: i2 = i1 + *src[1]; yading@10: *src[0]++ = i0; yading@10: *src[1]++ = i1; yading@10: *src[2]++ = i2; yading@10: } yading@10: } yading@10: } yading@10: yading@10: static int decode_tile(J2kDecoderContext *s, J2kTile *tile) yading@10: { yading@10: int compno, reslevelno, bandno; yading@10: int x, y, *src[4]; yading@10: uint8_t *line; yading@10: J2kT1Context t1; yading@10: yading@10: for (compno = 0; compno < s->ncomponents; compno++){ yading@10: J2kComponent *comp = tile->comp + compno; yading@10: J2kCodingStyle *codsty = tile->codsty + compno; yading@10: yading@10: for (reslevelno = 0; reslevelno < codsty->nreslevels; reslevelno++){ yading@10: J2kResLevel *rlevel = comp->reslevel + reslevelno; yading@10: for (bandno = 0; bandno < rlevel->nbands; bandno++){ yading@10: J2kBand *band = rlevel->band + bandno; yading@10: int cblkx, cblky, cblkno=0, xx0, x0, xx1, y0, yy0, yy1, bandpos; yading@10: yading@10: bandpos = bandno + (reslevelno > 0); yading@10: yading@10: yy0 = bandno == 0 ? 0 : comp->reslevel[reslevelno-1].coord[1][1] - comp->reslevel[reslevelno-1].coord[1][0]; yading@10: y0 = yy0; yading@10: yy1 = FFMIN(ff_j2k_ceildiv(band->coord[1][0] + 1, band->codeblock_height) * band->codeblock_height, yading@10: band->coord[1][1]) - band->coord[1][0] + yy0; yading@10: yading@10: if (band->coord[0][0] == band->coord[0][1] || band->coord[1][0] == band->coord[1][1]) yading@10: continue; yading@10: yading@10: for (cblky = 0; cblky < band->cblkny; cblky++){ yading@10: if (reslevelno == 0 || bandno == 1) yading@10: xx0 = 0; yading@10: else yading@10: xx0 = comp->reslevel[reslevelno-1].coord[0][1] - comp->reslevel[reslevelno-1].coord[0][0]; yading@10: x0 = xx0; yading@10: xx1 = FFMIN(ff_j2k_ceildiv(band->coord[0][0] + 1, band->codeblock_width) * band->codeblock_width, yading@10: band->coord[0][1]) - band->coord[0][0] + xx0; yading@10: yading@10: for (cblkx = 0; cblkx < band->cblknx; cblkx++, cblkno++){ yading@10: int y, x; yading@10: decode_cblk(s, codsty, &t1, band->cblk + cblkno, xx1 - xx0, yy1 - yy0, bandpos); yading@10: if (codsty->transform == FF_DWT53){ yading@10: for (y = yy0; y < yy1; y+=s->cdy[compno]){ yading@10: int *ptr = t1.data[y-yy0]; yading@10: for (x = xx0; x < xx1; x+=s->cdx[compno]){ yading@10: comp->data[(comp->coord[0][1] - comp->coord[0][0]) * y + x] = *ptr++ >> 1; yading@10: } yading@10: } yading@10: } else{ yading@10: for (y = yy0; y < yy1; y+=s->cdy[compno]){ yading@10: int *ptr = t1.data[y-yy0]; yading@10: for (x = xx0; x < xx1; x+=s->cdx[compno]){ yading@10: int tmp = ((int64_t)*ptr++) * ((int64_t)band->stepsize) >> 13, tmp2; yading@10: tmp2 = FFABS(tmp>>1) + (tmp&1); yading@10: comp->data[(comp->coord[0][1] - comp->coord[0][0]) * y + x] = tmp < 0 ? -tmp2 : tmp2; yading@10: } yading@10: } yading@10: } yading@10: xx0 = xx1; yading@10: xx1 = FFMIN(xx1 + band->codeblock_width, band->coord[0][1] - band->coord[0][0] + x0); yading@10: } yading@10: yy0 = yy1; yading@10: yy1 = FFMIN(yy1 + band->codeblock_height, band->coord[1][1] - band->coord[1][0] + y0); yading@10: } yading@10: } yading@10: } yading@10: ff_j2k_dwt_decode(&comp->dwt, comp->data); yading@10: src[compno] = comp->data; yading@10: } yading@10: if (tile->codsty[0].mct) yading@10: mct_decode(s, tile); yading@10: yading@10: if (s->precision <= 8) { yading@10: for (compno = 0; compno < s->ncomponents; compno++){ yading@10: y = tile->comp[compno].coord[1][0] - s->image_offset_y; yading@10: line = s->picture->data[0] + y * s->picture->linesize[0]; yading@10: for (; y < tile->comp[compno].coord[1][1] - s->image_offset_y; y += s->cdy[compno]){ yading@10: uint8_t *dst; yading@10: yading@10: x = tile->comp[compno].coord[0][0] - s->image_offset_x; yading@10: dst = line + x * s->ncomponents + compno; yading@10: yading@10: for (; x < tile->comp[compno].coord[0][1] - s->image_offset_x; x += s->cdx[compno]) { yading@10: *src[compno] += 1 << (s->cbps[compno]-1); yading@10: if (*src[compno] < 0) yading@10: *src[compno] = 0; yading@10: else if (*src[compno] >= (1 << s->cbps[compno])) yading@10: *src[compno] = (1 << s->cbps[compno]) - 1; yading@10: *dst = *src[compno]++; yading@10: dst += s->ncomponents; yading@10: } yading@10: line += s->picture->linesize[0]; yading@10: } yading@10: } yading@10: } else { yading@10: for (compno = 0; compno < s->ncomponents; compno++) { yading@10: y = tile->comp[compno].coord[1][0] - s->image_offset_y; yading@10: line = s->picture->data[0] + y * s->picture->linesize[0]; yading@10: for (; y < tile->comp[compno].coord[1][1] - s->image_offset_y; y += s->cdy[compno]) { yading@10: uint16_t *dst; yading@10: yading@10: x = tile->comp[compno].coord[0][0] - s->image_offset_x; yading@10: dst = (uint16_t *)(line + (x * s->ncomponents + compno) * 2); yading@10: for (; x < tile->comp[compno].coord[0][1] - s->image_offset_x; x += s-> cdx[compno]) { yading@10: int32_t val; yading@10: yading@10: val = *src[compno]++ << (16 - s->cbps[compno]); yading@10: val += 1 << 15; yading@10: val = av_clip(val, 0, (1 << 16) - 1); yading@10: *dst = val; yading@10: dst += s->ncomponents; yading@10: } yading@10: line += s->picture->linesize[0]; yading@10: } yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: static void cleanup(J2kDecoderContext *s) yading@10: { yading@10: int tileno, compno; yading@10: for (tileno = 0; tileno < s->numXtiles * s->numYtiles; tileno++){ yading@10: for (compno = 0; compno < s->ncomponents; compno++){ yading@10: J2kComponent *comp = s->tile[tileno].comp + compno; yading@10: J2kCodingStyle *codsty = s->tile[tileno].codsty + compno; yading@10: yading@10: ff_j2k_cleanup(comp, codsty); yading@10: } yading@10: av_freep(&s->tile[tileno].comp); yading@10: } yading@10: av_freep(&s->tile); yading@10: } yading@10: yading@10: static int decode_codestream(J2kDecoderContext *s) yading@10: { yading@10: J2kCodingStyle *codsty = s->codsty; yading@10: J2kQuantStyle *qntsty = s->qntsty; yading@10: uint8_t *properties = s->properties; yading@10: yading@10: for (;;){ yading@10: int oldpos, marker, len, ret = 0; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 2){ yading@10: av_log(s->avctx, AV_LOG_ERROR, "Missing EOC\n"); yading@10: break; yading@10: } yading@10: yading@10: marker = bytestream2_get_be16u(&s->g); yading@10: av_dlog(s->avctx, "marker 0x%.4X at pos 0x%x\n", marker, bytestream2_tell(&s->g) - 4); yading@10: oldpos = bytestream2_tell(&s->g); yading@10: yading@10: if (marker == J2K_SOD){ yading@10: J2kTile *tile = s->tile + s->curtileno; yading@10: if (ret = init_tile(s, s->curtileno)) { yading@10: av_log(s->avctx, AV_LOG_ERROR, "tile initialization failed\n"); yading@10: return ret; yading@10: } yading@10: if (ret = decode_packets(s, tile)) { yading@10: av_log(s->avctx, AV_LOG_ERROR, "packets decoding failed\n"); yading@10: return ret; yading@10: } yading@10: continue; yading@10: } yading@10: if (marker == J2K_EOC) yading@10: break; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 2) yading@10: return AVERROR(EINVAL); yading@10: len = bytestream2_get_be16u(&s->g); yading@10: switch (marker){ yading@10: case J2K_SIZ: yading@10: ret = get_siz(s); yading@10: break; yading@10: case J2K_COC: yading@10: ret = get_coc(s, codsty, properties); yading@10: break; yading@10: case J2K_COD: yading@10: ret = get_cod(s, codsty, properties); yading@10: break; yading@10: case J2K_QCC: yading@10: ret = get_qcc(s, len, qntsty, properties); yading@10: break; yading@10: case J2K_QCD: yading@10: ret = get_qcd(s, len, qntsty, properties); yading@10: break; yading@10: case J2K_SOT: yading@10: if (!(ret = get_sot(s))){ yading@10: codsty = s->tile[s->curtileno].codsty; yading@10: qntsty = s->tile[s->curtileno].qntsty; yading@10: properties = s->tile[s->curtileno].properties; yading@10: } yading@10: break; yading@10: case J2K_COM: yading@10: // the comment is ignored yading@10: bytestream2_skip(&s->g, len - 2); yading@10: break; yading@10: default: yading@10: av_log(s->avctx, AV_LOG_ERROR, "unsupported marker 0x%.4X at pos 0x%x\n", marker, bytestream2_tell(&s->g) - 4); yading@10: bytestream2_skip(&s->g, len - 2); yading@10: break; yading@10: } yading@10: if (bytestream2_tell(&s->g) - oldpos != len || ret){ yading@10: av_log(s->avctx, AV_LOG_ERROR, "error during processing marker segment %.4x\n", marker); yading@10: return ret ? ret : -1; yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: static int jp2_find_codestream(J2kDecoderContext *s) yading@10: { yading@10: uint32_t atom_size, atom; yading@10: int found_codestream = 0, search_range = 10; yading@10: yading@10: while(!found_codestream && search_range && bytestream2_get_bytes_left(&s->g) >= 8) { yading@10: atom_size = bytestream2_get_be32u(&s->g); yading@10: atom = bytestream2_get_be32u(&s->g); yading@10: if (atom == JP2_CODESTREAM) { yading@10: found_codestream = 1; yading@10: } else { yading@10: if (bytestream2_get_bytes_left(&s->g) < atom_size - 8) yading@10: return 0; yading@10: bytestream2_skipu(&s->g, atom_size - 8); yading@10: search_range--; yading@10: } yading@10: } yading@10: yading@10: if (found_codestream) yading@10: return 1; yading@10: return 0; yading@10: } yading@10: yading@10: static int decode_frame(AVCodecContext *avctx, yading@10: void *data, int *got_frame, yading@10: AVPacket *avpkt) yading@10: { yading@10: J2kDecoderContext *s = avctx->priv_data; yading@10: AVFrame *picture = data; yading@10: int tileno, ret; yading@10: yading@10: s->picture = picture; yading@10: yading@10: bytestream2_init(&s->g, avpkt->data, avpkt->size); yading@10: s->curtileno = -1; yading@10: yading@10: if (bytestream2_get_bytes_left(&s->g) < 2) { yading@10: ret = AVERROR(EINVAL); yading@10: goto err_out; yading@10: } yading@10: yading@10: // check if the image is in jp2 format yading@10: if (bytestream2_get_bytes_left(&s->g) >= 12 && yading@10: (bytestream2_get_be32u(&s->g) == 12) && yading@10: (bytestream2_get_be32u(&s->g) == JP2_SIG_TYPE) && yading@10: (bytestream2_get_be32u(&s->g) == JP2_SIG_VALUE)) { yading@10: if(!jp2_find_codestream(s)) { yading@10: av_log(avctx, AV_LOG_ERROR, "couldn't find jpeg2k codestream atom\n"); yading@10: ret = -1; yading@10: goto err_out; yading@10: } yading@10: } else { yading@10: bytestream2_seek(&s->g, 0, SEEK_SET); yading@10: } yading@10: yading@10: if (bytestream2_get_be16u(&s->g) != J2K_SOC){ yading@10: av_log(avctx, AV_LOG_ERROR, "SOC marker not present\n"); yading@10: ret = -1; yading@10: goto err_out; yading@10: } yading@10: if (ret = decode_codestream(s)) yading@10: goto err_out; yading@10: yading@10: for (tileno = 0; tileno < s->numXtiles * s->numYtiles; tileno++) yading@10: if (ret = decode_tile(s, s->tile + tileno)) yading@10: goto err_out; yading@10: yading@10: cleanup(s); yading@10: yading@10: *got_frame = 1; yading@10: yading@10: return bytestream2_tell(&s->g); yading@10: yading@10: err_out: yading@10: cleanup(s); yading@10: return ret; yading@10: } yading@10: yading@10: static av_cold int j2kdec_init(AVCodecContext *avctx) yading@10: { yading@10: J2kDecoderContext *s = avctx->priv_data; yading@10: yading@10: s->avctx = avctx; yading@10: yading@10: ff_j2k_init_tier1_luts(); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: AVCodec ff_jpeg2000_decoder = { yading@10: .name = "j2k", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_JPEG2000, yading@10: .priv_data_size = sizeof(J2kDecoderContext), yading@10: .init = j2kdec_init, yading@10: .decode = decode_frame, yading@10: .capabilities = CODEC_CAP_EXPERIMENTAL, yading@10: .long_name = NULL_IF_CONFIG_SMALL("JPEG 2000"), yading@10: };