tests/test_tile_decoder.c

/*
 * Copyright (c) 2008, Jerome Fimes, Communications & Systemes <jerome.fimes@c-s.fr>
 * 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.
 */
#define USE_OPJ_DEPRECATED
/* set this macro to enable profiling for the given test */
/* warning : in order to be effective, openjpeg must have been built with profiling enabled !! */
/*#define _PROFILE*/

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>

#ifdef _WIN32
#include <malloc.h>
#else
#include <stdlib.h>
#endif

#include "opj_config.h"
#include <stdlib.h>

#ifdef _WIN32
#include <windows.h>
#define strcasecmp _stricmp
#define strncasecmp _strnicmp
#else
#include <strings.h>
#endif /* _WIN32 */

#include "openjpeg.h"
#include "format_defs.h"


/* -------------------------------------------------------------------------- */
/* Declarations                                                               */
int get_file_format(const char *filename);
static int infile_format(const char *fname);

/* -------------------------------------------------------------------------- */
int get_file_format(const char *filename)
{
    unsigned int i;
    static const char *extension[] = {"pgx", "pnm", "pgm", "ppm", "bmp","tif", "raw", "rawl", "tga", "png", "j2k", "jp2", "jpt", "j2c", "jpc" };
    static const int format[] = { PGX_DFMT, PXM_DFMT, PXM_DFMT, PXM_DFMT, BMP_DFMT, TIF_DFMT, RAW_DFMT, RAWL_DFMT, TGA_DFMT, PNG_DFMT, J2K_CFMT, JP2_CFMT, JPT_CFMT, J2K_CFMT, J2K_CFMT };
    char * ext = strrchr(filename, '.');
    if (ext == NULL)
        return -1;
    ext++;
    if(ext) {
        for(i = 0; i < sizeof(format)/sizeof(*format); i++) {
            if(strcasecmp(ext, extension[i]) == 0) {
                return format[i];
            }
        }
    }

    return -1;
}

/* -------------------------------------------------------------------------- */
#define JP2_RFC3745_MAGIC "\x00\x00\x00\x0c\x6a\x50\x20\x20\x0d\x0a\x87\x0a"
#define JP2_MAGIC "\x0d\x0a\x87\x0a"
/* position 45: "\xff\x52" */
#define J2K_CODESTREAM_MAGIC "\xff\x4f\xff\x51"

static int infile_format(const char *fname)
{
    FILE *reader;
    const char *s, *magic_s;
    int ext_format, magic_format;
    unsigned char buf[12];
    unsigned int l_nb_read;

    reader = fopen(fname, "rb");

    if (reader == NULL)
        return -1;

    memset(buf, 0, 12);
    l_nb_read = (unsigned int)fread(buf, 1, 12, reader);
    fclose(reader);
    if (l_nb_read != 12)
        return -1;

    ext_format = get_file_format(fname);

    if (ext_format == JPT_CFMT)
        return JPT_CFMT;

    if (memcmp(buf, JP2_RFC3745_MAGIC, 12) == 0 || memcmp(buf, JP2_MAGIC, 4) == 0) {
        magic_format = JP2_CFMT;
        magic_s = ".jp2";
    } else if (memcmp(buf, J2K_CODESTREAM_MAGIC, 4) == 0) {
        magic_format = J2K_CFMT;
        magic_s = ".j2k or .jpc or .j2c";
    } else
        return -1;

    if (magic_format == ext_format)
        return ext_format;

    s = fname + strlen(fname) - 4;

    fputs("\n===========================================\n", stderr);
    fprintf(stderr, "The extension of this file is incorrect.\n"
            "FOUND %s. SHOULD BE %s\n", s, magic_s);
    fputs("===========================================\n", stderr);

    return magic_format;
}


/* -------------------------------------------------------------------------- */

/**
  sample error debug callback expecting no client object
 */
static void error_callback(const char *msg, void *client_data)
{
    (void)client_data;
    fprintf(stdout, "[ERROR] %s", msg);
}
/**
  sample warning debug callback expecting no client object
 */
static void warning_callback(const char *msg, void *client_data)
{
    (void)client_data;
    fprintf(stdout, "[WARNING] %s", msg);
}
/**
  sample debug callback expecting no client object
 */
static void info_callback(const char *msg, void *client_data)
{
    (void)client_data;
    fprintf(stdout, "[INFO] %s", msg);
}

/* -------------------------------------------------------------------------- */

