Merge pull request #45 from leicestersquare/color-corrrection

Added color correction nodes to composite

Author: leicestersquare

rct-graphics-helper/__init__.py

@@ -24,7 +24,7 @@
     "name": "RCT Graphics Helper",
     "description": "Render tool to replicate RCT graphics",
     "author": "Olivier Wervers",
-    "version": (0, 4, 6),
+    "version": (0, 6, 0),
     "blender": (2, 79, 0),
     "location": "Render",
     "support": "COMMUNITY",

rct-graphics-helper/builders/compositor_builder.py

@@ -48,11 +48,28 @@ def build(self, context):
         self.exit_branch_point()
 
         self.link(aa_mask, 0, combine_node, 3)
+        
+        #CurveRGB node
+        CurveRGB_composite_node = self.create_node("CompositorNodeCurveRGB")
+        curve_c = CurveRGB_composite_node.mapping.curves[3]
+        curve_c.points.new(.5,.5)
+        curve_c.points.new(.3,.2)
+        
+        self.link(mixed_anti_aliased, 0, CurveRGB_composite_node, 1)
+        
+        self.next_column()
+        
+        #HSV node
+        HueSat_composite_node = self.create_node("CompositorNodeHueSat")
+        HueSat_composite_node.inputs[2].default_value = 0.9
+        self.link(CurveRGB_composite_node, 0, HueSat_composite_node, 0)
+        
+        self.next_column()
 
         # Main output node
         output_composite_node = self.create_node("CompositorNodeComposite")
 
-        self.link(mixed_anti_aliased, 0, output_composite_node, 0)
+        self.link(HueSat_composite_node, 0, output_composite_node, 0)
         self.link(aa_mask, 0, output_composite_node, 1)
 
         # Metadata output node