From 47131500ad17f8eebc424312112d8cbc35678c2f Mon Sep 17 00:00:00 2001
From: Will Tatam <will@netmindz.net>
Date: Sat, 21 Oct 2023 21:34:17 +0100
Subject: [PATCH] Add gamma correction config option

---
 .../usermod_v2_animartrix.h                   | 51 +++++++++++++++++--
 1 file changed, 46 insertions(+), 5 deletions(-)

diff --git a/usermods/usermod_v2_animartrix/usermod_v2_animartrix.h b/usermods/usermod_v2_animartrix/usermod_v2_animartrix.h
index 1abed2f139..4ecf4ff630 100644
--- a/usermods/usermod_v2_animartrix/usermod_v2_animartrix.h
+++ b/usermods/usermod_v2_animartrix/usermod_v2_animartrix.h
@@ -107,19 +107,29 @@ class ANIMartRIXMod:public ANIMartRIX {
 	}
 
 	CRGB applyGamma(rgb pixel) {
-		uint8_t r = (uint8_t) pixel.red;
-		uint8_t g = (uint8_t) pixel.green;
-		uint8_t b = (uint8_t) pixel.blue;
-		return CRGB(gamma8(r), gamma8(g), gamma8(b));
+		if(enableGamma) {
+			uint8_t r = (uint8_t) pixel.red;
+			uint8_t g = (uint8_t) pixel.green;
+			uint8_t b = (uint8_t) pixel.blue;
+			return CRGB(gamma8(r), gamma8(g), gamma8(b));
+		}
+		else {
+			return CRGB(pixel.red, pixel.green, pixel.blue);
+		}
 	}
 
 	void setPixelColor(int index, rgb pixel) {
 		SEGMENT.setPixelColor(index, applyGamma(pixel));
   	}
 
+	void setEnableGamma(bool state) {
+		this->enableGamma = state;
+	}
 	// Add any extra custom effects not part of the ANIMartRIX libary here
 
 	private:
+
+	bool enableGamma = true;
 };
 ANIMartRIXMod anim;
 
@@ -390,7 +400,9 @@ class AnimartrixUsermod : public Usermod {
 
   public:
 
-    AnimartrixUsermod(const char *name, bool enabled):Usermod(name, enabled) {} //WLEDMM
+    AnimartrixUsermod(const char *name, bool enabled):Usermod(name, enabled) {
+		anim.setEnableGamma(enableGamma);
+	} //WLEDMM
 	
 
     void setup() {
@@ -474,13 +486,42 @@ class AnimartrixUsermod : public Usermod {
 
       String uiDomString = F("Animartrix requires the Creative Commons Attribution License CC BY-NC 3.0");
       infoArr.add(uiDomString);
+	  infoArr.add("Gamma Correction : " + enableGamma);
 	}
 
+
+    void addToConfig(JsonObject& root) {
+      JsonObject top = root.createNestedObject(FPSTR(_name)); // usermodname
+      top[FPSTR("enabled")] = enabled;
+	  top[FPSTR("enableGamma")] = enableGamma;
+    }
+
+    bool readFromConfig(JsonObject& root) {
+      JsonObject top = root[FPSTR(_name)];
+      if (top.isNull()) {
+        DEBUG_PRINT(FPSTR(_name));
+        DEBUG_PRINTLN(F(": No config found. (Using defaults.)"));
+        return false;
+      }
+
+      enabled        = top[FPSTR("enabled")] | enabled;
+      enableGamma    = top[FPSTR("enableGamma")] | enableGamma;
+      DEBUG_PRINT(FPSTR(_name));
+      DEBUG_PRINTLN(F(" config (re)loaded."));
+
+      anim.setEnableGamma(enableGamma);
+
+      return true;
+  }
+
+
     uint16_t getId()
     {
       return USERMOD_ID_ANIMARTRIX;
     }
 
+	private:
+	bool enableGamma;
 };