int main (int argc, char *argv[])
{
    opj_dparameters_t l_param;
    opj_codec_t * l_codec;
    opj_image_t * l_image;
    opj_stream_t * l_stream;
    OPJ_UINT32 l_data_size;
    OPJ_UINT32 l_max_data_size = 1000;
    OPJ_UINT32 l_tile_index;
    OPJ_BYTE * l_data = (OPJ_BYTE *) malloc(1000);
    OPJ_BOOL l_go_on = OPJ_TRUE;
    OPJ_UINT32 l_nb_comps=0 ;
    OPJ_INT32 l_current_tile_x0,l_current_tile_y0,l_current_tile_x1,l_current_tile_y1;

    int da_x0=0;
    int da_y0=0;
    int da_x1=1000;
    int da_y1=1000;
    char input_file[64];

    /* should be test_tile_decoder 0 0 1000 1000 tte1.j2k */
    if( argc == 6 ) {
        da_x0=atoi(argv[1]);
        da_y0=atoi(argv[2]);
        da_x1=atoi(argv[3]);
        da_y1=atoi(argv[4]);
        strcpy(input_file,argv[5]);

    } else {
        da_x0=0;
        da_y0=0;
        da_x1=1000;
        da_y1=1000;
        strcpy(input_file,"test.j2k");
    }

    if (! l_data) {
        return EXIT_FAILURE;
    }

    l_stream = opj_stream_create_default_file_stream(input_file,OPJ_TRUE);
    if (!l_stream) {
        free(l_data);
        fprintf(stderr, "ERROR -> failed to create the stream from the file\n");
        return EXIT_FAILURE;
    }

    /* Set the default decoding parameters */
    opj_set_default_decoder_parameters(&l_param);

    /* */
    l_param.decod_format = infile_format(input_file);

    /** you may here add custom decoding parameters */
    /* do not use layer decoding limitations */
    l_param.cp_layer = 0;

    /* do not use resolutions reductions */
    l_param.cp_reduce = 0;

    /* to decode only a part of the image data */
    /*opj_restrict_decoding(&l_param,0,0,1000,1000);*/


    switch(l_param.decod_format) {
    case J2K_CFMT: {	/* JPEG-2000 codestream */
        /* Get a decoder handle */
        l_codec = opj_create_decompress(OPJ_CODEC_J2K);
        break;
    }
    case JP2_CFMT: {	/* JPEG 2000 compressed image data */
        /* Get a decoder handle */
        l_codec = opj_create_decompress(OPJ_CODEC_JP2);
        break;
    }
    default: {
        fprintf(stderr, "ERROR -> Not a valid JPEG2000 file!\n");
        free(l_data);
        opj_stream_destroy(l_stream);
        return EXIT_FAILURE;
    }
    }

    /* catch events using our callbacks and give a local context */
    opj_set_info_handler(l_codec, info_callback,00);
    opj_set_warning_handler(l_codec, warning_callback,00);
    opj_set_error_handler(l_codec, error_callback,00);

    /* Setup the decoder decoding parameters using user parameters */
    if (! opj_setup_decoder(l_codec, &l_param)) {
        fprintf(stderr, "ERROR -> j2k_dump: failed to setup the decoder\n");
        free(l_data);
        opj_stream_destroy(l_stream);
        opj_destroy_codec(l_codec);
        return EXIT_FAILURE;
    }

    /* Read the main header of the codestream and if necessary the JP2 boxes*/
    if (! opj_read_header(l_stream, l_codec, &l_image)) {
        fprintf(stderr, "ERROR -> j2k_to_image: failed to read the header\n");
        free(l_data);
        opj_stream_destroy(l_stream);
        opj_destroy_codec(l_codec);
        return EXIT_FAILURE;
    }

    if (!opj_set_decode_area(l_codec, l_image, da_x0, da_y0,da_x1, da_y1)) {
        fprintf(stderr,	"ERROR -> j2k_to_image: failed to set the decoded area\n");
        free(l_data);
        opj_stream_destroy(l_stream);
        opj_destroy_codec(l_codec);
        opj_image_destroy(l_image);
        return EXIT_FAILURE;
    }


    while (l_go_on) {
        if (! opj_read_tile_header( l_codec,
                                    l_stream,
                                    &l_tile_index,
                                    &l_data_size,
                                    &l_current_tile_x0,
                                    &l_current_tile_y0,
                                    &l_current_tile_x1,
                                    &l_current_tile_y1,
                                    &l_nb_comps,
                                    &l_go_on)) {
            free(l_data);
            opj_stream_destroy(l_stream);
            opj_destroy_codec(l_codec);
            opj_image_destroy(l_image);
            return EXIT_FAILURE;
        }

        if (l_go_on) {
            if (l_data_size > l_max_data_size) {
                OPJ_BYTE *l_new_data = (OPJ_BYTE *) realloc(l_data, l_data_size);
                if (! l_new_data) {
                    free(l_new_data);
                    opj_stream_destroy(l_stream);
                    opj_destroy_codec(l_codec);
                    opj_image_destroy(l_image);
                    return EXIT_FAILURE;
                }
                l_data = l_new_data;
                l_max_data_size = l_data_size;
            }

            if (! opj_decode_tile_data(l_codec,l_tile_index,l_data,l_data_size,l_stream)) {
                free(l_data);
                opj_stream_destroy(l_stream);
                opj_destroy_codec(l_codec);
                opj_image_destroy(l_image);
                return EXIT_FAILURE;
            }
            /** now should inspect image to know the reduction factor and then how to behave with data */
        }
    }

    if (! opj_end_decompress(l_codec,l_stream)) {
        free(l_data);
        opj_stream_destroy(l_stream);
        opj_destroy_codec(l_codec);
        opj_image_destroy(l_image);
        return EXIT_FAILURE;
    }

    /* Free memory */
    free(l_data);
    opj_stream_destroy(l_stream);
    opj_destroy_codec(l_codec);
    opj_image_destroy(l_image);

    /* Print profiling*/
    /*PROFPRINT();*/

    return EXIT_SUCCESS;
}