blob: 57e483caf2969dc2ab623d3e54914d291d410f73 [file] [log] [blame]
// Copyright 2016 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
#include "core/fpdfapi/page/cpdf_color.h"
#include <utility>
#include "core/fpdfapi/page/cpdf_patterncs.h"
#include "third_party/base/check.h"
CPDF_Color::CPDF_Color() = default;
CPDF_Color::CPDF_Color(const CPDF_Color& that) {
*this = that;
}
CPDF_Color::~CPDF_Color() = default;
bool CPDF_Color::IsPattern() const {
return m_pCS && IsPatternInternal();
}
bool CPDF_Color::IsPatternInternal() const {
return m_pCS->GetFamily() == CPDF_ColorSpace::Family::kPattern;
}
void CPDF_Color::SetColorSpace(RetainPtr<CPDF_ColorSpace> colorspace) {
m_pCS = std::move(colorspace);
if (IsPatternInternal()) {
m_Buffer.clear();
m_pValue = std::make_unique<PatternValue>();
} else {
m_Buffer = m_pCS->CreateBufAndSetDefaultColor();
m_pValue.reset();
}
}
void CPDF_Color::SetValueForNonPattern(std::vector<float> values) {
DCHECK(!IsPatternInternal());
DCHECK(m_pCS->CountComponents() <= values.size());
m_Buffer = std::move(values);
}
void CPDF_Color::SetValueForPattern(RetainPtr<CPDF_Pattern> pattern,
pdfium::span<float> values) {
if (values.size() > kMaxPatternColorComps)
return;
if (!IsPattern()) {
SetColorSpace(
CPDF_ColorSpace::GetStockCS(CPDF_ColorSpace::Family::kPattern));
}
m_pValue->SetPattern(std::move(pattern));
m_pValue->SetComps(values);
}
CPDF_Color& CPDF_Color::operator=(const CPDF_Color& that) {
if (this == &that)
return *this;
m_Buffer = that.m_Buffer;
m_pValue =
that.m_pValue ? std::make_unique<PatternValue>(*that.m_pValue) : nullptr;
m_pCS = that.m_pCS;
return *this;
}
uint32_t CPDF_Color::CountComponents() const {
return m_pCS->CountComponents();
}
bool CPDF_Color::IsColorSpaceRGB() const {
return m_pCS ==
CPDF_ColorSpace::GetStockCS(CPDF_ColorSpace::Family::kDeviceRGB);
}
bool CPDF_Color::GetRGB(int* R, int* G, int* B) const {
float r = 0.0f;
float g = 0.0f;
float b = 0.0f;
bool result = false;
if (IsPatternInternal()) {
if (m_pValue) {
const CPDF_PatternCS* pPatternCS = m_pCS->AsPatternCS();
result = pPatternCS->GetPatternRGB(*m_pValue, &r, &g, &b);
}
} else {
if (!m_Buffer.empty())
result = m_pCS->GetRGB(m_Buffer, &r, &g, &b);
}
if (!result)
return false;
*R = static_cast<int32_t>(r * 255 + 0.5f);
*G = static_cast<int32_t>(g * 255 + 0.5f);
*B = static_cast<int32_t>(b * 255 + 0.5f);
return true;
}
RetainPtr<CPDF_Pattern> CPDF_Color::GetPattern() const {
DCHECK(IsPattern());
return m_pValue ? m_pValue->GetPattern() : nullptr;
}