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_colorstate.h"
8
9 #include <utility>
10
11 #include "core/fpdfapi/page/cpdf_colorspace.h"
12 #include "core/fpdfapi/page/cpdf_pattern.h"
13 #include "core/fpdfapi/page/cpdf_tilingpattern.h"
14 #include "core/fxge/dib/fx_dib.h"
15 #include "third_party/base/check.h"
16
17 CPDF_ColorState::CPDF_ColorState() = default;
18
CPDF_ColorState(const CPDF_ColorState & that)19 CPDF_ColorState::CPDF_ColorState(const CPDF_ColorState& that)
20 : m_Ref(that.m_Ref) {}
21
22 CPDF_ColorState::~CPDF_ColorState() = default;
23
Emplace()24 void CPDF_ColorState::Emplace() {
25 m_Ref.Emplace();
26 }
27
SetDefault()28 void CPDF_ColorState::SetDefault() {
29 m_Ref.GetPrivateCopy()->SetDefault();
30 }
31
GetFillColorRef() const32 FX_COLORREF CPDF_ColorState::GetFillColorRef() const {
33 return m_Ref.GetObject()->m_FillColorRef;
34 }
35
SetFillColorRef(FX_COLORREF colorref)36 void CPDF_ColorState::SetFillColorRef(FX_COLORREF colorref) {
37 m_Ref.GetPrivateCopy()->m_FillColorRef = colorref;
38 }
39
GetStrokeColorRef() const40 FX_COLORREF CPDF_ColorState::GetStrokeColorRef() const {
41 return m_Ref.GetObject()->m_StrokeColorRef;
42 }
43
SetStrokeColorRef(FX_COLORREF colorref)44 void CPDF_ColorState::SetStrokeColorRef(FX_COLORREF colorref) {
45 m_Ref.GetPrivateCopy()->m_StrokeColorRef = colorref;
46 }
47
GetFillColor() const48 const CPDF_Color* CPDF_ColorState::GetFillColor() const {
49 const ColorData* pData = m_Ref.GetObject();
50 return pData ? &pData->m_FillColor : nullptr;
51 }
52
GetMutableFillColor()53 CPDF_Color* CPDF_ColorState::GetMutableFillColor() {
54 return &m_Ref.GetPrivateCopy()->m_FillColor;
55 }
56
HasFillColor() const57 bool CPDF_ColorState::HasFillColor() const {
58 const CPDF_Color* pColor = GetFillColor();
59 return pColor && !pColor->IsNull();
60 }
61
GetStrokeColor() const62 const CPDF_Color* CPDF_ColorState::GetStrokeColor() const {
63 const ColorData* pData = m_Ref.GetObject();
64 return pData ? &pData->m_StrokeColor : nullptr;
65 }
66
GetMutableStrokeColor()67 CPDF_Color* CPDF_ColorState::GetMutableStrokeColor() {
68 return &m_Ref.GetPrivateCopy()->m_StrokeColor;
69 }
70
HasStrokeColor() const71 bool CPDF_ColorState::HasStrokeColor() const {
72 const CPDF_Color* pColor = GetStrokeColor();
73 return pColor && !pColor->IsNull();
74 }
75
SetFillColor(RetainPtr<CPDF_ColorSpace> colorspace,std::vector<float> values)76 void CPDF_ColorState::SetFillColor(RetainPtr<CPDF_ColorSpace> colorspace,
77 std::vector<float> values) {
78 ColorData* pData = m_Ref.GetPrivateCopy();
79 SetColor(std::move(colorspace), std::move(values), &pData->m_FillColor,
80 &pData->m_FillColorRef);
81 }
82
SetStrokeColor(RetainPtr<CPDF_ColorSpace> colorspace,std::vector<float> values)83 void CPDF_ColorState::SetStrokeColor(RetainPtr<CPDF_ColorSpace> colorspace,
84 std::vector<float> values) {
85 ColorData* pData = m_Ref.GetPrivateCopy();
86 SetColor(std::move(colorspace), std::move(values), &pData->m_StrokeColor,
87 &pData->m_StrokeColorRef);
88 }
89
SetFillPattern(RetainPtr<CPDF_Pattern> pattern,pdfium::span<float> values)90 void CPDF_ColorState::SetFillPattern(RetainPtr<CPDF_Pattern> pattern,
91 pdfium::span<float> values) {
92 ColorData* pData = m_Ref.GetPrivateCopy();
93 SetPattern(std::move(pattern), values, &pData->m_FillColor,
94 &pData->m_FillColorRef);
95 }
96
SetStrokePattern(RetainPtr<CPDF_Pattern> pattern,pdfium::span<float> values)97 void CPDF_ColorState::SetStrokePattern(RetainPtr<CPDF_Pattern> pattern,
98 pdfium::span<float> values) {
99 ColorData* pData = m_Ref.GetPrivateCopy();
100 SetPattern(std::move(pattern), values, &pData->m_StrokeColor,
101 &pData->m_StrokeColorRef);
102 }
103
SetColor(RetainPtr<CPDF_ColorSpace> colorspace,std::vector<float> values,CPDF_Color * color,FX_COLORREF * colorref)104 void CPDF_ColorState::SetColor(RetainPtr<CPDF_ColorSpace> colorspace,
105 std::vector<float> values,
106 CPDF_Color* color,
107 FX_COLORREF* colorref) {
108 DCHECK(color);
109 DCHECK(colorref);
110
111 if (colorspace) {
112 color->SetColorSpace(std::move(colorspace));
113 } else if (color->IsNull()) {
114 color->SetColorSpace(
115 CPDF_ColorSpace::GetStockCS(CPDF_ColorSpace::Family::kDeviceGray));
116 }
117 if (color->CountComponents() > values.size())
118 return;
119
120 if (!color->IsPattern())
121 color->SetValueForNonPattern(std::move(values));
122 int R;
123 int G;
124 int B;
125 *colorref = color->GetRGB(&R, &G, &B) ? FXSYS_BGR(B, G, R) : 0xFFFFFFFF;
126 }
127
SetPattern(RetainPtr<CPDF_Pattern> pattern,pdfium::span<float> values,CPDF_Color * color,FX_COLORREF * colorref)128 void CPDF_ColorState::SetPattern(RetainPtr<CPDF_Pattern> pattern,
129 pdfium::span<float> values,
130 CPDF_Color* color,
131 FX_COLORREF* colorref) {
132 DCHECK(color);
133 DCHECK(colorref);
134 color->SetValueForPattern(pattern, values);
135 int R;
136 int G;
137 int B;
138 if (color->GetRGB(&R, &G, &B)) {
139 *colorref = FXSYS_BGR(B, G, R);
140 return;
141 }
142 CPDF_TilingPattern* tiling = pattern->AsTilingPattern();
143 *colorref = tiling && tiling->colored() ? 0x00BFBFBF : 0xFFFFFFFF;
144 }
145
146 CPDF_ColorState::ColorData::ColorData() = default;
147
ColorData(const ColorData & src)148 CPDF_ColorState::ColorData::ColorData(const ColorData& src)
149 : m_FillColorRef(src.m_FillColorRef),
150 m_StrokeColorRef(src.m_StrokeColorRef),
151 m_FillColor(src.m_FillColor),
152 m_StrokeColor(src.m_StrokeColor) {}
153
154 CPDF_ColorState::ColorData::~ColorData() = default;
155
SetDefault()156 void CPDF_ColorState::ColorData::SetDefault() {
157 m_FillColorRef = 0;
158 m_StrokeColorRef = 0;
159 m_FillColor.SetColorSpace(
160 CPDF_ColorSpace::GetStockCS(CPDF_ColorSpace::Family::kDeviceGray));
161 m_StrokeColor.SetColorSpace(
162 CPDF_ColorSpace::GetStockCS(CPDF_ColorSpace::Family::kDeviceGray));
163 }
164
Clone() const165 RetainPtr<CPDF_ColorState::ColorData> CPDF_ColorState::ColorData::Clone()
166 const {
167 return pdfium::MakeRetain<CPDF_ColorState::ColorData>(*this);
168 }
169