1 // Copyright 2016 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "core/fpdfapi/render/cpdf_imagecacheentry.h"
8
9 #include <memory>
10 #include <utility>
11
12 #include "core/fpdfapi/page/cpdf_dib.h"
13 #include "core/fpdfapi/page/cpdf_image.h"
14 #include "core/fpdfapi/page/cpdf_page.h"
15 #include "core/fpdfapi/parser/cpdf_dictionary.h"
16 #include "core/fpdfapi/parser/cpdf_document.h"
17 #include "core/fpdfapi/parser/cpdf_stream.h"
18 #include "core/fpdfapi/render/cpdf_pagerendercache.h"
19 #include "core/fpdfapi/render/cpdf_rendercontext.h"
20 #include "core/fpdfapi/render/cpdf_renderstatus.h"
21 #include "core/fxge/dib/cfx_dibitmap.h"
22
23 namespace {
24
GetEstimatedImageSize(const RetainPtr<CFX_DIBBase> & pDIB)25 uint32_t GetEstimatedImageSize(const RetainPtr<CFX_DIBBase>& pDIB) {
26 if (!pDIB || !pDIB->GetBuffer())
27 return 0;
28
29 int height = pDIB->GetHeight();
30 ASSERT(pdfium::base::IsValueInRangeForNumericType<uint32_t>(height));
31 return static_cast<uint32_t>(height) * pDIB->GetPitch() +
32 pDIB->GetPaletteSize() * 4;
33 }
34
35 } // namespace
36
CPDF_ImageCacheEntry(CPDF_Document * pDoc,const RetainPtr<CPDF_Image> & pImage)37 CPDF_ImageCacheEntry::CPDF_ImageCacheEntry(CPDF_Document* pDoc,
38 const RetainPtr<CPDF_Image>& pImage)
39 : m_pDocument(pDoc), m_pImage(pImage) {}
40
41 CPDF_ImageCacheEntry::~CPDF_ImageCacheEntry() = default;
42
Reset()43 void CPDF_ImageCacheEntry::Reset() {
44 m_pCachedBitmap.Reset();
45 CalcSize();
46 }
47
DetachBitmap()48 RetainPtr<CFX_DIBBase> CPDF_ImageCacheEntry::DetachBitmap() {
49 return std::move(m_pCurBitmap);
50 }
51
DetachMask()52 RetainPtr<CFX_DIBBase> CPDF_ImageCacheEntry::DetachMask() {
53 return std::move(m_pCurMask);
54 }
55
StartGetCachedBitmap(CPDF_Dictionary * pPageResources,const CPDF_RenderStatus * pRenderStatus,bool bStdCS)56 CPDF_DIB::LoadState CPDF_ImageCacheEntry::StartGetCachedBitmap(
57 CPDF_Dictionary* pPageResources,
58 const CPDF_RenderStatus* pRenderStatus,
59 bool bStdCS) {
60 if (m_pCachedBitmap) {
61 m_pCurBitmap = m_pCachedBitmap;
62 m_pCurMask = m_pCachedMask;
63 return CPDF_DIB::LoadState::kSuccess;
64 }
65
66 m_pCurBitmap = pdfium::MakeRetain<CPDF_DIB>();
67 CPDF_DIB::LoadState ret = m_pCurBitmap.As<CPDF_DIB>()->StartLoadDIBBase(
68 m_pDocument.Get(), m_pImage->GetStream(), true,
69 pRenderStatus->GetFormResource(), pPageResources, bStdCS,
70 pRenderStatus->GetGroupFamily(), pRenderStatus->GetLoadMask());
71 if (ret == CPDF_DIB::LoadState::kContinue)
72 return CPDF_DIB::LoadState::kContinue;
73
74 if (ret == CPDF_DIB::LoadState::kSuccess)
75 ContinueGetCachedBitmap(pRenderStatus);
76 else
77 m_pCurBitmap.Reset();
78 return CPDF_DIB::LoadState::kFail;
79 }
80
Continue(PauseIndicatorIface * pPause,CPDF_RenderStatus * pRenderStatus)81 bool CPDF_ImageCacheEntry::Continue(PauseIndicatorIface* pPause,
82 CPDF_RenderStatus* pRenderStatus) {
83 CPDF_DIB::LoadState ret =
84 m_pCurBitmap.As<CPDF_DIB>()->ContinueLoadDIBBase(pPause);
85 if (ret == CPDF_DIB::LoadState::kContinue)
86 return true;
87
88 if (ret == CPDF_DIB::LoadState::kSuccess)
89 ContinueGetCachedBitmap(pRenderStatus);
90 else
91 m_pCurBitmap.Reset();
92 return false;
93 }
94
ContinueGetCachedBitmap(const CPDF_RenderStatus * pRenderStatus)95 void CPDF_ImageCacheEntry::ContinueGetCachedBitmap(
96 const CPDF_RenderStatus* pRenderStatus) {
97 m_MatteColor = m_pCurBitmap.As<CPDF_DIB>()->GetMatteColor();
98 m_pCurMask = m_pCurBitmap.As<CPDF_DIB>()->DetachMask();
99 CPDF_RenderContext* pContext = pRenderStatus->GetContext();
100 CPDF_PageRenderCache* pPageRenderCache = pContext->GetPageCache();
101 m_dwTimeCount = pPageRenderCache->GetTimeCount();
102 if (m_pCurBitmap->GetPitch() * m_pCurBitmap->GetHeight() < kHugeImageSize) {
103 m_pCachedBitmap = m_pCurBitmap->Clone(nullptr);
104 m_pCurBitmap.Reset();
105 } else {
106 m_pCachedBitmap = m_pCurBitmap;
107 }
108 if (m_pCurMask) {
109 m_pCachedMask = m_pCurMask->Clone(nullptr);
110 m_pCurMask.Reset();
111 }
112 m_pCurBitmap = m_pCachedBitmap;
113 m_pCurMask = m_pCachedMask;
114 CalcSize();
115 }
116
CalcSize()117 void CPDF_ImageCacheEntry::CalcSize() {
118 m_dwCacheSize = GetEstimatedImageSize(m_pCachedBitmap) +
119 GetEstimatedImageSize(m_pCachedMask);
120 }
121