Jpeg2KTool2.cpp

Go to the documentation of this file.

 
#include "Jpeg2KTool.h"
 
 
#ifdef ENABLE_OPENJPEG
 
#if OPENJPEG_VER >= JP2K_VER_20
 
 
using namespace jbxl;
 
 
void  JPEG2KImage::init(void)
{
    xs    = 0;
    ys    = 0;
    ws    = 0;
    hs    = 0;
    col   = 0;
    cmode = GRAPH_COLOR_RGBA;
    state = 0;
    image = NULL;
}
 
 
bool  JPEG2KImage::isNull(void)
{
    if (image!=NULL && image->comps!=NULL) return false;
    return true;
}
 
 
void  JPEG2KImage::free(void)
{
    if (image!=NULL) {
        opj_image_destroy(image);
    }
 
    init();
}
 
 
void  JPEG2KImage::fill(int v)
{
    if (!isNull()) {
        for (int k=0; k<col; k++) {
            if (image->comps[k].data!=NULL) {   
                for (int i=0; i<xs*ys; i++) {
                    image->comps[k].data[i] = (OPJ_INT32)v;
                }
            }
        }
    }
}
 
 
void  JPEG2KImage::setup_image(void)
{
    if (!isNull()) {
        //
        xs = image->x1 - image->x0;
        ys = image->y1 - image->y0;
 
        int fac = (int)image->comps->factor;
        ws = (xs + (1<<fac) -1)>>fac;
        hs = (ys + (1<<fac) -1)>>fac;
 
        col = (int)image->numcomps;
        /*
        if (image->color_space==OPJ_CLRSPC_SRGB) {          /// 1
            col = 3;
        }
        else if (image->color_space==OPJ_CLRSPC_GRAY) {     /// 2
            col = 1;
        }
        else if (image->color_space == OPJ_CLRSPC_SYCC || image->color_space == OPJ_CLRSPC_EYCC) {  /// 3, 4
            col = 3;
        }
        else if (image->color_space==OPJ_CLRSPC_CMYK) {     /// 5
            col = 4;
        }
        */
 
        // 設定されないものについては，未対応
        cmode = GRAPH_COLOR_UNKNOWN;
#if OPENJPEG_VER < JP2K_VER_25
        int depth = (int)image->comps->bpp;
#else
        int depth = (int)image->comps->prec;
#endif 
        DEBUG_MODE PRINT_MESG("JBXL::JPEG2KIMage::setup_image: INFO: xs  = %d, ys = %d, col = %d, depth = %d\n", xs, ys, col, depth);
        if (depth==0 || depth==8) {
            if      (col==1) cmode = GRAPH_COLOR_GRAY;
            else if (col==3) cmode = GRAPH_COLOR_RGB;
            else if (col==4) cmode = GRAPH_COLOR_RGBA;
        }
        if (cmode==GRAPH_COLOR_UNKNOWN) {
            PRINT_MESG("JBXL::JPEG2KIMage::setup_image: unknown color mode: col = %d, depth = %d\n", col, depth);
            cmode = GRAPH_COLOR_RGBA;
        } 
 
/*
        DEBUG_MODE {
            PRINT_MESG("JPEG2KImage::setup_image: OFFSET  %d %d %d %d\n", image->x0, image->y0, image->x1, image->y1);
            PRINT_MESG("JPEG2KImage::setup_image: NUMCOMP %d\n", image->numcomps);
            PRINT_MESG("JPEG2KImage::setup_image: COLORSP %d\n", image->color_space);
            PRINT_MESG("JPEG2KImage::setup_image: XS YS   %d %d\n", xs, ys);
            PRINT_MESG("JPEG2KImage::setup_image: COLOR   %d\n", col);
            PRINT_MESG("JPEG2KImage::setup_image: CMODE   %d\n", cmode);
            PRINT_MESG("\n");
            for (int i=0; i<(int)image->numcomps; i++) {
                PRINT_MESG("JPEG2KImage::setup_image: DX DY %d %d\n", image->comps[i].dx, image->comps[i].dy);
                PRINT_MESG("JPEG2KImage::setup_image: W  H  %d %d\n", image->comps[i].w,  image->comps[i].h);
                PRINT_MESG("JPEG2KImage::setup_image: X0 Y0 %d %d\n", image->comps[i].x0, image->comps[i].y0);
                PRINT_MESG("JPEG2KImage::setup_image: PREC  %d\n", image->comps[i].prec);
                PRINT_MESG("JPEG2KImage::setup_image: DEPTH %d\n", image->comps[i].bpp);
                PRINT_MESG("JPEG2KImage::setup_image: SGND  %d\n", image->comps[i].sgnd);
                PRINT_MESG("JPEG2KImage::setup_image: RESNO %d\n", image->comps[i].resno_decoded);
                PRINT_MESG("JPEG2KImage::setup_image: FACT  %d\n", image->comps[i].factor);
                PRINT_MESG("\n");
            }
        }
*/
    }
}
    
    
 
 
JPEG2KImage  jbxl::readJPEG2KFile(const char* fname)
{
    JPEG2KImage jp;
    //
    if (fname==NULL) {
        jp.state = JBXL_GRAPH_IVDARG_ERROR;
        return jp;
    }
 
    FILE* fp = fopen(fname, "rb");
    if (fp==NULL) {
        jp.state = JBXL_GRAPH_OPFILE_ERROR;
        return jp;
    }
 
    Buffer buf = read_Buffer_data(fp, 12);
    if (buf.vldsz<12) {
        free_Buffer(&buf);
        jp.state = JBXL_GRAPH_FILESZ_ERROR;
        fclose(fp);
        return jp;
    }
 
    fseek(fp, 0, 0);
    int format = isJPEG2KHeader(buf);
    if (format==JP2K_FMT_NONE) format = JP2K_FMT_JPT;
    free_Buffer(&buf);
 
#if OPENJPEG_VER < JP2K_VER_21
    jp = readJPEG2KData(fp, format);
    fclose(fp);
#else
    fclose(fp);
    jp = readJPEG2KData(fname, format);
#endif
 
    return jp;
}
 
 
#if OPENJPEG_VER < JP2K_VER_21
JPEG2KImage  jbxl::readJPEG2KData(FILE* fp, int format)
#else
JPEG2KImage  jbxl::readJPEG2KData(const char* fname, int format)
#endif
{
    JPEG2KImage jp;
 
    opj_stream_t* stream = NULL;
    opj_codec_t*  codec  = NULL;
 
    opj_dparameters_t parameters;   
    opj_set_default_decoder_parameters(&parameters);
 
#if OPENJPEG_VER < JP2K_VER_21
    stream = opj_stream_create_default_file_stream(fp, 1);
#else
    stream = opj_stream_create_default_file_stream(fname, 1);
#endif
    if (stream==NULL){
        jp.state = JBXL_GRAPH_RDFILE_ERROR;
        return jp;
    }
 
    if (format==JP2K_FMT_J2K) {         // JPEG 2000 codestream
        codec = opj_create_decompress(OPJ_CODEC_J2K);
    }   
    else if (format==JP2K_FMT_JP2) {    // JPEG 2000 compressed image data
        codec = opj_create_decompress(OPJ_CODEC_JP2);
    }
    else if (format==JP2K_FMT_JPT) {    // JPEG 2000 JPIP
        codec = opj_create_decompress(OPJ_CODEC_JPT);
    }
    else {
        PRINT_MESG("JBXL::readJPEG2KData: ERROR: unknown file format!\n");
        opj_stream_destroy(stream);
        return jp;
    }
 
    if (!opj_setup_decoder(codec, &parameters) ){
        opj_stream_destroy(stream);
        opj_destroy_codec(codec);
        jp.state = JBXL_GRAPH_ERROR;
        return jp;
    }
 
    // allow_partial
    if (!opj_decoder_set_strict_mode(codec, OPJ_FALSE)) {       // #define OPJ_FALSE 0
        opj_stream_destroy(stream);
        opj_destroy_codec(codec);
        jp.state = JBXL_GRAPH_ERROR;
        return jp;
    }
 
    /*
    if (!opj_codec_set_threads(codec, 1)) {
        opj_stream_destroy(stream);
        opj_destroy_codec(codec);
        jp.state = JBXL_GRAPH_ERROR;
        return jp;
    }*/
 
    if (!opj_read_header(stream, codec, &jp.image)){
        opj_stream_destroy(stream);
        opj_destroy_codec(codec);
        jp.state = JBXL_GRAPH_ERROR;
        return jp;
    }
 
//  if (!opj_set_decode_area(codec, jp.image, 0, 0, 0, 0)){
    if (!opj_set_decode_area(codec, jp.image, (OPJ_INT32)parameters.DA_x0, (OPJ_INT32)parameters.DA_y0, (OPJ_INT32)parameters.DA_x1, (OPJ_INT32)parameters.DA_y1)){
        opj_stream_destroy(stream);
        opj_destroy_codec(codec);
        jp.free();
        jp.state = JBXL_GRAPH_ERROR;
        return jp;
    }
 
    if (!(opj_decode(codec, stream, jp.image) && opj_end_decompress(codec, stream))) {
        opj_destroy_codec(codec);
        opj_stream_destroy(stream);
        jp.free();
        jp.state = JBXL_GRAPH_ERROR;
        return jp;
    }
 
    jp.setup_image();
 
    opj_stream_destroy(stream);
    opj_destroy_codec(codec);
 
    return jp;
}
 
 
#include "Jpeg2KToolCommon.cpp"
 
 
#endif      // OPENJPEG_VER >= JP2K_VER_20
#endif      // ENABLE_OPENJPEG