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 2007 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/reedsolomon/BC_ReedSolomonDecoder.h"
26 #include "xfa/src/fxbarcode/common/reedsolomon/BC_ReedSolomonGF256.h"
27 #include "BC_DataMatrixDecoder.h"
28 #include "BC_DataMatrixBitMatrixParser.h"
29 #include "BC_DataMatrixVersion.h"
30 #include "BC_DataMatrixDataBlock.h"
31 #include "BC_DataMatrixDecodedBitStreamParser.h"
CBC_DataMatrixDecoder()32 CBC_DataMatrixDecoder::CBC_DataMatrixDecoder() {
33 m_rsDecoder = NULL;
34 }
Init()35 void CBC_DataMatrixDecoder::Init() {
36 m_rsDecoder =
37 new CBC_ReedSolomonDecoder(CBC_ReedSolomonGF256::DataMatrixField);
38 }
~CBC_DataMatrixDecoder()39 CBC_DataMatrixDecoder::~CBC_DataMatrixDecoder() {
40 if (m_rsDecoder != NULL) {
41 delete m_rsDecoder;
42 }
43 m_rsDecoder = NULL;
44 }
Decode(CBC_CommonBitMatrix * bits,int32_t & e)45 CBC_CommonDecoderResult* CBC_DataMatrixDecoder::Decode(
46 CBC_CommonBitMatrix* bits,
47 int32_t& e) {
48 CBC_DataMatrixBitMatrixParser parser;
49 parser.Init(bits, e);
50 BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
51 CBC_DataMatrixVersion* version = parser.GetVersion();
52 CFX_ByteArray* byteTemp = parser.ReadCodewords(e);
53 BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
54 CBC_AutoPtr<CFX_ByteArray> codewords(byteTemp);
55 CFX_PtrArray* dataBlocks =
56 CBC_DataMatrixDataBlock::GetDataBlocks(codewords.get(), version, e);
57 BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
58 int32_t dataBlocksCount = dataBlocks->GetSize();
59 int32_t totalBytes = 0;
60 int32_t i, j;
61 for (i = 0; i < dataBlocksCount; i++) {
62 totalBytes +=
63 ((CBC_DataMatrixDataBlock*)(*dataBlocks)[i])->GetNumDataCodewords();
64 }
65 CFX_ByteArray resultBytes;
66 resultBytes.SetSize(totalBytes);
67 for (j = 0; j < dataBlocksCount; j++) {
68 CFX_ByteArray* codewordBytes =
69 ((CBC_DataMatrixDataBlock*)(*dataBlocks)[j])->GetCodewords();
70 int32_t numDataCodewords =
71 ((CBC_DataMatrixDataBlock*)(*dataBlocks)[j])->GetNumDataCodewords();
72 CorrectErrors(*codewordBytes, numDataCodewords, e);
73 if (e != BCExceptionNO) {
74 for (int32_t i = 0; i < dataBlocks->GetSize(); i++) {
75 delete (CBC_DataMatrixDataBlock*)(*dataBlocks)[i];
76 }
77 delete dataBlocks;
78 dataBlocks = NULL;
79 return NULL;
80 }
81 int32_t i;
82 for (i = 0; i < numDataCodewords; i++) {
83 resultBytes[i * dataBlocksCount + j] = (*codewordBytes)[i];
84 }
85 }
86 for (i = 0; i < (dataBlocks->GetSize()); i++) {
87 delete (CBC_DataMatrixDataBlock*)(*dataBlocks)[i];
88 }
89 delete dataBlocks;
90 dataBlocks = NULL;
91 CBC_CommonDecoderResult* resultR =
92 CBC_DataMatrixDecodedBitStreamParser::Decode(resultBytes, e);
93 BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
94 return resultR;
95 }
CorrectErrors(CFX_ByteArray & codewordBytes,int32_t numDataCodewords,int32_t & e)96 void CBC_DataMatrixDecoder::CorrectErrors(CFX_ByteArray& codewordBytes,
97 int32_t numDataCodewords,
98 int32_t& e) {
99 int32_t numCodewords = codewordBytes.GetSize();
100 CFX_Int32Array codewordsInts;
101 codewordsInts.SetSize(numCodewords);
102 int32_t i;
103 for (i = 0; i < numCodewords; i++) {
104 codewordsInts[i] = codewordBytes[i] & 0xFF;
105 }
106 int32_t numECCodewords = codewordBytes.GetSize() - numDataCodewords;
107 m_rsDecoder->Decode(&codewordsInts, numECCodewords, e);
108 if (e != BCExceptionNO) {
109 e = BCExceptionChecksumException;
110 return;
111 }
112 for (i = 0; i < numDataCodewords; i++) {
113 codewordBytes[i] = (uint8_t)codewordsInts[i];
114 }
115 }
116