DyeAtlas.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using System.IO;
using System.Reflection;
using System.Drawing.Imaging;
using System.Drawing.Drawing2D;

namespace DyeAtlas
{
    public partial class DyeAtlas : Form
    {
        public DyeAtlas()
        {
            InitializeComponent();
        }

        public enum GameType
        {
            Ark,
            Atlas
        }

        public Palette LoadPalette(GameType gt)
        {
            Stream s = Assembly.GetExecutingAssembly().GetManifestResourceStream($"DyeAtlas.palette_{gt.ToString().ToLowerInvariant()}.csv");

            Palette pal = Palette.LoadFromCSV(s);

            s.Dispose();

            return pal;
        }

        private Palette _palette = null;
        public Palette palette
        {
            get
            {
                if(_palette == null)
                {
                    _palette = LoadPalette((GameType)gamechoice.SelectedIndex);// should be ok to cast.. only 2 options lol
                }

                return _palette;
            }

            set
            {
                _palette = value;
            }
        }

        private void DyeAtlas_Load(object sender, EventArgs e)
        {
            Bitmap bk = new Bitmap(preview.Width, preview.Height);
            using (Graphics gfx = Graphics.FromImage(bk))
            {
                gfx.FillRectangle(new HatchBrush(HatchStyle.DarkDownwardDiagonal, Color.FromArgb(20, 20, 20), Color.FromArgb(128, 40, 128)), new Rectangle(0, 0, preview.Width, preview.Height));
            }
            preview.BackgroundImage = bk;

            resolution.SelectedIndex = 1;// 256x256

            // nothing is loaded yet, so nothing can be saved
            savePNGButton.Enabled = false;
            savePNTButton.Enabled = false;

            // load settings
            gamechoice.SelectedIndex = Properties.Settings.Default.GameType;
            mypaintings.Text = Properties.Settings.Default.MyPaintings;
        }

        private void DyeAtlas_FormClosed(object sender, FormClosedEventArgs e)
        {
            // save settings
            Properties.Settings.Default.GameType = gamechoice.SelectedIndex;
            Properties.Settings.Default.MyPaintings = mypaintings.Text;
        }

        // convenience
        public string MyPaintingsPath
        {
            get
            {
                return mypaintings.Text;
            }
        }

        private void Form1_DragEnter(object sender, DragEventArgs e)
        {
            if (!e.Data.GetDataPresent(DataFormats.FileDrop))
                return;

            e.Effect = DragDropEffects.Link;
        }

        public void SaveFile(string file)
        {
            try
            {
                Bitmap bmp = preview.Image as Bitmap;
                if (bmp == null)
                    return;

                if(Path.GetExtension(file).Equals(".pnt", StringComparison.InvariantCultureIgnoreCase))//IsFilePNT(file))
                {
                    PNTImage pnt = PNTImage.GenerateFromBitmap(palette, bmp, dithering.Checked, hsvcompare.Checked);

                    pnt.Save(file);
                }else
                {
                    bmp.Save(file);
                }

            }
            catch(Exception e)
            {
                MessageBox.Show(e.Message);
            }
        }

        public Size ExportDimensions
        {
            get
            {
                Size sz = new Size(256, 256);//default

                string[] parts = resolution.Text.Split(new char[] { 'x', 'X', ',', ' ' }, StringSplitOptions.RemoveEmptyEntries);
                if(parts.Length>=2)
                {
                    sz.Width = int.Parse(parts[0]);
                    sz.Height = int.Parse(parts[1]);
                }

                return sz;
            }
        }


        // for caching current bitmap  (not pnt)
        private string lastFilename = null;
        private DateTime lastFileWrite = new DateTime();
        private Size lastOutputSize = new Size(0, 0);
        private Bitmap lastBitmap = null;


