src/lib/openjp3d/bio.c

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