// 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 2008 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 "../barcode.h"
#include "../common/BC_CommonBitMatrix.h"
#include "../common/BC_CommonDecoderResult.h"
#include "../common/reedsolomon/BC_ReedSolomonDecoder.h"
#include "../common/reedsolomon/BC_ReedSolomonGF256.h"
#include "BC_QRBitMatrixParser.h"
#include "BC_QRDataBlock.h"
#include "BC_QRDecodedBitStreamParser.h"
#include "BC_QRCoderVersion.h"
#include "BC_QRCoderFormatInformation.h"
#include "BC_QRCoderDecoder.h"
CBC_QRCoderDecoder::CBC_QRCoderDecoder()
{
    m_rsDecoder = NULL;
}

void CBC_QRCoderDecoder::Init()
{
    m_rsDecoder = FX_NEW CBC_ReedSolomonDecoder(CBC_ReedSolomonGF256::QRCodeFild);
}
CBC_QRCoderDecoder::~CBC_QRCoderDecoder()
{
    if(m_rsDecoder != NULL) {
        delete m_rsDecoder;
    }
    m_rsDecoder = NULL;
}
CBC_CommonDecoderResult* CBC_QRCoderDecoder::Decode(FX_BOOL* image, FX_INT32 width, FX_INT32 height, FX_INT32 &e)
{
    CBC_CommonBitMatrix  bits;
    bits.Init(width);
    for(FX_INT32 i = 0; i < width; i++) {
        for(FX_INT32 j = 0; j < height; j++) {
            if(image[i * width + j]) {
                bits.Set(j, i);
            }
        }
    }
    CBC_CommonDecoderResult* cdr = Decode(&bits, height, e);
    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    return cdr;
}
CBC_CommonDecoderResult* CBC_QRCoderDecoder::Decode(CBC_CommonBitMatrix* bits, FX_INT32 byteModeDecode, FX_INT32 &e)
{
    CBC_QRBitMatrixParser parser;
    parser.Init(bits, e);
    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    CBC_QRCoderVersion *version = parser.ReadVersion(e);
    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    CBC_QRCoderFormatInformation* temp = parser.ReadFormatInformation(e);
    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    CBC_QRCoderErrorCorrectionLevel* ecLevel = temp->GetErrorCorrectionLevel();
    CFX_ByteArray* ba = parser.ReadCodewords(e);
    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    CBC_AutoPtr<CFX_ByteArray > codewords(ba);
    CFX_PtrArray *dataBlocks = CBC_QRDataBlock::GetDataBlocks(codewords.get(), version, ecLevel, e);
    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    FX_INT32 totalBytes = 0;
    for (FX_INT32 i = 0; i < dataBlocks->GetSize(); i++) {
        totalBytes += ((CBC_QRDataBlock*) ((*dataBlocks)[i]))->GetNumDataCodewords();
    }
    CFX_ByteArray resultBytes;
    FX_INT32 resultOffset = 0;
    for (FX_INT32 j = 0; j < dataBlocks->GetSize(); j++) {
        CBC_QRDataBlock *dataBlock = (CBC_QRDataBlock *)((*dataBlocks)[j]);
        CFX_ByteArray* codewordBytes = dataBlock->GetCodewords();
        FX_INT32 numDataCodewords = dataBlock->GetNumDataCodewords();
        CorrectErrors(codewordBytes, numDataCodewords, e);
        if (e != BCExceptionNO) {
            for(FX_INT32 k = 0; k < dataBlocks->GetSize(); k++) {
                delete (CBC_QRDataBlock*)(*dataBlocks)[k];
            }
            dataBlocks->RemoveAll();
            delete dataBlocks;
            dataBlocks = NULL;
            return NULL;
        }
        for(FX_INT32 i = 0; i < numDataCodewords; i++) {
            resultBytes.Add((*codewordBytes)[i]);
        }
    }
    for(FX_INT32 k = 0; k < dataBlocks->GetSize(); k++) {
        delete  (CBC_QRDataBlock*)(*dataBlocks)[k] ;
    }
    dataBlocks->RemoveAll();
    delete dataBlocks;
    dataBlocks = NULL;
    CBC_CommonDecoderResult* cdr = CBC_QRDecodedBitStreamParser::Decode(&resultBytes, version, ecLevel, byteModeDecode, e);
    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
    return cdr;
}
void CBC_QRCoderDecoder::CorrectErrors(CFX_ByteArray* codewordBytes, FX_INT32 numDataCodewords, FX_INT32 &e)
{
    FX_INT32 numCodewords = codewordBytes->GetSize();
    CFX_Int32Array codewordsInts;
    codewordsInts.SetSize(numCodewords);
    for(FX_INT32 i = 0; i < numCodewords; i++) {
        codewordsInts[i] = (FX_INT32)((*codewordBytes)[i] & 0xff);
    }
    FX_INT32 numECCodewords = codewordBytes->GetSize() - numDataCodewords;
    m_rsDecoder->Decode(&codewordsInts, numECCodewords, e);
    BC_EXCEPTION_CHECK_ReturnVoid(e);
    for(FX_INT32 k = 0; k < numDataCodewords; k++) {
        (*codewordBytes)[k] = (FX_BYTE) codewordsInts[k];
    }
}