1 files changed, 69 insertions, 5 deletions
diff --git a/core/fpdfapi/fpdf_page/cpdf_colorstatedata.cpp b/core/fpdfapi/fpdf_page/cpdf_colorstatedata.cpp
index 9dbcdec267..3a771ffaf0 100644
--- a/core/fpdfapi/fpdf_page/cpdf_colorstatedata.cpp
+++ b/core/fpdfapi/fpdf_page/cpdf_colorstatedata.cpp
@@ -6,15 +6,79 @@
 
 #include "core/fpdfapi/fpdf_page/cpdf_colorstatedata.h"
 
-CPDF_ColorStateData::CPDF_ColorStateData(const CPDF_ColorStateData& src) {
+#include "core/fpdfapi/fpdf_page/cpdf_tilingpattern.h"
+#include "core/fxge/include/fx_dib.h"
+
+CPDF_ColorStateData::CPDF_ColorStateData(const CPDF_ColorStateData& src)
+    : m_FillRGB(src.m_FillRGB), m_StrokeRGB(src.m_StrokeRGB) {
   m_FillColor.Copy(&src.m_FillColor);
-  m_FillRGB = src.m_FillRGB;
   m_StrokeColor.Copy(&src.m_StrokeColor);
-  m_StrokeRGB = src.m_StrokeRGB;
 }
 
-void CPDF_ColorStateData::Default() {
-  m_FillRGB = m_StrokeRGB = 0;
+void CPDF_ColorStateData::SetDefault() {
+  m_FillRGB = 0;
+  m_StrokeRGB = 0;
   m_FillColor.SetColorSpace(CPDF_ColorSpace::GetStockCS(PDFCS_DEVICEGRAY));
   m_StrokeColor.SetColorSpace(CPDF_ColorSpace::GetStockCS(PDFCS_DEVICEGRAY));
 }
+
+void CPDF_ColorStateData::SetFillColor(CPDF_ColorSpace* pCS,
+                                       FX_FLOAT* pValue,
+                                       uint32_t nValues) {
+  SetColor(m_FillColor, m_FillRGB, pCS, pValue, nValues);
+}
+
+void CPDF_ColorStateData::SetStrokeColor(CPDF_ColorSpace* pCS,
+                                         FX_FLOAT* pValue,
+                                         uint32_t nValues) {
+  SetColor(m_StrokeColor, m_StrokeRGB, pCS, pValue, nValues);
+}
+
+void CPDF_ColorStateData::SetFillPattern(CPDF_Pattern* pPattern,
+                                         FX_FLOAT* pValue,
+                                         uint32_t nValues) {
+  m_FillColor.SetValue(pPattern, pValue, nValues);
+  int R, G, B;
+  FX_BOOL ret = m_FillColor.GetRGB(R, G, B);
+  if (CPDF_TilingPattern* pTilingPattern = pPattern->AsTilingPattern()) {
+    if (!ret && pTilingPattern->colored()) {
+      m_FillRGB = 0x00BFBFBF;
+      return;
+    }
+  }
+  m_FillRGB = ret ? FXSYS_RGB(R, G, B) : (uint32_t)-1;
+}
+
+void CPDF_ColorStateData::SetStrokePattern(CPDF_Pattern* pPattern,
+                                           FX_FLOAT* pValue,
+                                           uint32_t nValues) {
+  m_StrokeColor.SetValue(pPattern, pValue, nValues);
+  int R, G, B;
+  FX_BOOL ret = m_StrokeColor.GetRGB(R, G, B);
+  if (CPDF_TilingPattern* pTilingPattern = pPattern->AsTilingPattern()) {
+    if (!ret && pTilingPattern->colored()) {
+      m_StrokeRGB = 0x00BFBFBF;
+      return;
+    }
+  }
+  m_StrokeRGB =
+      m_StrokeColor.GetRGB(R, G, B) ? FXSYS_RGB(R, G, B) : (uint32_t)-1;
+}
+
+void CPDF_ColorStateData::SetColor(CPDF_Color& color,
+                                   uint32_t& rgb,
+                                   CPDF_ColorSpace* pCS,
+                                   FX_FLOAT* pValue,
+                                   uint32_t nValues) {
+  if (pCS) {
+    color.SetColorSpace(pCS);
+  } else if (color.IsNull()) {
+    color.SetColorSpace(CPDF_ColorSpace::GetStockCS(PDFCS_DEVICEGRAY));
+  }
+  if (color.GetColorSpace()->CountComponents() > nValues)
+    return;
+
+  color.SetValue(pValue);
+  int R, G, B;
+  rgb = color.GetRGB(R, G, B) ? FXSYS_RGB(R, G, B) : (uint32_t)-1;
+}