1 // Copyright 2014 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 "public/fpdf_progressive.h"
8 
9 #include <memory>
10 #include <utility>
11 
12 #include "core/fpdfapi/page/cpdf_page.h"
13 #include "core/fpdfapi/render/cpdf_pagerendercontext.h"
14 #include "core/fpdfapi/render/cpdf_progressiverenderer.h"
15 #include "core/fxge/cfx_defaultrenderdevice.h"
16 #include "fpdfsdk/cpdfsdk_helpers.h"
17 #include "fpdfsdk/cpdfsdk_pauseadapter.h"
18 #include "fpdfsdk/cpdfsdk_renderpage.h"
19 #include "public/fpdfview.h"
20 
21 #if defined(_SKIA_SUPPORT_PATHS_)
22 #include "core/fxge/cfx_renderdevice.h"
23 #endif
24 
25 // These checks are here because core/ and public/ cannot depend on each other.
26 static_assert(CPDF_ProgressiveRenderer::kReady == FPDF_RENDER_READY,
27               "CPDF_ProgressiveRenderer::kReady value mismatch");
28 static_assert(CPDF_ProgressiveRenderer::kToBeContinued ==
29                   FPDF_RENDER_TOBECONTINUED,
30               "CPDF_ProgressiveRenderer::kToBeContinued value mismatch");
31 static_assert(CPDF_ProgressiveRenderer::kDone == FPDF_RENDER_DONE,
32               "CPDF_ProgressiveRenderer::kDone value mismatch");
33 static_assert(CPDF_ProgressiveRenderer::kFailed == FPDF_RENDER_FAILED,
34               "CPDF_ProgressiveRenderer::kFailed value mismatch");
35 
36 namespace {
37 
ToFPDFStatus(CPDF_ProgressiveRenderer::Status status)38 int ToFPDFStatus(CPDF_ProgressiveRenderer::Status status) {
39   return static_cast<int>(status);
40 }
41 
42 }  // namespace
43 
44 FPDF_EXPORT int FPDF_CALLCONV
FPDF_RenderPageBitmapWithColorScheme_Start(FPDF_BITMAP bitmap,FPDF_PAGE page,int start_x,int start_y,int size_x,int size_y,int rotate,int flags,const FPDF_COLORSCHEME * color_scheme,IFSDK_PAUSE * pause)45 FPDF_RenderPageBitmapWithColorScheme_Start(FPDF_BITMAP bitmap,
46                                            FPDF_PAGE page,
47                                            int start_x,
48                                            int start_y,
49                                            int size_x,
50                                            int size_y,
51                                            int rotate,
52                                            int flags,
53                                            const FPDF_COLORSCHEME* color_scheme,
54                                            IFSDK_PAUSE* pause) {
55   if (!bitmap || !pause || pause->version != 1)
56     return FPDF_RENDER_FAILED;
57 
58   CPDF_Page* pPage = CPDFPageFromFPDFPage(page);
59   if (!pPage)
60     return FPDF_RENDER_FAILED;
61 
62   auto pOwnedContext = std::make_unique<CPDF_PageRenderContext>();
63   CPDF_PageRenderContext* pContext = pOwnedContext.get();
64   pPage->SetRenderContext(std::move(pOwnedContext));
65 
66   RetainPtr<CFX_DIBitmap> pBitmap(CFXDIBitmapFromFPDFBitmap(bitmap));
67   auto pOwnedDevice = std::make_unique<CFX_DefaultRenderDevice>();
68   CFX_DefaultRenderDevice* pDevice = pOwnedDevice.get();
69   pContext->m_pDevice = std::move(pOwnedDevice);
70   pDevice->Attach(pBitmap, !!(flags & FPDF_REVERSE_BYTE_ORDER), nullptr, false);
71 
72   CPDFSDK_PauseAdapter pause_adapter(pause);
73   CPDFSDK_RenderPageWithContext(pContext, pPage, start_x, start_y, size_x,
74                                 size_y, rotate, flags, color_scheme,
75                                 /*need_to_restore=*/false, &pause_adapter);
76 
77 #if defined(_SKIA_SUPPORT_PATHS_)
78   pDevice->Flush(false);
79   pBitmap->UnPreMultiply();
80 #endif
81 
82   if (!pContext->m_pRenderer)
83     return FPDF_RENDER_FAILED;
84 
85   return ToFPDFStatus(pContext->m_pRenderer->GetStatus());
86 }
87 
FPDF_RenderPageBitmap_Start(FPDF_BITMAP bitmap,FPDF_PAGE page,int start_x,int start_y,int size_x,int size_y,int rotate,int flags,IFSDK_PAUSE * pause)88 FPDF_EXPORT int FPDF_CALLCONV FPDF_RenderPageBitmap_Start(FPDF_BITMAP bitmap,
89                                                           FPDF_PAGE page,
90                                                           int start_x,
91                                                           int start_y,
92                                                           int size_x,
93                                                           int size_y,
94                                                           int rotate,
95                                                           int flags,
96                                                           IFSDK_PAUSE* pause) {
97   return FPDF_RenderPageBitmapWithColorScheme_Start(
98       bitmap, page, start_x, start_y, size_x, size_y, rotate, flags,
99       /*color_scheme=*/nullptr, pause);
100 }
101 
FPDF_RenderPage_Continue(FPDF_PAGE page,IFSDK_PAUSE * pause)102 FPDF_EXPORT int FPDF_CALLCONV FPDF_RenderPage_Continue(FPDF_PAGE page,
103                                                        IFSDK_PAUSE* pause) {
104   if (!pause || pause->version != 1)
105     return FPDF_RENDER_FAILED;
106 
107   CPDF_Page* pPage = CPDFPageFromFPDFPage(page);
108   if (!pPage)
109     return FPDF_RENDER_FAILED;
110 
111   auto* pContext =
112       static_cast<CPDF_PageRenderContext*>(pPage->GetRenderContext());
113   if (!pContext || !pContext->m_pRenderer)
114     return FPDF_RENDER_FAILED;
115 
116   CPDFSDK_PauseAdapter pause_adapter(pause);
117   pContext->m_pRenderer->Continue(&pause_adapter);
118 #if defined(_SKIA_SUPPORT_PATHS_)
119   CFX_RenderDevice* pDevice = pContext->m_pDevice.get();
120   pDevice->Flush(false);
121   pDevice->GetBitmap()->UnPreMultiply();
122 #endif
123   return ToFPDFStatus(pContext->m_pRenderer->GetStatus());
124 }
125 
FPDF_RenderPage_Close(FPDF_PAGE page)126 FPDF_EXPORT void FPDF_CALLCONV FPDF_RenderPage_Close(FPDF_PAGE page) {
127   CPDF_Page* pPage = CPDFPageFromFPDFPage(page);
128   if (pPage) {
129 #if defined(_SKIA_SUPPORT_PATHS_)
130     auto* pContext =
131         static_cast<CPDF_PageRenderContext*>(pPage->GetRenderContext());
132     if (pContext && pContext->m_pRenderer) {
133       CFX_RenderDevice* pDevice = pContext->m_pDevice.get();
134       pDevice->Flush(true);
135       pDevice->GetBitmap()->UnPreMultiply();
136     }
137 #endif
138     pPage->SetRenderContext(nullptr);
139   }
140 }
141