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"
28CBC_QRDataBlock::CBC_QRDataBlock(int32_t numDataCodewords,
29                                 CFX_ByteArray* codewords)
30    : m_numDataCodewords(numDataCodewords), m_codewords(codewords) {}
31CBC_QRDataBlock::~CBC_QRDataBlock() {
32  if (m_codewords != NULL) {
33    delete m_codewords;
34    m_codewords = NULL;
35  }
36}
37int32_t CBC_QRDataBlock::GetNumDataCodewords() {
38  return m_numDataCodewords;
39}
40CFX_ByteArray* CBC_QRDataBlock::GetCodewords() {
41  return m_codewords;
42}
43CFX_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