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 "xfa/fxfa/parser/cxfa_layoutprocessor.h"
8
9 #include "third_party/base/ptr_util.h"
10 #include "xfa/fxfa/parser/cxfa_contentlayoutitem.h"
11 #include "xfa/fxfa/parser/cxfa_document.h"
12 #include "xfa/fxfa/parser/cxfa_layoutpagemgr.h"
13 #include "xfa/fxfa/parser/cxfa_measurement.h"
14 #include "xfa/fxfa/parser/xfa_document_datamerger_imp.h"
15 #include "xfa/fxfa/parser/xfa_layout_itemlayout.h"
16 #include "xfa/fxfa/parser/xfa_localemgr.h"
17 #include "xfa/fxfa/parser/xfa_object.h"
18 #include "xfa/fxfa/parser/xfa_utils.h"
19
CXFA_LayoutProcessor(CXFA_Document * pDocument)20 CXFA_LayoutProcessor::CXFA_LayoutProcessor(CXFA_Document* pDocument)
21 : m_pDocument(pDocument),
22 m_nProgressCounter(0),
23 m_bNeeLayout(true) {}
24
~CXFA_LayoutProcessor()25 CXFA_LayoutProcessor::~CXFA_LayoutProcessor() {}
26
GetDocument() const27 CXFA_Document* CXFA_LayoutProcessor::GetDocument() const {
28 return m_pDocument;
29 }
30
StartLayout(bool bForceRestart)31 int32_t CXFA_LayoutProcessor::StartLayout(bool bForceRestart) {
32 if (!bForceRestart && !IsNeedLayout())
33 return 100;
34
35 m_pRootItemLayoutProcessor.reset();
36 m_nProgressCounter = 0;
37 CXFA_Node* pFormPacketNode =
38 ToNode(m_pDocument->GetXFAObject(XFA_HASHCODE_Form));
39 if (!pFormPacketNode)
40 return -1;
41
42 CXFA_Node* pFormRoot =
43 pFormPacketNode->GetFirstChildByClass(XFA_Element::Subform);
44 if (!pFormRoot)
45 return -1;
46
47 if (!m_pLayoutPageMgr)
48 m_pLayoutPageMgr = pdfium::MakeUnique<CXFA_LayoutPageMgr>(this);
49 if (!m_pLayoutPageMgr->InitLayoutPage(pFormRoot))
50 return -1;
51
52 if (!m_pLayoutPageMgr->PrepareFirstPage(pFormRoot))
53 return -1;
54
55 m_pRootItemLayoutProcessor = pdfium::MakeUnique<CXFA_ItemLayoutProcessor>(
56 pFormRoot, m_pLayoutPageMgr.get());
57 m_nProgressCounter = 1;
58 return 0;
59 }
60
DoLayout(IFX_Pause * pPause)61 int32_t CXFA_LayoutProcessor::DoLayout(IFX_Pause* pPause) {
62 if (m_nProgressCounter < 1)
63 return -1;
64
65 XFA_ItemLayoutProcessorResult eStatus;
66 CXFA_Node* pFormNode = m_pRootItemLayoutProcessor->GetFormNode();
67 FX_FLOAT fPosX = pFormNode->GetMeasure(XFA_ATTRIBUTE_X).ToUnit(XFA_UNIT_Pt);
68 FX_FLOAT fPosY = pFormNode->GetMeasure(XFA_ATTRIBUTE_Y).ToUnit(XFA_UNIT_Pt);
69 do {
70 FX_FLOAT fAvailHeight = m_pLayoutPageMgr->GetAvailHeight();
71 eStatus = m_pRootItemLayoutProcessor->DoLayout(true, fAvailHeight,
72 fAvailHeight, nullptr);
73 if (eStatus != XFA_ItemLayoutProcessorResult::Done)
74 m_nProgressCounter++;
75
76 CXFA_ContentLayoutItem* pLayoutItem =
77 m_pRootItemLayoutProcessor->ExtractLayoutItem();
78 if (pLayoutItem)
79 pLayoutItem->m_sPos = CFX_PointF(fPosX, fPosY);
80
81 m_pLayoutPageMgr->SubmitContentItem(pLayoutItem, eStatus);
82 } while (eStatus != XFA_ItemLayoutProcessorResult::Done &&
83 (!pPause || !pPause->NeedToPauseNow()));
84
85 if (eStatus == XFA_ItemLayoutProcessorResult::Done) {
86 m_pLayoutPageMgr->FinishPaginatedPageSets();
87 m_pLayoutPageMgr->SyncLayoutData();
88 m_bNeeLayout = false;
89 m_rgChangedContainers.RemoveAll();
90 }
91 return 100 * (eStatus == XFA_ItemLayoutProcessorResult::Done
92 ? m_nProgressCounter
93 : m_nProgressCounter - 1) /
94 m_nProgressCounter;
95 }
96
IncrementLayout()97 bool CXFA_LayoutProcessor::IncrementLayout() {
98 if (m_bNeeLayout) {
99 StartLayout(true);
100 return DoLayout(nullptr) == 100;
101 }
102
103 for (int32_t i = 0, c = m_rgChangedContainers.GetSize(); i < c; i++) {
104 CXFA_Node* pNode = m_rgChangedContainers[i];
105 CXFA_Node* pParentNode =
106 pNode->GetNodeItem(XFA_NODEITEM_Parent, XFA_ObjectType::ContainerNode);
107 if (!pParentNode)
108 return false;
109 if (!CXFA_ItemLayoutProcessor::IncrementRelayoutNode(this, pNode,
110 pParentNode)) {
111 return false;
112 }
113 }
114 m_rgChangedContainers.RemoveAll();
115 return true;
116 }
117
CountPages() const118 int32_t CXFA_LayoutProcessor::CountPages() const {
119 return m_pLayoutPageMgr ? m_pLayoutPageMgr->GetPageCount() : 0;
120 }
121
GetPage(int32_t index) const122 CXFA_ContainerLayoutItem* CXFA_LayoutProcessor::GetPage(int32_t index) const {
123 return m_pLayoutPageMgr ? m_pLayoutPageMgr->GetPage(index) : nullptr;
124 }
125
GetLayoutItem(CXFA_Node * pFormItem)126 CXFA_LayoutItem* CXFA_LayoutProcessor::GetLayoutItem(CXFA_Node* pFormItem) {
127 return static_cast<CXFA_LayoutItem*>(
128 pFormItem->GetUserData(XFA_LAYOUTITEMKEY));
129 }
130
AddChangedContainer(CXFA_Node * pContainer)131 void CXFA_LayoutProcessor::AddChangedContainer(CXFA_Node* pContainer) {
132 if (m_rgChangedContainers.Find(pContainer) < 0)
133 m_rgChangedContainers.Add(pContainer);
134 }
135
GetRootLayoutItem() const136 CXFA_ContainerLayoutItem* CXFA_LayoutProcessor::GetRootLayoutItem() const {
137 return m_pLayoutPageMgr ? m_pLayoutPageMgr->GetRootLayoutItem() : nullptr;
138 }
139
IsNeedLayout()140 bool CXFA_LayoutProcessor::IsNeedLayout() {
141 return m_bNeeLayout || m_rgChangedContainers.GetSize() > 0;
142 }
143