From 056e58a29711a6557da5ad8796b1b47459058c44 Mon Sep 17 00:00:00 2001
From: Nadina Siddiqui <Nadina.Siddiqui@digitecgalaxus.ch>
Date: Sun, 29 Nov 2020 20:01:27 +0100
Subject: [PATCH] persist instructions

---
 LegoArt/MainWindow.xaml                      |  4 ++
 LegoArt/MainWindow.xaml.cs                   | 18 +++++--
 LegoArtTool/BuildingInstructionService.cs    | 47 +++++++++++++++++
 LegoArtTool/LegoArtImageGenerationService.cs | 53 +++++---------------
 4 files changed, 78 insertions(+), 44 deletions(-)
 create mode 100644 LegoArtTool/BuildingInstructionService.cs

diff --git a/LegoArt/MainWindow.xaml b/LegoArt/MainWindow.xaml
index a35584c..e8fbe00 100644
--- a/LegoArt/MainWindow.xaml
+++ b/LegoArt/MainWindow.xaml
@@ -22,6 +22,10 @@
             <TextBox Width="300" x:Name="tbImagePath" />
             <Button Width="50" Name="btnImageChooser" Content="..." Click="btnImageChooser_Click"/>
         </StackPanel>
+        
+        <StackPanel Grid.Row="0" Grid.Column="1" Orientation="Horizontal" Height="25" VerticalAlignment="Top" Margin="5, 0 , 0, 0">
+            <Button Width="200" Name="btnInstructionsPersister" Content="Save Instructions" Click="btnInstructionsPersister_Click"/>
+        </StackPanel>
 
         <StackPanel Grid.Row="2" Orientation="Vertical" Name="parentStackPanel" VerticalAlignment="Top">
         </StackPanel>
diff --git a/LegoArt/MainWindow.xaml.cs b/LegoArt/MainWindow.xaml.cs
index c2a89cf..dd45300 100644
--- a/LegoArt/MainWindow.xaml.cs
+++ b/LegoArt/MainWindow.xaml.cs
@@ -1,11 +1,12 @@
-using LegoArtTool;
+using System.Drawing;
+using LegoArtTool;
 using Microsoft.Win32;
 using System.Linq;
 using System.Windows;
 using System.Windows.Controls;
 using System.Windows.Input;
 using System.Windows.Media;
-using System.Windows.Shapes;
+using Rectangle = System.Windows.Shapes.Rectangle;
 
 namespace LegoArt
 {
@@ -17,7 +18,7 @@ namespace LegoArt
         private readonly LegoArtColorService legoArtColorService;
         private readonly BitmapHelperService bitmapHelperService;
         private readonly ImageHelperService imageHelperService;
-        private LegoArtImageGenerationService legoArtImageGenerationService;
+        private readonly LegoArtImageGenerationService legoArtImageGenerationService;
 
         public MainWindow()
         {
@@ -27,6 +28,8 @@ namespace LegoArt
             bitmapHelperService = new BitmapHelperService();
             imageHelperService = new ImageHelperService();
             legoArtImageGenerationService = new LegoArtImageGenerationService();
+
+            btnInstructionsPersister.Visibility = Visibility.Hidden;
         }
 
         private void btnImageChooser_Click(object sender, RoutedEventArgs e)
@@ -50,6 +53,13 @@ namespace LegoArt
             }
         }
 
