1 files changed, 24 insertions, 2 deletions
diff --git a/source/blender/compositor/intern/COM_CompositorContext.h b/source/blender/compositor/intern/COM_CompositorContext.h
index 2f6abf39985..2f5e8c0648d 100644
--- a/source/blender/compositor/intern/COM_CompositorContext.h
+++ b/source/blender/compositor/intern/COM_CompositorContext.h
@@ -27,6 +27,7 @@
 #include "BKE_text.h"
 #include <string>
 #include "DNA_node_types.h"
+#include "DNA_color_types.h"
 #include "BLI_rect.h"
 #include "DNA_scene_types.h"
 #include "COM_defines.h"
@@ -79,6 +80,9 @@ private:
 	 */
 	bool m_fastCalculation;
 
+	/* @brief color management settings */
+	const ColorManagedViewSettings *m_viewSettings;
+	const ColorManagedDisplaySettings *m_displaySettings;
 public:
 	/**
 	 * @brief constructor initializes the context with default values.
@@ -126,6 +130,26 @@ public:
 	const RenderData *getRenderData() const { return this->m_rd; }
 
 	/**
+	 * @brief set view settings of color color management
+	 */
+	void setViewSettings(const ColorManagedViewSettings *viewSettings) { this->m_viewSettings = viewSettings; }
+
+	/**
+	 * @brief get view settings of color color management
+	 */
+	const ColorManagedViewSettings *getViewSettings() const { return this->m_viewSettings; }
+
+	/**
+	 * @brief set display settings of color color management
+	 */
+	void setDisplaySettings(const ColorManagedDisplaySettings *displaySettings) { this->m_displaySettings = displaySettings; }
+
+	/**
+	 * @brief get display settings of color color management
+	 */
+	const ColorManagedDisplaySettings *getDisplaySettings() const { return this->m_displaySettings; }
+
+	/**
 	 * @brief set the quality
 	 */
 	void setQuality(CompositorQuality quality) { this->m_quality = quality; }
@@ -152,8 +176,6 @@ public:
 	
 	int getChunksize() { return this->getbNodeTree()->chunksize; }
 	
-	const int isColorManaged() const;
-	
 	void setFastCalculation(bool fastCalculation) {this->m_fastCalculation = fastCalculation;}
 	bool isFastCalculation() {return this->m_fastCalculation;}
 };