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