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 #include "third_party/base/numerics/safe_conversions.h"
23
24 namespace {
25
GetEstimatedImageSize(const RetainPtr<CFX_DIBBase> & pDIB)26 uint32_t GetEstimatedImageSize(const RetainPtr<CFX_DIBBase>& pDIB) {
27 if (!pDIB || !pDIB->GetBuffer())
28 return 0;
29
30 int height = pDIB->GetHeight();
31 ASSERT(pdfium::base::IsValueInRangeForNumericType<uint32_t>(height));
32 return static_cast<uint32_t>(height) * pDIB->GetPitch() +
33 pDIB->GetPaletteSize() * 4;
34 }
35
36 } // namespace
37
CPDF_ImageCacheEntry(CPDF_Document * pDoc,const RetainPtr<CPDF_Image> & pImage)38 CPDF_ImageCacheEntry::CPDF_ImageCacheEntry(CPDF_Document* pDoc,
39 const RetainPtr<CPDF_Image>& pImage)
40 : m_pDocument(pDoc), m_pImage(pImage) {}
41
42 CPDF_ImageCacheEntry::~CPDF_ImageCacheEntry() = default;
43
Reset()44 void CPDF_ImageCacheEntry::Reset() {
45 m_pCachedBitmap.Reset();
46 CalcSize();
47 }
48
DetachBitmap()49 RetainPtr<CFX_DIBBase> CPDF_ImageCacheEntry::DetachBitmap() {
50 return std::move(m_pCurBitmap);
51 }
52
DetachMask()53 RetainPtr<CFX_DIBBase> CPDF_ImageCacheEntry::DetachMask() {
54 return std::move(m_pCurMask);
55 }
56
StartGetCachedBitmap(const CPDF_Dictionary * pPageResources,const CPDF_RenderStatus * pRenderStatus,bool bStdCS)57 CPDF_DIB::LoadState CPDF_ImageCacheEntry::StartGetCachedBitmap(
58 const CPDF_Dictionary* pPageResources,
59 const CPDF_RenderStatus* pRenderStatus,
60 bool bStdCS) {
61 if (m_pCachedBitmap) {
62 m_pCurBitmap = m_pCachedBitmap;
63 m_pCurMask = m_pCachedMask;
64 return CPDF_DIB::LoadState::kSuccess;
65 }
66
67 m_pCurBitmap = pdfium::MakeRetain<CPDF_DIB>();
68 CPDF_DIB::LoadState ret = m_pCurBitmap.As<CPDF_DIB>()->StartLoadDIBBase(
69 m_pDocument.Get(), m_pImage->GetStream(), true,
70 pRenderStatus->GetFormResource(), pPageResources, bStdCS,
71 pRenderStatus->GetGroupFamily(), pRenderStatus->GetLoadMask());
72 if (ret == CPDF_DIB::LoadState::kContinue)
73 return CPDF_DIB::LoadState::kContinue;
74
75 if (ret == CPDF_DIB::LoadState::kSuccess)
76 ContinueGetCachedBitmap(pRenderStatus);
77 else
78 m_pCurBitmap.Reset();
79 return CPDF_DIB::LoadState::kFail;
80 }
81
Continue(PauseIndicatorIface * pPause,CPDF_RenderStatus * pRenderStatus)82 bool CPDF_ImageCacheEntry::Continue(PauseIndicatorIface* pPause,
83 CPDF_RenderStatus* pRenderStatus) {
84 CPDF_DIB::LoadState ret =
85 m_pCurBitmap.As<CPDF_DIB>()->ContinueLoadDIBBase(pPause);
86 if (ret == CPDF_DIB::LoadState::kContinue)
87 return true;
88
89 if (ret == CPDF_DIB::LoadState::kSuccess)
90 ContinueGetCachedBitmap(pRenderStatus);
91 else
92 m_pCurBitmap.Reset();
93 return false;
94 }
95
ContinueGetCachedBitmap(const CPDF_RenderStatus * pRenderStatus)96 void CPDF_ImageCacheEntry::ContinueGetCachedBitmap(
97 const CPDF_RenderStatus* pRenderStatus) {
98 m_MatteColor = m_pCurBitmap.As<CPDF_DIB>()->GetMatteColor();
99 m_pCurMask = m_pCurBitmap.As<CPDF_DIB>()->DetachMask();
100 CPDF_RenderContext* pContext = pRenderStatus->GetContext();
101 CPDF_PageRenderCache* pPageRenderCache = pContext->GetPageCache();
102 m_dwTimeCount = pPageRenderCache->GetTimeCount();
103 if (m_pCurBitmap->GetPitch() * m_pCurBitmap->GetHeight() < kHugeImageSize) {
104 m_pCachedBitmap = m_pCurBitmap->Clone(nullptr);
105 m_pCurBitmap.Reset();
106 } else {
107 m_pCachedBitmap = m_pCurBitmap;
108 }
109 if (m_pCurMask) {
110 m_pCachedMask = m_pCurMask->Clone(nullptr);
111 m_pCurMask.Reset();
112 }
113 m_pCurBitmap = m_pCachedBitmap;
114 m_pCurMask = m_pCachedMask;
115 CalcSize();
116 }
117
CalcSize()118 void CPDF_ImageCacheEntry::CalcSize() {
119 m_dwCacheSize = GetEstimatedImageSize(m_pCachedBitmap) +
120 GetEstimatedImageSize(m_pCachedMask);
121 }
122