        public void OpenFile(string file, bool autodetectResolution=true)
        {
            if (string.IsNullOrEmpty(file) || !File.Exists(file))
                return;

            try
            {
                PNTImage pnt = null;

                switch (Path.GetExtension(file).ToLowerInvariant())
                {
                    case ".pnt":
                        pnt = PNTImage.LoadPNT(palette, file);

                        if (autodetectResolution)
                        {
                            // autoselect resolution
                            if (pnt.width >= 256)
                                resolution.SelectedIndex = 1;//256x256
                            else
                                resolution.SelectedIndex = 0;//128x128   (shrug)
                        }
                        break;

                    case ".bmp":
                    case ".png":
                    case ".jpg":
                    case ".jpeg":
                    case ".gif":
                    case ".tga":
                    case ".tiff":
                        {
                            DateTime modifyTime = new FileInfo(file).LastWriteTimeUtc;

                            // do we already have a file open?
                            if (!string.IsNullOrEmpty(lastFilename))
                            {
                                // check if different/modified..  if so then invalidate our existing entry
                                if (!string.Equals(file, lastFilename, StringComparison.InvariantCultureIgnoreCase) ||
                                    !modifyTime.Equals(lastFileWrite) ||
                                    !ExportDimensions.Equals(lastOutputSize)/*hack to force reload if output dimensions change*/)
                                {
                                    lastFilename = null;
                                    lastFileWrite = new DateTime();
                                    lastOutputSize = new Size(0, 0);
                                    lastBitmap = null;
                                }
                            }


                            // only load if we need to
                            if (lastBitmap == null)
                            {
                                Bitmap bmp = (Bitmap)Bitmap.FromFile(file);

                                // powers-of-2 automatic rescale
                                int newWidth = (int)Math.Pow(2.0, Math.Ceiling(Math.Log((double)bmp.Width) / Math.Log(2.0)));
                                int newHeight = (int)Math.Pow(2.0, Math.Ceiling(Math.Log((double)bmp.Height) / Math.Log(2.0)));

                                // clamp to allowable export dimensions (assumed power-of-2)
                                newWidth = Math.Min(ExportDimensions.Width, newWidth);
                                newHeight = Math.Min(ExportDimensions.Height, newHeight);

                                // hold onto yerr hats, its time to rescale!
                                if (newWidth != bmp.Width || newHeight != bmp.Height)
                                {
                                    Bitmap newBmp = new Bitmap(newWidth, newHeight, PixelFormat.Format32bppArgb);
                                    newBmp.SetResolution(bmp.HorizontalResolution, bmp.VerticalResolution);

                                    using (Graphics gfx = Graphics.FromImage(newBmp))
                                    {
                                        gfx.CompositingMode = CompositingMode.SourceCopy;
                                        gfx.InterpolationMode = InterpolationMode.NearestNeighbor;
                                        gfx.PixelOffsetMode = PixelOffsetMode.HighQuality;

                                        using (ImageAttributes wrapMode = new ImageAttributes())
                                        {
                                            wrapMode.SetWrapMode(WrapMode.TileFlipXY);
                                            gfx.DrawImage(bmp, new Rectangle(0, 0, newWidth, newHeight), 0, 0, bmp.Width, bmp.Height, GraphicsUnit.Pixel, wrapMode);
                                        }
                                    }

                                    bmp.Dispose();
                                    bmp = newBmp;
                                }


                                lastFilename = file;
                                lastFileWrite = modifyTime;
                                lastOutputSize = bmp.Size;
                                lastBitmap = bmp;
                            }

                            pnt = PNTImage.GenerateFromBitmap(palette, lastBitmap, dithering.Checked, hsvcompare.Checked);
                        }
                        break;
                }

                if(pnt == null)
                {
                    // should we scold idiots? for dragging bad files
                    //Messagebo
                    return;
                }

                currentfile.Text = file;

                if(autodetectResolution)
                    resolution.Text = $"{pnt.width}x{pnt.height}";

                preview.Image = pnt.GenerateBitmap();


                // we got stuff, we could save
                savePNTButton.Enabled = true;
                savePNGButton.Enabled = true;
            }
            catch(Exception e)
            {
                MessageBox.Show(e.Message);
            }
            finally
            {

            }

        }

        private void Form1_DragDrop(object sender, DragEventArgs e)
        {
            if (!e.Data.GetDataPresent(DataFormats.FileDrop))
                return;

            string[] files = e.Data.GetData(DataFormats.FileDrop) as string[];

            OpenFile(files[0]);
        }

        private void openButton_Click(object sender, EventArgs e)
        {
            OpenFileDialog ofd = new OpenFileDialog();

            if (Directory.Exists(MyPaintingsPath))
                ofd.InitialDirectory = MyPaintingsPath;

            ofd.Filter = "Compatible Files (*.pnt;*.png;*.bmp;*.jpg)|*.pnt;*.png;*.bmp;*.jpg|Paint Files (*.pnt)|*.pnt|Image Files (*.png;*.bmp;*.jpg)|*.png;*.bmp;*.jpg|All Files (*.*)|*.*";

            if (ofd.ShowDialog() != DialogResult.OK)
                return;

            OpenFile(ofd.FileName);
        }

