1 // Copyright 2016 The PDFium Authors
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/page/cpdf_allstates.h"
8
9 #include <algorithm>
10 #include <utility>
11 #include <vector>
12
13 #include "core/fpdfapi/font/cpdf_font.h"
14 #include "core/fpdfapi/page/cpdf_pageobjectholder.h"
15 #include "core/fpdfapi/page/cpdf_streamcontentparser.h"
16 #include "core/fpdfapi/parser/cpdf_array.h"
17 #include "core/fpdfapi/parser/cpdf_dictionary.h"
18 #include "core/fpdfapi/parser/fpdf_parser_utility.h"
19 #include "core/fxcrt/bytestring.h"
20 #include "core/fxge/cfx_graphstatedata.h"
21
22 CPDF_AllStates::CPDF_AllStates() = default;
23
24 CPDF_AllStates::CPDF_AllStates(const CPDF_AllStates& that) = default;
25
26 CPDF_AllStates& CPDF_AllStates::operator=(const CPDF_AllStates& that) = default;
27
28 CPDF_AllStates::~CPDF_AllStates() = default;
29
SetDefaultStates()30 void CPDF_AllStates::SetDefaultStates() {
31 m_GraphicStates.SetDefaultStates();
32 }
33
SetLineDash(const CPDF_Array * pArray,float phase)34 void CPDF_AllStates::SetLineDash(const CPDF_Array* pArray, float phase) {
35 std::vector<float> dashes = ReadArrayElementsToVector(pArray, pArray->size());
36 mutable_graph_state().SetLineDash(std::move(dashes), phase);
37 }
38
ProcessExtGS(const CPDF_Dictionary * pGS,CPDF_StreamContentParser * pParser)39 void CPDF_AllStates::ProcessExtGS(const CPDF_Dictionary* pGS,
40 CPDF_StreamContentParser* pParser) {
41 CPDF_DictionaryLocker locker(pGS);
42 for (const auto& it : locker) {
43 RetainPtr<CPDF_Object> pObject = it.second->GetMutableDirect();
44 if (!pObject)
45 continue;
46
47 uint32_t key = it.first.GetID();
48 switch (key) {
49 case FXBSTR_ID('L', 'W', 0, 0):
50 mutable_graph_state().SetLineWidth(pObject->GetNumber());
51 break;
52 case FXBSTR_ID('L', 'C', 0, 0):
53 mutable_graph_state().SetLineCap(
54 static_cast<CFX_GraphStateData::LineCap>(pObject->GetInteger()));
55 break;
56 case FXBSTR_ID('L', 'J', 0, 0):
57 mutable_graph_state().SetLineJoin(
58 static_cast<CFX_GraphStateData::LineJoin>(pObject->GetInteger()));
59 break;
60 case FXBSTR_ID('M', 'L', 0, 0):
61 mutable_graph_state().SetMiterLimit(pObject->GetNumber());
62 break;
63 case FXBSTR_ID('D', 0, 0, 0): {
64 const CPDF_Array* pDash = pObject->AsArray();
65 if (!pDash)
66 break;
67
68 RetainPtr<const CPDF_Array> pArray = pDash->GetArrayAt(0);
69 if (!pArray)
70 break;
71
72 SetLineDash(pArray.Get(), pDash->GetFloatAt(1));
73 break;
74 }
75 case FXBSTR_ID('R', 'I', 0, 0):
76 mutable_general_state().SetRenderIntent(pObject->GetString());
77 break;
78 case FXBSTR_ID('F', 'o', 'n', 't'): {
79 const CPDF_Array* pFont = pObject->AsArray();
80 if (!pFont)
81 break;
82
83 mutable_text_state().SetFontSize(pFont->GetFloatAt(1));
84 mutable_text_state().SetFont(
85 pParser->FindFont(pFont->GetByteStringAt(0)));
86 break;
87 }
88 case FXBSTR_ID('T', 'R', 0, 0):
89 if (pGS->KeyExist("TR2")) {
90 continue;
91 }
92 [[fallthrough]];
93 case FXBSTR_ID('T', 'R', '2', 0):
94 mutable_general_state().SetTR(!pObject->IsName() ? std::move(pObject)
95 : nullptr);
96 break;
97 case FXBSTR_ID('B', 'M', 0, 0): {
98 const CPDF_Array* pArray = pObject->AsArray();
99 mutable_general_state().SetBlendMode(pArray ? pArray->GetByteStringAt(0)
100 : pObject->GetString());
101 if (general_state().GetBlendType() > BlendMode::kMultiply) {
102 pParser->GetPageObjectHolder()->SetBackgroundAlphaNeeded(true);
103 }
104 break;
105 }
106 case FXBSTR_ID('S', 'M', 'a', 's'): {
107 RetainPtr<CPDF_Dictionary> pMaskDict = ToDictionary(pObject);
108 mutable_general_state().SetSoftMask(pMaskDict);
109 if (pMaskDict)
110 mutable_general_state().SetSMaskMatrix(
111 pParser->GetCurStates()->m_CTM);
112 break;
113 }
114 case FXBSTR_ID('C', 'A', 0, 0):
115 mutable_general_state().SetStrokeAlpha(
116 std::clamp(pObject->GetNumber(), 0.0f, 1.0f));
117 break;
118 case FXBSTR_ID('c', 'a', 0, 0):
119 mutable_general_state().SetFillAlpha(
120 std::clamp(pObject->GetNumber(), 0.0f, 1.0f));
121 break;
122 case FXBSTR_ID('O', 'P', 0, 0):
123 mutable_general_state().SetStrokeOP(!!pObject->GetInteger());
124 if (!pGS->KeyExist("op"))
125 mutable_general_state().SetFillOP(!!pObject->GetInteger());
126 break;
127 case FXBSTR_ID('o', 'p', 0, 0):
128 mutable_general_state().SetFillOP(!!pObject->GetInteger());
129 break;
130 case FXBSTR_ID('O', 'P', 'M', 0):
131 mutable_general_state().SetOPMode(pObject->GetInteger());
132 break;
133 case FXBSTR_ID('B', 'G', 0, 0):
134 if (pGS->KeyExist("BG2")) {
135 continue;
136 }
137 [[fallthrough]];
138 case FXBSTR_ID('B', 'G', '2', 0):
139 mutable_general_state().SetBG(std::move(pObject));
140 break;
141 case FXBSTR_ID('U', 'C', 'R', 0):
142 if (pGS->KeyExist("UCR2")) {
143 continue;
144 }
145 [[fallthrough]];
146 case FXBSTR_ID('U', 'C', 'R', '2'):
147 mutable_general_state().SetUCR(std::move(pObject));
148 break;
149 case FXBSTR_ID('H', 'T', 0, 0):
150 mutable_general_state().SetHT(std::move(pObject));
151 break;
152 case FXBSTR_ID('F', 'L', 0, 0):
153 mutable_general_state().SetFlatness(pObject->GetNumber());
154 break;
155 case FXBSTR_ID('S', 'M', 0, 0):
156 mutable_general_state().SetSmoothness(pObject->GetNumber());
157 break;
158 case FXBSTR_ID('S', 'A', 0, 0):
159 mutable_general_state().SetStrokeAdjust(!!pObject->GetInteger());
160 break;
161 case FXBSTR_ID('A', 'I', 'S', 0):
162 mutable_general_state().SetAlphaSource(!!pObject->GetInteger());
163 break;
164 case FXBSTR_ID('T', 'K', 0, 0):
165 mutable_general_state().SetTextKnockout(!!pObject->GetInteger());
166 break;
167 }
168 }
169 }
170
ResetTextPosition()171 void CPDF_AllStates::ResetTextPosition() {
172 m_TextLinePos = CFX_PointF();
173 m_TextPos = CFX_PointF();
174 }
175
GetTransformedTextPosition() const176 CFX_PointF CPDF_AllStates::GetTransformedTextPosition() const {
177 return m_CTM.Transform(m_TextMatrix.Transform(
178 CFX_PointF(m_TextPos.x, m_TextPos.y + m_TextRise)));
179 }
180
MoveTextPoint(const CFX_PointF & point)181 void CPDF_AllStates::MoveTextPoint(const CFX_PointF& point) {
182 m_TextLinePos += point;
183 m_TextPos = m_TextLinePos;
184 }
185
MoveTextToNextLine()186 void CPDF_AllStates::MoveTextToNextLine() {
187 m_TextLinePos.y -= m_TextLeading;
188 m_TextPos = m_TextLinePos;
189 }
190
IncrementTextPositionX(float value)191 void CPDF_AllStates::IncrementTextPositionX(float value) {
192 m_TextPos.x += value;
193 }
194
IncrementTextPositionY(float value)195 void CPDF_AllStates::IncrementTextPositionY(float value) {
196 m_TextPos.y += value;
197 }
198