libopenjpeg/cio.c

   1 /*
   2  * Copyright (c) 2002-2007, Communications and Remote Sensing Laboratory, Universite catholique de Louvain (UCL), Belgium
   3  * Copyright (c) 2002-2007, Professor Benoit Macq
   4  * Copyright (c) 2001-2003, David Janssens
   5  * Copyright (c) 2002-2003, Yannick Verschueren
   6  * Copyright (c) 2003-2007, Francois-Olivier Devaux and Antonin Descampe
   7  * Copyright (c) 2005, Herve Drolon, FreeImage Team
   8  * All rights reserved.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
  20  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  23  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #include "opj_includes.h"
  33
  34 /* ----------------------------------------------------------------------- */
  35
  36 opj_cio_t* OPJ_CALLCONV opj_cio_open(opj_common_ptr cinfo, unsigned char *buffer, int length) {
  37         opj_cp_t *cp = NULL;
  38         opj_cio_t *cio = (opj_cio_t*)opj_malloc(sizeof(opj_cio_t));
  39         if(!cio) return NULL;
  40         cio->cinfo = cinfo;
  41         if(buffer && length) {
  42                 /* wrap a user buffer containing the encoded image */
  43                 cio->openmode = OPJ_STREAM_READ;
  44                 cio->buffer = buffer;
  45                 cio->length = length;
  46         }
  47         else if(!buffer && !length && cinfo) {
  48                 /* allocate a buffer for the encoded image */
  49                 cio->openmode = OPJ_STREAM_WRITE;
  50                 switch(cinfo->codec_format) {
  51                         case CODEC_J2K:
  52                                 cp = ((opj_j2k_t*)cinfo->j2k_handle)->cp;
  53                                 break;
  54                         case CODEC_JP2:
  55                                 cp = ((opj_jp2_t*)cinfo->jp2_handle)->j2k->cp;
  56                                 break;
  57                         default:
  58                                 opj_free(cio);
  59                                 return NULL;
  60                 }
  61                 cio->length = (unsigned int) (0.1625 * cp->img_size + 2000); /* 0.1625 = 1.3/8 and 2000 bytes as a minimum for headers */
  62                 cio->buffer = (unsigned char *)opj_malloc(cio->length);
  63                 if(!cio->buffer) {
  64                         opj_event_msg(cio->cinfo, EVT_ERROR, "Error allocating memory for compressed bitstream\n");
  65                         opj_free(cio);
  66                         return NULL;
  67                 }
  68         }
  69         else {
  70                 opj_free(cio);
  71                 return NULL;
  72         }
  73
  74         /* Initialize byte IO */
  75         cio->start = cio->buffer;
  76         cio->end = cio->buffer + cio->length;
  77         cio->bp = cio->buffer;
  78
  79         return cio;
  80 }
  81
  82 void OPJ_CALLCONV opj_cio_close(opj_cio_t *cio) {
  83         if(cio) {
  84                 if(cio->openmode == OPJ_STREAM_WRITE) {
  85                         /* destroy the allocated buffer */
  86                         opj_free(cio->buffer);
  87                 }
  88                 /* destroy the cio */
  89                 opj_free(cio);
  90         }
  91 }
  92
  93
  94 /* ----------------------------------------------------------------------- */
  95
  96 /*
  97  * Get position in byte stream.
  98  */
  99 int OPJ_CALLCONV cio_tell(opj_cio_t *cio) {
 100         return cio->bp - cio->start;
 101 }
 102
 103 /*
 104  * Set position in byte stream.
 105  *
 106  * pos : position, in number of bytes, from the beginning of the stream
 107  */
 108 void OPJ_CALLCONV cio_seek(opj_cio_t *cio, int pos) {
 109         cio->bp = cio->start + pos;
 110 }
 111
 112 /*
 113  * Number of bytes left before the end of the stream.
 114  */
 115 int cio_numbytesleft(opj_cio_t *cio) {
 116         return cio->end - cio->bp;
 117 }
 118
 119 /*
 120  * Get pointer to the current position in the stream.
 121  */
 122 unsigned char *cio_getbp(opj_cio_t *cio) {
 123         return cio->bp;
 124 }
 125
 126 /*
 127  * Write a byte.
 128  */
 129 opj_bool cio_byteout(opj_cio_t *cio, unsigned char v) {
 130         if (cio->bp >= cio->end) {
 131                 opj_event_msg(cio->cinfo, EVT_ERROR, "write error\n");
 132                 return OPJ_FALSE;
 133         }
 134         *cio->bp++ = v;
 135         return OPJ_TRUE;
 136 }
 137
 138 /*
 139  * Read a byte.
 140  */
 141 unsigned char cio_bytein(opj_cio_t *cio) {
 142         if (cio->bp >= cio->end) {
 143                 opj_event_msg(cio->cinfo, EVT_ERROR, "read error: passed the end of the codestream (start = %d, current = %d, end = %d\n", cio->start, cio->bp, cio->end);
 144                 return 0;
 145         }
 146         return *cio->bp++;
 147 }
 148
 149 /*
 150  * Write some bytes.
 151  *
 152  * v : value to write
 153  * n : number of bytes to write
 154  */
 155 unsigned int cio_write(opj_cio_t *cio, unsigned long long int v, int n) {
 156         int i;
 157         for (i = n - 1; i >= 0; i--) {
 158                 if( !cio_byteout(cio, (unsigned char) ((v >> (i << 3)) & 0xff)) )
 159                         return 0;
 160         }
 161         return n;
 162 }
 163
 164 /*
 165  * Read some bytes.
 166  *
 167  * n : number of bytes to read
 168  *
 169  * return : value of the n bytes read
 170  */
 171 unsigned int cio_read(opj_cio_t *cio, int n) {
 172         int i;
 173         unsigned int v;
 174         v = 0;
 175         for (i = n - 1; i >= 0; i--) {
 176                 v += cio_bytein(cio) << (i << 3);
 177         }
 178         return v;
 179 }
 180
 181 /*
 182  * Skip some bytes.
 183  *
 184  * n : number of bytes to skip
 185  */
 186 void cio_skip(opj_cio_t *cio, int n) {
 187         cio->bp += n;
 188 }
 189
 190
 191 /* ----------------------------------------------------------------------- */
 192
 193
 194 /**
 195  * Write some bytes to the given data buffer, this function is used in Big Endian cpus.
 196  * @param p_buffer              pointer the data buffer to write data to.
 197  * @param p_value               the value to write
 198  * @param p_nb_bytes    the number of bytes to write
 199 */
 200 void opj_write_bytes_BE (OPJ_BYTE * p_buffer, OPJ_UINT32 p_value, OPJ_UINT32 p_nb_bytes)
 201 {
 202         const OPJ_BYTE * l_data_ptr = ((const OPJ_BYTE *) &p_value) + p_nb_bytes;
 203
 204         assert(p_nb_bytes > 0 && p_nb_bytes <=  sizeof(OPJ_UINT32));
 205
 206         memcpy(p_buffer,l_data_ptr,p_nb_bytes);
 207 }
 208
 209 /**
 210  * Write some bytes to the given data buffer, this function is used in Little Endian cpus.
 211  * @param p_buffer              pointer the data buffer to write data to.
 212  * @param p_value               the value to write
 213  * @param p_nb_bytes    the number of bytes to write
 214  * @return                              the number of bytes written or -1 if an error occured
 215 */
 216 void opj_write_bytes_LE (OPJ_BYTE * p_buffer, OPJ_UINT32 p_value, OPJ_UINT32 p_nb_bytes)
 217 {
 218         const OPJ_BYTE * l_data_ptr = ((const OPJ_BYTE *) &p_value) + p_nb_bytes - 1;
 219         OPJ_UINT32 i;
 220
 221         assert(p_nb_bytes > 0 && p_nb_bytes <= sizeof(OPJ_UINT32));
 222
 223         for     (i=0;i<p_nb_bytes;++i) {
 224                 *(p_buffer++) = *(l_data_ptr--);
 225         }
 226 }
 227
 228 /**
 229  * Reads some bytes from the given data buffer, this function is used in Big Endian cpus.
 230  * @param p_buffer              pointer the data buffer to read data from.
 231  * @param p_value               pointer to the value that will store the data.
 232  * @param p_nb_bytes    the nb bytes to read.
 233  * @return                              the number of bytes read or -1 if an error occured.
 234  */
 235 void opj_read_bytes_BE(const OPJ_BYTE * p_buffer, OPJ_UINT32 * p_value, OPJ_UINT32 p_nb_bytes)
 236 {
 237         OPJ_BYTE * l_data_ptr = ((OPJ_BYTE *) p_value);
 238
 239         assert(p_nb_bytes > 0 && p_nb_bytes <= sizeof(OPJ_UINT32));
 240
 241         *p_value = 0;
 242         memcpy(l_data_ptr+4-p_nb_bytes,p_buffer,p_nb_bytes);
 243 }
 244
 245 /**
 246  * Reads some bytes from the given data buffer, this function is used in Little Endian cpus.
 247  * @param p_buffer              pointer the data buffer to read data from.
 248  * @param p_value               pointer to the value that will store the data.
 249  * @param p_nb_bytes    the nb bytes to read.
 250  * @return                              the number of bytes read or -1 if an error occured.
 251  */
 252 void opj_read_bytes_LE(const OPJ_BYTE * p_buffer, OPJ_UINT32 * p_value, OPJ_UINT32 p_nb_bytes)
 253 {
 254         OPJ_BYTE * l_data_ptr = ((OPJ_BYTE *) p_value) + p_nb_bytes-1;
 255         OPJ_UINT32 i;
 256
 257         assert(p_nb_bytes > 0 && p_nb_bytes <= sizeof(OPJ_UINT32));
 258
 259         *p_value = 0;
 260         for (i=0;i<p_nb_bytes;++i) {
 261                 *(l_data_ptr--) = *(p_buffer++);
 262         }
 263 }
 264
 265 /**
 266  * Write some bytes to the given data buffer, this function is used in Big Endian cpus.
 267  * @param p_buffer              pointer the data buffer to write data to.
 268  * @param p_value               the value to write
 269  * @return                              the number of bytes written or -1 if an error occured
 270  */
 271 void opj_write_double_BE(OPJ_BYTE * p_buffer, OPJ_FLOAT64 p_value)
 272 {
 273         const OPJ_BYTE * l_data_ptr = ((const OPJ_BYTE *) &p_value);
 274         memcpy(p_buffer,l_data_ptr,sizeof(OPJ_FLOAT64));
 275 }
 276
 277 /**
 278  * Write some bytes to the given data buffer, this function is used in Little Endian cpus.
 279  * @param p_buffer              pointer the data buffer to write data to.
 280  * @param p_value               the value to write
 281  */
 282 void opj_write_double_LE(OPJ_BYTE * p_buffer, OPJ_FLOAT64 p_value)
 283 {
 284         const OPJ_BYTE * l_data_ptr = ((const OPJ_BYTE *) &p_value) + sizeof(OPJ_FLOAT64) - 1;
 285         OPJ_UINT32 i;
 286         for     (i=0;i<sizeof(OPJ_FLOAT64);++i) {
 287                 *(p_buffer++) = *(l_data_ptr--);
 288         }
 289 }
 290
 291 /**
 292  * Reads some bytes from the given data buffer, this function is used in Big Endian cpus.
 293  * @param p_buffer              pointer the data buffer to read data from.
 294  * @param p_value               pointer to the value that will store the data.
 295  */
 296 void opj_read_double_BE(const OPJ_BYTE * p_buffer, OPJ_FLOAT64 * p_value)
 297 {
 298         OPJ_BYTE * l_data_ptr = ((OPJ_BYTE *) p_value);
 299         memcpy(l_data_ptr,p_buffer,sizeof(OPJ_FLOAT64));
 300 }
 301
 302
 303 /**
 304  * Reads some bytes from the given data buffer, this function is used in Little Endian cpus.
 305  * @param p_buffer              pointer the data buffer to read data from.
 306  * @param p_value               pointer to the value that will store the data.
 307  */
 308 void opj_read_double_LE(const OPJ_BYTE * p_buffer, OPJ_FLOAT64 * p_value)
 309 {
 310         OPJ_BYTE * l_data_ptr = ((OPJ_BYTE *) p_value) + sizeof(OPJ_FLOAT64)-1;
 311         OPJ_UINT32 i;
 312         for (i=0;i<sizeof(OPJ_FLOAT64);++i) {
 313                 *(l_data_ptr--) = *(p_buffer++);
 314         }
 315 }
 316
 317 /**
 318  * Write some bytes to the given data buffer, this function is used in Big Endian cpus.
 319  * @param p_buffer              pointer the data buffer to write data to.
 320  * @param p_value               the value to write
 321  * @return                              the number of bytes written or -1 if an error occured
 322  */
 323 void opj_write_float_BE(OPJ_BYTE * p_buffer, OPJ_FLOAT32 p_value)
 324 {
 325         const OPJ_BYTE * l_data_ptr = ((const OPJ_BYTE *) &p_value);
 326         memcpy(p_buffer,l_data_ptr,sizeof(OPJ_FLOAT32));
 327 }
 328
 329 /**
 330  * Write some bytes to the given data buffer, this function is used in Little Endian cpus.
 331  * @param p_buffer              pointer the data buffer to write data to.
 332  * @param p_value               the value to write
 333  */
 334 void opj_write_float_LE(OPJ_BYTE * p_buffer, OPJ_FLOAT32 p_value)
 335 {
 336         const OPJ_BYTE * l_data_ptr = ((const OPJ_BYTE *) &p_value) + sizeof(OPJ_FLOAT32) - 1;
 337         OPJ_UINT32 i;
 338         for     (i=0;i<sizeof(OPJ_FLOAT32);++i) {
 339                 *(p_buffer++) = *(l_data_ptr--);
 340         }
 341 }
 342
 343 /**
 344  * Reads some bytes from the given data buffer, this function is used in Big Endian cpus.
 345  * @param p_buffer              pointer the data buffer to read data from.
 346  * @param p_value               pointer to the value that will store the data.
 347  */
 348 void opj_read_float_BE(const OPJ_BYTE * p_buffer, OPJ_FLOAT32 * p_value)
 349 {
 350         OPJ_BYTE * l_data_ptr = ((OPJ_BYTE *) p_value);
 351         memcpy(l_data_ptr,p_buffer,sizeof(OPJ_FLOAT32));
 352 }
 353
 354
 355 /**
 356  * Reads some bytes from the given data buffer, this function is used in Little Endian cpus.
 357  * @param p_buffer              pointer the data buffer to read data from.
 358  * @param p_value               pointer to the value that will store the data.
 359  */
 360 void opj_read_float_LE(const OPJ_BYTE * p_buffer, OPJ_FLOAT32 * p_value)
 361 {
 362         OPJ_BYTE * l_data_ptr = ((OPJ_BYTE *) p_value) + sizeof(OPJ_FLOAT32)-1;
 363         OPJ_UINT32 i;
 364         for     (i=0;i<sizeof(OPJ_FLOAT32);++i) {
 365                 *(l_data_ptr--) = *(p_buffer++);
 366         }
 367 }
 368
 369
 370 /**
 371  * Creates an abstract stream. This function does nothing except allocating memory and initializing the abstract stream.
 372  * @return a stream object.
 373 */
 374 opj_stream_t* OPJ_CALLCONV opj_stream_create(OPJ_UINT32 p_size,opj_bool l_is_input)
 375 {
 376         opj_stream_private_t * l_stream = 00;
 377         l_stream = (opj_stream_private_t*) opj_malloc(sizeof(opj_stream_private_t));
 378         if (! l_stream) {
 379                 return 00;
 380         }
 381
 382         memset(l_stream,0,sizeof(opj_stream_private_t));
 383         l_stream->m_buffer_size = p_size;
 384         l_stream->m_stored_data = (OPJ_BYTE *) opj_malloc(p_size);
 385         if (! l_stream->m_stored_data) {
 386                 opj_free(l_stream);
 387                 return 00;
 388         }
 389
 390         l_stream->m_current_data = l_stream->m_stored_data;
 391
 392         if (l_is_input) {
 393                 l_stream->m_status |= opj_stream_e_input;
 394                 l_stream->m_opj_skip = opj_stream_read_skip;
 395                 l_stream->m_opj_seek = opj_stream_read_seek;
 396         }
 397         else {
 398                 l_stream->m_status |= opj_stream_e_output;
 399                 l_stream->m_opj_skip = opj_stream_write_skip;
 400                 l_stream->m_opj_seek = opj_stream_write_seek;
 401         }
 402
 403         l_stream->m_read_fn = opj_stream_default_read;
 404         l_stream->m_write_fn = opj_stream_default_write;
 405         l_stream->m_skip_fn = opj_stream_default_skip;
 406         l_stream->m_seek_fn = opj_stream_default_seek;
 407
 408         return (opj_stream_t *) l_stream;
 409 }
 410
 411 /**
 412  * Creates an abstract stream. This function does nothing except allocating memory and initializing the abstract stream.
 413  * @return a stream object.
 414 */
 415 opj_stream_t* OPJ_CALLCONV opj_stream_default_create(opj_bool l_is_input)
 416 {
 417         return opj_stream_create(J2K_STREAM_CHUNK_SIZE,l_is_input);
 418 }
 419
 420 /**
 421  * Destroys a stream created by opj_create_stream. This function does NOT close the abstract stream. If needed the user must
 422  * close its own implementation of the stream.
 423  */
 424 OPJ_API void OPJ_CALLCONV opj_stream_destroy(opj_stream_t* p_stream)
 425 {
 426         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
 427         if (l_stream) {
 428                 opj_free(l_stream->m_stored_data);
 429                 l_stream->m_stored_data = 00;
 430                 opj_free(l_stream);
 431         }
 432 }
 433
 434 /**
 435  * Sets the given function to be used as a read function.
 436  * @param               p_stream        the stream to modify
 437  * @param               p_function      the function to use a read function.
 438 */
 439 OPJ_API void OPJ_CALLCONV opj_stream_set_read_function(opj_stream_t* p_stream, opj_stream_read_fn p_function)
 440 {
 441         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
 442
 443         if ((!l_stream) || (! (l_stream->m_status & opj_stream_e_input))) {
 444                 return;
 445         }
 446
 447         l_stream->m_read_fn = p_function;
 448 }
 449
 450 OPJ_API void OPJ_CALLCONV opj_stream_set_seek_function(opj_stream_t* p_stream, opj_stream_seek_fn p_function)
 451 {
 452         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
 453         if
 454                 (!l_stream)
 455         {
 456                 return;
 457         }
 458         l_stream->m_seek_fn = p_function;
 459 }
 460
 461 /**
 462  * Sets the given function to be used as a write function.
 463  * @param               p_stream        the stream to modify
 464  * @param               p_function      the function to use a write function.
 465 */
 466 OPJ_API void OPJ_CALLCONV opj_stream_set_write_function(opj_stream_t* p_stream, opj_stream_write_fn p_function)
 467 {
 468         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
 469         if
 470                 ((!l_stream )|| (! (l_stream->m_status & opj_stream_e_output)))
 471         {
 472                 return;
 473         }
 474         l_stream->m_write_fn = p_function;
 475 }
 476
 477 /**
 478  * Sets the given function to be used as a skip function.
 479  * @param               p_stream        the stream to modify
 480  * @param               p_function      the function to use a skip function.
 481 */
 482 OPJ_API void OPJ_CALLCONV opj_stream_set_skip_function(opj_stream_t* p_stream, opj_stream_skip_fn p_function)
 483 {
 484         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
 485         if
 486                 (! l_stream)
 487         {
 488                 return;
 489         }
 490         l_stream->m_skip_fn = p_function;
 491 }
 492
 493 /**
 494  * Sets the given data to be used as a user data for the stream.
 495  * @param               p_stream        the stream to modify
 496  * @param               p_data          the data to set.
 497 */
 498 OPJ_API void OPJ_CALLCONV opj_stream_set_user_data(opj_stream_t* p_stream, void * p_data)
 499 {
 500         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
 501         l_stream->m_user_data = p_data;
 502 }
 503
 504 /**
 505  * Sets the given data to be used as a user data for the stream.
 506  * @param               p_stream        the stream to modify
 507  * @param               p_data          the data to set.
 508 */
 509 OPJ_API void OPJ_CALLCONV opj_stream_set_user_data_length(opj_stream_t* p_stream, OPJ_UINT32 data_length)
 510 {
 511         opj_stream_private_t* l_stream = (opj_stream_private_t*) p_stream;
 512
 513         l_stream->m_user_data_length = data_length;
 514 }
 515
 516 /**
 517  * Reads some bytes from the stream.
 518  * @param               p_stream        the stream to read data from.
 519  * @param               p_buffer        pointer to the data buffer that will receive the data.
 520  * @param               p_size          number of bytes to read.
 521  * @param               p_event_mgr     the user event manager to be notified of special events.
 522  * @return              the number of bytes read, or -1 if an error occured or if the stream is at the end.
 523  */
 524 OPJ_UINT32 opj_stream_read_data (opj_stream_private_t * p_stream,OPJ_BYTE * p_buffer, OPJ_UINT32 p_size, opj_event_mgr_t * p_event_mgr)
 525 {
 526         OPJ_UINT32 l_read_nb_bytes = 0;
 527         if (p_stream->m_bytes_in_buffer >= p_size) {
 528                 memcpy(p_buffer,p_stream->m_current_data,p_size);
 529                 p_stream->m_current_data += p_size;
 530                 p_stream->m_bytes_in_buffer -= p_size;
 531                 l_read_nb_bytes += p_size;
 532                 p_stream->m_byte_offset += p_size;
 533                 return l_read_nb_bytes;
 534         }
 535
 536         /* we are now in the case when the remaining data if not sufficient */
 537         if (p_stream->m_status & opj_stream_e_end) {
 538                 l_read_nb_bytes += p_stream->m_bytes_in_buffer;
 539                 memcpy(p_buffer,p_stream->m_current_data,p_stream->m_bytes_in_buffer);
 540                 p_stream->m_current_data += p_stream->m_bytes_in_buffer;
 541                 p_stream->m_byte_offset += p_stream->m_bytes_in_buffer;
 542                 p_stream->m_bytes_in_buffer = 0;
 543                 return l_read_nb_bytes ? l_read_nb_bytes : -1;
 544         }
 545
 546         /* the flag is not set, we copy data and then do an actual read on the stream */
 547         if (p_stream->m_bytes_in_buffer) {
 548                 l_read_nb_bytes += p_stream->m_bytes_in_buffer;
 549                 memcpy(p_buffer,p_stream->m_current_data,p_stream->m_bytes_in_buffer);
 550                 p_stream->m_current_data = p_stream->m_stored_data;
 551                 p_buffer += p_stream->m_bytes_in_buffer;
 552                 p_size -= p_stream->m_bytes_in_buffer;
 553                 p_stream->m_byte_offset += p_stream->m_bytes_in_buffer;
 554                 p_stream->m_bytes_in_buffer = 0;
 555         }
 556         else {
 557     /* case where we are already at the end of the buffer
 558        so reset the m_current_data to point to the start of the
 559        stored buffer to get ready to read from disk*/
 560                 p_stream->m_current_data = p_stream->m_stored_data;
 561         }
 562
 563
 564         while(1){
 565                 /* we should read less than a chunk -> read a chunk */
 566                 if (p_size < p_stream->m_buffer_size) {
 567                         /* we should do an actual read on the media */
 568                         p_stream->m_bytes_in_buffer = p_stream->m_read_fn(p_stream->m_stored_data,p_stream->m_buffer_size,p_stream->m_user_data);
 569
 570                         if (p_stream->m_bytes_in_buffer == -1) {
 571                                 /* end of stream */
 572                                 opj_event_msg_v2(p_event_mgr, EVT_INFO, "Stream reached its end !\n");
 573
 574                                 p_stream->m_bytes_in_buffer = 0;
 575                                 p_stream->m_status |= opj_stream_e_end;
 576                                 /* end of stream */
 577                                 return l_read_nb_bytes ? l_read_nb_bytes : -1;
 578                         }
 579                         else if (p_stream->m_bytes_in_buffer < p_size) {
 580                                 /* not enough data */
 581                                 l_read_nb_bytes += p_stream->m_bytes_in_buffer;
 582                                 memcpy(p_buffer,p_stream->m_current_data,p_stream->m_bytes_in_buffer);
 583                                 p_stream->m_current_data = p_stream->m_stored_data;
 584                                 p_buffer += p_stream->m_bytes_in_buffer;
 585                                 p_size -= p_stream->m_bytes_in_buffer;
 586                                 p_stream->m_byte_offset += p_stream->m_bytes_in_buffer;
 587                                 p_stream->m_bytes_in_buffer = 0;
 588                         }
 589                         else {
 590                                 l_read_nb_bytes += p_size;
 591                                 memcpy(p_buffer,p_stream->m_current_data,p_size);
 592                                 p_stream->m_current_data += p_size;
 593                                 p_stream->m_bytes_in_buffer -= p_size;
 594                                 p_stream->m_byte_offset += p_size;
 595                                 return l_read_nb_bytes;
 596                         }
 597                 }
 598                 else {
 599                         /* direct read on the dest buffer */
 600                         p_stream->m_bytes_in_buffer = p_stream->m_read_fn(p_buffer,p_size,p_stream->m_user_data);
 601
 602                         if (p_stream->m_bytes_in_buffer == -1) {
 603                                 /*  end of stream */
 604                                 opj_event_msg_v2(p_event_mgr, EVT_INFO, "Stream reached its end !\n");
 605
 606                                 p_stream->m_bytes_in_buffer = 0;
 607                                 p_stream->m_status |= opj_stream_e_end;
 608                                 /* end of stream */
 609                                 return l_read_nb_bytes ? l_read_nb_bytes : -1;
 610                         }
 611                         else if (p_stream->m_bytes_in_buffer < p_size) {
 612                                 /* not enough data */
 613                                 l_read_nb_bytes += p_stream->m_bytes_in_buffer;
 614                                 p_stream->m_current_data = p_stream->m_stored_data;
 615                                 p_buffer += p_stream->m_bytes_in_buffer;
 616                                 p_size -= p_stream->m_bytes_in_buffer;
 617                                 p_stream->m_byte_offset += p_stream->m_bytes_in_buffer;
 618                                 p_stream->m_bytes_in_buffer = 0;
 619                         }
 620                         else {
 621                                 /* we have read the exact size */
 622                                 l_read_nb_bytes += p_stream->m_bytes_in_buffer;
 623                                 p_stream->m_byte_offset += p_stream->m_bytes_in_buffer;
 624                                 p_stream->m_current_data = p_stream->m_stored_data;
 625                                 p_stream->m_bytes_in_buffer = 0;
 626                                 return l_read_nb_bytes;
 627                         }
 628                 }
 629         }
 630 }
 631
 632 /**
 633  * Writes some bytes from the stream.
 634  * @param               p_stream        the stream to write data to.
 635  * @param               p_buffer        pointer to the data buffer holds the data to be writtent.
 636  * @param               p_size          number of bytes to write.
 637  * @param               p_event_mgr     the user event manager to be notified of special events.
 638  * @return              the number of bytes writtent, or -1 if an error occured.
 639  */
 640 OPJ_UINT32 opj_stream_write_data (opj_stream_private_t * p_stream,const OPJ_BYTE * p_buffer,OPJ_UINT32 p_size, opj_event_mgr_t * p_event_mgr)
 641 {
 642         OPJ_UINT32 l_remaining_bytes = 0;
 643         OPJ_UINT32 l_write_nb_bytes = 0;
 644
 645         if
 646                 (p_stream->m_status & opj_stream_e_error)
 647         {
 648                 return -1;
 649         }
 650
 651         while(1)
 652         {
 653                 l_remaining_bytes = p_stream->m_buffer_size - p_stream->m_bytes_in_buffer;
 654                 // we have more memory than required
 655                 if
 656                         (l_remaining_bytes >= p_size)
 657                 {
 658                         memcpy(p_stream->m_current_data,p_buffer,p_size);
 659                         p_stream->m_current_data += p_size;
 660                         p_stream->m_bytes_in_buffer += p_size;
 661                         l_write_nb_bytes += p_size;
 662                         p_stream->m_byte_offset += p_size;
 663                         return l_write_nb_bytes;
 664                 }
 665
 666                 // we copy data and then do an actual read on the stream
 667                 if
 668                         (l_remaining_bytes)
 669                 {
 670                         l_write_nb_bytes += l_remaining_bytes;
 671                         memcpy(p_stream->m_current_data,p_buffer,l_remaining_bytes);
 672                         p_stream->m_current_data = p_stream->m_stored_data;
 673                         p_buffer += l_remaining_bytes;
 674                         p_size -= l_remaining_bytes;
 675                         p_stream->m_bytes_in_buffer += l_remaining_bytes;
 676                         p_stream->m_byte_offset += l_remaining_bytes;
 677                 }
 678                 if
 679                         (! opj_stream_flush(p_stream, p_event_mgr))
 680                 {
 681                         return -1;
 682                 }
 683         }
 684
 685 }
 686
 687 /**
 688  * Writes the content of the stream buffer to the stream.
 689  * @param               p_stream        the stream to write data to.
 690  * @param               p_event_mgr     the user event manager to be notified of special events.
 691  * @return              the number of bytes written, or -1 if an error occured.
 692  */
 693 opj_bool opj_stream_flush (opj_stream_private_t * p_stream, opj_event_mgr_t * p_event_mgr)
 694 {
 695         // the number of bytes written on the media.
 696         OPJ_UINT32 l_current_write_nb_bytes = 0;
 697         p_stream->m_current_data = p_stream->m_stored_data;
 698
 699         while
 700                 (p_stream->m_bytes_in_buffer)
 701         {
 702                 // we should do an actual write on the media
 703                 l_current_write_nb_bytes = p_stream->m_write_fn(p_stream->m_current_data,p_stream->m_bytes_in_buffer,p_stream->m_user_data);
 704                 if
 705                         (l_current_write_nb_bytes == -1)
 706                 {
 707                         p_stream->m_status |= opj_stream_e_error;
 708                         opj_event_msg_v2(p_event_mgr, EVT_INFO, "Error on writting stream!\n");
 709
 710                         return EXIT_FAILURE;
 711                 }
 712                 p_stream->m_current_data += l_current_write_nb_bytes;
 713                 p_stream->m_bytes_in_buffer -= l_current_write_nb_bytes;
 714         }
 715         p_stream->m_current_data = p_stream->m_stored_data;
 716         return EXIT_SUCCESS;
 717 }
 718
 719 /**
 720  * Skips a number of bytes from the stream.
 721  * @param               p_stream        the stream to skip data from.
 722  * @param               p_size          the number of bytes to skip.
 723  * @param               p_event_mgr     the user event manager to be notified of special events.
 724  * @return              the number of bytes skipped, or -1 if an error occured.
 725  */
 726 OPJ_SIZE_T opj_stream_read_skip (opj_stream_private_t * p_stream, OPJ_SIZE_T p_size, opj_event_mgr_t * p_event_mgr)
 727 {
 728         OPJ_SIZE_T l_skip_nb_bytes = 0;
 729         OPJ_SIZE_T l_current_skip_nb_bytes = 0;
 730
 731         if
 732                 (p_stream->m_bytes_in_buffer >= p_size)
 733         {
 734                 p_stream->m_current_data += p_size;
 735                 p_stream->m_bytes_in_buffer -= p_size;
 736                 l_skip_nb_bytes += p_size;
 737                 p_stream->m_byte_offset += l_skip_nb_bytes;
 738                 return l_skip_nb_bytes;
 739         }
 740
 741         // we are now in the case when the remaining data if not sufficient
 742         if
 743                 (p_stream->m_status & opj_stream_e_end)
 744         {
 745                 l_skip_nb_bytes += p_stream->m_bytes_in_buffer;
 746                 p_stream->m_current_data += p_stream->m_bytes_in_buffer;
 747                 p_stream->m_bytes_in_buffer = 0;
 748                 p_stream->m_byte_offset += l_skip_nb_bytes;
 749                 return l_skip_nb_bytes ? l_skip_nb_bytes : (OPJ_SIZE_T) -1;
 750         }
 751
 752         // the flag is not set, we copy data and then do an actual skip on the stream
 753         if
 754                 (p_stream->m_bytes_in_buffer)
 755         {
 756                 l_skip_nb_bytes += p_stream->m_bytes_in_buffer;
 757                 p_stream->m_current_data = p_stream->m_stored_data;
 758                 p_size -= p_stream->m_bytes_in_buffer;
 759                 p_stream->m_bytes_in_buffer = 0;
 760         }
 761
 762         while
 763                 (p_size > 0)
 764         {
 765                 // we should do an actual skip on the media
 766                 l_current_skip_nb_bytes = p_stream->m_skip_fn(p_size, p_stream->m_user_data);
 767                 if
 768                         (l_current_skip_nb_bytes == (OPJ_SIZE_T) -1)
 769                 {
 770                         opj_event_msg_v2(p_event_mgr, EVT_INFO, "Stream reached its end !\n");
 771
 772                         p_stream->m_status |= opj_stream_e_end;
 773                         p_stream->m_byte_offset += l_skip_nb_bytes;
 774                         // end if stream
 775                         return l_skip_nb_bytes ? l_skip_nb_bytes : (OPJ_SIZE_T) -1;
 776                 }
 777                 p_size -= l_current_skip_nb_bytes;
 778                 l_skip_nb_bytes += l_current_skip_nb_bytes;
 779         }
 780         p_stream->m_byte_offset += l_skip_nb_bytes;
 781         return l_skip_nb_bytes;
 782 }
 783
 784 /**
 785  * Skips a number of bytes from the stream.
 786  * @param               p_stream        the stream to skip data from.
 787  * @param               p_size          the number of bytes to skip.
 788  * @param               p_event_mgr     the user event manager to be notified of special events.
 789  * @return              the number of bytes skipped, or -1 if an error occured.
 790  */
 791 OPJ_SIZE_T opj_stream_write_skip (opj_stream_private_t * p_stream, OPJ_SIZE_T p_size, opj_event_mgr_t * p_event_mgr)
 792 {
 793         opj_bool l_is_written = 0;
 794         OPJ_SIZE_T l_current_skip_nb_bytes = 0;
 795         OPJ_SIZE_T l_skip_nb_bytes = 0;
 796
 797         if
 798                 (p_stream->m_status & opj_stream_e_error)
 799         {
 800                 return (OPJ_SIZE_T) -1;
 801         }
 802
 803         // we should flush data
 804         l_is_written = opj_stream_flush (p_stream, p_event_mgr);
 805         if
 806                 (! l_is_written)
 807         {
 808                 p_stream->m_status |= opj_stream_e_error;
 809                 p_stream->m_bytes_in_buffer = 0;
 810                 p_stream->m_current_data = p_stream->m_current_data;
 811                 return (OPJ_SIZE_T) -1;
 812         }
 813         // then skip
 814
 815         while
 816                 (p_size > 0)
 817         {
 818                 // we should do an actual skip on the media
 819                 l_current_skip_nb_bytes = p_stream->m_skip_fn(p_size, p_stream->m_user_data);
 820                 if
 821                         (l_current_skip_nb_bytes == (OPJ_SIZE_T)-1)
 822                 {
 823                         opj_event_msg_v2(p_event_mgr, EVT_INFO, "Stream error!\n");
 824
 825                         p_stream->m_status |= opj_stream_e_error;
 826                         p_stream->m_byte_offset += l_skip_nb_bytes;
 827                         // end if stream
 828                         return l_skip_nb_bytes ? l_skip_nb_bytes : (OPJ_SIZE_T)-1;
 829                 }
 830                 p_size -= l_current_skip_nb_bytes;
 831                 l_skip_nb_bytes += l_current_skip_nb_bytes;
 832         }
 833         p_stream->m_byte_offset += l_skip_nb_bytes;
 834         return l_skip_nb_bytes;
 835 }
 836
 837 /**
 838  * Tells the byte offset on the stream (similar to ftell).
 839  *
 840  * @param               p_stream        the stream to get the information from.
 841  *
 842  * @return              the current position of the stream.
 843  */
 844 OPJ_SIZE_T opj_stream_tell (const opj_stream_private_t * p_stream)
 845 {
 846         return p_stream->m_byte_offset;
 847 }
 848
 849
 850 /**
 851  * Get the number of bytes left before the end of the stream (similar to cio_numbytesleft).
 852  *
 853  * @param               p_stream        the stream to get the information from.
 854  *
 855  * @return              Number of bytes left before the end of the stream.
 856  */
 857 OPJ_SIZE_T opj_stream_get_number_byte_left (const opj_stream_private_t * p_stream)
 858 {
 859         return p_stream->m_user_data_length ?
 860                                 p_stream->m_user_data_length - p_stream->m_byte_offset :
 861                                 0;
 862 }
 863
 864 /**
 865  * Skips a number of bytes from the stream.
 866  * @param               p_stream        the stream to skip data from.
 867  * @param               p_size          the number of bytes to skip.
 868  * @param               p_event_mgr     the user event manager to be notified of special events.
 869  * @return              the number of bytes skipped, or -1 if an error occured.
 870  */
 871 OPJ_SIZE_T opj_stream_skip (opj_stream_private_t * p_stream, OPJ_SIZE_T p_size, opj_event_mgr_t * p_event_mgr)
 872 {
 873         return p_stream->m_opj_skip(p_stream,p_size,p_event_mgr);
 874 }
 875
 876
 877 /**
 878  * Skips a number of bytes from the stream.
 879  * @param               p_stream        the stream to skip data from.
 880  * @param               p_size          the number of bytes to skip.
 881  * @param               p_event_mgr     the user event manager to be notified of special events.
 882  * @return              the number of bytes skipped, or -1 if an error occured.
 883  */
 884 opj_bool opj_stream_read_seek (opj_stream_private_t * p_stream, OPJ_SIZE_T p_size, opj_event_mgr_t * p_event_mgr)
 885 {
 886         p_stream->m_current_data = p_stream->m_stored_data;
 887         p_stream->m_bytes_in_buffer = 0;
 888
 889         if( p_stream->m_seek_fn(p_size,p_stream->m_user_data)) {
 890                 p_stream->m_status |= opj_stream_e_end;
 891                 return EXIT_FAILURE;
 892         }
 893         else {
 894                 // reset stream status
 895                 p_stream->m_status &= (~opj_stream_e_end);
 896                 p_stream->m_byte_offset = p_size;
 897
 898         }
 899
 900         return EXIT_SUCCESS;
 901 }
 902
 903 /**
 904  * Skips a number of bytes from the stream.
 905  * @param               p_stream        the stream to skip data from.
 906  * @param               p_size          the number of bytes to skip.
 907  * @param               p_event_mgr     the user event manager to be notified of special events.
 908  * @return              the number of bytes skipped, or -1 if an error occured.
 909  */
 910 opj_bool opj_stream_write_seek (opj_stream_private_t * p_stream, OPJ_SIZE_T p_size, opj_event_mgr_t * p_event_mgr)
 911 {
 912         if
 913                 (! opj_stream_flush(p_stream,p_event_mgr))
 914         {
 915                 p_stream->m_status |= opj_stream_e_error;
 916                 return EXIT_FAILURE;
 917         }
 918
 919         p_stream->m_current_data = p_stream->m_stored_data;
 920         p_stream->m_bytes_in_buffer = 0;
 921
 922         if
 923                 (! p_stream->m_seek_fn(p_size,p_stream->m_user_data))
 924         {
 925                 p_stream->m_status |= opj_stream_e_error;
 926                 return EXIT_FAILURE;
 927         }
 928         else
 929         {
 930                 p_stream->m_byte_offset = p_size;
 931         }
 932         return EXIT_SUCCESS;
 933 }
 934
 935
 936 /**
 937  * Seeks a number of bytes from the stream.
 938  * @param               p_stream        the stream to skip data from.
 939  * @param               p_size          the number of bytes to skip.
 940  * @param               p_event_mgr     the user event manager to be notified of special events.
 941  * @return              true if the stream is seekable.
 942  */
 943 opj_bool opj_stream_seek (opj_stream_private_t * p_stream, OPJ_SIZE_T p_size, struct opj_event_mgr * p_event_mgr)
 944 {
 945         return p_stream->m_opj_seek(p_stream,p_size,p_event_mgr);
 946 }
 947
 948 /**
 949  * Tells if the given stream is seekable.
 950  */
 951 opj_bool opj_stream_has_seek (const opj_stream_private_t * p_stream)
 952 {
 953         return p_stream->m_seek_fn != opj_stream_default_seek;
 954 }
 955
 956
 957
 958
 959
 960 OPJ_UINT32 opj_stream_default_read (void * p_buffer, OPJ_UINT32 p_nb_bytes, void * p_user_data)
 961 {
 962         return (OPJ_UINT32) -1;
 963 }
 964 OPJ_UINT32 opj_stream_default_write (void * p_buffer, OPJ_UINT32 p_nb_bytes, void * p_user_data)
 965 {
 966         return (OPJ_UINT32) -1;
 967 }
 968 OPJ_SIZE_T opj_stream_default_skip (OPJ_SIZE_T p_nb_bytes, void * p_user_data)
 969 {
 970         return (OPJ_SIZE_T) -1;
 971 }
 972
 973 opj_bool opj_stream_default_seek (OPJ_SIZE_T p_nb_bytes, void * p_user_data)
 974 {
 975         return EXIT_FAILURE;
 976 }
 977
 978
 979
 980