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, Byte_t *cache_stream, OPJ_SIZE_T *streamlen)
  42 {
  43   Byte_t *stream = (Byte_t *)opj_malloc( (*streamlen)+newstreamlen);
  44   if( *streamlen > 0)
  45     memcpy( stream, cache_stream, *streamlen);
  46   memcpy( stream+(*streamlen), newstream, newstreamlen);
  47   *streamlen += newstreamlen;
  48
  49   if(cache_stream)
  50     opj_free( cache_stream);
  51
  52   return stream;
  53 }
  54
  55 void save_codestream( Byte_t *codestream, OPJ_SIZE_T streamlen, const char *fmt)
  56 {
  57   time_t timer;
  58   struct tm *t_st;
  59   char filename[20];
  60   FILE *fp;
  61
  62   time(&timer);
  63   t_st = localtime( &timer);
  64
  65   sprintf( filename, "%4d%02d%02d%02d%02d%02d.%.3s", t_st->tm_year+1900, t_st->tm_mon+1, t_st->tm_mday, t_st->tm_hour, t_st->tm_min, t_st->tm_sec, fmt);
  66
  67   fp = fopen( filename, "wb");
  68   if( fwrite( codestream, streamlen, 1, fp) != 1)
  69     fprintf( stderr, "Error: failed to write codestream to file %s\n", filename);
  70   fclose( fp);
  71 }
  72
  73
  74 Byte_t * jpipstream_to_pnm( Byte_t *jpipstream, msgqueue_param_t *msgqueue, Byte8_t csn, int fw, int fh, ihdrbox_param_t **ihdrbox)
  75 {
  76   Byte_t *pnmstream;
  77   Byte_t *j2kstream; /* j2k or jp2 codestream */
  78   Byte8_t j2klen;
  79   size_t retlen;
  80   FILE *fp;
  81   const char j2kfname[] = "tmp.j2k";
  82
  83   fp = fopen( j2kfname, "w+b");
  84   if( !fp )
  85     {
  86     return NULL;
  87     }
  88   j2kstream = recons_j2k( msgqueue, jpipstream, csn, fw, fh, &j2klen);
  89   if( !j2kstream )
  90     {
  91     fclose(fp);
  92     remove( j2kfname);
  93     return NULL;
  94     }
  95
  96   retlen = fwrite( j2kstream, 1, j2klen, fp);
  97   opj_free( j2kstream);
  98   fclose(fp);
  99   if( retlen != j2klen )
 100     {
 101     remove( j2kfname);
 102     return NULL;
 103     }
 104
 105   pnmstream = j2k_to_pnm( j2kfname, ihdrbox);
 106
 107   remove( j2kfname);
 108
 109   return pnmstream;
 110 }
 111
 112 ihdrbox_param_t * get_SIZ_from_jpipstream( Byte_t *jpipstream, msgqueue_param_t *msgqueue, Byte8_t csn)
 113 {
 114   ihdrbox_param_t *ihdrbox;
 115   Byte_t *j2kstream;
 116   Byte8_t j2klen;
 117   SIZmarker_param_t SIZ;
 118
 119   j2kstream = recons_j2kmainhead( msgqueue, jpipstream, csn, &j2klen);
 120   if( !get_mainheader_from_j2kstream( j2kstream, &SIZ, NULL)){
 121     opj_free( j2kstream);
 122     return NULL;
 123   }
 124
 125   ihdrbox = (ihdrbox_param_t *)opj_malloc( sizeof(ihdrbox_param_t));
 126
 127   ihdrbox->width = SIZ.Xsiz;
 128   ihdrbox->height = SIZ.Ysiz;
 129   ihdrbox->nc = SIZ.Csiz;
 130   ihdrbox->bpc = SIZ.Ssiz[0];
 131
 132   opj_free( j2kstream);
 133
 134   return ihdrbox;
 135 }