Merge pull request #3 from IvyCafe/lemon73/support-color-type-selector

feat: support color-type selection

Author: Lemon73-Computing

README.md

@@ -21,6 +21,28 @@ Arch:
 (To Be Write)
 -->
 
+## Usage
+
+```sh
+medicolor <image-path>
+
+# e.g.
+# medicolor ./sample.png
+
+# then, select color-type in your console
+# (output file name is "./output.png")
+```
+
+or
+
+```sh
+medicolor <image-path> <output-color-type>
+# <output-color-type>: (1 = Protanomaly, 2 = Deuteranomaly, 3 = Tritanomaly)
+
+# e.g.
+# medicolor ./sample 2
+```
+
 ## Acknowledgments
 
 Super Thanks:

src/Program.cs

@@ -46,17 +46,42 @@ static int Main(string[] args)
         using SKSurface surface = SKSurface.Create(info);
         SKCanvas canvas = surface.Canvas;
 
-        float[] colorMatrixTritanomaly10 =
-        [
-            1.255528f,  -0.076749f, -0.178779f, 0,  0,
-            -0.078411f, 0.930809f,  0.147602f,  0,  0,
-            0.004733f,  0.691367f,  0.303900f,  0,  0,
-            0,          0,          0,          1,  0
-        ];
+        // Check color type
+        string? typedNumber;
+        if (args.Length > 1)
+        {
+            typedNumber = args[1];
+        }
+        else
+        {
+            Console.WriteLine("Which type would you like to convert? (1 = Protanomaly, 2 = Deuteranomaly, 3 = Tritanomaly)");
+            Console.Write("Type the number here: ");
+            typedNumber = Console.ReadLine();
+        }
+
+        // check the input is number or not
+        if (!int.TryParse(typedNumber, out int returnNumber))
+        {
+            Console.ForegroundColor = ConsoleColor.Red;
+            Console.WriteLine($"Error: It can't convert {typedNumber} into number.");
+            Console.ResetColor();
+            return 1;
+        }
+
+        // check the number is in range or not
+        float[] colorMatrix = ColorMatrix(returnNumber);
+        if (colorMatrix.Length == 0)
+        {
+            Console.ForegroundColor = ConsoleColor.Red;
+            Console.WriteLine($"Error: Invalid color type {returnNumber}.");
+            Console.ResetColor();
+            return 1;
+        }
+        // end of check color type
 
         using (SKPaint paint = new())
         {
-            paint.ColorFilter = SKColorFilter.CreateColorMatrix(colorMatrixTritanomaly10);
+            paint.ColorFilter = SKColorFilter.CreateColorMatrix(colorMatrix);
             canvas.DrawBitmap(bitmap, new SKRect(0, 0, source.Width, source.Height), paint);
         }
 
@@ -69,4 +94,42 @@ static int Main(string[] args)
         Console.ResetColor();
         return 0;
     }
+
+    /// <summary>
+    /// The method to create color-matrix
+    /// </summary>
+    /// <param name="colorType">1 = Protanomaly, 2 = Deuteranomaly, 3 = Tritanomaly</param>
+    /// <returns>Matrix type</returns>
+    static float[] ColorMatrix(int colorType)
+    {
+        switch (colorType)
+        {
+            case 1:
+                return [
+                    0.152286f,  1.052583f,  -0.204868f, 0,  0,
+                    0.114503f,  0.786281f,  0.099216f,  0,  0,
+                    -0.003882f, -0.048116f, 1.051998f,  0,  0,
+                    0,          0,          0,          1,  0
+                ];
+
+            case 2:
+                return [
+                    0.367322f,  0.860646f,  -0.227968f, 0,  0,
+                    0.280085f,  0.672501f,  0.047413f,  0,  0,
+                    -0.011820f, 0.042940f,  0.968881f,  0,  0,
+                    0,          0,          0,          1,  0
+                ];
+
+            case 3:
+                return [
+                    1.255528f,  -0.076749f, -0.178779f, 0,  0,
+                    -0.078411f, 0.930809f,  0.147602f,  0,  0,
+                    0.004733f,  0.691367f,  0.303900f,  0,  0,
+                    0,          0,          0,          1,  0
+                ];
+
+            default:
+                return [];
+        }
+    }
 }