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 <optional>
10 #include <utility>
11
12 #include "core/fpdfapi/page/cpdf_colorspace.h"
13 #include "core/fpdfapi/page/cpdf_pattern.h"
14 #include "core/fpdfapi/page/cpdf_tilingpattern.h"
15 #include "core/fxge/dib/fx_dib.h"
16
17 CPDF_ColorState::CPDF_ColorState() = default;
18
19 CPDF_ColorState::CPDF_ColorState(const CPDF_ColorState& that) = default;
20
21 CPDF_ColorState::~CPDF_ColorState() = default;
22
Emplace()23 void CPDF_ColorState::Emplace() {
24 m_Ref.Emplace();
25 }
26
SetDefault()27 void CPDF_ColorState::SetDefault() {
28 m_Ref.GetPrivateCopy()->SetDefault();
29 }
30
GetFillColorRef() const31 FX_COLORREF CPDF_ColorState::GetFillColorRef() const {
32 return m_Ref.GetObject()->m_FillColorRef;
33 }
34
SetFillColorRef(FX_COLORREF colorref)35 void CPDF_ColorState::SetFillColorRef(FX_COLORREF colorref) {
36 if (!m_Ref || GetFillColorRef() != colorref) {
37 m_Ref.GetPrivateCopy()->m_FillColorRef = colorref;
38 }
39 }
40
GetStrokeColorRef() const41 FX_COLORREF CPDF_ColorState::GetStrokeColorRef() const {
42 return m_Ref.GetObject()->m_StrokeColorRef;
43 }
44
SetStrokeColorRef(FX_COLORREF colorref)45 void CPDF_ColorState::SetStrokeColorRef(FX_COLORREF colorref) {
46 if (!m_Ref || GetStrokeColorRef() != colorref) {
47 m_Ref.GetPrivateCopy()->m_StrokeColorRef = colorref;
48 }
49 }
50
GetFillColor() const51 const CPDF_Color* CPDF_ColorState::GetFillColor() const {
52 const ColorData* data = m_Ref.GetObject();
53 return data ? &data->m_FillColor : nullptr;
54 }
55
GetMutableFillColor()56 CPDF_Color* CPDF_ColorState::GetMutableFillColor() {
57 return &m_Ref.GetPrivateCopy()->m_FillColor;
58 }
59
HasFillColor() const60 bool CPDF_ColorState::HasFillColor() const {
61 const CPDF_Color* pColor = GetFillColor();
62 return pColor && !pColor->IsNull();
63 }
64
GetStrokeColor() const65 const CPDF_Color* CPDF_ColorState::GetStrokeColor() const {
66 const ColorData* data = m_Ref.GetObject();
67 return data ? &data->m_StrokeColor : nullptr;
68 }
69
GetMutableStrokeColor()70 CPDF_Color* CPDF_ColorState::GetMutableStrokeColor() {
71 return &m_Ref.GetPrivateCopy()->m_StrokeColor;
72 }
73
HasStrokeColor() const74 bool CPDF_ColorState::HasStrokeColor() const {
75 const CPDF_Color* pColor = GetStrokeColor();
76 return pColor && !pColor->IsNull();
77 }
78
SetFillColor(RetainPtr<CPDF_ColorSpace> colorspace,std::vector<float> values)79 void CPDF_ColorState::SetFillColor(RetainPtr<CPDF_ColorSpace> colorspace,
80 std::vector<float> values) {
81 ColorData* data = m_Ref.GetPrivateCopy();
82 std::optional<FX_COLORREF> colorref =
83 SetColor(std::move(colorspace), std::move(values), data->m_FillColor);
84 if (colorref.has_value()) {
85 data->m_FillColorRef = colorref.value();
86 }
87 }
88
SetStrokeColor(RetainPtr<CPDF_ColorSpace> colorspace,std::vector<float> values)89 void CPDF_ColorState::SetStrokeColor(RetainPtr<CPDF_ColorSpace> colorspace,
90 std::vector<float> values) {
91 ColorData* data = m_Ref.GetPrivateCopy();
92 std::optional<FX_COLORREF> colorref =
93 SetColor(std::move(colorspace), std::move(values), data->m_StrokeColor);
94 if (colorref.has_value()) {
95 data->m_StrokeColorRef = colorref.value();
96 }
97 }
98
SetFillPattern(RetainPtr<CPDF_Pattern> pattern,pdfium::span<float> values)99 void CPDF_ColorState::SetFillPattern(RetainPtr<CPDF_Pattern> pattern,
100 pdfium::span<float> values) {
101 ColorData* data = m_Ref.GetPrivateCopy();
102 data->m_FillColorRef =
103 SetPattern(std::move(pattern), values, data->m_FillColor);
104 }
105
SetStrokePattern(RetainPtr<CPDF_Pattern> pattern,pdfium::span<float> values)106 void CPDF_ColorState::SetStrokePattern(RetainPtr<CPDF_Pattern> pattern,
107 pdfium::span<float> values) {
108 ColorData* data = m_Ref.GetPrivateCopy();
109 data->m_StrokeColorRef =
110 SetPattern(std::move(pattern), values, data->m_StrokeColor);
111 }
112
SetColor(RetainPtr<CPDF_ColorSpace> colorspace,std::vector<float> values,CPDF_Color & color)113 std::optional<FX_COLORREF> CPDF_ColorState::SetColor(
114 RetainPtr<CPDF_ColorSpace> colorspace,
115 std::vector<float> values,
116 CPDF_Color& color) {
117 if (colorspace) {
118 color.SetColorSpace(std::move(colorspace));
119 } else if (color.IsNull()) {
120 color.SetColorSpace(
121 CPDF_ColorSpace::GetStockCS(CPDF_ColorSpace::Family::kDeviceGray));
122 }
123 if (color.ComponentCount() > values.size()) {
124 return std::nullopt;
125 }
126
127 if (!color.IsPattern()) {
128 color.SetValueForNonPattern(std::move(values));
129 }
130 return color.GetColorRef().value_or(0xFFFFFFFF);
131 }
132
SetPattern(RetainPtr<CPDF_Pattern> pattern,pdfium::span<float> values,CPDF_Color & color)133 FX_COLORREF CPDF_ColorState::SetPattern(RetainPtr<CPDF_Pattern> pattern,
134 pdfium::span<float> values,
135 CPDF_Color& color) {
136 color.SetValueForPattern(pattern, values);
137 std::optional<FX_COLORREF> colorref = color.GetColorRef();
138 if (colorref.has_value()) {
139 return colorref.value();
140 }
141
142 CPDF_TilingPattern* tiling = pattern->AsTilingPattern();
143 return 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