| // 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 |
| |
| #include "render_int.h" |
| |
| #include <vector> |
| |
| #include "../fpdf_page/pageint.h" |
| #include "core/include/fpdfapi/fpdf_module.h" |
| #include "core/include/fpdfapi/fpdf_pageobj.h" |
| #include "core/include/fpdfapi/fpdf_render.h" |
| #include "core/include/fxcodec/fx_codec.h" |
| #include "core/include/fxcrt/fx_safe_types.h" |
| #include "core/include/fxge/fx_ge.h" |
| #include "third_party/base/nonstd_unique_ptr.h" |
| |
| namespace { |
| |
| unsigned int GetBits8(const uint8_t* pData, uint64_t bitpos, size_t nbits) { |
| ASSERT(nbits == 1 || nbits == 2 || nbits == 4 || nbits == 8 || nbits == 16); |
| ASSERT((bitpos & (nbits - 1)) == 0); |
| unsigned int byte = pData[bitpos / 8]; |
| if (nbits == 8) { |
| return byte; |
| } |
| if (nbits == 4) { |
| return (bitpos % 8) ? (byte & 0x0f) : (byte >> 4); |
| } |
| if (nbits == 2) { |
| return (byte >> (6 - bitpos % 8)) & 0x03; |
| } |
| if (nbits == 1) { |
| return (byte >> (7 - bitpos % 8)) & 0x01; |
| } |
| if (nbits == 16) { |
| return byte * 256 + pData[bitpos / 8 + 1]; |
| } |
| return 0; |
| } |
| |
| FX_SAFE_DWORD CalculatePitch8(FX_DWORD bpc, FX_DWORD components, int width) { |
| FX_SAFE_DWORD pitch = bpc; |
| pitch *= components; |
| pitch *= width; |
| pitch += 7; |
| pitch /= 8; |
| return pitch; |
| } |
| |
| FX_SAFE_DWORD CalculatePitch32(int bpp, int width) { |
| FX_SAFE_DWORD pitch = bpp; |
| pitch *= width; |
| pitch += 31; |
| pitch /= 32; // quantized to number of 32-bit words. |
| pitch *= 4; // and then back to bytes, (not just /8 in one step). |
| return pitch; |
| } |
| |
| // Wrapper class to use with nonstd::unique_ptr for CJPX_Decoder. |
| class JpxBitMapContext { |
| public: |
| explicit JpxBitMapContext(ICodec_JpxModule* jpx_module) |
| : jpx_module_(jpx_module), decoder_(nullptr) {} |
| |
| ~JpxBitMapContext() { |
| jpx_module_->DestroyDecoder(decoder_); |
| } |
| |
| // Takes ownership of |decoder|. |
| void set_decoder(CJPX_Decoder* decoder) { decoder_ = decoder; } |
| |
| CJPX_Decoder* decoder() { return decoder_; } |
| |
| private: |
| ICodec_JpxModule* const jpx_module_; // Weak pointer. |
| CJPX_Decoder* decoder_; // Decoder, owned. |
| |
| // Disallow evil constructors |
| JpxBitMapContext(const JpxBitMapContext&); |
| void operator=(const JpxBitMapContext&); |
| }; |
| |
| const int kMaxImageDimension = 0x01FFFF; |
| |
| } // namespace |
| |
| CFX_DIBSource* CPDF_Image::LoadDIBSource(CFX_DIBSource** ppMask, |
| FX_DWORD* pMatteColor, |
| FX_BOOL bStdCS, |
| FX_DWORD GroupFamily, |
| FX_BOOL bLoadMask) const { |
| CPDF_DIBSource* pSource = new CPDF_DIBSource; |
| if (pSource->Load(m_pDocument, m_pStream, (CPDF_DIBSource**)ppMask, |
| pMatteColor, NULL, NULL, bStdCS, GroupFamily, bLoadMask)) { |
| return pSource; |
| } |
| delete pSource; |
| return NULL; |
| } |
| CFX_DIBSource* CPDF_Image::DetachBitmap() { |
| CFX_DIBSource* pBitmap = m_pDIBSource; |
| m_pDIBSource = NULL; |
| return pBitmap; |
| } |
| CFX_DIBSource* CPDF_Image::DetachMask() { |
| CFX_DIBSource* pBitmap = m_pMask; |
| m_pMask = NULL; |
| return pBitmap; |
| } |
| FX_BOOL CPDF_Image::StartLoadDIBSource(CPDF_Dictionary* pFormResource, |
| CPDF_Dictionary* pPageResource, |
| FX_BOOL bStdCS, |
| FX_DWORD GroupFamily, |
| FX_BOOL bLoadMask) { |
| m_pDIBSource = new CPDF_DIBSource; |
| int ret = |
| ((CPDF_DIBSource*)m_pDIBSource) |
| ->StartLoadDIBSource(m_pDocument, m_pStream, TRUE, pFormResource, |
| pPageResource, bStdCS, GroupFamily, bLoadMask); |
| if (ret == 2) { |
| return TRUE; |
| } |
| if (!ret) { |
| delete m_pDIBSource; |
| m_pDIBSource = NULL; |
| return FALSE; |
| } |
| m_pMask = ((CPDF_DIBSource*)m_pDIBSource)->DetachMask(); |
| m_MatteColor = ((CPDF_DIBSource*)m_pDIBSource)->m_MatteColor; |
| return FALSE; |
| } |
| FX_BOOL CPDF_Image::Continue(IFX_Pause* pPause) { |
| int ret = ((CPDF_DIBSource*)m_pDIBSource)->ContinueLoadDIBSource(pPause); |
| if (ret == 2) { |
| return TRUE; |
| } |
| if (!ret) { |
| delete m_pDIBSource; |
| m_pDIBSource = NULL; |
| return FALSE; |
| } |
| m_pMask = ((CPDF_DIBSource*)m_pDIBSource)->DetachMask(); |
| m_MatteColor = ((CPDF_DIBSource*)m_pDIBSource)->m_MatteColor; |
| return FALSE; |
| } |
| CPDF_DIBSource::CPDF_DIBSource() { |
| m_pDocument = NULL; |
| m_pStreamAcc = NULL; |
| m_pDict = NULL; |
| m_bpp = 0; |
| m_Width = m_Height = 0; |
| m_pColorSpace = NULL; |
| m_bDefaultDecode = TRUE; |
| m_bImageMask = FALSE; |
| m_bDoBpcCheck = TRUE; |
| m_pPalette = NULL; |
| m_pCompData = NULL; |
| m_bColorKey = FALSE; |
| m_pMaskedLine = m_pLineBuf = NULL; |
| m_pDecoder = NULL; |
| m_nComponents = 0; |
| m_bpc = 0; |
| m_bLoadMask = FALSE; |
| m_Family = 0; |
| m_pMask = NULL; |
| m_MatteColor = 0; |
| m_pJbig2Context = NULL; |
| m_pGlobalStream = NULL; |
| m_bStdCS = FALSE; |
| m_pMaskStream = NULL; |
| m_Status = 0; |
| m_bHasMask = FALSE; |
| } |
| CPDF_DIBSource::~CPDF_DIBSource() { |
| delete m_pStreamAcc; |
| FX_Free(m_pMaskedLine); |
| FX_Free(m_pLineBuf); |
| m_pCachedBitmap.reset(); |
| delete m_pDecoder; |
| FX_Free(m_pCompData); |
| CPDF_ColorSpace* pCS = m_pColorSpace; |
| if (pCS && m_pDocument) { |
| m_pDocument->GetPageData()->ReleaseColorSpace(pCS->GetArray()); |
| } |
| if (m_pJbig2Context) { |
| ICodec_Jbig2Module* pJbig2Module = CPDF_ModuleMgr::Get()->GetJbig2Module(); |
| pJbig2Module->DestroyJbig2Context(m_pJbig2Context); |
| } |
| delete m_pGlobalStream; |
| } |
| CFX_DIBitmap* CPDF_DIBSource::GetBitmap() const { |
| return m_pCachedBitmap ? m_pCachedBitmap.get() : Clone(); |
| } |
| void CPDF_DIBSource::ReleaseBitmap(CFX_DIBitmap* pBitmap) const { |
| if (pBitmap && pBitmap != m_pCachedBitmap) { |
| delete pBitmap; |
| } |
| } |
| FX_BOOL CPDF_DIBSource::Load(CPDF_Document* pDoc, |
| const CPDF_Stream* pStream, |
| CPDF_DIBSource** ppMask, |
| FX_DWORD* pMatteColor, |
| CPDF_Dictionary* pFormResources, |
| CPDF_Dictionary* pPageResources, |
| FX_BOOL bStdCS, |
| FX_DWORD GroupFamily, |
| FX_BOOL bLoadMask) { |
| if (pStream == NULL) { |
| return FALSE; |
| } |
| m_pDocument = pDoc; |
| m_pDict = pStream->GetDict(); |
| if (m_pDict == NULL) { |
| return FALSE; |
| } |
| m_pStream = pStream; |
| m_Width = m_pDict->GetInteger(FX_BSTRC("Width")); |
| m_Height = m_pDict->GetInteger(FX_BSTRC("Height")); |
| if (m_Width <= 0 || m_Height <= 0 || m_Width > kMaxImageDimension || |
| m_Height > kMaxImageDimension) { |
| return FALSE; |
| } |
| m_GroupFamily = GroupFamily; |
| m_bLoadMask = bLoadMask; |
| if (!LoadColorInfo(m_pStream->GetObjNum() != 0 ? NULL : pFormResources, |
| pPageResources)) { |
| return FALSE; |
| } |
| if (m_bDoBpcCheck && (m_bpc == 0 || m_nComponents == 0)) { |
| return FALSE; |
| } |
| FX_SAFE_DWORD src_size = |
| CalculatePitch8(m_bpc, m_nComponents, m_Width) * m_Height; |
| if (!src_size.IsValid()) { |
| return FALSE; |
| } |
| m_pStreamAcc = new CPDF_StreamAcc; |
| m_pStreamAcc->LoadAllData(pStream, FALSE, src_size.ValueOrDie(), TRUE); |
| if (m_pStreamAcc->GetSize() == 0 || m_pStreamAcc->GetData() == NULL) { |
| return FALSE; |
| } |
| if (!CreateDecoder()) { |
| return FALSE; |
| } |
| if (m_bImageMask) { |
| m_bpp = 1; |
| m_bpc = 1; |
| m_nComponents = 1; |
| m_AlphaFlag = 1; |
| } else if (m_bpc * m_nComponents == 1) { |
| m_bpp = 1; |
| } else if (m_bpc * m_nComponents <= 8) { |
| m_bpp = 8; |
| } else { |
| m_bpp = 24; |
| } |
| FX_SAFE_DWORD pitch = CalculatePitch32(m_bpp, m_Width); |
| if (!pitch.IsValid()) { |
| return FALSE; |
| } |
| m_pLineBuf = FX_Alloc(uint8_t, pitch.ValueOrDie()); |
| if (m_pColorSpace && bStdCS) { |
| m_pColorSpace->EnableStdConversion(TRUE); |
| } |
| LoadPalette(); |
| if (m_bColorKey) { |
| m_bpp = 32; |
| m_AlphaFlag = 2; |
| pitch = CalculatePitch32(m_bpp, m_Width); |
| if (!pitch.IsValid()) { |
| return FALSE; |
| } |
| m_pMaskedLine = FX_Alloc(uint8_t, pitch.ValueOrDie()); |
| } |
| m_Pitch = pitch.ValueOrDie(); |
| if (ppMask) { |
| *ppMask = LoadMask(*pMatteColor); |
| } |
| if (m_pColorSpace && bStdCS) { |
| m_pColorSpace->EnableStdConversion(FALSE); |
| } |
| return TRUE; |
| } |
| int CPDF_DIBSource::ContinueToLoadMask() { |
| if (m_bImageMask) { |
| m_bpp = 1; |
| m_bpc = 1; |
| m_nComponents = 1; |
| m_AlphaFlag = 1; |
| } else if (m_bpc * m_nComponents == 1) { |
| m_bpp = 1; |
| } else if (m_bpc * m_nComponents <= 8) { |
| m_bpp = 8; |
| } else { |
| m_bpp = 24; |
| } |
| if (!m_bpc || !m_nComponents) { |
| return 0; |
| } |
| FX_SAFE_DWORD pitch = CalculatePitch32(m_bpp, m_Width); |
| if (!pitch.IsValid()) { |
| return 0; |
| } |
| m_pLineBuf = FX_Alloc(uint8_t, pitch.ValueOrDie()); |
| if (m_pColorSpace && m_bStdCS) { |
| m_pColorSpace->EnableStdConversion(TRUE); |
| } |
| LoadPalette(); |
| if (m_bColorKey) { |
| m_bpp = 32; |
| m_AlphaFlag = 2; |
| pitch = CalculatePitch32(m_bpp, m_Width); |
| if (!pitch.IsValid()) { |
| return 0; |
| } |
| m_pMaskedLine = FX_Alloc(uint8_t, pitch.ValueOrDie()); |
| } |
| m_Pitch = pitch.ValueOrDie(); |
| return 1; |
| } |
| int CPDF_DIBSource::StartLoadDIBSource(CPDF_Document* pDoc, |
| const CPDF_Stream* pStream, |
| FX_BOOL bHasMask, |
| CPDF_Dictionary* pFormResources, |
| CPDF_Dictionary* pPageResources, |
| FX_BOOL bStdCS, |
| FX_DWORD GroupFamily, |
| FX_BOOL bLoadMask) { |
| if (pStream == NULL) { |
| return 0; |
| } |
| m_pDocument = pDoc; |
| m_pDict = pStream->GetDict(); |
| m_pStream = pStream; |
| m_bStdCS = bStdCS; |
| m_bHasMask = bHasMask; |
| m_Width = m_pDict->GetInteger(FX_BSTRC("Width")); |
| m_Height = m_pDict->GetInteger(FX_BSTRC("Height")); |
| if (m_Width <= 0 || m_Height <= 0 || m_Width > kMaxImageDimension || |
| m_Height > kMaxImageDimension) { |
| return 0; |
| } |
| m_GroupFamily = GroupFamily; |
| m_bLoadMask = bLoadMask; |
| if (!LoadColorInfo(m_pStream->GetObjNum() != 0 ? NULL : pFormResources, |
| pPageResources)) { |
| return 0; |
| } |
| if (m_bDoBpcCheck && (m_bpc == 0 || m_nComponents == 0)) { |
| return 0; |
| } |
| FX_SAFE_DWORD src_size = |
| CalculatePitch8(m_bpc, m_nComponents, m_Width) * m_Height; |
| if (!src_size.IsValid()) { |
| return 0; |
| } |
| m_pStreamAcc = new CPDF_StreamAcc; |
| m_pStreamAcc->LoadAllData(pStream, FALSE, src_size.ValueOrDie(), TRUE); |
| if (m_pStreamAcc->GetSize() == 0 || m_pStreamAcc->GetData() == NULL) { |
| return 0; |
| } |
| int ret = CreateDecoder(); |
| if (ret != 1) { |
| if (!ret) { |
| return ret; |
| } |
| if (!ContinueToLoadMask()) { |
| return 0; |
| } |
| if (m_bHasMask) { |
| StratLoadMask(); |
| } |
| return ret; |
| } |
| if (!ContinueToLoadMask()) { |
| return 0; |
| } |
| if (m_bHasMask) { |
| ret = StratLoadMask(); |
| } |
| if (ret == 2) { |
| return ret; |
| } |
| if (m_pColorSpace && m_bStdCS) { |
| m_pColorSpace->EnableStdConversion(FALSE); |
| } |
| return ret; |
| } |
| int CPDF_DIBSource::ContinueLoadDIBSource(IFX_Pause* pPause) { |
| FXCODEC_STATUS ret; |
| if (m_Status == 1) { |
| const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder(); |
| if (decoder == FX_BSTRC("JPXDecode")) { |
| return 0; |
| } |
| ICodec_Jbig2Module* pJbig2Module = CPDF_ModuleMgr::Get()->GetJbig2Module(); |
| if (m_pJbig2Context == NULL) { |
| m_pJbig2Context = pJbig2Module->CreateJbig2Context(); |
| if (m_pStreamAcc->GetImageParam()) { |
| CPDF_Stream* pGlobals = |
| m_pStreamAcc->GetImageParam()->GetStream(FX_BSTRC("JBIG2Globals")); |
| if (pGlobals) { |
| m_pGlobalStream = new CPDF_StreamAcc; |
| m_pGlobalStream->LoadAllData(pGlobals, FALSE); |
| } |
| } |
| ret = pJbig2Module->StartDecode(m_pJbig2Context, m_pDocument, m_Width, |
| m_Height, m_pStreamAcc, m_pGlobalStream, |
| m_pCachedBitmap->GetBuffer(), |
| m_pCachedBitmap->GetPitch(), pPause); |
| if (ret < 0) { |
| m_pCachedBitmap.reset(); |
| delete m_pGlobalStream; |
| m_pGlobalStream = NULL; |
| pJbig2Module->DestroyJbig2Context(m_pJbig2Context); |
| m_pJbig2Context = NULL; |
| return 0; |
| } |
| if (ret == FXCODEC_STATUS_DECODE_TOBECONTINUE) { |
| return 2; |
| } |
| int ret1 = 1; |
| if (m_bHasMask) { |
| ret1 = ContinueLoadMaskDIB(pPause); |
| m_Status = 2; |
| } |
| if (ret1 == 2) { |
| return ret1; |
| } |
| if (m_pColorSpace && m_bStdCS) { |
| m_pColorSpace->EnableStdConversion(FALSE); |
| } |
| return ret1; |
| } |
| FXCODEC_STATUS ret = pJbig2Module->ContinueDecode(m_pJbig2Context, pPause); |
| if (ret < 0) { |
| m_pCachedBitmap.reset(); |
| delete m_pGlobalStream; |
| m_pGlobalStream = NULL; |
| pJbig2Module->DestroyJbig2Context(m_pJbig2Context); |
| m_pJbig2Context = NULL; |
| return 0; |
| } |
| if (ret == FXCODEC_STATUS_DECODE_TOBECONTINUE) { |
| return 2; |
| } |
| int ret1 = 1; |
| if (m_bHasMask) { |
| ret1 = ContinueLoadMaskDIB(pPause); |
| m_Status = 2; |
| } |
| if (ret1 == 2) { |
| return ret1; |
| } |
| if (m_pColorSpace && m_bStdCS) { |
| m_pColorSpace->EnableStdConversion(FALSE); |
| } |
| return ret1; |
| } |
| if (m_Status == 2) { |
| return ContinueLoadMaskDIB(pPause); |
| } |
| return 0; |
| } |
| FX_BOOL CPDF_DIBSource::LoadColorInfo(CPDF_Dictionary* pFormResources, |
| CPDF_Dictionary* pPageResources) { |
| m_bpc_orig = m_pDict->GetInteger(FX_BSTRC("BitsPerComponent")); |
| if (m_pDict->GetInteger("ImageMask")) { |
| m_bImageMask = TRUE; |
| } |
| if (m_bImageMask || !m_pDict->KeyExist(FX_BSTRC("ColorSpace"))) { |
| if (!m_bImageMask) { |
| CPDF_Object* pFilter = m_pDict->GetElementValue(FX_BSTRC("Filter")); |
| if (pFilter) { |
| CFX_ByteString filter; |
| if (pFilter->IsName()) { |
| filter = pFilter->GetString(); |
| if (filter == FX_BSTRC("JPXDecode")) { |
| m_bDoBpcCheck = FALSE; |
| return TRUE; |
| } |
| } else if (CPDF_Array* pArray = pFilter->AsArray()) { |
| if (pArray->GetString(pArray->GetCount() - 1) == |
| FX_BSTRC("JPXDecode")) { |
| m_bDoBpcCheck = FALSE; |
| return TRUE; |
| } |
| } |
| } |
| } |
| m_bImageMask = TRUE; |
| m_bpc = m_nComponents = 1; |
| CPDF_Array* pDecode = m_pDict->GetArray(FX_BSTRC("Decode")); |
| m_bDefaultDecode = pDecode == NULL || pDecode->GetInteger(0) == 0; |
| return TRUE; |
| } |
| CPDF_Object* pCSObj = m_pDict->GetElementValue(FX_BSTRC("ColorSpace")); |
| if (pCSObj == NULL) { |
| return FALSE; |
| } |
| CPDF_DocPageData* pDocPageData = m_pDocument->GetPageData(); |
| if (pFormResources) { |
| m_pColorSpace = pDocPageData->GetColorSpace(pCSObj, pFormResources); |
| } |
| if (m_pColorSpace == NULL) { |
| m_pColorSpace = pDocPageData->GetColorSpace(pCSObj, pPageResources); |
| } |
| if (m_pColorSpace == NULL) { |
| return FALSE; |
| } |
| m_Family = m_pColorSpace->GetFamily(); |
| m_nComponents = m_pColorSpace->CountComponents(); |
| if (m_Family == PDFCS_ICCBASED && pCSObj->IsName()) { |
| CFX_ByteString cs = pCSObj->GetString(); |
| if (cs == FX_BSTRC("DeviceGray")) { |
| m_nComponents = 1; |
| } else if (cs == FX_BSTRC("DeviceRGB")) { |
| m_nComponents = 3; |
| } else if (cs == FX_BSTRC("DeviceCMYK")) { |
| m_nComponents = 4; |
| } |
| } |
| ValidateDictParam(); |
| m_pCompData = GetDecodeAndMaskArray(m_bDefaultDecode, m_bColorKey); |
| if (m_pCompData == NULL) { |
| return FALSE; |
| } |
| return TRUE; |
| } |
| DIB_COMP_DATA* CPDF_DIBSource::GetDecodeAndMaskArray(FX_BOOL& bDefaultDecode, |
| FX_BOOL& bColorKey) { |
| if (m_pColorSpace == NULL) { |
| return NULL; |
| } |
| DIB_COMP_DATA* pCompData = FX_Alloc(DIB_COMP_DATA, m_nComponents); |
| int max_data = (1 << m_bpc) - 1; |
| CPDF_Array* pDecode = m_pDict->GetArray(FX_BSTRC("Decode")); |
| if (pDecode) { |
| for (FX_DWORD i = 0; i < m_nComponents; i++) { |
| pCompData[i].m_DecodeMin = pDecode->GetNumber(i * 2); |
| FX_FLOAT max = pDecode->GetNumber(i * 2 + 1); |
| pCompData[i].m_DecodeStep = (max - pCompData[i].m_DecodeMin) / max_data; |
| FX_FLOAT def_value, def_min, def_max; |
| m_pColorSpace->GetDefaultValue(i, def_value, def_min, def_max); |
| if (m_Family == PDFCS_INDEXED) { |
| def_max = (FX_FLOAT)max_data; |
| } |
| if (def_min != pCompData[i].m_DecodeMin || def_max != max) { |
| bDefaultDecode = FALSE; |
| } |
| } |
| } else { |
| for (FX_DWORD i = 0; i < m_nComponents; i++) { |
| FX_FLOAT def_value; |
| m_pColorSpace->GetDefaultValue(i, def_value, pCompData[i].m_DecodeMin, |
| pCompData[i].m_DecodeStep); |
| if (m_Family == PDFCS_INDEXED) { |
| pCompData[i].m_DecodeStep = (FX_FLOAT)max_data; |
| } |
| pCompData[i].m_DecodeStep = |
| (pCompData[i].m_DecodeStep - pCompData[i].m_DecodeMin) / max_data; |
| } |
| } |
| if (!m_pDict->KeyExist(FX_BSTRC("SMask"))) { |
| CPDF_Object* pMask = m_pDict->GetElementValue(FX_BSTRC("Mask")); |
| if (pMask == NULL) { |
| return pCompData; |
| } |
| if (CPDF_Array* pArray = pMask->AsArray()) { |
| if (pArray->GetCount() >= m_nComponents * 2) { |
| for (FX_DWORD i = 0; i < m_nComponents; i++) { |
| int min_num = pArray->GetInteger(i * 2); |
| int max_num = pArray->GetInteger(i * 2 + 1); |
| pCompData[i].m_ColorKeyMin = FX_MAX(min_num, 0); |
| pCompData[i].m_ColorKeyMax = FX_MIN(max_num, max_data); |
| } |
| } |
| bColorKey = TRUE; |
| } |
| } |
| return pCompData; |
| } |
| ICodec_ScanlineDecoder* FPDFAPI_CreateFaxDecoder( |
| const uint8_t* src_buf, |
| FX_DWORD src_size, |
| int width, |
| int height, |
| const CPDF_Dictionary* pParams); |
| ICodec_ScanlineDecoder* FPDFAPI_CreateFlateDecoder( |
| const uint8_t* src_buf, |
| FX_DWORD src_size, |
| int width, |
| int height, |
| int nComps, |
| int bpc, |
| const CPDF_Dictionary* pParams); |
| int CPDF_DIBSource::CreateDecoder() { |
| const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder(); |
| if (decoder.IsEmpty()) { |
| return 1; |
| } |
| if (m_bDoBpcCheck && m_bpc == 0) { |
| return 0; |
| } |
| const uint8_t* src_data = m_pStreamAcc->GetData(); |
| FX_DWORD src_size = m_pStreamAcc->GetSize(); |
| const CPDF_Dictionary* pParams = m_pStreamAcc->GetImageParam(); |
| if (decoder == FX_BSTRC("CCITTFaxDecode")) { |
| m_pDecoder = FPDFAPI_CreateFaxDecoder(src_data, src_size, m_Width, m_Height, |
| pParams); |
| } else if (decoder == FX_BSTRC("DCTDecode")) { |
| m_pDecoder = CPDF_ModuleMgr::Get()->GetJpegModule()->CreateDecoder( |
| src_data, src_size, m_Width, m_Height, m_nComponents, |
| pParams ? pParams->GetInteger("ColorTransform", 1) : 1); |
| if (!m_pDecoder) { |
| FX_BOOL bTransform = FALSE; |
| int comps, bpc; |
| ICodec_JpegModule* pJpegModule = CPDF_ModuleMgr::Get()->GetJpegModule(); |
| if (pJpegModule->LoadInfo(src_data, src_size, m_Width, m_Height, comps, |
| bpc, bTransform)) { |
| if (m_nComponents != comps) { |
| FX_Free(m_pCompData); |
| m_nComponents = comps; |
| if (m_Family == PDFCS_LAB && m_nComponents != 3) { |
| m_pCompData = NULL; |
| return 0; |
| } |
| m_pCompData = GetDecodeAndMaskArray(m_bDefaultDecode, m_bColorKey); |
| if (m_pCompData == NULL) { |
| return 0; |
| } |
| } |
| m_bpc = bpc; |
| m_pDecoder = CPDF_ModuleMgr::Get()->GetJpegModule()->CreateDecoder( |
| src_data, src_size, m_Width, m_Height, m_nComponents, bTransform); |
| } |
| } |
| } else if (decoder == FX_BSTRC("FlateDecode")) { |
| m_pDecoder = FPDFAPI_CreateFlateDecoder( |
| src_data, src_size, m_Width, m_Height, m_nComponents, m_bpc, pParams); |
| } else if (decoder == FX_BSTRC("JPXDecode")) { |
| LoadJpxBitmap(); |
| return m_pCachedBitmap ? 1 : 0; |
| } else if (decoder == FX_BSTRC("JBIG2Decode")) { |
| m_pCachedBitmap.reset(new CFX_DIBitmap); |
| if (!m_pCachedBitmap->Create( |
| m_Width, m_Height, m_bImageMask ? FXDIB_1bppMask : FXDIB_1bppRgb)) { |
| m_pCachedBitmap.reset(); |
| return 0; |
| } |
| m_Status = 1; |
| return 2; |
| } else if (decoder == FX_BSTRC("RunLengthDecode")) { |
| m_pDecoder = CPDF_ModuleMgr::Get() |
| ->GetCodecModule() |
| ->GetBasicModule() |
| ->CreateRunLengthDecoder(src_data, src_size, m_Width, |
| m_Height, m_nComponents, m_bpc); |
| } |
| if (!m_pDecoder) |
| return 0; |
| |
| FX_SAFE_DWORD requested_pitch = |
| CalculatePitch8(m_bpc, m_nComponents, m_Width); |
| if (!requested_pitch.IsValid()) { |
| return 0; |
| } |
| FX_SAFE_DWORD provided_pitch = CalculatePitch8( |
| m_pDecoder->GetBPC(), m_pDecoder->CountComps(), m_pDecoder->GetWidth()); |
| if (!provided_pitch.IsValid()) { |
| return 0; |
| } |
| if (provided_pitch.ValueOrDie() < requested_pitch.ValueOrDie()) { |
| return 0; |
| } |
| return 1; |
| } |
| void CPDF_DIBSource::LoadJpxBitmap() { |
| ICodec_JpxModule* pJpxModule = CPDF_ModuleMgr::Get()->GetJpxModule(); |
| if (!pJpxModule) |
| return; |
| |
| nonstd::unique_ptr<JpxBitMapContext> context( |
| new JpxBitMapContext(pJpxModule)); |
| context->set_decoder(pJpxModule->CreateDecoder(m_pStreamAcc->GetData(), |
| m_pStreamAcc->GetSize(), |
| m_pColorSpace == nullptr)); |
| if (!context->decoder()) |
| return; |
| |
| FX_DWORD width = 0; |
| FX_DWORD height = 0; |
| FX_DWORD components = 0; |
| pJpxModule->GetImageInfo(context->decoder(), &width, &height, &components); |
| if ((int)width < m_Width || (int)height < m_Height) |
| return; |
| |
| FX_BOOL bSwapRGB = FALSE; |
| if (m_pColorSpace) { |
| if (components != (FX_DWORD)m_pColorSpace->CountComponents()) |
| return; |
| |
| if (m_pColorSpace == CPDF_ColorSpace::GetStockCS(PDFCS_DEVICERGB)) { |
| bSwapRGB = TRUE; |
| m_pColorSpace = nullptr; |
| } |
| } else { |
| if (components == 3) { |
| bSwapRGB = TRUE; |
| } else if (components == 4) { |
| m_pColorSpace = CPDF_ColorSpace::GetStockCS(PDFCS_DEVICECMYK); |
| } |
| m_nComponents = components; |
| } |
| |
| FXDIB_Format format; |
| if (components == 1) { |
| format = FXDIB_8bppRgb; |
| } else if (components <= 3) { |
| format = FXDIB_Rgb; |
| } else if (components == 4) { |
| format = FXDIB_Rgb32; |
| } else { |
| width = (width * components + 2) / 3; |
| format = FXDIB_Rgb; |
| } |
| |
| m_pCachedBitmap.reset(new CFX_DIBitmap); |
| if (!m_pCachedBitmap->Create(width, height, format)) { |
| m_pCachedBitmap.reset(); |
| return; |
| } |
| m_pCachedBitmap->Clear(0xFFFFFFFF); |
| std::vector<uint8_t> output_offsets(components); |
| for (FX_DWORD i = 0; i < components; ++i) |
| output_offsets[i] = i; |
| if (bSwapRGB) { |
| output_offsets[0] = 2; |
| output_offsets[2] = 0; |
| } |
| if (!pJpxModule->Decode(context->decoder(), m_pCachedBitmap->GetBuffer(), |
| m_pCachedBitmap->GetPitch(), output_offsets)) { |
| m_pCachedBitmap.reset(); |
| return; |
| } |
| if (m_pColorSpace && m_pColorSpace->GetFamily() == PDFCS_INDEXED && |
| m_bpc < 8) { |
| int scale = 8 - m_bpc; |
| for (FX_DWORD row = 0; row < height; ++row) { |
| uint8_t* scanline = (uint8_t*)m_pCachedBitmap->GetScanline(row); |
| for (FX_DWORD col = 0; col < width; ++col) { |
| *scanline = (*scanline) >> scale; |
| ++scanline; |
| } |
| } |
| } |
| m_bpc = 8; |
| } |
| CPDF_DIBSource* CPDF_DIBSource::LoadMask(FX_DWORD& MatteColor) { |
| MatteColor = 0xFFFFFFFF; |
| CPDF_Stream* pSoftMask = m_pDict->GetStream(FX_BSTRC("SMask")); |
| if (pSoftMask) { |
| CPDF_Array* pMatte = pSoftMask->GetDict()->GetArray(FX_BSTRC("Matte")); |
| if (pMatte && m_pColorSpace && |
| (FX_DWORD)m_pColorSpace->CountComponents() <= m_nComponents) { |
| FX_FLOAT* pColor = FX_Alloc(FX_FLOAT, m_nComponents); |
| for (FX_DWORD i = 0; i < m_nComponents; i++) { |
| pColor[i] = pMatte->GetFloat(i); |
| } |
| FX_FLOAT R, G, B; |
| m_pColorSpace->GetRGB(pColor, R, G, B); |
| FX_Free(pColor); |
| MatteColor = FXARGB_MAKE(0, FXSYS_round(R * 255), FXSYS_round(G * 255), |
| FXSYS_round(B * 255)); |
| } |
| return LoadMaskDIB(pSoftMask); |
| } |
| |
| if (CPDF_Stream* pStream = |
| ToStream(m_pDict->GetElementValue(FX_BSTRC("Mask")))) |
| return LoadMaskDIB(pStream); |
| |
| return nullptr; |
| } |
| int CPDF_DIBSource::StratLoadMask() { |
| m_MatteColor = 0XFFFFFFFF; |
| m_pMaskStream = m_pDict->GetStream(FX_BSTRC("SMask")); |
| if (m_pMaskStream) { |
| CPDF_Array* pMatte = m_pMaskStream->GetDict()->GetArray(FX_BSTRC("Matte")); |
| if (pMatte != NULL && m_pColorSpace && |
| (FX_DWORD)m_pColorSpace->CountComponents() <= m_nComponents) { |
| FX_FLOAT R, G, B; |
| FX_FLOAT* pColor = FX_Alloc(FX_FLOAT, m_nComponents); |
| for (FX_DWORD i = 0; i < m_nComponents; i++) { |
| pColor[i] = pMatte->GetFloat(i); |
| } |
| m_pColorSpace->GetRGB(pColor, R, G, B); |
| FX_Free(pColor); |
| m_MatteColor = FXARGB_MAKE(0, FXSYS_round(R * 255), FXSYS_round(G * 255), |
| FXSYS_round(B * 255)); |
| } |
| return StartLoadMaskDIB(); |
| } |
| |
| m_pMaskStream = ToStream(m_pDict->GetElementValue(FX_BSTRC("Mask"))); |
| return m_pMaskStream ? StartLoadMaskDIB() : 1; |
| } |
| int CPDF_DIBSource::ContinueLoadMaskDIB(IFX_Pause* pPause) { |
| if (m_pMask == NULL) { |
| return 1; |
| } |
| int ret = m_pMask->ContinueLoadDIBSource(pPause); |
| if (ret == 2) { |
| return ret; |
| } |
| if (m_pColorSpace && m_bStdCS) { |
| m_pColorSpace->EnableStdConversion(FALSE); |
| } |
| if (!ret) { |
| delete m_pMask; |
| m_pMask = NULL; |
| return ret; |
| } |
| return 1; |
| } |
| CPDF_DIBSource* CPDF_DIBSource::DetachMask() { |
| CPDF_DIBSource* pDIBSource = m_pMask; |
| m_pMask = NULL; |
| return pDIBSource; |
| } |
| CPDF_DIBSource* CPDF_DIBSource::LoadMaskDIB(CPDF_Stream* pMask) { |
| CPDF_DIBSource* pMaskSource = new CPDF_DIBSource; |
| if (!pMaskSource->Load(m_pDocument, pMask, NULL, NULL, NULL, NULL, TRUE)) { |
| delete pMaskSource; |
| return NULL; |
| } |
| return pMaskSource; |
| } |
| int CPDF_DIBSource::StartLoadMaskDIB() { |
| m_pMask = new CPDF_DIBSource; |
| int ret = m_pMask->StartLoadDIBSource(m_pDocument, m_pMaskStream, FALSE, |
| nullptr, nullptr, TRUE); |
| if (ret == 2) { |
| if (m_Status == 0) |
| m_Status = 2; |
| return 2; |
| } |
| if (!ret) { |
| delete m_pMask; |
| m_pMask = nullptr; |
| return 1; |
| } |
| return 1; |
| } |
| void CPDF_DIBSource::LoadPalette() { |
| if (m_bpc == 0) { |
| return; |
| } |
| if (m_bpc * m_nComponents > 8) { |
| return; |
| } |
| if (m_pColorSpace == NULL) { |
| return; |
| } |
| if (m_bpc * m_nComponents == 1) { |
| if (m_bDefaultDecode && |
| (m_Family == PDFCS_DEVICEGRAY || m_Family == PDFCS_DEVICERGB)) { |
| return; |
| } |
| if (m_pColorSpace->CountComponents() > 3) { |
| return; |
| } |
| FX_FLOAT color_values[3]; |
| color_values[0] = m_pCompData[0].m_DecodeMin; |
| color_values[1] = color_values[2] = color_values[0]; |
| FX_FLOAT R = 0.0f, G = 0.0f, B = 0.0f; |
| m_pColorSpace->GetRGB(color_values, R, G, B); |
| FX_ARGB argb0 = ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), |
| FXSYS_round(B * 255)); |
| color_values[0] += m_pCompData[0].m_DecodeStep; |
| color_values[1] += m_pCompData[0].m_DecodeStep; |
| color_values[2] += m_pCompData[0].m_DecodeStep; |
| m_pColorSpace->GetRGB(color_values, R, G, B); |
| FX_ARGB argb1 = ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), |
| FXSYS_round(B * 255)); |
| if (argb0 != 0xFF000000 || argb1 != 0xFFFFFFFF) { |
| SetPaletteArgb(0, argb0); |
| SetPaletteArgb(1, argb1); |
| } |
| return; |
| } |
| if (m_pColorSpace == CPDF_ColorSpace::GetStockCS(PDFCS_DEVICEGRAY) && |
| m_bpc == 8 && m_bDefaultDecode) { |
| } else { |
| int palette_count = 1 << (m_bpc * m_nComponents); |
| CFX_FixedBufGrow<FX_FLOAT, 16> color_values(m_nComponents); |
| FX_FLOAT* color_value = color_values; |
| for (int i = 0; i < palette_count; i++) { |
| int color_data = i; |
| for (FX_DWORD j = 0; j < m_nComponents; j++) { |
| int encoded_component = color_data % (1 << m_bpc); |
| color_data /= 1 << m_bpc; |
| color_value[j] = m_pCompData[j].m_DecodeMin + |
| m_pCompData[j].m_DecodeStep * encoded_component; |
| } |
| FX_FLOAT R = 0, G = 0, B = 0; |
| if (m_nComponents == 1 && m_Family == PDFCS_ICCBASED && |
| m_pColorSpace->CountComponents() > 1) { |
| int nComponents = m_pColorSpace->CountComponents(); |
| FX_FLOAT* temp_buf = FX_Alloc(FX_FLOAT, nComponents); |
| for (int i = 0; i < nComponents; i++) { |
| temp_buf[i] = *color_value; |
| } |
| m_pColorSpace->GetRGB(temp_buf, R, G, B); |
| FX_Free(temp_buf); |
| } else { |
| m_pColorSpace->GetRGB(color_value, R, G, B); |
| } |
| SetPaletteArgb(i, ArgbEncode(255, FXSYS_round(R * 255), |
| FXSYS_round(G * 255), FXSYS_round(B * 255))); |
| } |
| } |
| } |
| void CPDF_DIBSource::ValidateDictParam() { |
| m_bpc = m_bpc_orig; |
| CPDF_Object* pFilter = m_pDict->GetElementValue(FX_BSTRC("Filter")); |
| if (pFilter) { |
| if (pFilter->IsName()) { |
| CFX_ByteString filter = pFilter->GetString(); |
| if (filter == FX_BSTRC("CCITTFaxDecode") || |
| filter == FX_BSTRC("JBIG2Decode")) { |
| m_bpc = 1; |
| m_nComponents = 1; |
| } |
| if (filter == FX_BSTRC("RunLengthDecode")) { |
| if (m_bpc != 1) { |
| m_bpc = 8; |
| } |
| } else if (filter == FX_BSTRC("DCTDecode")) { |
| m_bpc = 8; |
| } |
| } else if (CPDF_Array* pArray = pFilter->AsArray()) { |
| if (pArray->GetString(pArray->GetCount() - 1) == |
| FX_BSTRC("CCITTFaxDecode") || |
| pArray->GetString(pArray->GetCount() - 1) == |
| FX_BSTRC("JBIG2Decode")) { |
| m_bpc = 1; |
| m_nComponents = 1; |
| } |
| if (pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("DCTDecode")) { |
| // Previously, pArray->GetString(pArray->GetCount() - 1) == |
| // FX_BSTRC("RunLengthDecode") was checked in the "if" statement as |
| // well, |
| // but too many documents don't conform to it. |
| m_bpc = 8; |
| } |
| } |
| } |
| if (m_bpc != 1 && m_bpc != 2 && m_bpc != 4 && m_bpc != 8 && m_bpc != 16) { |
| m_bpc = 0; |
| } |
| } |
| #define NORMALCOLOR_MAX(color, max) \ |
| (color) > (max) ? (max) : (color) < 0 ? 0 : (color); |
| void CPDF_DIBSource::TranslateScanline24bpp(uint8_t* dest_scan, |
| const uint8_t* src_scan) const { |
| if (m_bpc == 0) { |
| return; |
| } |
| int max_data = (1 << m_bpc) - 1; |
| if (m_bDefaultDecode) { |
| if (m_Family == PDFCS_DEVICERGB || m_Family == PDFCS_CALRGB) { |
| const uint8_t* src_pos = src_scan; |
| switch (m_bpc) { |
| case 16: |
| for (int col = 0; col < m_Width; col++) { |
| *dest_scan++ = src_pos[4]; |
| *dest_scan++ = src_pos[2]; |
| *dest_scan++ = *src_pos; |
| src_pos += 6; |
| } |
| break; |
| case 8: |
| for (int column = 0; column < m_Width; column++) { |
| *dest_scan++ = src_pos[2]; |
| *dest_scan++ = src_pos[1]; |
| *dest_scan++ = *src_pos; |
| src_pos += 3; |
| } |
| break; |
| default: |
| int src_bit_pos = 0; |
| int dest_byte_pos = 0; |
| for (int column = 0; column < m_Width; column++) { |
| int R = GetBits8(src_scan, src_bit_pos, m_bpc); |
| src_bit_pos += m_bpc; |
| int G = GetBits8(src_scan, src_bit_pos, m_bpc); |
| src_bit_pos += m_bpc; |
| int B = GetBits8(src_scan, src_bit_pos, m_bpc); |
| src_bit_pos += m_bpc; |
| R = NORMALCOLOR_MAX(R, max_data); |
| G = NORMALCOLOR_MAX(G, max_data); |
| B = NORMALCOLOR_MAX(B, max_data); |
| dest_scan[dest_byte_pos] = B * 255 / max_data; |
| dest_scan[dest_byte_pos + 1] = G * 255 / max_data; |
| dest_scan[dest_byte_pos + 2] = R * 255 / max_data; |
| dest_byte_pos += 3; |
| } |
| break; |
| } |
| return; |
| } |
| if (m_bpc == 8) { |
| if (m_nComponents == m_pColorSpace->CountComponents()) |
| m_pColorSpace->TranslateImageLine(dest_scan, src_scan, m_Width, m_Width, |
| m_Height, TransMask()); |
| return; |
| } |
| } |
| CFX_FixedBufGrow<FX_FLOAT, 16> color_values1(m_nComponents); |
| FX_FLOAT* color_values = color_values1; |
| FX_FLOAT R = 0.0f, G = 0.0f, B = 0.0f; |
| if (m_bpc == 8) { |
| int src_byte_pos = 0; |
| int dest_byte_pos = 0; |
| for (int column = 0; column < m_Width; column++) { |
| for (FX_DWORD color = 0; color < m_nComponents; color++) { |
| int data = src_scan[src_byte_pos++]; |
| color_values[color] = m_pCompData[color].m_DecodeMin + |
| m_pCompData[color].m_DecodeStep * data; |
| } |
| if (TransMask()) { |
| FX_FLOAT k = 1.0f - color_values[3]; |
| R = (1.0f - color_values[0]) * k; |
| G = (1.0f - color_values[1]) * k; |
| B = (1.0f - color_values[2]) * k; |
| } else { |
| m_pColorSpace->GetRGB(color_values, R, G, B); |
| } |
| R = NORMALCOLOR_MAX(R, 1); |
| G = NORMALCOLOR_MAX(G, 1); |
| B = NORMALCOLOR_MAX(B, 1); |
| dest_scan[dest_byte_pos] = (int32_t)(B * 255); |
| dest_scan[dest_byte_pos + 1] = (int32_t)(G * 255); |
| dest_scan[dest_byte_pos + 2] = (int32_t)(R * 255); |
| dest_byte_pos += 3; |
| } |
| } else { |
| int src_bit_pos = 0; |
| int dest_byte_pos = 0; |
| for (int column = 0; column < m_Width; column++) { |
| for (FX_DWORD color = 0; color < m_nComponents; color++) { |
| int data = GetBits8(src_scan, src_bit_pos, m_bpc); |
| color_values[color] = m_pCompData[color].m_DecodeMin + |
| m_pCompData[color].m_DecodeStep * data; |
| src_bit_pos += m_bpc; |
| } |
| if (TransMask()) { |
| FX_FLOAT k = 1.0f - color_values[3]; |
| R = (1.0f - color_values[0]) * k; |
| G = (1.0f - color_values[1]) * k; |
| B = (1.0f - color_values[2]) * k; |
| } else { |
| m_pColorSpace->GetRGB(color_values, R, G, B); |
| } |
| R = NORMALCOLOR_MAX(R, 1); |
| G = NORMALCOLOR_MAX(G, 1); |
| B = NORMALCOLOR_MAX(B, 1); |
| dest_scan[dest_byte_pos] = (int32_t)(B * 255); |
| dest_scan[dest_byte_pos + 1] = (int32_t)(G * 255); |
| dest_scan[dest_byte_pos + 2] = (int32_t)(R * 255); |
| dest_byte_pos += 3; |
| } |
| } |
| } |
| uint8_t* CPDF_DIBSource::GetBuffer() const { |
| if (m_pCachedBitmap) { |
| return m_pCachedBitmap->GetBuffer(); |
| } |
| return NULL; |
| } |
| const uint8_t* CPDF_DIBSource::GetScanline(int line) const { |
| if (m_bpc == 0) { |
| return nullptr; |
| } |
| FX_SAFE_DWORD src_pitch = CalculatePitch8(m_bpc, m_nComponents, m_Width); |
| if (!src_pitch.IsValid()) |
| return nullptr; |
| FX_DWORD src_pitch_value = src_pitch.ValueOrDie(); |
| const uint8_t* pSrcLine = nullptr; |
| if (m_pCachedBitmap && src_pitch_value <= m_pCachedBitmap->GetPitch()) { |
| if (line >= m_pCachedBitmap->GetHeight()) { |
| line = m_pCachedBitmap->GetHeight() - 1; |
| } |
| pSrcLine = m_pCachedBitmap->GetScanline(line); |
| } else if (m_pDecoder) { |
| pSrcLine = m_pDecoder->GetScanline(line); |
| } else { |
| if (m_pStreamAcc->GetSize() >= (line + 1) * src_pitch_value) { |
| pSrcLine = m_pStreamAcc->GetData() + line * src_pitch_value; |
| } |
| } |
| if (!pSrcLine) { |
| uint8_t* pLineBuf = m_pMaskedLine ? m_pMaskedLine : m_pLineBuf; |
| FXSYS_memset(pLineBuf, 0xFF, m_Pitch); |
| return pLineBuf; |
| } |
| if (m_bpc * m_nComponents == 1) { |
| if (m_bImageMask && m_bDefaultDecode) { |
| for (FX_DWORD i = 0; i < src_pitch_value; i++) { |
| m_pLineBuf[i] = ~pSrcLine[i]; |
| } |
| } else if (m_bColorKey) { |
| FX_DWORD reset_argb, set_argb; |
| reset_argb = m_pPalette ? m_pPalette[0] : 0xFF000000; |
| set_argb = m_pPalette ? m_pPalette[1] : 0xFFFFFFFF; |
| if (m_pCompData[0].m_ColorKeyMin == 0) { |
| reset_argb = 0; |
| } |
| if (m_pCompData[0].m_ColorKeyMax == 1) { |
| set_argb = 0; |
| } |
| set_argb = FXARGB_TODIB(set_argb); |
| reset_argb = FXARGB_TODIB(reset_argb); |
| FX_DWORD* dest_scan = (FX_DWORD*)m_pMaskedLine; |
| for (int col = 0; col < m_Width; col++) { |
| if (pSrcLine[col / 8] & (1 << (7 - col % 8))) { |
| *dest_scan = set_argb; |
| } else { |
| *dest_scan = reset_argb; |
| } |
| dest_scan++; |
| } |
| return m_pMaskedLine; |
| } else { |
| FXSYS_memcpy(m_pLineBuf, pSrcLine, src_pitch_value); |
| } |
| return m_pLineBuf; |
| } |
| if (m_bpc * m_nComponents <= 8) { |
| if (m_bpc == 8) { |
| FXSYS_memcpy(m_pLineBuf, pSrcLine, src_pitch_value); |
| } else { |
| int src_bit_pos = 0; |
| for (int col = 0; col < m_Width; col++) { |
| int color_index = 0; |
| for (FX_DWORD color = 0; color < m_nComponents; color++) { |
| int data = GetBits8(pSrcLine, src_bit_pos, m_bpc); |
| color_index |= data << (color * m_bpc); |
| src_bit_pos += m_bpc; |
| } |
| m_pLineBuf[col] = color_index; |
| } |
| } |
| if (m_bColorKey) { |
| uint8_t* pDestPixel = m_pMaskedLine; |
| const uint8_t* pSrcPixel = m_pLineBuf; |
| for (int col = 0; col < m_Width; col++) { |
| uint8_t index = *pSrcPixel++; |
| if (m_pPalette) { |
| *pDestPixel++ = FXARGB_B(m_pPalette[index]); |
| *pDestPixel++ = FXARGB_G(m_pPalette[index]); |
| *pDestPixel++ = FXARGB_R(m_pPalette[index]); |
| } else { |
| *pDestPixel++ = index; |
| *pDestPixel++ = index; |
| *pDestPixel++ = index; |
| } |
| *pDestPixel = (index < m_pCompData[0].m_ColorKeyMin || |
| index > m_pCompData[0].m_ColorKeyMax) |
| ? 0xFF |
| : 0; |
| pDestPixel++; |
| } |
| return m_pMaskedLine; |
| } |
| return m_pLineBuf; |
| } |
| if (m_bColorKey) { |
| if (m_nComponents == 3 && m_bpc == 8) { |
| uint8_t* alpha_channel = m_pMaskedLine + 3; |
| for (int col = 0; col < m_Width; col++) { |
| const uint8_t* pPixel = pSrcLine + col * 3; |
| alpha_channel[col * 4] = (pPixel[0] < m_pCompData[0].m_ColorKeyMin || |
| pPixel[0] > m_pCompData[0].m_ColorKeyMax || |
| pPixel[1] < m_pCompData[1].m_ColorKeyMin || |
| pPixel[1] > m_pCompData[1].m_ColorKeyMax || |
| pPixel[2] < m_pCompData[2].m_ColorKeyMin || |
| pPixel[2] > m_pCompData[2].m_ColorKeyMax) |
| ? 0xFF |
| : 0; |
| } |
| } else { |
| FXSYS_memset(m_pMaskedLine, 0xFF, m_Pitch); |
| } |
| } |
| if (m_pColorSpace) { |
| TranslateScanline24bpp(m_pLineBuf, pSrcLine); |
| pSrcLine = m_pLineBuf; |
| } |
| if (m_bColorKey) { |
| const uint8_t* pSrcPixel = pSrcLine; |
| uint8_t* pDestPixel = m_pMaskedLine; |
| for (int col = 0; col < m_Width; col++) { |
| *pDestPixel++ = *pSrcPixel++; |
| *pDestPixel++ = *pSrcPixel++; |
| *pDestPixel++ = *pSrcPixel++; |
| pDestPixel++; |
| } |
| return m_pMaskedLine; |
| } |
| return pSrcLine; |
| } |
| |
| FX_BOOL CPDF_DIBSource::SkipToScanline(int line, IFX_Pause* pPause) const { |
| return m_pDecoder && m_pDecoder->SkipToScanline(line, pPause); |
| } |
| |
| void CPDF_DIBSource::DownSampleScanline(int line, |
| uint8_t* dest_scan, |
| int dest_bpp, |
| int dest_width, |
| FX_BOOL bFlipX, |
| int clip_left, |
| int clip_width) const { |
| if (line < 0 || !dest_scan || dest_bpp <= 0 || dest_width <= 0 || |
| clip_left < 0 || clip_width <= 0) { |
| return; |
| } |
| |
| FX_DWORD src_width = m_Width; |
| FX_SAFE_DWORD pitch = CalculatePitch8(m_bpc, m_nComponents, m_Width); |
| if (!pitch.IsValid()) |
| return; |
| |
| const uint8_t* pSrcLine = nullptr; |
| if (m_pCachedBitmap) { |
| pSrcLine = m_pCachedBitmap->GetScanline(line); |
| } else if (m_pDecoder) { |
| pSrcLine = m_pDecoder->GetScanline(line); |
| } else { |
| FX_DWORD src_pitch = pitch.ValueOrDie(); |
| pitch *= (line + 1); |
| if (!pitch.IsValid()) { |
| return; |
| } |
| |
| if (m_pStreamAcc->GetSize() >= pitch.ValueOrDie()) { |
| pSrcLine = m_pStreamAcc->GetData() + line * src_pitch; |
| } |
| } |
| int orig_Bpp = m_bpc * m_nComponents / 8; |
| int dest_Bpp = dest_bpp / 8; |
| if (!pSrcLine) { |
| FXSYS_memset(dest_scan, 0xFF, dest_Bpp * clip_width); |
| return; |
| } |
| |
| FX_SAFE_INT32 max_src_x = clip_left; |
| max_src_x += clip_width - 1; |
| max_src_x *= src_width; |
| max_src_x /= dest_width; |
| if (!max_src_x.IsValid()) |
| return; |
| |
| if (m_bpc * m_nComponents == 1) { |
| DownSampleScanline1Bit(orig_Bpp, dest_Bpp, src_width, pSrcLine, dest_scan, |
| dest_width, bFlipX, clip_left, clip_width); |
| } else if (m_bpc * m_nComponents <= 8) { |
| DownSampleScanline8Bit(orig_Bpp, dest_Bpp, src_width, pSrcLine, dest_scan, |
| dest_width, bFlipX, clip_left, clip_width); |
| } else { |
| DownSampleScanline32Bit(orig_Bpp, dest_Bpp, src_width, pSrcLine, dest_scan, |
| dest_width, bFlipX, clip_left, clip_width); |
| } |
| } |
| |
| void CPDF_DIBSource::DownSampleScanline1Bit(int orig_Bpp, |
| int dest_Bpp, |
| FX_DWORD src_width, |
| const uint8_t* pSrcLine, |
| uint8_t* dest_scan, |
| int dest_width, |
| FX_BOOL bFlipX, |
| int clip_left, |
| int clip_width) const { |
| FX_DWORD set_argb = (FX_DWORD)-1; |
| FX_DWORD reset_argb = 0; |
| if (m_bImageMask) { |
| if (m_bDefaultDecode) { |
| set_argb = 0; |
| reset_argb = (FX_DWORD)-1; |
| } |
| } else if (m_bColorKey) { |
| reset_argb = m_pPalette ? m_pPalette[0] : 0xFF000000; |
| set_argb = m_pPalette ? m_pPalette[1] : 0xFFFFFFFF; |
| if (m_pCompData[0].m_ColorKeyMin == 0) { |
| reset_argb = 0; |
| } |
| if (m_pCompData[0].m_ColorKeyMax == 1) { |
| set_argb = 0; |
| } |
| set_argb = FXARGB_TODIB(set_argb); |
| reset_argb = FXARGB_TODIB(reset_argb); |
| for (int i = 0; i < clip_width; i++) { |
| FX_DWORD src_x = (clip_left + i) * src_width / dest_width; |
| if (bFlipX) { |
| src_x = src_width - src_x - 1; |
| } |
| src_x %= src_width; |
| if (pSrcLine[src_x / 8] & (1 << (7 - src_x % 8))) { |
| ((FX_DWORD*)dest_scan)[i] = set_argb; |
| } else { |
| ((FX_DWORD*)dest_scan)[i] = reset_argb; |
| } |
| } |
| return; |
| } else { |
| if (dest_Bpp == 1) { |
| } else if (m_pPalette) { |
| reset_argb = m_pPalette[0]; |
| set_argb = m_pPalette[1]; |
| } |
| } |
| for (int i = 0; i < clip_width; i++) { |
| FX_DWORD src_x = (clip_left + i) * src_width / dest_width; |
| if (bFlipX) { |
| src_x = src_width - src_x - 1; |
| } |
| src_x %= src_width; |
| int dest_pos = i * dest_Bpp; |
| if (pSrcLine[src_x / 8] & (1 << (7 - src_x % 8))) { |
| if (dest_Bpp == 1) { |
| dest_scan[dest_pos] = (uint8_t)set_argb; |
| } else if (dest_Bpp == 3) { |
| dest_scan[dest_pos] = FXARGB_B(set_argb); |
| dest_scan[dest_pos + 1] = FXARGB_G(set_argb); |
| dest_scan[dest_pos + 2] = FXARGB_R(set_argb); |
| } else { |
| *(FX_DWORD*)(dest_scan + dest_pos) = set_argb; |
| } |
| } else { |
| if (dest_Bpp == 1) { |
| dest_scan[dest_pos] = (uint8_t)reset_argb; |
| } else if (dest_Bpp == 3) { |
| dest_scan[dest_pos] = FXARGB_B(reset_argb); |
| dest_scan[dest_pos + 1] = FXARGB_G(reset_argb); |
| dest_scan[dest_pos + 2] = FXARGB_R(reset_argb); |
| } else { |
| *(FX_DWORD*)(dest_scan + dest_pos) = reset_argb; |
| } |
| } |
| } |
| } |
| |
| void CPDF_DIBSource::DownSampleScanline8Bit(int orig_Bpp, |
| int dest_Bpp, |
| FX_DWORD src_width, |
| const uint8_t* pSrcLine, |
| uint8_t* dest_scan, |
| int dest_width, |
| FX_BOOL bFlipX, |
| int clip_left, |
| int clip_width) const { |
| if (m_bpc < 8) { |
| int src_bit_pos = 0; |
| for (FX_DWORD col = 0; col < src_width; col++) { |
| int color_index = 0; |
| for (FX_DWORD color = 0; color < m_nComponents; color++) { |
| int data = GetBits8(pSrcLine, src_bit_pos, m_bpc); |
| color_index |= data << (color * m_bpc); |
| src_bit_pos += m_bpc; |
| } |
| m_pLineBuf[col] = color_index; |
| } |
| pSrcLine = m_pLineBuf; |
| } |
| if (m_bColorKey) { |
| for (int i = 0; i < clip_width; i++) { |
| FX_DWORD src_x = (clip_left + i) * src_width / dest_width; |
| if (bFlipX) { |
| src_x = src_width - src_x - 1; |
| } |
| src_x %= src_width; |
| uint8_t* pDestPixel = dest_scan + i * 4; |
| uint8_t index = pSrcLine[src_x]; |
| if (m_pPalette) { |
| *pDestPixel++ = FXARGB_B(m_pPalette[index]); |
| *pDestPixel++ = FXARGB_G(m_pPalette[index]); |
| *pDestPixel++ = FXARGB_R(m_pPalette[index]); |
| } else { |
| *pDestPixel++ = index; |
| *pDestPixel++ = index; |
| *pDestPixel++ = index; |
| } |
| *pDestPixel = (index < m_pCompData[0].m_ColorKeyMin || |
| index > m_pCompData[0].m_ColorKeyMax) |
| ? 0xFF |
| : 0; |
| } |
| return; |
| } |
| for (int i = 0; i < clip_width; i++) { |
| FX_DWORD src_x = (clip_left + i) * src_width / dest_width; |
| if (bFlipX) { |
| src_x = src_width - src_x - 1; |
| } |
| src_x %= src_width; |
| uint8_t index = pSrcLine[src_x]; |
| if (dest_Bpp == 1) { |
| dest_scan[i] = index; |
| } else { |
| int dest_pos = i * dest_Bpp; |
| FX_ARGB argb = m_pPalette[index]; |
| dest_scan[dest_pos] = FXARGB_B(argb); |
| dest_scan[dest_pos + 1] = FXARGB_G(argb); |
| dest_scan[dest_pos + 2] = FXARGB_R(argb); |
| } |
| } |
| } |
| |
| void CPDF_DIBSource::DownSampleScanline32Bit(int orig_Bpp, |
| int dest_Bpp, |
| FX_DWORD src_width, |
| const uint8_t* pSrcLine, |
| uint8_t* dest_scan, |
| int dest_width, |
| FX_BOOL bFlipX, |
| int clip_left, |
| int clip_width) const { |
| int last_src_x = -1; |
| FX_ARGB last_argb = FXARGB_MAKE(0xFF, 0xFF, 0xFF, 0xFF); |
| FX_FLOAT unit_To8Bpc = 255.0f / ((1 << m_bpc) - 1); |
| for (int i = 0; i < clip_width; i++) { |
| int dest_x = clip_left + i; |
| FX_DWORD src_x = (bFlipX ? (dest_width - dest_x - 1) : dest_x) * |
| (int64_t)src_width / dest_width; |
| src_x %= src_width; |
| |
| // No need to check for 32-bit overflow, as |src_x| is bounded by |
| // |src_width| and DownSampleScanline already checked for overflow with the |
| // pitch calculation. |
| const uint8_t* pSrcPixel = nullptr; |
| size_t bit_offset = 0; |
| if (m_bpc % 8 == 0) { |
| pSrcPixel = pSrcLine + src_x * orig_Bpp; |
| } else { |
| size_t num_bits = src_x * m_bpc * m_nComponents; |
| pSrcPixel = pSrcLine + num_bits / 8; |
| bit_offset = num_bits % 8; |
| } |
| |
| uint8_t* pDestPixel = dest_scan + i * dest_Bpp; |
| FX_ARGB argb; |
| if (src_x == last_src_x) { |
| argb = last_argb; |
| } else { |
| CFX_FixedBufGrow<uint8_t, 128> extracted_components(m_nComponents); |
| if (m_bpc % 8 != 0) { |
| uint64_t src_bit_pos = bit_offset; |
| for (FX_DWORD j = 0; j < m_nComponents; ++j) { |
| extracted_components[j] = static_cast<uint8_t>( |
| GetBits8(pSrcPixel, src_bit_pos, m_bpc) * unit_To8Bpc); |
| src_bit_pos += m_bpc; |
| } |
| pSrcPixel = extracted_components; |
| } |
| |
| if (m_pColorSpace) { |
| uint8_t color[4]; |
| const FX_BOOL bTransMask = TransMask(); |
| if (m_bDefaultDecode) { |
| m_pColorSpace->TranslateImageLine(color, pSrcPixel, 1, 0, 0, |
| bTransMask); |
| } else { |
| for (FX_DWORD j = 0; j < m_nComponents; ++j) { |
| FX_FLOAT component_value = |
| static_cast<FX_FLOAT>(extracted_components[j]); |
| int color_value = |
| (int)((m_pCompData[j].m_DecodeMin + |
| m_pCompData[j].m_DecodeStep * component_value) * |
| 255.0f + |
| 0.5f); |
| extracted_components[j] = |
| color_value > 255 ? 255 : (color_value < 0 ? 0 : color_value); |
| } |
| m_pColorSpace->TranslateImageLine(color, extracted_components, 1, 0, |
| 0, bTransMask); |
| } |
| argb = FXARGB_MAKE(0xFF, color[2], color[1], color[0]); |
| } else { |
| argb = FXARGB_MAKE(0xFF, pSrcPixel[2], pSrcPixel[1], pSrcPixel[0]); |
| } |
| if (m_bColorKey) { |
| int alpha = 0xFF; |
| if (m_nComponents == 3 && m_bpc == 8) { |
| alpha = (pSrcPixel[0] < m_pCompData[0].m_ColorKeyMin || |
| pSrcPixel[0] > m_pCompData[0].m_ColorKeyMax || |
| pSrcPixel[1] < m_pCompData[1].m_ColorKeyMin || |
| pSrcPixel[1] > m_pCompData[1].m_ColorKeyMax || |
| pSrcPixel[2] < m_pCompData[2].m_ColorKeyMin || |
| pSrcPixel[2] > m_pCompData[2].m_ColorKeyMax) |
| ? 0xFF |
| : 0; |
| } |
| argb &= 0xFFFFFF; |
| argb |= alpha << 24; |
| } |
| last_src_x = src_x; |
| last_argb = argb; |
| } |
| if (dest_Bpp == 4) { |
| *(FX_DWORD*)pDestPixel = FXARGB_TODIB(argb); |
| } else { |
| *pDestPixel++ = FXARGB_B(argb); |
| *pDestPixel++ = FXARGB_G(argb); |
| *pDestPixel = FXARGB_R(argb); |
| } |
| } |
| } |
| |
| FX_BOOL CPDF_DIBSource::TransMask() const { |
| return m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && |
| m_Family == PDFCS_DEVICECMYK; |
| } |
| |
| void CPDF_DIBSource::SetDownSampleSize(int dest_width, int dest_height) const { |
| if (m_pDecoder) { |
| m_pDecoder->DownScale(dest_width, dest_height); |
| ((CPDF_DIBSource*)this)->m_Width = m_pDecoder->GetWidth(); |
| ((CPDF_DIBSource*)this)->m_Height = m_pDecoder->GetHeight(); |
| } |
| } |
| |
| void CPDF_DIBSource::ClearImageData() { |
| if (m_pDecoder) { |
| m_pDecoder->ClearImageData(); |
| } |
| } |
| |
| CPDF_ProgressiveImageLoaderHandle::CPDF_ProgressiveImageLoaderHandle() { |
| m_pImageLoader = NULL; |
| m_pCache = NULL; |
| m_pImage = NULL; |
| } |
| CPDF_ProgressiveImageLoaderHandle::~CPDF_ProgressiveImageLoaderHandle() {} |
| FX_BOOL CPDF_ProgressiveImageLoaderHandle::Start( |
| CPDF_ImageLoader* pImageLoader, |
| const CPDF_ImageObject* pImage, |
| CPDF_PageRenderCache* pCache, |
| FX_BOOL bStdCS, |
| FX_DWORD GroupFamily, |
| FX_BOOL bLoadMask, |
| CPDF_RenderStatus* pRenderStatus, |
| int32_t nDownsampleWidth, |
| int32_t nDownsampleHeight) { |
| m_pImageLoader = pImageLoader; |
| m_pCache = pCache; |
| m_pImage = (CPDF_ImageObject*)pImage; |
| m_nDownsampleWidth = nDownsampleWidth; |
| m_nDownsampleHeight = nDownsampleHeight; |
| FX_BOOL ret; |
| if (pCache) { |
| ret = pCache->StartGetCachedBitmap(pImage->m_pImage->GetStream(), bStdCS, |
| GroupFamily, bLoadMask, pRenderStatus, |
| m_nDownsampleWidth, m_nDownsampleHeight); |
| if (ret == FALSE) { |
| m_pImageLoader->m_bCached = TRUE; |
| m_pImageLoader->m_pBitmap = pCache->m_pCurImageCache->DetachBitmap(); |
| m_pImageLoader->m_pMask = pCache->m_pCurImageCache->DetachMask(); |
| m_pImageLoader->m_MatteColor = pCache->m_pCurImageCache->m_MatteColor; |
| } |
| } else { |
| ret = pImage->m_pImage->StartLoadDIBSource(pRenderStatus->m_pFormResource, |
| pRenderStatus->m_pPageResource, |
| bStdCS, GroupFamily, bLoadMask); |
| if (ret == FALSE) { |
| m_pImageLoader->m_bCached = FALSE; |
| m_pImageLoader->m_pBitmap = m_pImage->m_pImage->DetachBitmap(); |
| m_pImageLoader->m_pMask = m_pImage->m_pImage->DetachMask(); |
| m_pImageLoader->m_MatteColor = m_pImage->m_pImage->m_MatteColor; |
| } |
| } |
| return ret; |
| } |
| FX_BOOL CPDF_ProgressiveImageLoaderHandle::Continue(IFX_Pause* pPause) { |
| FX_BOOL ret; |
| if (m_pCache) { |
| ret = m_pCache->Continue(pPause); |
| if (ret == FALSE) { |
| m_pImageLoader->m_bCached = TRUE; |
| m_pImageLoader->m_pBitmap = m_pCache->m_pCurImageCache->DetachBitmap(); |
| m_pImageLoader->m_pMask = m_pCache->m_pCurImageCache->DetachMask(); |
| m_pImageLoader->m_MatteColor = m_pCache->m_pCurImageCache->m_MatteColor; |
| } |
| } else { |
| ret = m_pImage->m_pImage->Continue(pPause); |
| if (ret == FALSE) { |
| m_pImageLoader->m_bCached = FALSE; |
| m_pImageLoader->m_pBitmap = m_pImage->m_pImage->DetachBitmap(); |
| m_pImageLoader->m_pMask = m_pImage->m_pImage->DetachMask(); |
| m_pImageLoader->m_MatteColor = m_pImage->m_pImage->m_MatteColor; |
| } |
| } |
| return ret; |
| } |
| FX_BOOL CPDF_ImageLoader::Load(const CPDF_ImageObject* pImage, |
| CPDF_PageRenderCache* pCache, |
| FX_BOOL bStdCS, |
| FX_DWORD GroupFamily, |
| FX_BOOL bLoadMask, |
| CPDF_RenderStatus* pRenderStatus) { |
| if (pImage == NULL) { |
| return FALSE; |
| } |
| if (pCache) { |
| pCache->GetCachedBitmap(pImage->m_pImage->GetStream(), m_pBitmap, m_pMask, |
| m_MatteColor, bStdCS, GroupFamily, bLoadMask, |
| pRenderStatus, m_nDownsampleWidth, |
| m_nDownsampleHeight); |
| m_bCached = TRUE; |
| } else { |
| m_pBitmap = pImage->m_pImage->LoadDIBSource(&m_pMask, &m_MatteColor, bStdCS, |
| GroupFamily, bLoadMask); |
| m_bCached = FALSE; |
| } |
| return FALSE; |
| } |
| FX_BOOL CPDF_ImageLoader::StartLoadImage(const CPDF_ImageObject* pImage, |
| CPDF_PageRenderCache* pCache, |
| void*& LoadHandle, |
| FX_BOOL bStdCS, |
| FX_DWORD GroupFamily, |
| FX_BOOL bLoadMask, |
| CPDF_RenderStatus* pRenderStatus, |
| int32_t nDownsampleWidth, |
| int32_t nDownsampleHeight) { |
| m_nDownsampleWidth = nDownsampleWidth; |
| m_nDownsampleHeight = nDownsampleHeight; |
| CPDF_ProgressiveImageLoaderHandle* pLoaderHandle = |
| new CPDF_ProgressiveImageLoaderHandle; |
| FX_BOOL ret = pLoaderHandle->Start(this, pImage, pCache, bStdCS, GroupFamily, |
| bLoadMask, pRenderStatus, |
| m_nDownsampleWidth, m_nDownsampleHeight); |
| LoadHandle = pLoaderHandle; |
| return ret; |
| } |
| FX_BOOL CPDF_ImageLoader::Continue(void* LoadHandle, IFX_Pause* pPause) { |
| return ((CPDF_ProgressiveImageLoaderHandle*)LoadHandle)->Continue(pPause); |
| } |
| CPDF_ImageLoader::~CPDF_ImageLoader() { |
| if (!m_bCached) { |
| delete m_pBitmap; |
| delete m_pMask; |
| } |
| } |
