// 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 #include "core/fxcodec/codec/ccodec_bmpmodule.h" #include "core/fxcodec/codec/ccodec_gifmodule.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/cfx_retain_ptr.h" #include "core/fxcrt/fx_system.h" #include "core/fxge/fx_dib.h" class CCodec_JpegModule; class CCodec_ModuleMgr; class CFX_DIBAttribute; class IFX_SeekableReadStream; struct FXJPEG_Context; 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 CFX_RetainPtr& 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); FXCODEC_STATUS GetFrames(int32_t& frames); FXCODEC_STATUS StartDecode(const CFX_RetainPtr& pDIBitmap, int start_x, int start_y, int size_x, int size_y, int32_t frames = 0, bool bInterpol = true); 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, bool bInterpol); PixelWeight* GetPixelWeight(int pixel) { return reinterpret_cast(m_pWeightTables.data() + (pixel - m_DestMin) * m_ItemSize); } int m_DestMin; int m_ItemSize; std::vector m_pWeightTables; }; class CFXCODEC_HorzTable { public: CFXCODEC_HorzTable(); ~CFXCODEC_HorzTable(); void Calc(int dest_len, int src_len, bool bInterpol); PixelWeight* GetPixelWeight(int pixel) { return reinterpret_cast(m_pWeightTables.data() + pixel * m_ItemSize); } int m_ItemSize; std::vector 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(m_pWeightTables.data() + pixel * m_ItemSize); } int m_ItemSize; std::vector m_pWeightTables; }; CFX_RetainPtr m_pFile; CCodec_ModuleMgr* m_pCodecMgr; FXJPEG_Context* m_pJpegContext; FXPNG_Context* m_pPngContext; FXGIF_Context* m_pGifContext; FXBMP_Context* m_pBmpContext; CCodec_TiffContext* 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; CFX_RetainPtr m_pDeviceBitmap; bool m_bInterpol; 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; int m_FrameNumber; int 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; // 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*& src_buf) 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(int32_t row_num, uint8_t* row_buf) override; protected: bool BmpReadMoreData(CCodec_BmpModule* pBmpModule, FXCODEC_STATUS& err_status); bool GifReadMoreData(CCodec_GifModule* pGifModule, FXCODEC_STATUS& err_status); void GifDoubleLineResampleVert( const CFX_RetainPtr& pDeviceBitmap, double scale_y, int des_row); void PngOneOneMapResampleHorz( const CFX_RetainPtr& pDeviceBitmap, int32_t des_line, uint8_t* src_scan, FXCodec_Format src_format); bool DetectImageType(FXCODEC_IMAGE_TYPE imageType, CFX_DIBAttribute* pAttribute); void GetDownScale(int& down_scale); void GetTransMethod(FXDIB_Format des_format, FXCodec_Format src_format); void ReSampleScanline(const CFX_RetainPtr& pDeviceBitmap, int32_t des_line, uint8_t* src_scan, FXCodec_Format src_format); void Resample(const CFX_RetainPtr& pDeviceBitmap, int32_t src_line, uint8_t* src_scan, FXCodec_Format src_format); void ResampleVert(const CFX_RetainPtr& pDeviceBitmap, double scale_y, int des_row); bool JpegReadMoreData(CCodec_JpegModule* pJpegModule, FXCODEC_STATUS& err_status); void ResampleVertBT(const CFX_RetainPtr& pDeviceBitmap, double scale_y, int des_row); }; #endif // CORE_FXCODEC_CODEC_CCODEC_PROGRESSIVEDECODER_H_