+        private void btnInstructionsPersister_Click(object sender, RoutedEventArgs e)
+        {
+            var d = new DataObject(DataFormats.Bitmap, scaledImage.Source, true);
+            var bitmap = d.GetData(typeof(Bitmap)) as Bitmap;
+            BuildingInstructionService.PersistBuildingInstructions(bitmap, tbImagePath.Text);
+        }
+
         private void LoadImage(string path)
         {
             var bitmap = new System.Drawing.Bitmap(path);
@@ -64,6 +74,8 @@ namespace LegoArt
                 var convertedBitmap = bitmapHelperService.ConvertToPixelMatrix(sourceBitmap, 20);
                 scaledImage.Source = imageHelperService.LoadToImage(convertedBitmap);
 
+                btnInstructionsPersister.Visibility = Visibility.Visible;
+                
                 parentStackPanel.Children.Clear();
 
                 AddLine(System.Drawing.Color.White, "Have", "Needed", "RGB", "#", "Difference", System.Drawing.Color.White);
diff --git a/LegoArtTool/BuildingInstructionService.cs b/LegoArtTool/BuildingInstructionService.cs
new file mode 100644
index 0000000..e02e561
--- /dev/null
+++ b/LegoArtTool/BuildingInstructionService.cs
@@ -0,0 +1,47 @@
+using System.Drawing;
+using System.Drawing.Imaging;
+using System.IO;
+
+namespace LegoArtTool
+{
+    public class BuildingInstructionService
+    {
+        public static void PersistBuildingInstructions(Bitmap bitmap, string path)
+        {
+            var segmentSize = bitmap.Width / 3;
+            var outPutDirectory = CreateOutPutDirectory(path);
+
+            var segmentNumber = 1;
+            for (var h = 0; h < 3; h++)
+            {
+                for (var w = 0; w < 3; w++)
+                {
+                    var rectangle = new Rectangle(w * segmentSize, h * segmentSize, segmentSize, segmentSize);
+                    var segment = bitmap.Clone(rectangle, bitmap.PixelFormat);
+                    segment.Save($"{outPutDirectory}/{segmentNumber++}.png", ImageFormat.Png);
+                }
+            }
+        }
+
+        private static string CreateOutPutDirectory(string path)
+        {
+            var increment = 0;
+            
+            while (Directory.Exists(GetOutputDirectoryPath(increment, path)))
+            {
+                increment++;
+            }
+
+            var outputDirectoryPath = GetOutputDirectoryPath(increment, path);
+            Directory.CreateDirectory(outputDirectoryPath);
+            return outputDirectoryPath;
+        }
+
+        private static string GetOutputDirectoryPath(int increment, string path)
+        {
+            var parent = Directory.GetParent(path);
+            var filename = Path.GetFileNameWithoutExtension(path);
+            return increment == 0 ? $"{parent}/legoArtInstructions" : $"{parent}/legoArtInstructions_{filename} ({increment})";
+        }
+    }
+}
\ No newline at end of file
diff --git a/LegoArtTool/LegoArtImageGenerationService.cs b/LegoArtTool/LegoArtImageGenerationService.cs
index ea9116c..15b605b 100644
--- a/LegoArtTool/LegoArtImageGenerationService.cs
+++ b/LegoArtTool/LegoArtImageGenerationService.cs
@@ -35,19 +35,19 @@ namespace LegoArtTool
             var reducedBitmap = new Bitmap(bitmap.Width, bitmap.Height);
             
             var remainingColors = legoArtColorInfos.ToDictionary(k => k.LegoNumber, v => v.HaveCount);
-
-
+            
             while (pixels.Count(p => !p.IsAssigned) > 0)
             {
                 var minOffset = pixels.Where(p => !p.IsAssigned).SelectMany(p => p.LegoArtColorOffsets.Where(o => o.IsAvailable)
                         .Select(o => o.ColorOffset))
                         .Min(x => x);
 
-                var pixel = pixels.Where(p => !p.IsAssigned).First(p => p.LegoArtColorOffsets.Any(o => o.ColorOffset == minOffset));
+                var pixel = pixels.Where(p => !p.IsAssigned).First(p => p.LegoArtColorOffsets.Where(o => o.IsAvailable)
+                    .Any(o => o.ColorOffset == minOffset));
                 var legoArtColorInfo = pixel.LegoArtColorOffsets.First(o => o.ColorOffset == minOffset).LegoArtColorInfo;
+               
                 reducedBitmap.SetPixel(pixel.X, pixel.Y, legoArtColorInfo.Color);
-                
-                
+
                 remainingColors[legoArtColorInfo.LegoNumber]--;
                 pixel.IsAssigned = true;
 
@@ -62,50 +62,21 @@ namespace LegoArtTool
             return reducedBitmap;
         }
 
-        private Pixel MapPixel(
-            Color inputColor,
-            int xPosition,
-            int yPosition)
+        private Pixel MapPixel(Color inputColor, int xPosition, int yPosition)
         {
-            var legoArtColorOffsets = legoArtColorInfos.Select(i => ColorOffset(inputColor, i)).ToList();
+            var legoArtColorOffsets = legoArtColorInfos.Select(i => ColorDistance(inputColor, i)).ToList();
             return new Pixel(xPosition, yPosition, legoArtColorOffsets);
         }
         
-        private static LegoArtColorOffset ColourDistance(Color inputColor, LegoArtColorInfo colorInfo)
+        private static LegoArtColorOffset ColorDistance(Color inputColor, LegoArtColorInfo colorInfo)
         {
             var targetColor = colorInfo.Color;
-            var rmean = ((long)inputColor.R + (long)targetColor.R) / 2;
-            var r = (long)inputColor.R - (long)targetColor.R;
-            var g = (long)inputColor.G - (long)targetColor.G;
-            var b = (long)inputColor.B - (long)targetColor.B;
+            var rmean = (inputColor.R + (long)targetColor.R) / 2;
+            var r = inputColor.R - (long)targetColor.R;
+            var g = inputColor.G - (long)targetColor.G;
+            var b = inputColor.B - (long)targetColor.B;
             var offset = (decimal) Math.Sqrt((((512 + rmean) * r * r) >> 8) + 4 * g * g + (((767 - rmean) * b * b) >> 8));
             return new LegoArtColorOffset((decimal) offset, colorInfo);
         }
-
-        private LegoArtColorOffset ColorOffset(Color inputColor, LegoArtColorInfo colorInfo)
-        {
-            var offset = Math.Abs(
-                       ColorNum(inputColor) - ColorNum(colorInfo.Color)) +
-                   GetHueDistance(inputColor.GetHue(), colorInfo.Color.GetHue());
-            
-            return new LegoArtColorOffset((decimal) offset, colorInfo);
-        }
-
-        private static float GetBrightness(Color c)
-        {
-            return (c.R * 0.299f + c.G * 0.587f + c.B *0.114f) / 256f;
-        }
-
-        private static float GetHueDistance(float hue1, float hue2)
-        { 
-            var d = Math.Abs(hue1 - hue2); 
-            return d > 180 ? 360 - d : d;
-            
-        }
-
-        private static float ColorNum(Color c)
-        {
-            return c.GetSaturation() + GetBrightness(c);
-        }
     }
 }
\ No newline at end of file