forked from bartvdbraak/blender
ca8aa8c901
This adds redcode (the file format of RED one, R3D) support to blender. Seems to work fine with the footage I found on the web, but keep in mind, that because of the unoptimized nature of libopenjpeg, frame decoding isn't that fast. It is also a rather challenging task, to make 4k-float-footage realtime :)
711 lines
19 KiB
C
711 lines
19 KiB
C
/*
|
||
* Copyright (c) 2002-2007, Communications and Remote Sensing Laboratory, Universite catholique de Louvain (UCL), Belgium
|
||
* Copyright (c) 2002-2007, Professor Benoit Macq
|
||
* Copyright (c) 2001-2003, David Janssens
|
||
* Copyright (c) 2002-2003, Yannick Verschueren
|
||
* Copyright (c) 2003-2007, Francois-Olivier Devaux and Antonin Descampe
|
||
* Copyright (c) 2005, Herve Drolon, FreeImage Team
|
||
* All rights reserved.
|
||
*
|
||
* Redistribution and use in source and binary forms, with or without
|
||
* modification, are permitted provided that the following conditions
|
||
* are met:
|
||
* 1. Redistributions of source code must retain the above copyright
|
||
* notice, this list of conditions and the following disclaimer.
|
||
* 2. Redistributions in binary form must reproduce the above copyright
|
||
* notice, this list of conditions and the following disclaimer in the
|
||
* documentation and/or other materials provided with the distribution.
|
||
*
|
||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
|
||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||
* POSSIBILITY OF SUCH DAMAGE.
|
||
*/
|
||
|
||
#include "opj_includes.h"
|
||
|
||
/** @defgroup JP2 JP2 - JPEG-2000 file format reader/writer */
|
||
/*@{*/
|
||
|
||
/** @name Local static functions */
|
||
/*@{*/
|
||
|
||
/**
|
||
Read box headers
|
||
@param cinfo Codec context info
|
||
@param cio Input stream
|
||
@param box
|
||
@return Returns true if successful, returns false otherwise
|
||
*/
|
||
static bool jp2_read_boxhdr(opj_common_ptr cinfo, opj_cio_t *cio, opj_jp2_box_t *box);
|
||
/*static void jp2_write_url(opj_cio_t *cio, char *Idx_file);*/
|
||
/**
|
||
Read the IHDR box - Image Header box
|
||
@param jp2 JP2 handle
|
||
@param cio Input buffer stream
|
||
@return Returns true if successful, returns false otherwise
|
||
*/
|
||
static bool jp2_read_ihdr(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
static void jp2_write_ihdr(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
static void jp2_write_bpcc(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
static bool jp2_read_bpcc(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
static void jp2_write_colr(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
static bool jp2_read_colr(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
/**
|
||
Write the FTYP box - File type box
|
||
@param jp2 JP2 handle
|
||
@param cio Output buffer stream
|
||
*/
|
||
static void jp2_write_ftyp(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
/**
|
||
Read the FTYP box - File type box
|
||
@param jp2 JP2 handle
|
||
@param cio Input buffer stream
|
||
@return Returns true if successful, returns false otherwise
|
||
*/
|
||
static bool jp2_read_ftyp(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
static int jp2_write_jp2c(opj_jp2_t *jp2, opj_cio_t *cio, opj_image_t *image, opj_codestream_info_t *cstr_info);
|
||
static bool jp2_read_jp2c(opj_jp2_t *jp2, opj_cio_t *cio, unsigned int *j2k_codestream_length, unsigned int *j2k_codestream_offset);
|
||
static void jp2_write_jp(opj_cio_t *cio);
|
||
/**
|
||
Read the JP box - JPEG 2000 signature
|
||
@param jp2 JP2 handle
|
||
@param cio Input buffer stream
|
||
@return Returns true if successful, returns false otherwise
|
||
*/
|
||
static bool jp2_read_jp(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
/**
|
||
Decode the structure of a JP2 file
|
||
@param jp2 JP2 handle
|
||
@param cio Input buffer stream
|
||
@return Returns true if successful, returns false otherwise
|
||
*/
|
||
static bool jp2_read_struct(opj_jp2_t *jp2, opj_cio_t *cio);
|
||
|
||
/*@}*/
|
||
|
||
/*@}*/
|
||
|
||
/* ----------------------------------------------------------------------- */
|
||
|
||
static bool jp2_read_boxhdr(opj_common_ptr cinfo, opj_cio_t *cio, opj_jp2_box_t *box) {
|
||
box->init_pos = cio_tell(cio);
|
||
box->length = cio_read(cio, 4);
|
||
box->type = cio_read(cio, 4);
|
||
if (box->length == 1) {
|
||
if (cio_read(cio, 4) != 0) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Cannot handle box sizes higher than 2^32\n");
|
||
return false;
|
||
}
|
||
box->length = cio_read(cio, 4);
|
||
if (box->length == 0)
|
||
box->length = cio_numbytesleft(cio) + 12;
|
||
}
|
||
else if (box->length == 0) {
|
||
box->length = cio_numbytesleft(cio) + 8;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
#if 0
|
||
static void jp2_write_url(opj_cio_t *cio, char *Idx_file) {
|
||
unsigned int i;
|
||
opj_jp2_box_t box;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_URL, 4); /* DBTL */
|
||
cio_write(cio, 0, 1); /* VERS */
|
||
cio_write(cio, 0, 3); /* FLAG */
|
||
|
||
if(Idx_file) {
|
||
for (i = 0; i < strlen(Idx_file); i++) {
|
||
cio_write(cio, Idx_file[i], 1);
|
||
}
|
||
}
|
||
|
||
box.length = cio_tell(cio) - box.init_pos;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
}
|
||
#endif
|
||
|
||
static bool jp2_read_ihdr(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
|
||
opj_common_ptr cinfo = jp2->cinfo;
|
||
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
if (JP2_IHDR != box.type) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Expected IHDR Marker\n");
|
||
return false;
|
||
}
|
||
|
||
jp2->h = cio_read(cio, 4); /* HEIGHT */
|
||
jp2->w = cio_read(cio, 4); /* WIDTH */
|
||
jp2->numcomps = cio_read(cio, 2); /* NC */
|
||
jp2->comps = (opj_jp2_comps_t*) opj_malloc(jp2->numcomps * sizeof(opj_jp2_comps_t));
|
||
|
||
jp2->bpc = cio_read(cio, 1); /* BPC */
|
||
|
||
jp2->C = cio_read(cio, 1); /* C */
|
||
jp2->UnkC = cio_read(cio, 1); /* UnkC */
|
||
jp2->IPR = cio_read(cio, 1); /* IPR */
|
||
|
||
if (cio_tell(cio) - box.init_pos != box.length) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with IHDR Box\n");
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
static void jp2_write_ihdr(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_IHDR, 4); /* IHDR */
|
||
|
||
cio_write(cio, jp2->h, 4); /* HEIGHT */
|
||
cio_write(cio, jp2->w, 4); /* WIDTH */
|
||
cio_write(cio, jp2->numcomps, 2); /* NC */
|
||
|
||
cio_write(cio, jp2->bpc, 1); /* BPC */
|
||
|
||
cio_write(cio, jp2->C, 1); /* C : Always 7 */
|
||
cio_write(cio, jp2->UnkC, 1); /* UnkC, colorspace unknown */
|
||
cio_write(cio, jp2->IPR, 1); /* IPR, no intellectual property */
|
||
|
||
box.length = cio_tell(cio) - box.init_pos;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
}
|
||
|
||
static void jp2_write_bpcc(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
unsigned int i;
|
||
opj_jp2_box_t box;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_BPCC, 4); /* BPCC */
|
||
|
||
for (i = 0; i < jp2->numcomps; i++) {
|
||
cio_write(cio, jp2->comps[i].bpcc, 1);
|
||
}
|
||
|
||
box.length = cio_tell(cio) - box.init_pos;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
}
|
||
|
||
|
||
static bool jp2_read_bpcc(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
unsigned int i;
|
||
opj_jp2_box_t box;
|
||
|
||
opj_common_ptr cinfo = jp2->cinfo;
|
||
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
if (JP2_BPCC != box.type) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Expected BPCC Marker\n");
|
||
return false;
|
||
}
|
||
|
||
for (i = 0; i < jp2->numcomps; i++) {
|
||
jp2->comps[i].bpcc = cio_read(cio, 1);
|
||
}
|
||
|
||
if (cio_tell(cio) - box.init_pos != box.length) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with BPCC Box\n");
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
static void jp2_write_colr(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_COLR, 4); /* COLR */
|
||
|
||
cio_write(cio, jp2->meth, 1); /* METH */
|
||
cio_write(cio, jp2->precedence, 1); /* PRECEDENCE */
|
||
cio_write(cio, jp2->approx, 1); /* APPROX */
|
||
|
||
if (jp2->meth == 1) {
|
||
cio_write(cio, jp2->enumcs, 4); /* EnumCS */
|
||
} else {
|
||
cio_write(cio, 0, 1); /* PROFILE (??) */
|
||
}
|
||
|
||
box.length = cio_tell(cio) - box.init_pos;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
}
|
||
|
||
static bool jp2_read_colr(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
int skip_len;
|
||
|
||
opj_common_ptr cinfo = jp2->cinfo;
|
||
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
do {
|
||
if (JP2_COLR != box.type) {
|
||
cio_skip(cio, box.length - 8);
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
}
|
||
} while(JP2_COLR != box.type);
|
||
|
||
jp2->meth = cio_read(cio, 1); /* METH */
|
||
jp2->precedence = cio_read(cio, 1); /* PRECEDENCE */
|
||
jp2->approx = cio_read(cio, 1); /* APPROX */
|
||
|
||
if (jp2->meth == 1) {
|
||
jp2->enumcs = cio_read(cio, 4); /* EnumCS */
|
||
} else {
|
||
/* skip PROFILE */
|
||
skip_len = box.init_pos + box.length - cio_tell(cio);
|
||
if (skip_len < 0) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with JP2H box size\n");
|
||
return false;
|
||
}
|
||
cio_skip(cio, box.init_pos + box.length - cio_tell(cio));
|
||
}
|
||
|
||
if (cio_tell(cio) - box.init_pos != box.length) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with BPCC Box\n");
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
void jp2_write_jp2h(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_JP2H, 4); /* JP2H */
|
||
|
||
jp2_write_ihdr(jp2, cio);
|
||
|
||
if (jp2->bpc == 255) {
|
||
jp2_write_bpcc(jp2, cio);
|
||
}
|
||
jp2_write_colr(jp2, cio);
|
||
|
||
box.length = cio_tell(cio) - box.init_pos;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
}
|
||
|
||
bool jp2_read_jp2h(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
int skip_len;
|
||
|
||
opj_common_ptr cinfo = jp2->cinfo;
|
||
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
do {
|
||
if (JP2_JP2H != box.type) {
|
||
if (box.type == JP2_JP2C) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Expected JP2H Marker\n");
|
||
return false;
|
||
}
|
||
cio_skip(cio, box.length - 8);
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
}
|
||
} while(JP2_JP2H != box.type);
|
||
|
||
if (!jp2_read_ihdr(jp2, cio))
|
||
return false;
|
||
|
||
if (jp2->bpc == 255) {
|
||
if (!jp2_read_bpcc(jp2, cio))
|
||
return false;
|
||
}
|
||
if (!jp2_read_colr(jp2, cio))
|
||
return false;
|
||
|
||
skip_len = box.init_pos + box.length - cio_tell(cio);
|
||
if (skip_len < 0) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with JP2H Box\n");
|
||
return false;
|
||
}
|
||
cio_skip(cio, box.init_pos + box.length - cio_tell(cio));
|
||
|
||
return true;
|
||
}
|
||
|
||
static void jp2_write_ftyp(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
unsigned int i;
|
||
opj_jp2_box_t box;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_FTYP, 4); /* FTYP */
|
||
|
||
cio_write(cio, jp2->brand, 4); /* BR */
|
||
cio_write(cio, jp2->minversion, 4); /* MinV */
|
||
|
||
for (i = 0; i < jp2->numcl; i++) {
|
||
cio_write(cio, jp2->cl[i], 4); /* CL */
|
||
}
|
||
|
||
box.length = cio_tell(cio) - box.init_pos;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
}
|
||
|
||
static bool jp2_read_ftyp(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
int i;
|
||
opj_jp2_box_t box;
|
||
|
||
opj_common_ptr cinfo = jp2->cinfo;
|
||
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
|
||
if (JP2_FTYP != box.type) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Expected FTYP Marker\n");
|
||
return false;
|
||
}
|
||
|
||
jp2->brand = cio_read(cio, 4); /* BR */
|
||
jp2->minversion = cio_read(cio, 4); /* MinV */
|
||
jp2->numcl = (box.length - 16) / 4;
|
||
jp2->cl = (unsigned int *) opj_malloc(jp2->numcl * sizeof(unsigned int));
|
||
|
||
for (i = 0; i < (int)jp2->numcl; i++) {
|
||
jp2->cl[i] = cio_read(cio, 4); /* CLi */
|
||
}
|
||
|
||
if (cio_tell(cio) - box.init_pos != box.length) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with FTYP Box\n");
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
static int jp2_write_jp2c(opj_jp2_t *jp2, opj_cio_t *cio, opj_image_t *image, opj_codestream_info_t *cstr_info) {
|
||
unsigned int j2k_codestream_offset, j2k_codestream_length;
|
||
opj_jp2_box_t box;
|
||
|
||
opj_j2k_t *j2k = jp2->j2k;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_JP2C, 4); /* JP2C */
|
||
|
||
/* J2K encoding */
|
||
j2k_codestream_offset = cio_tell(cio);
|
||
if(!j2k_encode(j2k, cio, image, cstr_info)) {
|
||
opj_event_msg(j2k->cinfo, EVT_ERROR, "Failed to encode image\n");
|
||
return 0;
|
||
}
|
||
j2k_codestream_length = cio_tell(cio) - j2k_codestream_offset;
|
||
|
||
jp2->j2k_codestream_offset = j2k_codestream_offset;
|
||
jp2->j2k_codestream_length = j2k_codestream_length;
|
||
|
||
box.length = 8 + jp2->j2k_codestream_length;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
|
||
return box.length;
|
||
}
|
||
|
||
static bool jp2_read_jp2c(opj_jp2_t *jp2, opj_cio_t *cio, unsigned int *j2k_codestream_length, unsigned int *j2k_codestream_offset) {
|
||
opj_jp2_box_t box;
|
||
|
||
opj_common_ptr cinfo = jp2->cinfo;
|
||
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
do {
|
||
if(JP2_JP2C != box.type) {
|
||
cio_skip(cio, box.length - 8);
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
}
|
||
} while(JP2_JP2C != box.type);
|
||
|
||
*j2k_codestream_offset = cio_tell(cio);
|
||
*j2k_codestream_length = box.length - 8;
|
||
|
||
return true;
|
||
}
|
||
|
||
static void jp2_write_jp(opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
|
||
box.init_pos = cio_tell(cio);
|
||
cio_skip(cio, 4);
|
||
cio_write(cio, JP2_JP, 4); /* JP2 signature */
|
||
cio_write(cio, 0x0d0a870a, 4);
|
||
|
||
box.length = cio_tell(cio) - box.init_pos;
|
||
cio_seek(cio, box.init_pos);
|
||
cio_write(cio, box.length, 4); /* L */
|
||
cio_seek(cio, box.init_pos + box.length);
|
||
}
|
||
|
||
static bool jp2_read_jp(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
opj_jp2_box_t box;
|
||
|
||
opj_common_ptr cinfo = jp2->cinfo;
|
||
|
||
jp2_read_boxhdr(cinfo, cio, &box);
|
||
if (JP2_JP != box.type) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Expected JP Marker\n");
|
||
return false;
|
||
}
|
||
if (0x0d0a870a != cio_read(cio, 4)) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with JP Marker\n");
|
||
return false;
|
||
}
|
||
if (cio_tell(cio) - box.init_pos != box.length) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Error with JP Box size\n");
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
|
||
static bool jp2_read_struct(opj_jp2_t *jp2, opj_cio_t *cio) {
|
||
if (!jp2_read_jp(jp2, cio))
|
||
return false;
|
||
if (!jp2_read_ftyp(jp2, cio))
|
||
return false;
|
||
if (!jp2_read_jp2h(jp2, cio))
|
||
return false;
|
||
if (!jp2_read_jp2c(jp2, cio, &jp2->j2k_codestream_length, &jp2->j2k_codestream_offset))
|
||
return false;
|
||
|
||
return true;
|
||
}
|
||
|
||
/* ----------------------------------------------------------------------- */
|
||
/* JP2 decoder interface */
|
||
/* ----------------------------------------------------------------------- */
|
||
|
||
opj_jp2_t* jp2_create_decompress(opj_common_ptr cinfo) {
|
||
opj_jp2_t *jp2 = (opj_jp2_t*) opj_calloc(1, sizeof(opj_jp2_t));
|
||
if(jp2) {
|
||
jp2->cinfo = cinfo;
|
||
/* create the J2K codec */
|
||
jp2->j2k = j2k_create_decompress(cinfo);
|
||
if(jp2->j2k == NULL) {
|
||
jp2_destroy_decompress(jp2);
|
||
return NULL;
|
||
}
|
||
}
|
||
return jp2;
|
||
}
|
||
|
||
void jp2_destroy_decompress(opj_jp2_t *jp2) {
|
||
if(jp2) {
|
||
/* destroy the J2K codec */
|
||
j2k_destroy_decompress(jp2->j2k);
|
||
|
||
if(jp2->comps) {
|
||
opj_free(jp2->comps);
|
||
}
|
||
if(jp2->cl) {
|
||
opj_free(jp2->cl);
|
||
}
|
||
opj_free(jp2);
|
||
}
|
||
}
|
||
|
||
void jp2_setup_decoder(opj_jp2_t *jp2, opj_dparameters_t *parameters) {
|
||
/* setup the J2K codec */
|
||
j2k_setup_decoder(jp2->j2k, parameters);
|
||
/* further JP2 initializations go here */
|
||
}
|
||
|
||
opj_image_t* jp2_decode(opj_jp2_t *jp2, opj_cio_t *cio, opj_codestream_info_t *cstr_info) {
|
||
opj_common_ptr cinfo;
|
||
opj_image_t *image = NULL;
|
||
|
||
if(!jp2 || !cio) {
|
||
return NULL;
|
||
}
|
||
|
||
cinfo = jp2->cinfo;
|
||
|
||
/* JP2 decoding */
|
||
if(!jp2_read_struct(jp2, cio)) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Failed to decode jp2 structure\n");
|
||
return NULL;
|
||
}
|
||
|
||
/* J2K decoding */
|
||
image = j2k_decode(jp2->j2k, cio, cstr_info);
|
||
if(!image) {
|
||
opj_event_msg(cinfo, EVT_ERROR, "Failed to decode J2K image\n");
|
||
}
|
||
|
||
/* Set Image Color Space */
|
||
if (jp2->enumcs == 16)
|
||
image->color_space = CLRSPC_SRGB;
|
||
else if (jp2->enumcs == 17)
|
||
image->color_space = CLRSPC_GRAY;
|
||
else if (jp2->enumcs == 18)
|
||
image->color_space = CLRSPC_SYCC;
|
||
else
|
||
image->color_space = CLRSPC_UNKNOWN;
|
||
|
||
return image;
|
||
}
|
||
|
||
/* ----------------------------------------------------------------------- */
|
||
/* JP2 encoder interface */
|
||
/* ----------------------------------------------------------------------- */
|
||
|
||
opj_jp2_t* jp2_create_compress(opj_common_ptr cinfo) {
|
||
opj_jp2_t *jp2 = (opj_jp2_t*)opj_malloc(sizeof(opj_jp2_t));
|
||
if(jp2) {
|
||
jp2->cinfo = cinfo;
|
||
/* create the J2K codec */
|
||
jp2->j2k = j2k_create_compress(cinfo);
|
||
if(jp2->j2k == NULL) {
|
||
jp2_destroy_compress(jp2);
|
||
return NULL;
|
||
}
|
||
}
|
||
return jp2;
|
||
}
|
||
|
||
void jp2_destroy_compress(opj_jp2_t *jp2) {
|
||
if(jp2) {
|
||
/* destroy the J2K codec */
|
||
j2k_destroy_compress(jp2->j2k);
|
||
|
||
if(jp2->comps) {
|
||
opj_free(jp2->comps);
|
||
}
|
||
if(jp2->cl) {
|
||
opj_free(jp2->cl);
|
||
}
|
||
opj_free(jp2);
|
||
}
|
||
}
|
||
|
||
void jp2_setup_encoder(opj_jp2_t *jp2, opj_cparameters_t *parameters, opj_image_t *image) {
|
||
int i;
|
||
int depth_0, sign;
|
||
|
||
if(!jp2 || !parameters || !image)
|
||
return;
|
||
|
||
/* setup the J2K codec */
|
||
/* ------------------- */
|
||
|
||
/* Check if number of components respects standard */
|
||
if (image->numcomps < 1 || image->numcomps > 16384) {
|
||
opj_event_msg(jp2->cinfo, EVT_ERROR, "Invalid number of components specified while setting up JP2 encoder\n");
|
||
return;
|
||
}
|
||
|
||
j2k_setup_encoder(jp2->j2k, parameters, image);
|
||
|
||
/* setup the JP2 codec */
|
||
/* ------------------- */
|
||
|
||
/* Profile box */
|
||
|
||
jp2->brand = JP2_JP2; /* BR */
|
||
jp2->minversion = 0; /* MinV */
|
||
jp2->numcl = 1;
|
||
jp2->cl = (unsigned int*) opj_malloc(jp2->numcl * sizeof(unsigned int));
|
||
jp2->cl[0] = JP2_JP2; /* CL0 : JP2 */
|
||
|
||
/* Image Header box */
|
||
|
||
jp2->numcomps = image->numcomps; /* NC */
|
||
jp2->comps = (opj_jp2_comps_t*) opj_malloc(jp2->numcomps * sizeof(opj_jp2_comps_t));
|
||
jp2->h = image->y1 - image->y0; /* HEIGHT */
|
||
jp2->w = image->x1 - image->x0; /* WIDTH */
|
||
/* BPC */
|
||
depth_0 = image->comps[0].prec - 1;
|
||
sign = image->comps[0].sgnd;
|
||
jp2->bpc = depth_0 + (sign << 7);
|
||
for (i = 1; i < image->numcomps; i++) {
|
||
int depth = image->comps[i].prec - 1;
|
||
sign = image->comps[i].sgnd;
|
||
if (depth_0 != depth)
|
||
jp2->bpc = 255;
|
||
}
|
||
jp2->C = 7; /* C : Always 7 */
|
||
jp2->UnkC = 0; /* UnkC, colorspace specified in colr box */
|
||
jp2->IPR = 0; /* IPR, no intellectual property */
|
||
|
||
/* BitsPerComponent box */
|
||
|
||
for (i = 0; i < image->numcomps; i++) {
|
||
jp2->comps[i].bpcc = image->comps[i].prec - 1 + (image->comps[i].sgnd << 7);
|
||
}
|
||
|
||
/* Colour Specification box */
|
||
|
||
if ((image->numcomps == 1 || image->numcomps == 3) && (jp2->bpc != 255)) {
|
||
jp2->meth = 1; /* METH: Enumerated colourspace */
|
||
} else {
|
||
jp2->meth = 2; /* METH: Restricted ICC profile */
|
||
}
|
||
if (jp2->meth == 1) {
|
||
if (image->color_space == 1)
|
||
jp2->enumcs = 16; /* sRGB as defined by IEC 61966<36>2<EFBFBD>1 */
|
||
else if (image->color_space == 2)
|
||
jp2->enumcs = 17; /* greyscale */
|
||
else if (image->color_space == 3)
|
||
jp2->enumcs = 18; /* YUV */
|
||
} else {
|
||
jp2->enumcs = 0; /* PROFILE (??) */
|
||
}
|
||
jp2->precedence = 0; /* PRECEDENCE */
|
||
jp2->approx = 0; /* APPROX */
|
||
|
||
}
|
||
|
||
bool jp2_encode(opj_jp2_t *jp2, opj_cio_t *cio, opj_image_t *image, opj_codestream_info_t *cstr_info) {
|
||
|
||
/* JP2 encoding */
|
||
|
||
/* JPEG 2000 Signature box */
|
||
jp2_write_jp(cio);
|
||
/* File Type box */
|
||
jp2_write_ftyp(jp2, cio);
|
||
/* JP2 Header box */
|
||
jp2_write_jp2h(jp2, cio);
|
||
|
||
/* J2K encoding */
|
||
|
||
if(!jp2_write_jp2c(jp2, cio, image, cstr_info)) {
|
||
opj_event_msg(jp2->cinfo, EVT_ERROR, "Failed to encode image\n");
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
|