v2/libopenjpeg/bio.c

   1 /*\r
   2  * Copyright (c) 2002-2007, Communications and Remote Sensing Laboratory, Universite catholique de Louvain (UCL), Belgium\r
   3  * Copyright (c) 2002-2007, Professor Benoit Macq\r
   4  * Copyright (c) 2001-2003, David Janssens\r
   5  * Copyright (c) 2002-2003, Yannick Verschueren\r
   6  * Copyright (c) 2003-2007, Francois-Olivier Devaux and Antonin Descampe\r
   7  * Copyright (c) 2005, Herve Drolon, FreeImage Team\r
   8  * Copyright (c) 2008, Jerome Fimes, Communications & Systemes <jerome.fimes@c-s.fr>\r
   9  * All rights reserved.\r
  10  *\r
  11  * Redistribution and use in source and binary forms, with or without\r
  12  * modification, are permitted provided that the following conditions\r
  13  * are met:\r
  14  * 1. Redistributions of source code must retain the above copyright\r
  15  *    notice, this list of conditions and the following disclaimer.\r
  16  * 2. Redistributions in binary form must reproduce the above copyright\r
  17  *    notice, this list of conditions and the following disclaimer in the\r
  18  *    documentation and/or other materials provided with the distribution.\r
  19  *\r
  20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'\r
  21  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE\r
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE\r
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE\r
  24  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR\r
  25  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF\r
  26  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS\r
  27  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN\r
  28  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)\r
  29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE\r
  30  * POSSIBILITY OF SUCH DAMAGE.\r
  31  */\r
  32 \r
  33 #include "bio.h"\r
  34 #include "opj_malloc.h"\r
  35 \r
  36 /** @defgroup BIO BIO - Individual bit input-output stream */\r
  37 /*@{*/\r
  38 \r
  39 /** @name Local static functions */\r
  40 /*@{*/\r
  41 \r
  42 /**\r
  43 Write a bit\r
  44 @param bio BIO handle\r
  45 @param b Bit to write (0 or 1)\r
  46 */\r
  47 static void bio_putbit(opj_bio_t *bio, OPJ_UINT32 b);\r
  48 /**\r
  49 Read a bit\r
  50 @param bio BIO handle\r
  51 @return Returns the read bit\r
  52 */\r
  53 static OPJ_UINT32 bio_getbit(opj_bio_t *bio);\r
  54 /**\r
  55 Write a byte\r
  56 @param bio BIO handle\r
  57 @return Returns 0 if successful, returns 1 otherwise\r
  58 */\r
  59 static bool bio_byteout(opj_bio_t *bio);\r
  60 /**\r
  61 Read a byte\r
  62 @param bio BIO handle\r
  63 @return Returns 0 if successful, returns 1 otherwise\r
  64 */\r
  65 static bool bio_bytein(opj_bio_t *bio);\r
  66 \r
  67 /*@}*/\r
  68 \r
  69 /*@}*/\r
  70 \r
  71 /* \r
  72 ==========================================================\r
  73    local functions\r
  74 ==========================================================\r
  75 */\r
  76 \r
  77 static bool bio_byteout(opj_bio_t *bio) {\r
  78         bio->buf = (bio->buf << 8) & 0xffff;\r
  79         bio->ct = bio->buf == 0xff00 ? 7 : 8;\r
  80         if (bio->bp >= bio->end) {\r
  81                 return true;\r
  82         }\r
  83         *bio->bp++ = bio->buf >> 8;\r
  84         return false;\r
  85 }\r
  86 \r
  87 static bool bio_bytein(opj_bio_t *bio) {\r
  88         bio->buf = (bio->buf << 8) & 0xffff;\r
  89         bio->ct = bio->buf == 0xff00 ? 7 : 8;\r
  90         if (bio->bp >= bio->end) {\r
  91                 return true;\r
  92         }\r
  93         bio->buf |= *bio->bp++;\r
  94         return false;\r
  95 }\r
  96 \r
  97 static void bio_putbit(opj_bio_t *bio, OPJ_UINT32 b) {\r
  98         if (bio->ct == 0) {\r
  99                 bio_byteout(bio);\r
 100         }\r
 101         bio->ct--;\r
 102         bio->buf |= b << bio->ct;\r
 103 }\r
 104 \r
 105 static OPJ_UINT32 bio_getbit(opj_bio_t *bio) {\r
 106         if (bio->ct == 0) {\r
 107                 bio_bytein(bio);\r
 108         }\r
 109         bio->ct--;\r
 110         return (bio->buf >> bio->ct) & 1;\r
 111 }\r
 112 \r
 113 /* \r
 114 ==========================================================\r
 115    Bit Input/Output interface\r
 116 ==========================================================\r
 117 */\r
 118 \r
 119 opj_bio_t* bio_create(void) {\r
 120         opj_bio_t *bio = (opj_bio_t*)opj_malloc(sizeof(opj_bio_t));\r
 121         return bio;\r
 122 }\r
 123 \r
 124 void bio_destroy(opj_bio_t *bio) {\r
 125         if(bio) {\r
 126                 opj_free(bio);\r
 127         }\r
 128 }\r
 129 \r
 130 OPJ_UINT32 bio_numbytes(opj_bio_t *bio) {\r
 131         return (bio->bp - bio->start);\r
 132 }\r
 133 \r
 134 void bio_init_enc(opj_bio_t *bio, OPJ_BYTE *bp, OPJ_UINT32 len) {\r
 135         bio->start = bp;\r
 136         bio->end = bp + len;\r
 137         bio->bp = bp;\r
 138         bio->buf = 0;\r
 139         bio->ct = 8;\r
 140 }\r
 141 \r
 142 void bio_init_dec(opj_bio_t *bio, OPJ_BYTE *bp, OPJ_UINT32 len) {\r
 143         bio->start = bp;\r
 144         bio->end = bp + len;\r
 145         bio->bp = bp;\r
 146         bio->buf = 0;\r
 147         bio->ct = 0;\r
 148 }\r
 149 \r
 150 void bio_write(opj_bio_t *bio, OPJ_UINT32 v, OPJ_UINT32 n) {\r
 151         OPJ_UINT32 i;\r
 152         for (i = n - 1; i != -1 ; --i) {\r
 153                 bio_putbit(bio, (v >> i) & 1);\r
 154         }\r
 155 }\r
 156 \r
 157 OPJ_UINT32 bio_read(opj_bio_t *bio, OPJ_UINT32 n) {\r
 158         OPJ_UINT32 i, v;\r
 159         v = 0;\r
 160         for (i = n - 1; i != -1 ; --i) {\r
 161                 v += bio_getbit(bio) << i;\r
 162         }\r
 163         return v;\r
 164 }\r
 165 \r
 166 bool bio_flush(opj_bio_t *bio) {\r
 167         bio->ct = 0;\r
 168         if (bio_byteout(bio)) {\r
 169                 return true;\r
 170         }\r
 171         if (bio->ct == 7) {\r
 172                 bio->ct = 0;\r
 173                 if (bio_byteout(bio)) {\r
 174                         return true;\r
 175                 }\r
 176         }\r
 177         return false;\r
 178 }\r
 179 \r
 180 bool bio_inalign(opj_bio_t *bio) {\r
 181         bio->ct = 0;\r
 182         if ((bio->buf & 0xff) == 0xff) {\r
 183                 if (bio_bytein(bio)) {\r
 184                         return true;\r
 185                 }\r
 186                 bio->ct = 0;\r
 187         }\r
 188         return false;\r
 189 }\r