From e17cccd5c9774fd69bf142d75e1fe3e240010a6b Mon Sep 17 00:00:00 2001
From: Devine Lu Linvega <aliceffekt@gmail.com>
Date: Wed, 8 Nov 2017 13:44:08 +1300
Subject: [PATCH] *

---
 TODO.md                           |  1 -
 sources/scripts/modules/filter.js | 38 +++++++++++++++++++++++++++----
 2 files changed, 33 insertions(+), 6 deletions(-)

diff --git a/TODO.md b/TODO.md
index 5157316..46f2c77 100644
--- a/TODO.md
+++ b/TODO.md
@@ -25,7 +25,6 @@
 
 ### Brush
 - Finish Port draft
-- Color picker
 
 ### Line
 - Preview support
\ No newline at end of file
diff --git a/sources/scripts/modules/filter.js b/sources/scripts/modules/filter.js
index 1a98f5e..d16cee3 100644
--- a/sources/scripts/modules/filter.js
+++ b/sources/scripts/modules/filter.js
@@ -17,8 +17,8 @@ function Filter()
 
     ronin.render.context().putImageData(originalData, 0, 0);
   });
-  
-  this.methods.balance = new Method("balance","#ff0033","Filter color balance.",function(q){
+
+  this.methods.saturation = new Method("saturation","0.5,0.5","Filter color saturation.",function(q){
 
     var color = new Color(q).floats();
 
@@ -26,11 +26,39 @@ function Filter()
     var data = originalData.data;
 
     for(var i = 0; i < data.length; i += 4) {
-      data[i]     = data[i] * (color.r + 0.5);
-      data[i + 1] = data[i + 1] * (color.g + 0.5);
-      data[i + 2] = data[i + 2] * (color.b + 0.5);
+      var r = data[i];
+      var g = data[i+1];
+      var b = data[i+2];
+      var v = color.r*r + color.g*g + color.b*b;
+      data[i] = data[i+1] = data[i+2] = v
     }
 
     ronin.render.context().putImageData(originalData, 0, 0);
   });
+
+  this.preview = function(q)
+  {
+    if(!q.methods.saturation){ return; }
+
+    ronin.preview.clear();
+
+    // var color = new Color(q).floats();
+
+    var x = q.methods.saturation.x/ronin.frame.width;
+
+    var originalData = ronin.render.context().getImageData(0, 0, ronin.frame.width*2, ronin.frame.height*2);
+    var data = originalData.data;
+
+    for(var i = 0; i < data.length; i += 4) {
+      var r = data[i];
+      var g = data[i+1];
+      var b = data[i+2];
+      var v = (r + g + b)/3;
+      data[i] = (r*x)+(v*(1-x))
+      data[i+1] = (g*x)+(v*(1-x))
+      data[i+2] = (b*x)+(v*(1-x))
+    }
+
+    ronin.preview.context().putImageData(originalData, 0, 0);
+  }
 }
\ No newline at end of file