Related to Issue 205, but not the direct issue: read both copies of the format info...
[zxing.git] / core / src / com / google / zxing / qrcode / decoder / FormatInformation.java
1 /*
2  * Copyright 2007 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.decoder;
18
19 /**
20  * <p>Encapsulates a QR Code's format information, including the data mask used and
21  * error correction level.</p>
22  *
23  * @author Sean Owen
24  * @see DataMask
25  * @see ErrorCorrectionLevel
26  */
27 final class FormatInformation {
28
29   private static final int FORMAT_INFO_MASK_QR = 0x5412;
30
31   /**
32    * See ISO 18004:2006, Annex C, Table C.1
33    */
34   private static final int[][] FORMAT_INFO_DECODE_LOOKUP = {
35       {0x5412, 0x00},
36       {0x5125, 0x01},
37       {0x5E7C, 0x02},
38       {0x5B4B, 0x03},
39       {0x45F9, 0x04},
40       {0x40CE, 0x05},
41       {0x4F97, 0x06},
42       {0x4AA0, 0x07},
43       {0x77C4, 0x08},
44       {0x72F3, 0x09},
45       {0x7DAA, 0x0A},
46       {0x789D, 0x0B},
47       {0x662F, 0x0C},
48       {0x6318, 0x0D},
49       {0x6C41, 0x0E},
50       {0x6976, 0x0F},
51       {0x1689, 0x10},
52       {0x13BE, 0x11},
53       {0x1CE7, 0x12},
54       {0x19D0, 0x13},
55       {0x0762, 0x14},
56       {0x0255, 0x15},
57       {0x0D0C, 0x16},
58       {0x083B, 0x17},
59       {0x355F, 0x18},
60       {0x3068, 0x19},
61       {0x3F31, 0x1A},
62       {0x3A06, 0x1B},
63       {0x24B4, 0x1C},
64       {0x2183, 0x1D},
65       {0x2EDA, 0x1E},
66       {0x2BED, 0x1F},
67   };
68
69   /**
70    * Offset i holds the number of 1 bits in the binary representation of i
71    */
72   private static final int[] BITS_SET_IN_HALF_BYTE =
73       {0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4};
74
75   private final ErrorCorrectionLevel errorCorrectionLevel;
76   private final byte dataMask;
77
78   private FormatInformation(int formatInfo) {
79     // Bits 3,4
80     errorCorrectionLevel = ErrorCorrectionLevel.forBits((formatInfo >> 3) & 0x03);
81     // Bottom 3 bits
82     dataMask = (byte) (formatInfo & 0x07);
83   }
84
85   static int numBitsDiffering(int a, int b) {
86     a ^= b; // a now has a 1 bit exactly where its bit differs with b's
87     // Count bits set quickly with a series of lookups:
88     return BITS_SET_IN_HALF_BYTE[a & 0x0F] +
89         BITS_SET_IN_HALF_BYTE[(a >>> 4 & 0x0F)] +
90         BITS_SET_IN_HALF_BYTE[(a >>> 8 & 0x0F)] +
91         BITS_SET_IN_HALF_BYTE[(a >>> 12 & 0x0F)] +
92         BITS_SET_IN_HALF_BYTE[(a >>> 16 & 0x0F)] +
93         BITS_SET_IN_HALF_BYTE[(a >>> 20 & 0x0F)] +
94         BITS_SET_IN_HALF_BYTE[(a >>> 24 & 0x0F)] +
95         BITS_SET_IN_HALF_BYTE[(a >>> 28 & 0x0F)];
96   }
97
98   /**
99    * @param maskedFormatInfo1 format info indicator, with mask still applied
100    * @param maskedFormatInfo2 second copy of same info; both are checked at the same time
101    *  to establish best match
102    * @return information about the format it specifies, or <code>null</code>
103    *  if doesn't seem to match any known pattern
104    */
105   static FormatInformation decodeFormatInformation(int maskedFormatInfo1, int maskedFormatInfo2) {
106     FormatInformation formatInfo = doDecodeFormatInformation(maskedFormatInfo1, maskedFormatInfo2);
107     if (formatInfo != null) {
108       return formatInfo;
109     }
110     // Should return null, but, some QR codes apparently
111     // do not mask this info. Try again by actually masking the pattern
112     // first
113     return doDecodeFormatInformation(maskedFormatInfo1 ^ FORMAT_INFO_MASK_QR,
114                                      maskedFormatInfo2 ^ FORMAT_INFO_MASK_QR);
115   }
116
117   private static FormatInformation doDecodeFormatInformation(int maskedFormatInfo1, int maskedFormatInfo2) {
118     // Find the int in FORMAT_INFO_DECODE_LOOKUP with fewest bits differing
119     int bestDifference = Integer.MAX_VALUE;
120     int bestFormatInfo = 0;
121     for (int i = 0; i < FORMAT_INFO_DECODE_LOOKUP.length; i++) {
122       int[] decodeInfo = FORMAT_INFO_DECODE_LOOKUP[i];
123       int targetInfo = decodeInfo[0];
124       if (targetInfo == maskedFormatInfo1 || targetInfo == maskedFormatInfo2) {
125         // Found an exact match
126         return new FormatInformation(decodeInfo[1]);
127       }
128       int bitsDifference = numBitsDiffering(maskedFormatInfo1, targetInfo);
129       if (bitsDifference < bestDifference) {
130         bestFormatInfo = decodeInfo[1];
131         bestDifference = bitsDifference;
132       }
133       if (maskedFormatInfo1 != maskedFormatInfo2) {
134         // also try the other option
135         bitsDifference = numBitsDiffering(maskedFormatInfo2, targetInfo);
136         if (bitsDifference < bestDifference) {
137           bestFormatInfo = decodeInfo[1];
138           bestDifference = bitsDifference;
139         }
140       }
141     }
142     // Hamming distance of the 32 masked codes is 7, by construction, so <= 3 bits
143     // differing means we found a match
144     if (bestDifference <= 3) {
145       return new FormatInformation(bestFormatInfo);
146     }
147     return null;
148   }
149
150   ErrorCorrectionLevel getErrorCorrectionLevel() {
151     return errorCorrectionLevel;
152   }
153
154   byte getDataMask() {
155     return dataMask;
156   }
157
158   public int hashCode() {
159     return (errorCorrectionLevel.ordinal() << 3) | (int) dataMask;
160   }
161
162   public boolean equals(Object o) {
163     if (!(o instanceof FormatInformation)) {
164       return false;
165     }
166     FormatInformation other = (FormatInformation) o;
167     return this.errorCorrectionLevel == other.errorCorrectionLevel &&
168         this.dataMask == other.dataMask;
169   }
170
171 }