// 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
|
|
#ifndef CORE_FXCODEC_CODEC_CCODEC_PROGRESSIVEDECODER_H_
|
#define CORE_FXCODEC_CODEC_CCODEC_PROGRESSIVEDECODER_H_
|
|
#include <memory>
|
#include <utility>
|
#include <vector>
|
|
#include "core/fxcodec/codec/ccodec_bmpmodule.h"
|
#include "core/fxcodec/codec/ccodec_gifmodule.h"
|
#include "core/fxcodec/codec/ccodec_jpegmodule.h"
|
#include "core/fxcodec/codec/ccodec_pngmodule.h"
|
#include "core/fxcodec/codec/ccodec_tiffmodule.h"
|
#include "core/fxcodec/fx_codec_def.h"
|
#include "core/fxcrt/fx_system.h"
|
#include "core/fxcrt/retain_ptr.h"
|
#include "core/fxcrt/unowned_ptr.h"
|
#include "core/fxge/fx_dib.h"
|
|
class CCodec_ModuleMgr;
|
class CFX_DIBAttribute;
|
class IFX_SeekableReadStream;
|
|
class CCodec_ProgressiveDecoder : public CCodec_BmpModule::Delegate,
|
public CCodec_GifModule::Delegate,
|
public CCodec_PngModule::Delegate {
|
public:
|
enum FXCodec_Format {
|
FXCodec_Invalid = 0,
|
FXCodec_1bppGray = 0x101,
|
FXCodec_1bppRgb = 0x001,
|
FXCodec_8bppGray = 0x108,
|
FXCodec_8bppRgb = 0x008,
|
FXCodec_Rgb = 0x018,
|
FXCodec_Rgb32 = 0x020,
|
FXCodec_Argb = 0x220,
|
FXCodec_Cmyk = 0x120
|
};
|
|
explicit CCodec_ProgressiveDecoder(CCodec_ModuleMgr* pCodecMgr);
|
virtual ~CCodec_ProgressiveDecoder();
|
|
FXCODEC_STATUS LoadImageInfo(const RetainPtr<IFX_SeekableReadStream>& pFile,
|
FXCODEC_IMAGE_TYPE imageType,
|
CFX_DIBAttribute* pAttribute,
|
bool bSkipImageTypeCheck);
|
|
FXCODEC_IMAGE_TYPE GetType() const { return m_imagType; }
|
int32_t GetWidth() const { return m_SrcWidth; }
|
int32_t GetHeight() const { return m_SrcHeight; }
|
int32_t GetNumComponents() const { return m_SrcComponents; }
|
int32_t GetBPC() const { return m_SrcBPC; }
|
void SetClipBox(FX_RECT* clip);
|
|
std::pair<FXCODEC_STATUS, size_t> GetFrames();
|
FXCODEC_STATUS StartDecode(const RetainPtr<CFX_DIBitmap>& pDIBitmap,
|
int start_x,
|
int start_y,
|
int size_x,
|
int size_y);
|
|
FXCODEC_STATUS ContinueDecode();
|
|
struct PixelWeight {
|
int m_SrcStart;
|
int m_SrcEnd;
|
int m_Weights[1];
|
};
|
|
class CFXCODEC_WeightTable {
|
public:
|
CFXCODEC_WeightTable();
|
~CFXCODEC_WeightTable();
|
|
void Calc(int dest_len,
|
int dest_min,
|
int dest_max,
|
int src_len,
|
int src_min,
|
int src_max);
|
PixelWeight* GetPixelWeight(int pixel) {
|
return reinterpret_cast<PixelWeight*>(m_pWeightTables.data() +
|
(pixel - m_DestMin) * m_ItemSize);
|
}
|
|
int m_DestMin;
|
int m_ItemSize;
|
std::vector<uint8_t> m_pWeightTables;
|
};
|
|
class CFXCODEC_HorzTable {
|
public:
|
CFXCODEC_HorzTable();
|
~CFXCODEC_HorzTable();
|
|
void Calc(int dest_len, int src_len);
|
PixelWeight* GetPixelWeight(int pixel) {
|
return reinterpret_cast<PixelWeight*>(m_pWeightTables.data() +
|
pixel * m_ItemSize);
|
}
|
|
int m_ItemSize;
|
std::vector<uint8_t> m_pWeightTables;
|
};
|
|
class CFXCODEC_VertTable {
|
public:
|
CFXCODEC_VertTable();
|
~CFXCODEC_VertTable();
|
|
void Calc(int dest_len, int src_len);
|
PixelWeight* GetPixelWeight(int pixel) {
|
return reinterpret_cast<PixelWeight*>(m_pWeightTables.data() +
|
pixel * m_ItemSize);
|
}
|
int m_ItemSize;
|
std::vector<uint8_t> m_pWeightTables;
|
};
|
|
// CCodec_PngModule::Delegate
|
bool PngReadHeader(int width,
|
int height,
|
int bpc,
|
int pass,
|
int* color_type,
|
double* gamma) override;
|
bool PngAskScanlineBuf(int line, uint8_t** pSrcBuf) override;
|
void PngFillScanlineBufCompleted(int pass, int line) override;
|
|
// CCodec_GifModule::Delegate
|
void GifRecordCurrentPosition(uint32_t& cur_pos) override;
|
bool GifInputRecordPositionBuf(uint32_t rcd_pos,
|
const FX_RECT& img_rc,
|
int32_t pal_num,
|
void* pal_ptr,
|
int32_t delay_time,
|
bool user_input,
|
int32_t trans_index,
|
int32_t disposal_method,
|
bool interlace) override;
|
void GifReadScanline(int32_t row_num, uint8_t* row_buf) override;
|
|
// CCodec_BmpModule::Delegate
|
bool BmpInputImagePositionBuf(uint32_t rcd_pos) override;
|
void BmpReadScanline(uint32_t row_num,
|
const std::vector<uint8_t>& row_buf) override;
|
|
private:
|
bool BmpReadMoreData(CCodec_BmpModule* pBmpModule,
|
FXCODEC_STATUS& err_status);
|
bool GifReadMoreData(CCodec_GifModule* pGifModule,
|
FXCODEC_STATUS& err_status);
|
bool JpegReadMoreData(CCodec_JpegModule* pJpegModule,
|
FXCODEC_STATUS& err_status);
|
void PngOneOneMapResampleHorz(const RetainPtr<CFX_DIBitmap>& pDeviceBitmap,
|
int32_t des_line,
|
uint8_t* src_scan,
|
FXCodec_Format src_format);
|
bool DetectImageType(FXCODEC_IMAGE_TYPE imageType,
|
CFX_DIBAttribute* pAttribute);
|
bool BmpDetectImageType(CFX_DIBAttribute* pAttribute, uint32_t size);
|
bool JpegDetectImageType(CFX_DIBAttribute* pAttribute, uint32_t size);
|
bool PngDetectImageType(CFX_DIBAttribute* pAttribute, uint32_t size);
|
bool GifDetectImageType(CFX_DIBAttribute* pAttribute, uint32_t size);
|
bool TifDetectImageType(CFX_DIBAttribute* pAttribute, uint32_t size);
|
|
void GetDownScale(int& down_scale);
|
void GetTransMethod(FXDIB_Format des_format, FXCodec_Format src_format);
|
|
void ReSampleScanline(const RetainPtr<CFX_DIBitmap>& pDeviceBitmap,
|
int32_t des_line,
|
uint8_t* src_scan,
|
FXCodec_Format src_format);
|
void Resample(const RetainPtr<CFX_DIBitmap>& pDeviceBitmap,
|
int32_t src_line,
|
uint8_t* src_scan,
|
FXCodec_Format src_format);
|
void ResampleVert(const RetainPtr<CFX_DIBitmap>& pDeviceBitmap,
|
double scale_y,
|
int des_row);
|
void ResampleVertBT(const RetainPtr<CFX_DIBitmap>& pDeviceBitmap,
|
double scale_y,
|
int des_row);
|
void GifDoubleLineResampleVert(const RetainPtr<CFX_DIBitmap>& pDeviceBitmap,
|
double scale_y,
|
int des_row);
|
|
FXCODEC_STATUS JpegStartDecode(const RetainPtr<CFX_DIBitmap>& pDIBitmap);
|
FXCODEC_STATUS PngStartDecode(const RetainPtr<CFX_DIBitmap>& pDIBitmap);
|
FXCODEC_STATUS GifStartDecode(const RetainPtr<CFX_DIBitmap>& pDIBitmap);
|
FXCODEC_STATUS BmpStartDecode(const RetainPtr<CFX_DIBitmap>& pDIBitmap);
|
|
FXCODEC_STATUS JpegContinueDecode();
|
FXCODEC_STATUS PngContinueDecode();
|
FXCODEC_STATUS GifContinueDecode();
|
FXCODEC_STATUS BmpContinueDecode();
|
FXCODEC_STATUS TifContinueDecode();
|
|
RetainPtr<IFX_SeekableReadStream> m_pFile;
|
RetainPtr<CFX_DIBitmap> m_pDeviceBitmap;
|
UnownedPtr<CCodec_ModuleMgr> m_pCodecMgr;
|
std::unique_ptr<CCodec_JpegModule::Context> m_pJpegContext;
|
std::unique_ptr<CCodec_PngModule::Context> m_pPngContext;
|
std::unique_ptr<CCodec_GifModule::Context> m_pGifContext;
|
std::unique_ptr<CCodec_BmpModule::Context> m_pBmpContext;
|
std::unique_ptr<CCodec_TiffModule::Context> m_pTiffContext;
|
FXCODEC_IMAGE_TYPE m_imagType;
|
uint32_t m_offSet;
|
uint8_t* m_pSrcBuf;
|
uint32_t m_SrcSize;
|
uint8_t* m_pDecodeBuf;
|
int m_ScanlineSize;
|
CFXCODEC_WeightTable m_WeightHorz;
|
CFXCODEC_VertTable m_WeightVert;
|
CFXCODEC_HorzTable m_WeightHorzOO;
|
int m_SrcWidth;
|
int m_SrcHeight;
|
int m_SrcComponents;
|
int m_SrcBPC;
|
FX_RECT m_clipBox;
|
int m_startX;
|
int m_startY;
|
int m_sizeX;
|
int m_sizeY;
|
int m_TransMethod;
|
FX_ARGB* m_pSrcPalette;
|
int m_SrcPaletteNumber;
|
int m_SrcRow;
|
FXCodec_Format m_SrcFormat;
|
int m_SrcPassNumber;
|
size_t m_FrameNumber;
|
size_t m_FrameCur;
|
int m_GifBgIndex;
|
uint8_t* m_pGifPalette;
|
int32_t m_GifPltNumber;
|
int m_GifTransIndex;
|
FX_RECT m_GifFrameRect;
|
bool m_BmpIsTopBottom;
|
FXCODEC_STATUS m_status;
|
};
|
|
#endif // CORE_FXCODEC_CODEC_CCODEC_PROGRESSIVEDECODER_H_
|