OLD | NEW |
| (Empty) |
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/qrcode/BC_QRDataBlock.h" | |
24 | |
25 #include <memory> | |
26 | |
27 #include "xfa/src/fxbarcode/qrcode/BC_QRCoderECB.h" | |
28 #include "xfa/src/fxbarcode/qrcode/BC_QRCoderECBlocks.h" | |
29 #include "xfa/src/fxbarcode/qrcode/BC_QRCoderVersion.h" | |
30 #include "xfa/src/fxbarcode/utils.h" | |
31 | |
32 CBC_QRDataBlock::CBC_QRDataBlock(int32_t numDataCodewords, | |
33 CFX_ByteArray* codewords) | |
34 : m_numDataCodewords(numDataCodewords), m_codewords(codewords) {} | |
35 CBC_QRDataBlock::~CBC_QRDataBlock() { | |
36 delete m_codewords; | |
37 } | |
38 int32_t CBC_QRDataBlock::GetNumDataCodewords() { | |
39 return m_numDataCodewords; | |
40 } | |
41 CFX_ByteArray* CBC_QRDataBlock::GetCodewords() { | |
42 return m_codewords; | |
43 } | |
44 CFX_PtrArray* CBC_QRDataBlock::GetDataBlocks( | |
45 CFX_ByteArray* rawCodewords, | |
46 CBC_QRCoderVersion* version, | |
47 CBC_QRCoderErrorCorrectionLevel* ecLevel, | |
48 int32_t& e) { | |
49 if (rawCodewords->GetSize() != version->GetTotalCodeWords()) { | |
50 e = BCExceptionIllegalArgument; | |
51 BC_EXCEPTION_CHECK_ReturnValue(e, NULL); | |
52 } | |
53 CBC_QRCoderECBlocks* ecBlocks = version->GetECBlocksForLevel(ecLevel); | |
54 int32_t totalBlocks = 0; | |
55 CFX_PtrArray* ecBlockArray = ecBlocks->GetECBlocks(); | |
56 int32_t i = 0; | |
57 for (i = 0; i < ecBlockArray->GetSize(); i++) { | |
58 totalBlocks += ((CBC_QRCoderECB*)(*ecBlockArray)[i])->GetCount(); | |
59 } | |
60 std::unique_ptr<CFX_PtrArray> result(new CFX_PtrArray()); | |
61 result->SetSize(totalBlocks); | |
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 } | |
OLD | NEW |