libopenjp3d/bio.c

   1 /*\r
   2  * Copyright (c) 2001-2003, David Janssens\r
   3  * Copyright (c) 2002-2003, Yannick Verschueren\r
   4  * Copyright (c) 2003-2005, Francois Devaux and Antonin Descampe\r
   5  * Copyright (c) 2005, Herve Drolon, FreeImage Team\r
   6  * Copyright (c) 2002-2005, Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium\r
   7  * All rights reserved.\r
   8  *\r
   9  * Redistribution and use in source and binary forms, with or without\r
  10  * modification, are permitted provided that the following conditions\r
  11  * are met:\r
  12  * 1. Redistributions of source code must retain the above copyright\r
  13  *    notice, this list of conditions and the following disclaimer.\r
  14  * 2. Redistributions in binary form must reproduce the above copyright\r
  15  *    notice, this list of conditions and the following disclaimer in the\r
  16  *    documentation and/or other materials provided with the distribution.\r
  17  *\r
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'\r
  19  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE\r
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE\r
  21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE\r
  22  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR\r
  23  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF\r
  24  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS\r
  25  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN\r
  26  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)\r
  27  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE\r
  28  * POSSIBILITY OF SUCH DAMAGE.\r
  29  */\r
  30 \r
  31 #include "opj_includes.h"\r
  32 \r
  33 /** @defgroup BIO BIO - Individual bit input-output stream */\r
  34 /*@{*/\r
  35 \r
  36 /** @name Local static functions */\r
  37 /*@{*/\r
  38 \r
  39 /**\r
  40 Write a bit\r
  41 @param bio BIO handle\r
  42 @param b Bit to write (0 or 1)\r
  43 */\r
  44 static void bio_putbit(opj_bio_t *bio, int b);\r
  45 /**\r
  46 Read a bit\r
  47 @param bio BIO handle\r
  48 @return Returns the read bit\r
  49 */\r
  50 static int bio_getbit(opj_bio_t *bio);\r
  51 /**\r
  52 Write a byte\r
  53 @param bio BIO handle\r
  54 @return Returns 0 if successful, returns 1 otherwise\r
  55 */\r
  56 static int bio_byteout(opj_bio_t *bio);\r
  57 /**\r
  58 Read a byte\r
  59 @param bio BIO handle\r
  60 @return Returns 0 if successful, returns 1 otherwise\r
  61 */\r
  62 static int bio_bytein(opj_bio_t *bio);\r
  63 \r
  64 /*@}*/\r
  65 \r
  66 /*@}*/\r
  67 \r
  68 \r
  69 /* \r
  70 ==========================================================\r
  71    local functions\r
  72 ==========================================================\r
  73 */\r
  74 \r
  75 static int bio_byteout(opj_bio_t *bio) {\r
  76         bio->buf = (bio->buf << 8) & 0xffff;\r
  77         bio->ct = bio->buf == 0xff00 ? 7 : 8;\r
  78         if (bio->bp >= bio->end) {\r
  79                 return 1;\r
  80         }\r
  81         *bio->bp++ = bio->buf >> 8;\r
  82         return 0;\r
  83 }\r
  84 \r
  85 static int bio_bytein(opj_bio_t *bio) {\r
  86         bio->buf = (bio->buf << 8) & 0xffff;\r
  87         bio->ct = bio->buf == 0xff00 ? 7 : 8;\r
  88         if (bio->bp >= bio->end) {\r
  89                 return 1;\r
  90         }\r
  91         bio->buf |= *bio->bp++;\r
  92         return 0;\r
  93 }\r
  94 \r
  95 static void bio_putbit(opj_bio_t *bio, int b) {\r
  96         if (bio->ct == 0) {\r
  97                 bio_byteout(bio);\r
  98         }\r
  99         bio->ct--;\r
 100         bio->buf |= b << bio->ct;\r
 101 }\r
 102 \r
 103 /* MOD antonin */\r
 104 static int bio_getbit(opj_bio_t *bio) {\r
 105 /* DOM */\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() {\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 int 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, unsigned char *bp, int 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, unsigned char *bp, int 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, int v, int n) {\r
 151         int i;\r
 152         for (i = n - 1; i >= 0; i--) {\r
 153                 bio_putbit(bio, (v >> i) & 1);\r
 154         }\r
 155 }\r
 156 \r
 157 int bio_read(opj_bio_t *bio, int n) {\r
 158         int i, v;\r
 159         v = 0;\r
 160         for (i = n - 1; i >= 0; i--) {\r
 161                 v += bio_getbit(bio) << i;\r
 162         }\r
 163         return v;\r
 164 }\r
 165 \r
 166 int bio_flush(opj_bio_t *bio) {\r
 167         bio->ct = 0;\r
 168         if (bio_byteout(bio)) {\r
 169                 return 1;\r
 170         }\r
 171         if (bio->ct == 7) {\r
 172                 bio->ct = 0;\r
 173                 if (bio_byteout(bio)) {\r
 174                         return 1;\r
 175                 }\r
 176         }\r
 177         return 0;\r
 178 }\r
 179 \r
 180 int 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 1;\r
 185                 }\r
 186                 bio->ct = 0;\r
 187         }\r
 188         return 0;\r
 189 }\r