/* -*-mode:java; c-basic-offset:2; -*- */
/* JOrbis
* Copyright (C) 2000 ymnk, JCraft,Inc.
*
* Written by: 2000 ymnk<ymnk@jcraft.com>
*
* Many thanks to
* Monty <monty@xiph.org> and
* The XIPHOPHORUS Company http://www.xiph.org/ .
* JOrbis has been based on their awesome works, Vorbis codec.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public License
* as published by the Free Software Foundation; either version 2 of
* the License, or (at your option) any later version.
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
package com.jcraft.jogg;
public class Buffer{
private static final int BUFFER_INCREMENT=256;
private static final int[] mask={
0x00000000,0x00000001,0x00000003,0x00000007,0x0000000f,
0x0000001f,0x0000003f,0x0000007f,0x000000ff,0x000001ff,
0x000003ff,0x000007ff,0x00000fff,0x00001fff,0x00003fff,
0x00007fff,0x0000ffff,0x0001ffff,0x0003ffff,0x0007ffff,
0x000fffff,0x001fffff,0x003fffff,0x007fffff,0x00ffffff,
0x01ffffff,0x03ffffff,0x07ffffff,0x0fffffff,0x1fffffff,
0x3fffffff,0x7fffffff,0xffffffff
};
int ptr=0;
byte[] buffer=null;
int endbit=0;
int endbyte=0;
int storage=0;
public void writeinit(){
buffer=new byte[BUFFER_INCREMENT];
ptr=0;
buffer[0]=(byte)'\0';
storage=BUFFER_INCREMENT;
}
public void write(byte[] s){
for(int i=0; i<s.length; i++){
if(s[i]==0)break;
write(s[i],8);
}
}
public void read(byte[] s, int bytes){
int i=0;
while(bytes--!=0){
s[i++]=(byte)(read(8));
}
}
void reset(){
ptr=0;
buffer[0]=(byte)'\0';
endbit=endbyte=0;
}
public void writeclear(){
buffer=null;
}
public void readinit(byte[] buf, int bytes){
readinit(buf, 0, bytes);
}
public void readinit(byte[] buf, int start, int bytes){
//System.err.println("readinit: start="+start+", bytes="+bytes);
//for(int i=0;i<bytes; i++){
//System.err.println(i+": "+Integer.toHexString(buf[i+start]));
//}
ptr=start;
buffer=buf;
endbit=endbyte=0;
storage=bytes;
}
public void write(int value, int bits){
//System.err.println("write: "+Integer.toHexString(value)+", bits="+bits+" ptr="+ptr+", storage="+storage+", endbyte="+endbyte);
if(endbyte+4>=storage){
byte[] foo=new byte[storage+BUFFER_INCREMENT];
System.arraycopy(buffer, 0, foo, 0, storage);
buffer=foo;
storage+=BUFFER_INCREMENT;
}
value&=mask[bits];
bits+=endbit;
buffer[ptr]|=(byte)(value<<endbit);
if(bits>=8){
buffer[ptr+1]=(byte)(value>>>(8-endbit));
if(bits>=16){
buffer[ptr+2]=(byte)(value>>>(16-endbit));
if(bits>=24){
buffer[ptr+3]=(byte)(value>>>(24-endbit));
if(bits>=32){
if(endbit>0)
buffer[ptr+4]=(byte)(value>>>(32-endbit));
else
buffer[ptr+4]=0;
}
}
}
}
endbyte+=bits/8;
ptr+=bits/8;
endbit=bits&7;
}
public int look(int bits){
int ret;
int m=mask[bits];
bits+=endbit;
//System.err.println("look ptr:"+ptr+", bits="+bits+", endbit="+endbit+", storage="+storage);
if(endbyte+4>=storage){
if(endbyte+(bits-1)/8>=storage)return(-1);
}
ret=((buffer[ptr])&0xff)>>>endbit;
// ret=((byte)(buffer[ptr]))>>>endbit;
if(bits>8){
ret|=((buffer[ptr+1])&0xff)<<(8-endbit);
// ret|=((byte)(buffer[ptr+1]))<<(8-endbit);
if(bits>16){
ret|=((buffer[ptr+2])&0xff)<<(16-endbit);
// ret|=((byte)(buffer[ptr+2]))<<(16-endbit);
if(bits>24){
ret|=((buffer[ptr+3])&0xff)<<(24-endbit);
//System.err.print("ret="+Integer.toHexString(ret)+", ((byte)(buffer[ptr+3]))="+Integer.toHexString(((buffer[ptr+3])&0xff)));
// ret|=((byte)(buffer[ptr+3]))<<(24-endbit);
//System.err.println(" ->ret="+Integer.toHexString(ret));
if(bits>32 && endbit!=0){
ret|=((buffer[ptr+4])&0xff)<<(32-endbit);
// ret|=((byte)(buffer[ptr+4]))<<(32-endbit);
}
}
}
}
return(m&ret);
}
public int look1(){
if(endbyte>=storage)return(-1);
return((buffer[ptr]>>endbit)&1);
}
public void adv(int bits){
bits+=endbit;
ptr+=bits/8;
endbyte+=bits/8;
endbit=bits&7;
}
public void adv1(){
++endbit;
if(endbit>7){
endbit=0;
ptr++;
endbyte++;
}
}
public int read(int bits){
//System.err.println(this+" read: bits="+bits+", storage="+storage+", endbyte="+endbyte);
//System.err.println(this+" read: bits="+bits+", storage="+storage+", endbyte="+endbyte+
// ", ptr="+ptr+", endbit="+endbit+", buf[ptr]="+buffer[ptr]);
int ret;
int m=mask[bits];
bits+=endbit;
if(endbyte+4>=storage){
ret=-1;
if(endbyte+(bits-1)/8>=storage){
ptr+=bits/8;
endbyte+=bits/8;
endbit=bits&7;
return(ret);
}
}
/*
ret=(byte)(buffer[ptr]>>>endbit);
if(bits>8){
ret|=(buffer[ptr+1]<<(8-endbit));
if(bits>16){
ret|=(buffer[ptr+2]<<(16-endbit));
if(bits>24){
ret|=(buffer[ptr+3]<<(24-endbit));
if(bits>32 && endbit>0){
ret|=(buffer[ptr+4]<<(32-endbit));
}
}
}
}
*/
ret=((buffer[ptr])&0xff)>>>endbit;
if(bits>8){
ret|=((buffer[ptr+1])&0xff)<<(8-endbit);
// ret|=((byte)(buffer[ptr+1]))<<(8-endbit);
if(bits>16){
ret|=((buffer[ptr+2])&0xff)<<(16-endbit);
// ret|=((byte)(buffer[ptr+2]))<<(16-endbit);
if(bits>24){
ret|=((buffer[ptr+3])&0xff)<<(24-endbit);
// ret|=((byte)(buffer[ptr+3]))<<(24-endbit);
if(bits>32 && endbit!=0){
ret|=((buffer[ptr+4])&0xff)<<(32-endbit);
// ret|=((byte)(buffer[ptr+4]))<<(32-endbit);
}
}
}
}
ret&=m;
ptr+=bits/8;
// ptr=bits/8;
endbyte+=bits/8;
// endbyte=bits/8;
endbit=bits&7;
return(ret);
}
public int readB(int bits){
//System.err.println(this+" read: bits="+bits+", storage="+storage+", endbyte="+endbyte+
// ", ptr="+ptr+", endbit="+endbit+", buf[ptr]="+buffer[ptr]);
int ret;
int m=32-bits;
bits+=endbit;
if(endbyte+4>=storage){
/* not the main path */
ret=-1;
if(endbyte*8+bits>storage*8) {
ptr+=bits/8;
endbyte+=bits/8;
endbit=bits&7;
return(ret);
}
}
ret=(buffer[ptr]&0xff)<<(24+endbit);
if(bits>8){
ret|=(buffer[ptr+1]&0xff)<<(16+endbit);
if(bits>16){
ret|=(buffer[ptr+2]&0xff)<<(8+endbit);
if(bits>24){
ret|=(buffer[ptr+3]&0xff)<<(endbit);
if(bits>32 && (endbit != 0))
ret|=(buffer[ptr+4]&0xff)>>(8-endbit);
}
}
}
ret=(ret>>>(m>>1))>>>((m+1)>>1);
ptr+=bits/8;
endbyte+=bits/8;
endbit=bits&7;
return(ret);
}
public int read1(){
int ret;
if(endbyte>=storage){
ret=-1;
endbit++;
if(endbit>7){
endbit=0;
ptr++;
endbyte++;
}
return(ret);
}
ret=(buffer[ptr]>>endbit)&1;
endbit++;
if(endbit>7){
endbit=0;
ptr++;
endbyte++;
}
return(ret);
}
public int bytes(){
return(endbyte+(endbit+7)/8);
}
public int bits(){
return(endbyte*8+endbit);
}
public byte[] buffer(){
return(buffer);
}
public static int ilog(int v){
int ret=0;
while(v>0){
ret++;
v>>>=1;
}
return(ret);
}
public static void report(String in){
System.err.println(in);
System.exit(1);
}
/*
static void cliptest(int[] b, int vals, int bits, int[] comp, int compsize){
int bytes;
byte[] buffer;
o.reset();
for(int i=0;i<vals;i++){
o.write(b[i],((bits!=0)?bits:ilog(b[i])));
}
buffer=o.buffer();
bytes=o.bytes();
System.err.println("cliptest: bytes="+bytes);
if(bytes!=compsize)report("wrong number of bytes!\n");
for(int i=0;i<bytes;i++){
if(buffer[i]!=(byte)comp[i]){
for(int j=0;j<bytes;j++){
System.err.println(j+": "+Integer.toHexString(buffer[j])+" "+
Integer.toHexString(comp[j]));
}
report("wrote incorrect value!\n");
}
}
System.err.println("bits: "+bits);
r.readinit(buffer,bytes);
for(int i=0;i<vals;i++){
int tbit=(bits!=0)?bits:ilog(b[i]);
System.err.println(Integer.toHexString(b[i])+" tbit: "+tbit);
if(r.look(tbit)==-1){
report("out of data!\n");
}
if(r.look(tbit)!=(b[i]&mask[tbit])){
report(i+" looked at incorrect value! "+Integer.toHexString(r.look(tbit))+", "+Integer.toHexString(b[i]&mask[tbit])+":"+b[i]+" bit="+tbit);
}
if(tbit==1){
if(r.look1()!=(b[i]&mask[tbit])){
report("looked at single bit incorrect value!\n");
}
}
if(tbit==1){
if(r.read1()!=(b[i]&mask[tbit])){
report("read incorrect single bit value!\n");
}
}
else{
if(r.read(tbit)!=(b[i]&mask[tbit])){
report("read incorrect value!\n");
}
}
}
if(r.bytes()!=bytes){
report("leftover bytes after read!\n");
}
}
static int[] testbuffer1=
{18,12,103948,4325,543,76,432,52,3,65,4,56,32,42,34,21,1,23,32,546,456,7,
567,56,8,8,55,3,52,342,341,4,265,7,67,86,2199,21,7,1,5,1,4};
static int test1size=43;
static int[] testbuffer2=
{216531625,1237861823,56732452,131,3212421,12325343,34547562,12313212,
1233432,534,5,346435231,14436467,7869299,76326614,167548585,
85525151,0,12321,1,349528352};
static int test2size=21;
static int[] large=
{2136531625,2137861823,56732452,131,3212421,12325343,34547562,12313212,
1233432,534,5,2146435231,14436467,7869299,76326614,167548585,
85525151,0,12321,1,2146528352};
static int[] testbuffer3=
{1,0,14,0,1,0,12,0,1,0,0,0,1,1,0,1,0,1,0,1,0,1,0,1,0,1,0,0,1,1,1,1,1,0,0,1,
0,1,30,1,1,1,0,0,1,0,0,0,12,0,11,0,1,0,0,1};
static int test3size=56;
static int onesize=33;
static int[] one={146,25,44,151,195,15,153,176,233,131,196,65,85,172,47,40,
34,242,223,136,35,222,211,86,171,50,225,135,214,75,172,
223,4};
static int twosize=6;
static int[] two={61,255,255,251,231,29};
static int threesize=54;
static int[] three={169,2,232,252,91,132,156,36,89,13,123,176,144,32,254,
142,224,85,59,121,144,79,124,23,67,90,90,216,79,23,83,
58,135,196,61,55,129,183,54,101,100,170,37,127,126,10,
100,52,4,14,18,86,77,1};
static int foursize=38;
static int[] four={18,6,163,252,97,194,104,131,32,1,7,82,137,42,129,11,72,
132,60,220,112,8,196,109,64,179,86,9,137,195,208,122,169,
28,2,133,0,1};
static int fivesize=45;
static int[] five={169,2,126,139,144,172,30,4,80,72,240,59,130,218,73,62,
241,24,210,44,4,20,0,248,116,49,135,100,110,130,181,169,
84,75,159,2,1,0,132,192,8,0,0,18,22};
static int sixsize=7;
static int[] six={17,177,170,242,169,19,148};
static Buffer o=new Buffer();
static Buffer r=new Buffer();
public static void main(String[] arg){
byte[] buffer;
int bytes;
// o=new Buffer();
// r=new Buffer();
o.writeinit();
System.err.print("\nSmall preclipped packing: ");
cliptest(testbuffer1,test1size,0,one,onesize);
System.err.print("ok.");
System.err.print("\nNull bit call: ");
cliptest(testbuffer3,test3size,0,two,twosize);
System.err.print("ok.");
System.err.print("\nLarge preclipped packing: ");
cliptest(testbuffer2,test2size,0,three,threesize);
System.err.print("ok.");
System.err.print("\n32 bit preclipped packing: ");
o.reset();
for(int i=0;i<test2size;i++)
o.write(large[i],32);
buffer=o.buffer();
bytes=o.bytes();
r.readinit(buffer,bytes);
for(int i=0;i<test2size;i++){
if(r.look(32)==-1){
report("out of data. failed!");
}
if(r.look(32)!=large[i]){
System.err.print(r.look(32)+" != "+large[i]+" ("+
Integer.toHexString(r.look(32))+"!="+
Integer.toHexString(large[i])+")");
report("read incorrect value!\n");
}
r.adv(32);
}
if(r.bytes()!=bytes)report("leftover bytes after read!\n");
System.err.print("ok.");
System.err.print("\nSmall unclipped packing: ");
cliptest(testbuffer1,test1size,7,four,foursize);
System.err.print("ok.");
System.err.print("\nLarge unclipped packing: ");
cliptest(testbuffer2,test2size,17,five,fivesize);
System.err.print("ok.");
System.err.print("\nSingle bit unclicpped packing: ");
cliptest(testbuffer3,test3size,1,six,sixsize);
System.err.print("ok.");
System.err.print("\nTesting read past end: ");
r.readinit("\0\0\0\0\0\0\0\0".getBytes(),8);
for(int i=0;i<64;i++){
if(r.read(1)!=0){
System.err.print("failed; got -1 prematurely.\n");
System.exit(1);
}
}
if(r.look(1)!=-1 ||
r.read(1)!=-1){
System.err.print("failed; read past end without -1.\n");
System.exit(1);
}
r.readinit("\0\0\0\0\0\0\0\0".getBytes(),8);
if(r.read(30)!=0 || r.read(16)!=0){
System.err.print("failed 2; got -1 prematurely.\n");
System.exit(1);
}
if(r.look(18)!=0 ||
r.look(18)!=0){
System.err.print("failed 3; got -1 prematurely.\n");
System.exit(1);
}
if(r.look(19)!=-1 ||
r.look(19)!=-1){
System.err.print("failed; read past end without -1.\n");
System.exit(1);
}
if(r.look(32)!=-1 ||
r.look(32)!=-1){
System.err.print("failed; read past end without -1.\n");
System.exit(1);
}
System.err.print("ok.\n\n");
}
*/
}