        private void dithering_CheckedChanged(object sender, EventArgs e)
        {
            Reprocess();
        }


        HSVSlidersPopup hsvPopup = null;

        private void hsvcompare_CheckedChanged(object sender, EventArgs e)
        {
            if(hsvcompare.Checked)
            {
                // if enabling, provide popup
                hsvPopup = new HSVSlidersPopup(this);

                // manual spawn location for dialog.. right next to button
                hsvPopup.Location = PointToScreen(hsvcompare.Location + new Size(hsvcompare.Width, 0));


                // non-modal;  let it live
                hsvPopup.Show(this);

            }
            else
            {
                // if destroying, should kill popup
                if (hsvPopup != null)
                {
                    hsvPopup.Close();
                    hsvPopup = null;
                }
            }

            Reprocess();
        }

        private void savePNTButton_Click(object sender, EventArgs e)
        {
            // dont do popup if image is blank; prolly button should be disabled
            if (preview.Image == null)
                return;

            SaveFileDialog sfd = new SaveFileDialog();

            if (Directory.Exists(MyPaintingsPath))
                sfd.InitialDirectory = MyPaintingsPath;

            sfd.Filter = "Paint File (*.pnt)|*.pnt|All Files (*.*)|*.*";


            // default some save filename
            sfd.FileName = Path.GetFileNameWithoutExtension(currentfile.Text) + ".pnt";

            if (sfd.ShowDialog() != DialogResult.OK)
                return;

            SaveFile(sfd.FileName);
        }

        private void savePNGButton_Click(object sender, EventArgs e)
        {
            // dont do popup if image is blank; prolly button should be disabled
            if (preview.Image == null)
                return;

            SaveFileDialog sfd = new SaveFileDialog();

            if (Directory.Exists(MyPaintingsPath))
                sfd.InitialDirectory = MyPaintingsPath;

            sfd.RestoreDirectory = true;// if we are exporting image to desktop or whatever then we should not mess with open file
            sfd.Filter = "PNG File (*.png)|*.png|Image Files (*.png;*.bmp;*.jpg)|*.png;*.bmp;*.jpg|All Files (*.*)|*.*";

            // default some save filename
            sfd.FileName = Path.GetFileNameWithoutExtension(currentfile.Text) + ".png";

            if (sfd.ShowDialog() != DialogResult.OK)
                return;

            SaveFile(sfd.FileName);
        }

        private void resolution_TextChanged(object sender, EventArgs e)
        {
            Reprocess();
        }

        private void mypaintingsbrowse_Click(object sender, EventArgs e)
        {
            FolderBrowserDialog fld = new FolderBrowserDialog();

            fld.SelectedPath = mypaintings.Text;
            fld.Description = @"This is usually like:   [GAME INSTALL]\ShooterGame\Saved\MyPaintings";

            if (fld.ShowDialog() != DialogResult.OK)
                return;

            mypaintings.Text = fld.SelectedPath;
        }

        private void openfolder_Click(object sender, EventArgs e)
        {
            if (string.IsNullOrEmpty(mypaintings.Text) || !Directory.Exists(mypaintings.Text))
                return;

            System.Diagnostics.Process.Start("explorer", mypaintings.Text);
        }

        private void gamechoice_SelectedIndexChanged(object sender, EventArgs e)
        {
            // invalidate palette
            palette = null;

            Reprocess();
        }

        public void Reprocess()
        {
            // reload current file
            OpenFile(currentfile.Text, false);
        }


        #region RGB<->HSV
        // https://stackoverflow.com/questions/359612/how-to-convert-rgb-color-to-hsv
        public static class ColorConversion
        {
            public static void ColorToHSV(Color color, out double hue, out double saturation, out double value)
            {
                int max = Math.Max(color.R, Math.Max(color.G, color.B));
                int min = Math.Min(color.R, Math.Min(color.G, color.B));

                hue = color.GetHue();
                saturation = (max == 0) ? 0 : 1d - (1d * min / max);
                value = max / 255d;
            }

