// 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 2009 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 "BC_LuminanceSource.h"
#include "BC_BufferedImageLuminanceSource.h"
class CBC_Pause : public IFX_Pause
{
public:
    virtual FX_BOOL	NeedToPauseNow()
    {
        return TRUE;
    }
};
static CFX_DIBitmap* CreateDIBSource(IFX_FileRead* fileread)
{
    CFX_DIBitmap* bitmap = NULL;
    CCodec_ModuleMgr* pCodecMgr	= NULL;
    ICodec_ProgressiveDecoder* pImageCodec = NULL;
    pCodecMgr = CCodec_ModuleMgr::Create();
    pImageCodec = pCodecMgr->CreateProgressiveDecoder();
    FXCODEC_STATUS status = FXCODEC_STATUS_DECODE_FINISH;
    status = pImageCodec->LoadImageInfo(fileread, FXCODEC_IMAGE_UNKNOWN);
    if (status != FXCODEC_STATUS_FRAME_READY) {
        return NULL;
    }
    bitmap = FX_NEW CFX_DIBitmap;
    bitmap->Create(pImageCodec->GetWidth(), pImageCodec->GetHeight(), FXDIB_Argb);
    bitmap->Clear(FXARGB_MAKE(0xFF, 0xFF, 0xFF, 0xFF));
    CBC_Pause pause;
    FX_INT32 frames;
    status = pImageCodec->GetFrames(frames, &pause);
    while (status == FXCODEC_STATUS_FRAME_TOBECONTINUE) {
        status = pImageCodec->GetFrames(frames, &pause);
    }
    if (status != FXCODEC_STATUS_DECODE_READY) {
        goto except;
    }
    status = pImageCodec->StartDecode(bitmap,
                                      0,
                                      0,
                                      bitmap->GetWidth(),
                                      bitmap->GetHeight(),
                                      0,
                                      FALSE);
    if (status == FXCODEC_STATUS_ERR_PARAMS) {
        goto except;
    }
    if (status != FXCODEC_STATUS_DECODE_TOBECONTINUE) {
        goto except;
    }
    while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) {
        status = pImageCodec->ContinueDecode(&pause);
    }
    if (status != FXCODEC_STATUS_DECODE_FINISH) {
        goto except;
    }
    if (pImageCodec) {
        delete pImageCodec;
        pImageCodec = NULL;
    }
    if (pCodecMgr) {
        pCodecMgr->Destroy();
        pCodecMgr = NULL;
    }
    return bitmap;
except:
    if (pImageCodec) {
        delete pImageCodec;
        pImageCodec = NULL;
    }
    if (pCodecMgr) {
        pCodecMgr->Destroy();
        pCodecMgr = NULL;
    }
    if (bitmap) {
        delete bitmap;
    }
    return NULL;
}
CBC_BufferedImageLuminanceSource::CBC_BufferedImageLuminanceSource(const CFX_WideString &filename): CBC_LuminanceSource(0, 0), m_filename(filename)
{
    m_height = 0;
    m_width = 0;
    m_bytesPerLine = 0;
    m_top = 0;
    m_left = 0;
}
void CBC_BufferedImageLuminanceSource::Init(FX_INT32 &e)
{
    IFX_FileRead* fileread = FX_CreateFileRead(m_filename);
    m_pBitmap = CreateDIBSource(fileread);
    if (m_pBitmap == NULL) {
        e = BCExceptionLoadFile;
        return;
    }
    m_pBitmap->ConvertFormat(FXDIB_Argb);
    m_height = m_pBitmap->GetHeight();
    m_width = m_pBitmap->GetWidth();
    m_rgbData.SetSize(m_height * m_width);
    m_bytesPerLine = m_width * 4;
    m_top = 0;
    m_left = 0;
}
CBC_BufferedImageLuminanceSource::CBC_BufferedImageLuminanceSource(CFX_DIBitmap *pBitmap): CBC_LuminanceSource(0, 0)
{
    m_pBitmap = pBitmap->Clone();
    m_pBitmap->ConvertFormat(FXDIB_Argb);
    m_height = m_pBitmap->GetHeight();
    m_width = m_pBitmap->GetWidth();
    m_rgbData.SetSize(m_height * m_width);
    m_bytesPerLine = m_width * 4;
    m_top = 0;
    m_left = 0;
}
CBC_BufferedImageLuminanceSource::~CBC_BufferedImageLuminanceSource()
{
    delete m_pBitmap;
    m_pBitmap = NULL;
}
CFX_ByteArray *CBC_BufferedImageLuminanceSource::GetRow(FX_INT32 y, CFX_ByteArray &row, FX_INT32 &e)
{
    if (y < 0 || y >= m_height) {
        e = BCExceptionRequestedRowIsOutSizeTheImage;
        return NULL;
    }
    FX_INT32 width = m_width;
    if(row.GetSize() == 0 || row.GetSize() < width) {
        row.SetSize(width);
    }
    if(m_rgbData.GetSize() == 0 || m_rgbData.GetSize() < width) {
        m_rgbData.SetSize(width);
    }
    FX_INT32* rowLine = (FX_INT32*)m_pBitmap->GetScanline(y);
    FX_INT32 x;
    for (x = 0; x < width; x++) {
        FX_INT32 pixel = rowLine[x];
        FX_INT32 luminance = (306 * ((pixel >> 16) & 0xFF) +
                              601 * ((pixel >> 8) & 0xFF) +
                              117 * (pixel & 0xFF)) >> 10;
        row[x] = (FX_BYTE) luminance;
    }
    return &row;
}
CFX_ByteArray *CBC_BufferedImageLuminanceSource::GetMatrix()
{
    CFX_ByteArray *matirx = FX_NEW CFX_ByteArray();
    matirx->SetSize(m_bytesPerLine * m_height);
    FX_INT32 *rgb = (FX_INT32*)m_pBitmap->GetBuffer();
    FX_INT32 y;
    for(y = 0; y < m_height; y++) {
        FX_INT32 offset = y * m_width;
        FX_INT32 x;
        for(x = 0; x < m_width; x++) {
            FX_INT32 pixel = rgb[offset + x];
            FX_INT32 luminance = (306 * ((pixel >> 16) & 0xFF) +
                                  601 * ((pixel >> 8) & 0xFF) +
                                  117 * (pixel & 0xFF)) >> 10;
            (*matirx)[offset + x] = (FX_BYTE) luminance;
        }
    }
    return matirx;
}
FX_BOOL CBC_BufferedImageLuminanceSource::IsCropSupported()
{
    return TRUE;
}
FX_BOOL CBC_BufferedImageLuminanceSource::IsRotateSupported()
{
    return TRUE;
}
CBC_LuminanceSource *CBC_BufferedImageLuminanceSource::Crop(FX_INT32 left, FX_INT32 top, FX_INT32 width, FX_INT32 height)
{
    return NULL;
}
CBC_LuminanceSource *CBC_BufferedImageLuminanceSource::RotateCounterClockwise(FX_INT32 &e)
{
    if (!IsRotateSupported()) {
        e = BCExceptionRotateNotSupported;
        return NULL;
    }
    FX_INT32 sourceWidth = m_width;
    FX_INT32 sourceHeight = m_height;
    return NULL;
}