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 "BC_QRCoderECBlocks.h"
25 #include "BC_QRCoderECB.h"
26 #include "BC_QRDataBlock.h"
27 #include "BC_QRCoderVersion.h"
CBC_QRDataBlock(int32_t numDataCodewords,CFX_ByteArray * codewords)28 CBC_QRDataBlock::CBC_QRDataBlock(int32_t numDataCodewords,
29 CFX_ByteArray* codewords)
30 : m_numDataCodewords(numDataCodewords), m_codewords(codewords) {}
~CBC_QRDataBlock()31 CBC_QRDataBlock::~CBC_QRDataBlock() {
32 if (m_codewords != NULL) {
33 delete m_codewords;
34 m_codewords = NULL;
35 }
36 }
GetNumDataCodewords()37 int32_t CBC_QRDataBlock::GetNumDataCodewords() {
38 return m_numDataCodewords;
39 }
GetCodewords()40 CFX_ByteArray* CBC_QRDataBlock::GetCodewords() {
41 return m_codewords;
42 }
GetDataBlocks(CFX_ByteArray * rawCodewords,CBC_QRCoderVersion * version,CBC_QRCoderErrorCorrectionLevel * ecLevel,int32_t & e)43 CFX_PtrArray* CBC_QRDataBlock::GetDataBlocks(
44 CFX_ByteArray* rawCodewords,
45 CBC_QRCoderVersion* version,
46 CBC_QRCoderErrorCorrectionLevel* ecLevel,
47 int32_t& e) {
48 if (rawCodewords->GetSize() != version->GetTotalCodeWords()) {
49 e = BCExceptionIllegalArgument;
50 BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
51 }
52 CBC_QRCoderECBlocks* ecBlocks = version->GetECBlocksForLevel(ecLevel);
53 int32_t totalBlocks = 0;
54 CFX_PtrArray* ecBlockArray = ecBlocks->GetECBlocks();
55 int32_t i = 0;
56 for (i = 0; i < ecBlockArray->GetSize(); i++) {
57 totalBlocks += ((CBC_QRCoderECB*)(*ecBlockArray)[i])->GetCount();
58 }
59 CFX_PtrArray* datablock = new CFX_PtrArray();
60 datablock->SetSize(totalBlocks);
61 CBC_AutoPtr<CFX_PtrArray> result(datablock);
62 int32_t numResultBlocks = 0;
63 for (int32_t j = 0; j < ecBlockArray->GetSize(); j++) {
64 CBC_QRCoderECB* ecBlock = (CBC_QRCoderECB*)(*ecBlockArray)[j];
65 for (int32_t k = 0; k < ecBlock->GetCount(); k++) {
66 int32_t numDataCodewords = ecBlock->GetDataCodeWords();
67 int32_t numBlockCodewords =
68 ecBlocks->GetECCodeWordsPerBlock() + numDataCodewords;
69 CFX_ByteArray* bytearray = new CFX_ByteArray();
70 bytearray->SetSize(numBlockCodewords);
71 (*result)[numResultBlocks++] =
72 new CBC_QRDataBlock(numDataCodewords, bytearray);
73 }
74 }
75 int32_t shorterBlocksTotalCodewords =
76 ((CBC_QRDataBlock*)(*result)[0])->m_codewords->GetSize();
77 int32_t longerBlocksStartAt = result->GetSize() - 1;
78 while (longerBlocksStartAt >= 0) {
79 int32_t numCodewords = ((CBC_QRDataBlock*)(*result)[longerBlocksStartAt])
80 ->m_codewords->GetSize();
81 if (numCodewords == shorterBlocksTotalCodewords) {
82 break;
83 }
84 longerBlocksStartAt--;
85 }
86 longerBlocksStartAt++;
87 int32_t shorterBlocksNumDataCodewords =
88 shorterBlocksTotalCodewords - ecBlocks->GetECCodeWordsPerBlock();
89 int32_t rawCodewordsOffset = 0;
90 int32_t x = 0;
91 for (int32_t k = 0; k < shorterBlocksNumDataCodewords; k++) {
92 for (x = 0; x < numResultBlocks; x++) {
93 (*(((CBC_QRDataBlock*)(*result)[x])->m_codewords))[k] =
94 (*rawCodewords)[rawCodewordsOffset++];
95 }
96 }
97 for (x = longerBlocksStartAt; x < numResultBlocks; x++) {
98 (*(((CBC_QRDataBlock*)(*result)[x])
99 ->m_codewords))[shorterBlocksNumDataCodewords] =
100 (*rawCodewords)[rawCodewordsOffset++];
101 }
102 int32_t max = ((CBC_QRDataBlock*)(*result)[0])->m_codewords->GetSize();
103 for (i = shorterBlocksNumDataCodewords; i < max; i++) {
104 for (int32_t y = 0; y < numResultBlocks; y++) {
105 int32_t iOffset = y < longerBlocksStartAt ? i : i + 1;
106 (*(((CBC_QRDataBlock*)(*result)[y])->m_codewords))[iOffset] =
107 (*rawCodewords)[rawCodewordsOffset++];
108 }
109 }
110 return result.release();
111 }
112