// Copyright 2014 PDFium Authors. All rights reserved. | |
// Use of this source code is governed by a BSD-style license that can be | |
// found in the LICENSE file. | |
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | |
// Original code is licensed as follows: | |
/* | |
* Copyright 2007 ZXing authors | |
* | |
* Licensed under the Apache License, Version 2.0 (the "License"); | |
* you may not use this file except in compliance with the License. | |
* You may obtain a copy of the License at | |
* | |
* http://www.apache.org/licenses/LICENSE-2.0 | |
* | |
* Unless required by applicable law or agreed to in writing, software | |
* distributed under the License is distributed on an "AS IS" BASIS, | |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
* See the License for the specific language governing permissions and | |
* limitations under the License. | |
*/ | |
#include "xfa/src/fxbarcode/barcode.h" | |
#include "BC_QRCoderECBlocks.h" | |
#include "BC_QRCoderECB.h" | |
#include "BC_QRDataBlock.h" | |
#include "BC_QRCoderVersion.h" | |
CBC_QRDataBlock::CBC_QRDataBlock(int32_t numDataCodewords, | |
CFX_ByteArray* codewords) | |
: m_numDataCodewords(numDataCodewords), m_codewords(codewords) {} | |
CBC_QRDataBlock::~CBC_QRDataBlock() { | |
if (m_codewords != NULL) { | |
delete m_codewords; | |
m_codewords = NULL; | |
} | |
} | |
int32_t CBC_QRDataBlock::GetNumDataCodewords() { | |
return m_numDataCodewords; | |
} | |
CFX_ByteArray* CBC_QRDataBlock::GetCodewords() { | |
return m_codewords; | |
} | |
CFX_PtrArray* CBC_QRDataBlock::GetDataBlocks( | |
CFX_ByteArray* rawCodewords, | |
CBC_QRCoderVersion* version, | |
CBC_QRCoderErrorCorrectionLevel* ecLevel, | |
int32_t& e) { | |
if (rawCodewords->GetSize() != version->GetTotalCodeWords()) { | |
e = BCExceptionIllegalArgument; | |
BC_EXCEPTION_CHECK_ReturnValue(e, NULL); | |
} | |
CBC_QRCoderECBlocks* ecBlocks = version->GetECBlocksForLevel(ecLevel); | |
int32_t totalBlocks = 0; | |
CFX_PtrArray* ecBlockArray = ecBlocks->GetECBlocks(); | |
int32_t i = 0; | |
for (i = 0; i < ecBlockArray->GetSize(); i++) { | |
totalBlocks += ((CBC_QRCoderECB*)(*ecBlockArray)[i])->GetCount(); | |
} | |
CFX_PtrArray* datablock = new CFX_PtrArray(); | |
datablock->SetSize(totalBlocks); | |
CBC_AutoPtr<CFX_PtrArray> result(datablock); | |
int32_t numResultBlocks = 0; | |
for (int32_t j = 0; j < ecBlockArray->GetSize(); j++) { | |
CBC_QRCoderECB* ecBlock = (CBC_QRCoderECB*)(*ecBlockArray)[j]; | |
for (int32_t k = 0; k < ecBlock->GetCount(); k++) { | |
int32_t numDataCodewords = ecBlock->GetDataCodeWords(); | |
int32_t numBlockCodewords = | |
ecBlocks->GetECCodeWordsPerBlock() + numDataCodewords; | |
CFX_ByteArray* bytearray = new CFX_ByteArray(); | |
bytearray->SetSize(numBlockCodewords); | |
(*result)[numResultBlocks++] = | |
new CBC_QRDataBlock(numDataCodewords, bytearray); | |
} | |
} | |
int32_t shorterBlocksTotalCodewords = | |
((CBC_QRDataBlock*)(*result)[0])->m_codewords->GetSize(); | |
int32_t longerBlocksStartAt = result->GetSize() - 1; | |
while (longerBlocksStartAt >= 0) { | |
int32_t numCodewords = ((CBC_QRDataBlock*)(*result)[longerBlocksStartAt]) | |
->m_codewords->GetSize(); | |
if (numCodewords == shorterBlocksTotalCodewords) { | |
break; | |
} | |
longerBlocksStartAt--; | |
} | |
longerBlocksStartAt++; | |
int32_t shorterBlocksNumDataCodewords = | |
shorterBlocksTotalCodewords - ecBlocks->GetECCodeWordsPerBlock(); | |
int32_t rawCodewordsOffset = 0; | |
int32_t x = 0; | |
for (int32_t k = 0; k < shorterBlocksNumDataCodewords; k++) { | |
for (x = 0; x < numResultBlocks; x++) { | |
(*(((CBC_QRDataBlock*)(*result)[x])->m_codewords))[k] = | |
(*rawCodewords)[rawCodewordsOffset++]; | |
} | |
} | |
for (x = longerBlocksStartAt; x < numResultBlocks; x++) { | |
(*(((CBC_QRDataBlock*)(*result)[x]) | |
->m_codewords))[shorterBlocksNumDataCodewords] = | |
(*rawCodewords)[rawCodewordsOffset++]; | |
} | |
int32_t max = ((CBC_QRDataBlock*)(*result)[0])->m_codewords->GetSize(); | |
for (i = shorterBlocksNumDataCodewords; i < max; i++) { | |
for (int32_t y = 0; y < numResultBlocks; y++) { | |
int32_t iOffset = y < longerBlocksStartAt ? i : i + 1; | |
(*(((CBC_QRDataBlock*)(*result)[y])->m_codewords))[iOffset] = | |
(*rawCodewords)[rawCodewordsOffset++]; | |
} | |
} | |
return result.release(); | |
} |