src/lib/openjpip/jpipstream_manager.c

   1 /*
   2  * $Id$
   3  *
   4  * Copyright (c) 2002-2014, Universite catholique de Louvain (UCL), Belgium
   5  * Copyright (c) 2002-2014, Professor Benoit Macq
   6  * Copyright (c) 2010-2011, Kaori Hagihara
   7  * All rights reserved.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
  19  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  22  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  23  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  24  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  25  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  26  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  27  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 #include <time.h>
  35 #include "jpipstream_manager.h"
  36 #include "jp2k_encoder.h"
  37 #include "jp2k_decoder.h"
  38 #include "ihdrbox_manager.h"
  39 #include "j2kheader_manager.h"
  40
  41 Byte_t * update_JPIPstream(Byte_t *newstream, OPJ_SIZE_T newstreamlen,
  42                            Byte_t *cache_stream, OPJ_SIZE_T *streamlen)
  43 {
  44     Byte_t *stream = (Byte_t *)opj_malloc((*streamlen) + newstreamlen);
  45     if (*streamlen > 0) {
  46         memcpy(stream, cache_stream, *streamlen);
  47     }
  48     memcpy(stream + (*streamlen), newstream, newstreamlen);
  49     *streamlen += newstreamlen;
  50
  51     if (cache_stream) {
  52         opj_free(cache_stream);
  53     }
  54
  55     return stream;
  56 }
  57
  58 void save_codestream(Byte_t *codestream, OPJ_SIZE_T streamlen, const char *fmt)
  59 {
  60     time_t timer;
  61     struct tm *t_st;
  62     char filename[20];
  63     FILE *fp;
  64
  65     time(&timer);
  66     t_st = localtime(&timer);
  67
  68     sprintf(filename, "%4d%02d%02d%02d%02d%02d.%.3s", t_st->tm_year + 1900,
  69             t_st->tm_mon + 1, t_st->tm_mday, t_st->tm_hour, t_st->tm_min, t_st->tm_sec,
  70             fmt);
  71
  72     fp = fopen(filename, "wb");
  73     if (fwrite(codestream, streamlen, 1, fp) != 1) {
  74         fprintf(stderr, "Error: failed to write codestream to file %s\n", filename);
  75     }
  76     fclose(fp);
  77 }
  78
  79
  80 Byte_t * jpipstream_to_pnm(Byte_t *jpipstream, msgqueue_param_t *msgqueue,
  81                            Byte8_t csn, int fw, int fh, ihdrbox_param_t **ihdrbox)
  82 {
  83     Byte_t *pnmstream;
  84     Byte_t *j2kstream; /* j2k or jp2 codestream */
  85     Byte8_t j2klen;
  86     size_t retlen;
  87     FILE *fp;
  88     const char j2kfname[] = "tmp.j2k";
  89
  90     fp = fopen(j2kfname, "w+b");
  91     if (!fp) {
  92         return NULL;
  93     }
  94     j2kstream = recons_j2k(msgqueue, jpipstream, csn, fw, fh, &j2klen);
  95     if (!j2kstream) {
  96         fclose(fp);
  97         remove(j2kfname);
  98         return NULL;
  99     }
 100
 101     retlen = fwrite(j2kstream, 1, j2klen, fp);
 102     opj_free(j2kstream);
 103     fclose(fp);
 104     if (retlen != j2klen) {
 105         remove(j2kfname);
 106         return NULL;
 107     }
 108
 109     pnmstream = j2k_to_pnm(j2kfname, ihdrbox);
 110
 111     remove(j2kfname);
 112
 113     return pnmstream;
 114 }
 115
 116 ihdrbox_param_t * get_SIZ_from_jpipstream(Byte_t *jpipstream,
 117         msgqueue_param_t *msgqueue, Byte8_t csn)
 118 {
 119     ihdrbox_param_t *ihdrbox;
 120     Byte_t *j2kstream;
 121     Byte8_t j2klen;
 122     SIZmarker_param_t SIZ;
 123
 124     j2kstream = recons_j2kmainhead(msgqueue, jpipstream, csn, &j2klen);
 125     if (!get_mainheader_from_j2kstream(j2kstream, &SIZ, NULL)) {
 126         opj_free(j2kstream);
 127         return NULL;
 128     }
 129
 130     ihdrbox = (ihdrbox_param_t *)opj_malloc(sizeof(ihdrbox_param_t));
 131
 132     ihdrbox->width = SIZ.Xsiz;
 133     ihdrbox->height = SIZ.Ysiz;
 134     ihdrbox->nc = SIZ.Csiz;
 135     ihdrbox->bpc = SIZ.Ssiz[0];
 136
 137     opj_free(j2kstream);
 138
 139     return ihdrbox;
 140 }