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 junit.framework.TestCase;
  20
  21 /**
  22  * @author satorux@google.com (Satoru Takabayashi) - creator
  23  * @author dswitkin@google.com (Daniel Switkin) - ported from C++
  24  */
  25 public class BitVectorTestCase extends TestCase {
  26
  27   private static int getUnsignedByte(BitVector v, int index) {
  28     return v.getArray()[index] & 0xff;
  29   }
  30
  31   public void testAppendBit() {
  32     BitVector v = new BitVector();
  33     assertEquals(0, v.sizeInBytes());
  34     // 1
  35     v.appendBit(1);
  36     assertEquals(1, v.size());
  37     assertEquals(0x80, getUnsignedByte(v, 0));
  38     // 10
  39     v.appendBit(0);
  40     assertEquals(2, v.size());
  41     assertEquals(0x80, getUnsignedByte(v, 0));
  42     // 101
  43     v.appendBit(1);
  44     assertEquals(3, v.size());
  45     assertEquals(0xa0, getUnsignedByte(v, 0));
  46     // 1010
  47     v.appendBit(0);
  48     assertEquals(4, v.size());
  49     assertEquals(0xa0, getUnsignedByte(v, 0));
  50     // 10101
  51     v.appendBit(1);
  52     assertEquals(5, v.size());
  53     assertEquals(0xa8, getUnsignedByte(v, 0));
  54     // 101010
  55     v.appendBit(0);
  56     assertEquals(6, v.size());
  57     assertEquals(0xa8, getUnsignedByte(v, 0));
  58     // 1010101
  59     v.appendBit(1);
  60     assertEquals(7, v.size());
  61     assertEquals(0xaa, getUnsignedByte(v, 0));
  62     // 10101010
  63     v.appendBit(0);
  64     assertEquals(8, v.size());
  65     assertEquals(0xaa, getUnsignedByte(v, 0));
  66     // 10101010 1
  67     v.appendBit(1);
  68     assertEquals(9, v.size());
  69     assertEquals(0xaa, getUnsignedByte(v, 0));
  70     assertEquals(0x80, getUnsignedByte(v, 1));
  71     // 10101010 10
  72     v.appendBit(0);
  73     assertEquals(10, v.size());
  74     assertEquals(0xaa, getUnsignedByte(v, 0));
  75     assertEquals(0x80, getUnsignedByte(v, 1));
  76   }
  77
  78   public void testAppendBits() {
  79     {
  80       BitVector v = new BitVector();
  81       v.appendBits(0x1, 1);
  82       assertEquals(1, v.size());
  83       assertEquals(0x80, getUnsignedByte(v, 0));
  84     }
  85     {
  86       BitVector v = new BitVector();
  87       v.appendBits(0xff, 8);
  88       assertEquals(8, v.size());
  89       assertEquals(0xff, getUnsignedByte(v, 0));
  90     }
  91     {
  92       BitVector v = new BitVector();
  93       v.appendBits(0xff7, 12);
  94       assertEquals(12, v.size());
  95       assertEquals(0xff, getUnsignedByte(v, 0));
  96       assertEquals(0x70, getUnsignedByte(v, 1));
  97     }
  98   }
  99
 100   public void testNumBytes() {
 101     BitVector v = new BitVector();
 102     assertEquals(0, v.sizeInBytes());
 103     v.appendBit(0);
 104     // 1 bit was added in the vector, so 1 byte should be consumed.
 105     assertEquals(1, v.sizeInBytes());
 106     v.appendBits(0, 7);
 107     assertEquals(1, v.sizeInBytes());
 108     v.appendBits(0, 8);
 109     assertEquals(2, v.sizeInBytes());
 110     v.appendBits(0, 1);
 111     // We now have 17 bits, so 3 bytes should be consumed.
 112     assertEquals(3, v.sizeInBytes());
 113   }
 114
 115   public void testAppendBitVector() {
 116     BitVector v1 = new BitVector();
 117     v1.appendBits(0xbe, 8);
 118     BitVector v2 = new BitVector();
 119     v2.appendBits(0xef, 8);
 120     v1.appendBitVector(v2);
 121     // beef = 1011 1110 1110 1111
 122     assertEquals("1011111011101111", v1.toString());
 123   }
 124
 125   public void testXOR() {
 126     {
 127       BitVector v1 = new BitVector();
 128       v1.appendBits(0x5555aaaa, 32);
 129       BitVector v2 = new BitVector();
 130       v2.appendBits(0xaaaa5555, 32);
 131       v1.xor(v2);
 132       assertEquals(0xff, getUnsignedByte(v1, 0));
 133       assertEquals(0xff, getUnsignedByte(v1, 1));
 134       assertEquals(0xff, getUnsignedByte(v1, 2));
 135       assertEquals(0xff, getUnsignedByte(v1, 3));
 136     }
 137     {
 138       BitVector v1 = new BitVector();
 139       v1.appendBits(0x2a, 7);  // 010 1010
 140       BitVector v2 = new BitVector();
 141       v2.appendBits(0x55, 7);  // 101 0101
 142       v1.xor(v2);
 143       assertEquals(0xfe, getUnsignedByte(v1, 0));  // 1111 1110
 144     }
 145   }
 146
 147   public void testAt() {
 148     BitVector v = new BitVector();
 149     v.appendBits(0xdead, 16);  // 1101 1110 1010 1101
 150     assertEquals(1, v.at(0));
 151     assertEquals(1, v.at(1));
 152     assertEquals(0, v.at(2));
 153     assertEquals(1, v.at(3));
 154
 155     assertEquals(1, v.at(4));
 156     assertEquals(1, v.at(5));
 157     assertEquals(1, v.at(6));
 158     assertEquals(0, v.at(7));
 159
 160     assertEquals(1, v.at(8));
 161     assertEquals(0, v.at(9));
 162     assertEquals(1, v.at(10));
 163     assertEquals(0, v.at(11));
 164
 165     assertEquals(1, v.at(12));
 166     assertEquals(1, v.at(13));
 167     assertEquals(0, v.at(14));
 168     assertEquals(1, v.at(15));
 169   }
 170
 171   public void testToString() {
 172     BitVector v = new BitVector();
 173     v.appendBits(0xdead, 16);  // 1101 1110 1010 1101
 174     assertEquals("1101111010101101", v.toString());
 175   }
 176
 177 }