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         bio->buf = (bio->buf << 8) & 0xffff;
  82         bio->ct = bio->buf == 0xff00 ? 7 : 8;
  83         if (bio->bp >= bio->end) {
  84                 return 1;
  85         }
  86         *bio->bp++ = bio->buf >> 8;
  87         return 0;
  88 }
  89
  90 static int bio_bytein(opj_bio_t *bio) {
  91         bio->buf = (bio->buf << 8) & 0xffff;
  92         bio->ct = bio->buf == 0xff00 ? 7 : 8;
  93         if (bio->bp >= bio->end) {
  94                 return 1;
  95         }
  96         bio->buf |= *bio->bp++;
  97         return 0;
  98 }
  99
 100 static void bio_putbit(opj_bio_t *bio, int b) {
 101         if (bio->ct == 0) {
 102                 bio_byteout(bio);
 103         }
 104         bio->ct--;
 105         bio->buf |= b << bio->ct;
 106 }
 107
 108 /* MOD antonin */
 109 static int bio_getbit(opj_bio_t *bio) {
 110 /* DOM */
 111         if (bio->ct == 0) {
 112                 bio_bytein(bio);
 113         }
 114         bio->ct--;
 115         return (bio->buf >> bio->ct) & 1;
 116 }
 117
 118 /*
 119 ==========================================================
 120    Bit Input/Output interface
 121 ==========================================================
 122 */
 123
 124 opj_bio_t* bio_create() {
 125         opj_bio_t *bio = (opj_bio_t*)opj_malloc(sizeof(opj_bio_t));
 126         return bio;
 127 }
 128
 129 void bio_destroy(opj_bio_t *bio) {
 130         if(bio) {
 131                 opj_free(bio);
 132         }
 133 }
 134
 135 int bio_numbytes(opj_bio_t *bio) {
 136         return (bio->bp - bio->start);
 137 }
 138
 139 void bio_init_enc(opj_bio_t *bio, unsigned char *bp, int len) {
 140         bio->start = bp;
 141         bio->end = bp + len;
 142         bio->bp = bp;
 143         bio->buf = 0;
 144         bio->ct = 8;
 145 }
 146
 147 void bio_init_dec(opj_bio_t *bio, unsigned char *bp, int len) {
 148         bio->start = bp;
 149         bio->end = bp + len;
 150         bio->bp = bp;
 151         bio->buf = 0;
 152         bio->ct = 0;
 153 }
 154
 155 void bio_write(opj_bio_t *bio, int v, int n) {
 156         int i;
 157         for (i = n - 1; i >= 0; i--) {
 158                 bio_putbit(bio, (v >> i) & 1);
 159         }
 160 }
 161
 162 int bio_read(opj_bio_t *bio, int n) {
 163         int i, v;
 164         v = 0;
 165         for (i = n - 1; i >= 0; i--) {
 166                 v += bio_getbit(bio) << i;
 167         }
 168         return v;
 169 }
 170
 171 int bio_flush(opj_bio_t *bio) {
 172         bio->ct = 0;
 173         if (bio_byteout(bio)) {
 174                 return 1;
 175         }
 176         if (bio->ct == 7) {
 177                 bio->ct = 0;
 178                 if (bio_byteout(bio)) {
 179                         return 1;
 180                 }
 181         }
 182         return 0;
 183 }
 184
 185 int bio_inalign(opj_bio_t *bio) {
 186         bio->ct = 0;
 187         if ((bio->buf & 0xff) == 0xff) {
 188                 if (bio_bytein(bio)) {
 189                         return 1;
 190                 }
 191                 bio->ct = 0;
 192         }
 193         return 0;
 194 }