87cc4aea6238eb1ac1971e32b9540dd36f930278
[zxing.git] / core / test / src / com / google / zxing / qrcode / encoder / BitVectorTestCase.java
1 /*
2  * Copyright 2008 ZXing authors
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package com.google.zxing.qrcode.encoder;
18
19 import com.google.zxing.common.BitArray;
20 import junit.framework.TestCase;
21
22 /**
23  * @author satorux@google.com (Satoru Takabayashi) - creator
24  * @author dswitkin@google.com (Daniel Switkin) - ported from C++
25  */
26 public class BitVectorTestCase extends TestCase {
27
28   private static long getUnsignedInt(BitArray v, int index) {
29     long result = 0L;
30     for (int i = 0, offset = index << 3; i < 32; i++) {
31       if (v.get(offset + i)) {
32         result |= 1L << (31 - i);
33       }
34     }
35     return result;
36   }
37
38   public void testAppendBit() {
39     BitArray v = new BitArray();
40     assertEquals(0, v.getSizeInBytes());
41     // 1
42     v.appendBit(true);
43     assertEquals(1, v.getSize());
44     assertEquals(0x80000000L, getUnsignedInt(v, 0));
45     // 10
46     v.appendBit(false);
47     assertEquals(2, v.getSize());
48     assertEquals(0x80000000L, getUnsignedInt(v, 0));
49     // 101
50     v.appendBit(true);
51     assertEquals(3, v.getSize());
52     assertEquals(0xa0000000L, getUnsignedInt(v, 0));
53     // 1010
54     v.appendBit(false);
55     assertEquals(4, v.getSize());
56     assertEquals(0xa0000000L, getUnsignedInt(v, 0));
57     // 10101
58     v.appendBit(true);
59     assertEquals(5, v.getSize());
60     assertEquals(0xa8000000L, getUnsignedInt(v, 0));
61     // 101010
62     v.appendBit(false);
63     assertEquals(6, v.getSize());
64     assertEquals(0xa8000000L, getUnsignedInt(v, 0));
65     // 1010101
66     v.appendBit(true);
67     assertEquals(7, v.getSize());
68     assertEquals(0xaa000000L, getUnsignedInt(v, 0));
69     // 10101010
70     v.appendBit(false);
71     assertEquals(8, v.getSize());
72     assertEquals(0xaa000000L, getUnsignedInt(v, 0));
73     // 10101010 1
74     v.appendBit(true);
75     assertEquals(9, v.getSize());
76     assertEquals(0xaa800000L, getUnsignedInt(v, 0));
77     // 10101010 10
78     v.appendBit(false);
79     assertEquals(10, v.getSize());
80     assertEquals(0xaa800000L, getUnsignedInt(v, 0));
81   }
82
83   public void testAppendBits() {
84     {
85       BitArray v = new BitArray();
86       v.appendBits(0x1, 1);
87       assertEquals(1, v.getSize());
88       assertEquals(0x80000000L, getUnsignedInt(v, 0));
89     }
90     {
91       BitArray v = new BitArray();
92       v.appendBits(0xff, 8);
93       assertEquals(8, v.getSize());
94       assertEquals(0xff000000L, getUnsignedInt(v, 0));
95     }
96     {
97       BitArray v = new BitArray();
98       v.appendBits(0xff7, 12);
99       assertEquals(12, v.getSize());
100       assertEquals(0xff700000L, getUnsignedInt(v, 0));
101     }
102   }
103
104   public void testNumBytes() {
105     BitArray v = new BitArray();
106     assertEquals(0, v.getSizeInBytes());
107     v.appendBit(false);
108     // 1 bit was added in the vector, so 1 byte should be consumed.
109     assertEquals(1, v.getSizeInBytes());
110     v.appendBits(0, 7);
111     assertEquals(1, v.getSizeInBytes());
112     v.appendBits(0, 8);
113     assertEquals(2, v.getSizeInBytes());
114     v.appendBits(0, 1);
115     // We now have 17 bits, so 3 bytes should be consumed.
116     assertEquals(3, v.getSizeInBytes());
117   }
118
119   public void testAppendBitVector() {
120     BitArray v1 = new BitArray();
121     v1.appendBits(0xbe, 8);
122     BitArray v2 = new BitArray();
123     v2.appendBits(0xef, 8);
124     v1.appendBitArray(v2);
125     // beef = 1011 1110 1110 1111
126     assertEquals(" X.XXXXX. XXX.XXXX", v1.toString());
127   }
128
129   public void testXOR() {
130     {
131       BitArray v1 = new BitArray();
132       v1.appendBits(0x5555aaaa, 32);
133       BitArray v2 = new BitArray();
134       v2.appendBits(0xaaaa5555, 32);
135       v1.xor(v2);
136       assertEquals(0xffffffffL, getUnsignedInt(v1, 0));
137     }
138     {
139       BitArray v1 = new BitArray();
140       v1.appendBits(0x2a, 7);  // 010 1010
141       BitArray v2 = new BitArray();
142       v2.appendBits(0x55, 7);  // 101 0101
143       v1.xor(v2);
144       assertEquals(0xfe000000L, getUnsignedInt(v1, 0));  // 1111 1110
145     }
146   }
147
148   public void testAt() {
149     BitArray v = new BitArray();
150     v.appendBits(0xdead, 16);  // 1101 1110 1010 1101
151     assertTrue(v.get(0));
152     assertTrue(v.get(1));
153     assertFalse(v.get(2));
154     assertTrue(v.get(3));
155
156     assertTrue(v.get(4));
157     assertTrue(v.get(5));
158     assertTrue(v.get(6));
159     assertFalse(v.get(7));
160
161     assertTrue(v.get(8));
162     assertFalse(v.get(9));
163     assertTrue(v.get(10));
164     assertFalse(v.get(11));
165
166     assertTrue(v.get(12));
167     assertTrue(v.get(13));
168     assertFalse(v.get(14));
169     assertTrue(v.get(15));
170   }
171
172   public void testToString() {
173     BitArray v = new BitArray();
174     v.appendBits(0xdead, 16);  // 1101 1110 1010 1101
175     assertEquals(" XX.XXXX. X.X.XX.X", v.toString());
176   }
177
178 }