1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6 // Original code is licensed as follows:
7 /*
8  * Copyright 2008 ZXing authors
9  *
10  * Licensed under the Apache License, Version 2.0 (the "License");
11  * you may not use this file except in compliance with the License.
12  * You may obtain a copy of the License at
13  *
14  *      http://www.apache.org/licenses/LICENSE-2.0
15  *
16  * Unless required by applicable law or agreed to in writing, software
17  * distributed under the License is distributed on an "AS IS" BASIS,
18  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
19  * See the License for the specific language governing permissions and
20  * limitations under the License.
21  */
22 
23 #include "xfa/src/fxbarcode/barcode.h"
24 #include "xfa/src/fxbarcode/common/BC_CommonBitMatrix.h"
25 #include "xfa/src/fxbarcode/common/BC_CommonDecoderResult.h"
26 #include "xfa/src/fxbarcode/common/reedsolomon/BC_ReedSolomonDecoder.h"
27 #include "xfa/src/fxbarcode/common/reedsolomon/BC_ReedSolomonGF256.h"
28 #include "BC_QRBitMatrixParser.h"
29 #include "BC_QRDataBlock.h"
30 #include "BC_QRDecodedBitStreamParser.h"
31 #include "BC_QRCoderVersion.h"
32 #include "BC_QRCoderFormatInformation.h"
33 #include "BC_QRCoderDecoder.h"
CBC_QRCoderDecoder()34 CBC_QRCoderDecoder::CBC_QRCoderDecoder() {
35   m_rsDecoder = NULL;
36 }
37 
Init()38 void CBC_QRCoderDecoder::Init() {
39   m_rsDecoder = new CBC_ReedSolomonDecoder(CBC_ReedSolomonGF256::QRCodeFild);
40 }
~CBC_QRCoderDecoder()41 CBC_QRCoderDecoder::~CBC_QRCoderDecoder() {
42   if (m_rsDecoder != NULL) {
43     delete m_rsDecoder;
44   }
45   m_rsDecoder = NULL;
46 }
Decode(FX_BOOL * image,int32_t width,int32_t height,int32_t & e)47 CBC_CommonDecoderResult* CBC_QRCoderDecoder::Decode(FX_BOOL* image,
48                                                     int32_t width,
49                                                     int32_t height,
50                                                     int32_t& e) {
51   CBC_CommonBitMatrix bits;
52   bits.Init(width);
53   for (int32_t i = 0; i < width; i++) {
54     for (int32_t j = 0; j < height; j++) {
55       if (image[i * width + j]) {
56         bits.Set(j, i);
57       }
58     }
59   }
60   CBC_CommonDecoderResult* cdr = Decode(&bits, height, e);
61   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
62   return cdr;
63 }
Decode(CBC_CommonBitMatrix * bits,int32_t byteModeDecode,int32_t & e)64 CBC_CommonDecoderResult* CBC_QRCoderDecoder::Decode(CBC_CommonBitMatrix* bits,
65                                                     int32_t byteModeDecode,
66                                                     int32_t& e) {
67   CBC_QRBitMatrixParser parser;
68   parser.Init(bits, e);
69   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
70   CBC_QRCoderVersion* version = parser.ReadVersion(e);
71   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
72   CBC_QRCoderFormatInformation* temp = parser.ReadFormatInformation(e);
73   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
74   CBC_QRCoderErrorCorrectionLevel* ecLevel = temp->GetErrorCorrectionLevel();
75   CFX_ByteArray* ba = parser.ReadCodewords(e);
76   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
77   CBC_AutoPtr<CFX_ByteArray> codewords(ba);
78   CFX_PtrArray* dataBlocks =
79       CBC_QRDataBlock::GetDataBlocks(codewords.get(), version, ecLevel, e);
80   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
81   int32_t totalBytes = 0;
82   for (int32_t i = 0; i < dataBlocks->GetSize(); i++) {
83     totalBytes += ((CBC_QRDataBlock*)((*dataBlocks)[i]))->GetNumDataCodewords();
84   }
85   CFX_ByteArray resultBytes;
86   for (int32_t j = 0; j < dataBlocks->GetSize(); j++) {
87     CBC_QRDataBlock* dataBlock = (CBC_QRDataBlock*)((*dataBlocks)[j]);
88     CFX_ByteArray* codewordBytes = dataBlock->GetCodewords();
89     int32_t numDataCodewords = dataBlock->GetNumDataCodewords();
90     CorrectErrors(codewordBytes, numDataCodewords, e);
91     if (e != BCExceptionNO) {
92       for (int32_t k = 0; k < dataBlocks->GetSize(); k++) {
93         delete (CBC_QRDataBlock*)(*dataBlocks)[k];
94       }
95       dataBlocks->RemoveAll();
96       delete dataBlocks;
97       dataBlocks = NULL;
98       return NULL;
99     }
100     for (int32_t i = 0; i < numDataCodewords; i++) {
101       resultBytes.Add((*codewordBytes)[i]);
102     }
103   }
104   for (int32_t k = 0; k < dataBlocks->GetSize(); k++) {
105     delete (CBC_QRDataBlock*)(*dataBlocks)[k];
106   }
107   dataBlocks->RemoveAll();
108   delete dataBlocks;
109   dataBlocks = NULL;
110   CBC_CommonDecoderResult* cdr = CBC_QRDecodedBitStreamParser::Decode(
111       &resultBytes, version, ecLevel, byteModeDecode, e);
112   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
113   return cdr;
114 }
CorrectErrors(CFX_ByteArray * codewordBytes,int32_t numDataCodewords,int32_t & e)115 void CBC_QRCoderDecoder::CorrectErrors(CFX_ByteArray* codewordBytes,
116                                        int32_t numDataCodewords,
117                                        int32_t& e) {
118   int32_t numCodewords = codewordBytes->GetSize();
119   CFX_Int32Array codewordsInts;
120   codewordsInts.SetSize(numCodewords);
121   for (int32_t i = 0; i < numCodewords; i++) {
122     codewordsInts[i] = (int32_t)((*codewordBytes)[i] & 0xff);
123   }
124   int32_t numECCodewords = codewordBytes->GetSize() - numDataCodewords;
125   m_rsDecoder->Decode(&codewordsInts, numECCodewords, e);
126   BC_EXCEPTION_CHECK_ReturnVoid(e);
127   for (int32_t k = 0; k < numDataCodewords; k++) {
128     (*codewordBytes)[k] = (uint8_t)codewordsInts[k];
129   }
130 }
131