1 /* ***** BEGIN LICENSE BLOCK *****
2 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
3 *
4 * The contents of this file are subject to the Mozilla Public License Version
5 * 1.1 (the "License"); you may not use this file except in compliance with
6 * the License. You may obtain a copy of the License at
7 * http://www.mozilla.org/MPL/
8 *
9 * Software distributed under the License is distributed on an "AS IS" basis,
10 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
11 * for the specific language governing rights and limitations under the
12 * License.
13 *
14 * The Original Code is "SMS Library for the Java platform".
15 *
16 * The Initial Developer of the Original Code is Markus Eriksson.
17 * Portions created by the Initial Developer are Copyright (C) 2002
18 * the Initial Developer. All Rights Reserved.
19 *
20 * Contributor(s):
21 *
22 * Alternatively, the contents of this file may be used under the terms of
23 * either the GNU General Public License Version 2 or later (the "GPL"), or
24 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
25 * in which case the provisions of the GPL or the LGPL are applicable instead
26 * of those above. If you wish to allow use of your version of this file only
27 * under the terms of either the GPL or the LGPL, and not to allow others to
28 * use your version of this file under the terms of the MPL, indicate your
29 * decision by deleting the provisions above and replace them with the notice
30 * and other provisions required by the GPL or the LGPL. If you do not delete
31 * the provisions above, a recipient may use your version of this file under
32 * the terms of any one of the MPL, the GPL or the LGPL.
33 *
34 * ***** END LICENSE BLOCK ***** */
35 package org.marre.util;
36
37 import java.io.*;
38
39 /***
40 *
41 */
42 public class BitArrayOutputStream extends ByteArrayOutputStream
43 {
44 private int myBitOffset;
45 private int myBuffer;
46
47 /***
48 * Default constructor.
49 */
50 public BitArrayOutputStream(int size)
51 {
52 super(size);
53 resetBitCounter();
54 }
55
56 public synchronized byte[] toByteArray()
57 {
58 flushByte();
59 return super.toByteArray();
60 }
61
62 private synchronized void resetBitCounter()
63 {
64 myBitOffset = 0;
65 myBuffer = 0x00;
66 }
67
68 public synchronized void reset()
69 {
70 super.reset();
71 resetBitCounter();
72 }
73
74 public synchronized void flushByte()
75 {
76 if (myBitOffset > 0)
77 {
78 super.write(myBuffer);
79 resetBitCounter();
80 }
81 }
82
83 public synchronized void writeBits( byte[] data, int nBits )
84 {
85 int bitsLeft = nBits;
86
87 for (int i = 0; bitsLeft > 0; i++)
88 {
89 writeBits(data[i], Math.min(bitsLeft, 8));
90 bitsLeft -= 8;
91 }
92 }
93
94 public synchronized void writeBits( int data, int nBits )
95 {
96 int bitsLeft = nBits;
97 int dataLeft = data;
98
99 while (bitsLeft > 0)
100 {
101 writeBit(dataLeft & 0x01);
102 dataLeft >>= 1;
103 bitsLeft--;
104 }
105 }
106
107 public synchronized void writeBit( int bit )
108 {
109 myBuffer |= ((bit & 0x01) << myBitOffset);
110 myBitOffset++;
111
112 if (myBitOffset == 8)
113 {
114 flushByte();
115 }
116 }
117
118 public synchronized void write(int data)
119 {
120 writeBits(data, 8);
121 }
122
123 public synchronized void write(byte[] data)
124 {
125 writeBits(data, 8 * data.length);
126 }
127
128 public synchronized void write(byte[] data, int off, int len)
129 {
130 throw new RuntimeException("Not supported yet");
131 }
132
133 public synchronized void close()
134 throws IOException
135 {
136 flushByte();
137 super.close();
138 }
139 }