Home | History | Annotate | Download | only in datamatrix 
      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 "xfa/src/fxbarcode/common/BC_CommonBitMatrix.h"
     25 #include "xfa/src/fxbarcode/common/reedsolomon/BC_ReedSolomonDecoder.h"
     26 #include "xfa/src/fxbarcode/common/reedsolomon/BC_ReedSolomonGF256.h"
     27 #include "BC_DataMatrixDecoder.h"
     28 #include "BC_DataMatrixBitMatrixParser.h"
     29 #include "BC_DataMatrixVersion.h"
     30 #include "BC_DataMatrixDataBlock.h"
     31 #include "BC_DataMatrixDecodedBitStreamParser.h"
     32 CBC_DataMatrixDecoder::CBC_DataMatrixDecoder() {
     33   m_rsDecoder = NULL;
     34 }
     35 void CBC_DataMatrixDecoder::Init() {
     36   m_rsDecoder =
     37       new CBC_ReedSolomonDecoder(CBC_ReedSolomonGF256::DataMatrixField);
     38 }
     39 CBC_DataMatrixDecoder::~CBC_DataMatrixDecoder() {
     40   if (m_rsDecoder != NULL) {
     41     delete m_rsDecoder;
     42   }
     43   m_rsDecoder = NULL;
     44 }
     45 CBC_CommonDecoderResult* CBC_DataMatrixDecoder::Decode(
     46     CBC_CommonBitMatrix* bits,
     47     int32_t& e) {
     48   CBC_DataMatrixBitMatrixParser parser;
     49   parser.Init(bits, e);
     50   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
     51   CBC_DataMatrixVersion* version = parser.GetVersion();
     52   CFX_ByteArray* byteTemp = parser.ReadCodewords(e);
     53   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
     54   CBC_AutoPtr<CFX_ByteArray> codewords(byteTemp);
     55   CFX_PtrArray* dataBlocks =
     56       CBC_DataMatrixDataBlock::GetDataBlocks(codewords.get(), version, e);
     57   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
     58   int32_t dataBlocksCount = dataBlocks->GetSize();
     59   int32_t totalBytes = 0;
     60   int32_t i, j;
     61   for (i = 0; i < dataBlocksCount; i++) {
     62     totalBytes +=
     63         ((CBC_DataMatrixDataBlock*)(*dataBlocks)[i])->GetNumDataCodewords();
     64   }
     65   CFX_ByteArray resultBytes;
     66   resultBytes.SetSize(totalBytes);
     67   for (j = 0; j < dataBlocksCount; j++) {
     68     CFX_ByteArray* codewordBytes =
     69         ((CBC_DataMatrixDataBlock*)(*dataBlocks)[j])->GetCodewords();
     70     int32_t numDataCodewords =
     71         ((CBC_DataMatrixDataBlock*)(*dataBlocks)[j])->GetNumDataCodewords();
     72     CorrectErrors(*codewordBytes, numDataCodewords, e);
     73     if (e != BCExceptionNO) {
     74       for (int32_t i = 0; i < dataBlocks->GetSize(); i++) {
     75         delete (CBC_DataMatrixDataBlock*)(*dataBlocks)[i];
     76       }
     77       delete dataBlocks;
     78       dataBlocks = NULL;
     79       return NULL;
     80     }
     81     int32_t i;
     82     for (i = 0; i < numDataCodewords; i++) {
     83       resultBytes[i * dataBlocksCount + j] = (*codewordBytes)[i];
     84     }
     85   }
     86   for (i = 0; i < (dataBlocks->GetSize()); i++) {
     87     delete (CBC_DataMatrixDataBlock*)(*dataBlocks)[i];
     88   }
     89   delete dataBlocks;
     90   dataBlocks = NULL;
     91   CBC_CommonDecoderResult* resultR =
     92       CBC_DataMatrixDecodedBitStreamParser::Decode(resultBytes, e);
     93   BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
     94   return resultR;
     95 }
     96 void CBC_DataMatrixDecoder::CorrectErrors(CFX_ByteArray& codewordBytes,
     97                                           int32_t numDataCodewords,
     98                                           int32_t& e) {
     99   int32_t numCodewords = codewordBytes.GetSize();
    100   CFX_Int32Array codewordsInts;
    101   codewordsInts.SetSize(numCodewords);
    102   int32_t i;
    103   for (i = 0; i < numCodewords; i++) {
    104     codewordsInts[i] = codewordBytes[i] & 0xFF;
    105   }
    106   int32_t numECCodewords = codewordBytes.GetSize() - numDataCodewords;
    107   m_rsDecoder->Decode(&codewordsInts, numECCodewords, e);
    108   if (e != BCExceptionNO) {
    109     e = BCExceptionChecksumException;
    110     return;
    111   }
    112   for (i = 0; i < numDataCodewords; i++) {
    113     codewordBytes[i] = (uint8_t)codewordsInts[i];
    114   }
    115 }
    116