            public static Color ColorFromHSV(double hue, double saturation, double value)
            {
                int hi = Convert.ToInt32(Math.Floor(hue / 60)) % 6;
                double f = hue / 60 - Math.Floor(hue / 60);

                value = value * 255;
                int v = Convert.ToInt32(value);
                int p = Convert.ToInt32(value * (1 - saturation));
                int q = Convert.ToInt32(value * (1 - f * saturation));
                int t = Convert.ToInt32(value * (1 - (1 - f) * saturation));

                if (hi == 0)
                    return Color.FromArgb(255, v, t, p);
                else if (hi == 1)
                    return Color.FromArgb(255, q, v, p);
                else if (hi == 2)
                    return Color.FromArgb(255, p, v, t);
                else if (hi == 3)
                    return Color.FromArgb(255, p, q, v);
                else if (hi == 4)
                    return Color.FromArgb(255, t, p, v);
                else
                    return Color.FromArgb(255, v, p, q);
            }
        }
        #endregion

        #region AngleDiff
        public static class AngleDiff
        {
            //https://stackoverflow.com/questions/7570808/how-do-i-calculate-the-difference-of-two-angle-measures
            /**
            * Shortest distance (angular) between two angles.
            * It will be in range [0, 180].
            */
            public static int distance(int alpha, int beta)
            {
                int phi = Math.Abs(beta - alpha) % 360;       // This is either the distance or 360 - distance
                int distance = phi > 180 ? 360 - phi : phi;
                return distance;
            }

            public static double distance(double alpha, double beta)
            {
                return (double)distance((int)Math.Round(alpha), (int)Math.Round(beta));
            }
        }
        #endregion

        private void batchToImageButton_Click(object sender, EventArgs e)
        {
            FolderBrowserDialog folderIn = new FolderBrowserDialog();

            folderIn.Description = "Input: select a folder containing .PNT files";
            if (Directory.Exists(MyPaintingsPath))
                folderIn.SelectedPath = MyPaintingsPath;
            folderIn.ShowNewFolderButton = false;

            if (folderIn.ShowDialog() != DialogResult.OK)
                return;

            if (!Directory.Exists(folderIn.SelectedPath))
                return;


            List<string> inputFiles = new List<string>();
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.pnt"));

            if(inputFiles.Count == 0)
            {
                MessageBox.Show("No .PNT files found in this directory!");
                return;
            }



            FolderBrowserDialog folderOut = new FolderBrowserDialog();

            folderOut.Description = "Output: select a folder to store images";
            folderOut.ShowNewFolderButton = true;

            if (folderOut.ShowDialog() != DialogResult.OK)
                return;

            try
            {
                Cursor.Current = Cursors.WaitCursor;
                Enabled = false;

                foreach (string inFile in inputFiles)
                {
                    string outFile = Path.Combine(folderOut.SelectedPath, Path.GetFileNameWithoutExtension(inFile) + ".png");

                    OpenFile(inFile);
                    Application.DoEvents();
                    SaveFile(outFile);
                }
            }
            finally
            {
                Cursor.Current = Cursors.Default;
                Enabled = true;
            }
        }

        private void batchToPNTButton_Click(object sender, EventArgs e)
        {
            FolderBrowserDialog folderIn = new FolderBrowserDialog();

            folderIn.Description = "Input: select a folder containing image files";
            folderIn.ShowNewFolderButton = false;

            if (folderIn.ShowDialog() != DialogResult.OK)
                return;

            if (!Directory.Exists(folderIn.SelectedPath))
                return;


            List<string> inputFiles = new List<string>();
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.jpg"));
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.jpeg"));
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.bmp"));
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.png"));
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.gif"));
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.tga"));
            inputFiles.AddRange(Directory.GetFiles(folderIn.SelectedPath, "*.tiff"));

            if (inputFiles.Count == 0)
            {
                MessageBox.Show("No image files found in this directory!");
                return;
            }



            FolderBrowserDialog folderOut = new FolderBrowserDialog();

            folderOut.Description = "Output: select a folder to store .PNT files";
            if (Directory.Exists(MyPaintingsPath))
                folderOut.SelectedPath = MyPaintingsPath;
            folderOut.ShowNewFolderButton = true;

            if (folderOut.ShowDialog() != DialogResult.OK)
                return;

            try
            {
                Cursor.Current = Cursors.WaitCursor;
                Enabled = false;

                foreach (string inFile in inputFiles)
                {
                    string outFile = Path.Combine(folderOut.SelectedPath, Path.GetFileNameWithoutExtension(inFile) + ".pnt");

                    OpenFile(inFile);
                    Application.DoEvents();
                    SaveFile(outFile);
                }
            }
            finally
            {
                Cursor.Current = Cursors.Default;
                Enabled = true;
            }
        }
    }
}