com.mycomm.itool.compress.huffman.s2.WriteBuffer Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ITools Show documentation
ITools
The newest version!
package com.mycomm.itool.compress.huffman.s2;

/**
  * WriteBuffer Class
  * Written By: Mitchell Pomery (21130887)
  * Buffers either bit or byte data and writes it out in bulk amounts
  * Also pads bit data nicely.
 **/

import java.io.OutputStream;
import java.io.IOException;

class WriteBuffer {
	int pos = 0; //the number of bits in the buffer
	boolean[] buffer = new boolean[8];
	byte[] outStream = new byte[12288];
	int outLength = 0; //the number of bytes waiting to be written
	OutputStream out;
	boolean booleandata = true; //If we are expecting bits
	
	
	public WriteBuffer(OutputStream out, boolean bitdata) {
		this.out = out;
		booleandata = bitdata;
	}
	
	
	public void write(boolean[] b) throws IOException{
		if (booleandata) {
			for (int i = 0; i < b.length; i++) {
				buffer[pos] = b[i];
				pos++;
				if (pos > 7) {
				//character |= 0x80;
					byte character = BitByteConverter.booleanArrayToByte(buffer);
					pos -= 8;
					outStream[outLength] = character;
					outLength++;
				}
				if (outLength > 12200) {
					out.write(outStream, 0, outLength);
					outLength = 0;
				}
			}
		}
	}
	
	
	public void write(byte[] b) throws IOException{
		if (!booleandata) {
			for (int i = 0; i < b.length; i++) {
				outStream[outLength] = b[i];
				outLength++;
				if (outLength > 12200) {
					out.write(outStream, 0, outLength);
					outLength = 0;
				}
			}
		}
	}
	
	
	public void write(byte b) throws IOException{
		if (!booleandata) {
			outStream[outLength] = b;
			outLength++;
			if (outLength > 12200) {
				out.write(outStream, 0, outLength);
				outLength = 0;
			}
		}
	}
	
	
	public void flush() throws IOException{
		if (booleandata) {
			boolean[] padding = new boolean[8 + (8 - pos)];
			int poscopy = pos;
			while (poscopy < 8) {
				padding[padding.length - (poscopy + 1)] = true;
				poscopy++;
			}
			write(padding);
		}
		out.write(outStream, 0, outLength);
		outLength = 0;
	}
}