xfa/src/fxbarcode/src/BC_QRDataBlock.cpp

// 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

#include "barcode.h"
#include "include/BC_QRCoderECBlocks.h"
#include "include/BC_QRCoderECB.h"
#include "include/BC_QRDataBlock.h"
#include "include/BC_QRCoderVersion.h"
CBC_QRDataBlock::CBC_QRDataBlock(FX_INT32 numDataCodewords, CFX_ByteArray *codewords)
    : m_numDataCodewords(numDataCodewords)
    , m_codewords(codewords)
{
}
CBC_QRDataBlock::~CBC_QRDataBlock()
{
    if(m_codewords != NULL) {
        delete m_codewords;
        m_codewords = NULL;
    }
}
FX_INT32 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, FX_INT32 &e)
{
    if(rawCodewords->GetSize() != version->GetTotalCodeWords()) {
        e = BCExceptionIllegalArgument;
        BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    }
    CBC_QRCoderECBlocks *ecBlocks = version->GetECBlocksForLevel(ecLevel);
    FX_INT32 totalBlocks = 0;
    CFX_PtrArray* ecBlockArray = ecBlocks->GetECBlocks();
    FX_INT32 i = 0;
    for(i = 0; i < ecBlockArray->GetSize(); i++) {
        totalBlocks += ((CBC_QRCoderECB*)(*ecBlockArray)[i])->GetCount();
    }
    CFX_PtrArray *datablock = FX_NEW CFX_PtrArray();
    datablock->SetSize(totalBlocks);
    CBC_AutoPtr<CFX_PtrArray > result(datablock);
    FX_INT32 numResultBlocks = 0;
    for(FX_INT32 j = 0; j < ecBlockArray->GetSize(); j++) {
        CBC_QRCoderECB *ecBlock = (CBC_QRCoderECB*)(*ecBlockArray)[j];
        for(FX_INT32 k = 0; k < ecBlock->GetCount(); k++) {
            FX_INT32 numDataCodewords = ecBlock->GetDataCodeWords();
            FX_INT32 numBlockCodewords = ecBlocks->GetECCodeWordsPerBlock() + numDataCodewords;
            CFX_ByteArray *bytearray = FX_NEW CFX_ByteArray();
            bytearray->SetSize(numBlockCodewords);
            (*result)[numResultBlocks++] = FX_NEW CBC_QRDataBlock(numDataCodewords, bytearray);
        }
    }
    FX_INT32 shorterBlocksTotalCodewords = ((CBC_QRDataBlock*)(*result)[0])->m_codewords->GetSize();
    FX_INT32 longerBlocksStartAt = result->GetSize() - 1;
    while(longerBlocksStartAt >= 0) {
        FX_INT32 numCodewords = ((CBC_QRDataBlock*)(*result)[longerBlocksStartAt])->m_codewords->GetSize();
        if(numCodewords == shorterBlocksTotalCodewords) {
            break;
        }
        longerBlocksStartAt--;
    }
    longerBlocksStartAt++;
    FX_INT32 shorterBlocksNumDataCodewords = shorterBlocksTotalCodewords - ecBlocks->GetECCodeWordsPerBlock();
    FX_INT32 rawCodewordsOffset = 0;
    FX_INT32 x = 0;
    for(FX_INT32 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++];
    }
    FX_INT32 max = ((CBC_QRDataBlock*)(*result)[0])->m_codewords->GetSize();
    for(i = shorterBlocksNumDataCodewords; i < max; i++) {
        for(FX_INT32 y = 0; y < numResultBlocks; y++) {
            FX_INT32 iOffset = y < longerBlocksStartAt ? i : i + 1;
            (*(((CBC_QRDataBlock*)(*result)[y])->m_codewords))[iOffset] = (*rawCodewords)[rawCodewordsOffset++];
        }
    }
    return result.release();
}