Browse Source

Implemented config settings for color corrector

Bepis 7 years ago
parent
commit
3fac22817a
1 changed files with 16 additions and 7 deletions
  1. 16 7
      Plugins/ColorCorrector/ColorCorrector.cs

+ 16 - 7
Plugins/ColorCorrector/ColorCorrector.cs

@@ -12,6 +12,12 @@ namespace ColorCorrector
     public class ColorCorrector : BaseUnityPlugin
     {
         public override string Name => "Color Filter Remover";
+
+        private bool CorrectorEnabled
+        {
+            get => bool.Parse(BepInEx.Config.GetEntry("colorcorrector-enabled", "False"));
+            set => BepInEx.Config.SetEntry("colorcorrector-enabled", value.ToString());
+        }
         
         AmplifyColorEffect amplifyComponent;
         BloomAndFlares bloomComponent;
@@ -20,8 +26,10 @@ namespace ColorCorrector
         {
             if (Camera.main != null && Camera.main?.gameObject != null)
             {
-                amplifyComponent = Camera.main.gameObject.GetComponent<AmplifyColorEffect>(); ;
+                amplifyComponent = Camera.main.gameObject.GetComponent<AmplifyColorEffect>();
                 bloomComponent = Camera.main.gameObject.GetComponent<BloomAndFlares>();
+
+                SetEffects(!CorrectorEnabled);
             }
         }
 
@@ -29,16 +37,17 @@ namespace ColorCorrector
         {
             if (UnityEngine.Input.GetKeyDown(UnityEngine.KeyCode.F6))
             {
-                ToggleEffects();
+                CorrectorEnabled = !CorrectorEnabled;
+                SetEffects(!CorrectorEnabled);
             }
         }
 
-        void ToggleEffects()
+        void SetEffects(bool filterEnabled)
         {
-            amplifyComponent.enabled = !amplifyComponent.enabled;
-            bloomComponent.enabled = !bloomComponent.enabled;
-            Console.WriteLine($"Amplify Filter Enabled: {amplifyComponent.enabled}");
-            Console.WriteLine($"Bloom Filter Enabled: {bloomComponent.enabled}");
+            amplifyComponent.enabled = filterEnabled;
+            bloomComponent.enabled = filterEnabled;
+            Console.WriteLine($"Amplify Filter Enabled: {filterEnabled}");
+            Console.WriteLine($"Bloom Filter Enabled: {filterEnabled}");
         }
     }
 }