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