UVtools/UVtools.GUI/Controls/Tools/CtrlToolLayerImport.cs

/*
 *                     GNU AFFERO GENERAL PUBLIC LICENSE
 *                       Version 3, 19 November 2007
 *  Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
 *  Everyone is permitted to copy and distribute verbatim copies
 *  of this license document, but changing it is not allowed.
 */

using System;
using System.Collections.Concurrent;
using System.Diagnostics;
using System.Drawing;
using System.Windows.Forms;
using UVtools.Core.Objects;
using UVtools.Core.Operations;

namespace UVtools.GUI.Controls.Tools
{
    public partial class CtrlToolLayerImport : CtrlToolWindowContent
    {
        public OperationLayerImport Operation { get; }

        public CtrlToolLayerImport()
        {
            InitializeComponent();
            Operation = new OperationLayerImport(Program.FrmMain.ActualLayerImage.Size);
            SetOperation(Operation);

            nmInsertAfterLayer.Maximum = Program.SlicerFile.LayerCount-1;

            nmInsertAfterLayer.Value = Program.FrmMain.ActualLayer;
            nmInsertAfterLayer_ValueChanged(nmInsertAfterLayer, EventArgs.Empty);
        }

        public override bool UpdateOperation()
        {
            Operation.InsertAfterLayerIndex = (uint)nmInsertAfterLayer.Value;
            Operation.ReplaceStartLayer = cbReplaceStartLayer.Checked;
            Operation.ReplaceSubsequentLayers = cbReplaceSubsequentLayers.Checked;
            Operation.DiscardRemainingLayers = cbDiscardRemainingLayers.Checked;
            Operation.MergeImages = cbMergeImages.Checked;
            return true;
        }


        public override bool ValidateForm()
        {
            UpdateOperation();
            var message = Operation.Validate();
            if (message is null) return true;


            message.Content += "\nDo you want to remove all invalid files from list?";
            if (MessageBoxError(message.ToString(), MessageBoxButtons.YesNo) == DialogResult.Yes)
            {
                ConcurrentBag<StringTag> result = (ConcurrentBag<StringTag>)message.Tag;
                foreach (var file in result)
                {
                    Operation.Files.Remove(file);
                }
                UpdateListBox();
            }

            return false;
        }

        private void nmInsertAfterLayer_ValueChanged(object sender, EventArgs e)
        {
            lbHeight.Text = $"({Program.SlicerFile.GetHeightFromLayer((uint) nmInsertAfterLayer.Value)}mm)";
            UpdateResultText();
        }

        private void EventClick(object sender, EventArgs e)
        {
            if (ReferenceEquals(sender, cbReplaceStartLayer) ||
                ReferenceEquals(sender, cbReplaceSubsequentLayers) ||
                ReferenceEquals(sender, cbDiscardRemainingLayers))
            {
                if (ReferenceEquals(sender, cbReplaceSubsequentLayers))
                {
                    cbDiscardRemainingLayers.Enabled = cbDiscardRemainingLayers.Enabled = cbReplaceSubsequentLayers.Checked;
                    cbMergeImages.Enabled = cbReplaceSubsequentLayers.Checked;
                    if (!cbReplaceSubsequentLayers.Checked)
                    {
                        cbDiscardRemainingLayers.Checked = false;
                        cbMergeImages.Checked = false;
                    }
                }

                UpdateResultText();
                return;
            }

            if (ReferenceEquals(sender, btnAdd))
            {
                using (var fileOpen = new OpenFileDialog
                {
                    Multiselect = true,
                    CheckFileExists = true,
                    Filter = "Image Files(*.PNG;*.BMP;*.JPEG;*.JPG;*.GIF)|*.PNG;*.BMP;*.JPEG;*.JPG;*.GIF"
                })
                {
                    if (fileOpen.ShowDialog() != DialogResult.OK) return;

                    foreach (var filename in fileOpen.FileNames)
                    {
                        Operation.AddFile(filename);
                    }

                    if (cbAutoSort.Checked)
                    {
                        Operation.Sort();
                    }

                    UpdateListBox();
                }

                return;
            }

            if (ReferenceEquals(sender, btnRemove))
            {
                foreach (StringTag selectedItem in lbFiles.SelectedItems)
                {
                    Operation.Files.Remove(selectedItem);
                }
                UpdateListBox();
                
                return;
            }

            if (ReferenceEquals(sender, btnSort))
            {
                Operation.Sort();
                UpdateListBox();
                return;
            }

            if (ReferenceEquals(sender, btnClear))
            {
                Operation.Files.Clear();
                UpdateListBox();
                return;
            }
        }

        private void lbFiles_SelectedIndexChanged(object sender, EventArgs e)
        {
            if (btnRemove.Enabled = lbFiles.SelectedIndex >= 0)
            {
                var file = lbFiles.SelectedItem as StringTag;
                pbSelectedImage.Image = new Bitmap(file.TagString);
            }
            else
            {
                pbSelectedImage.Image = null;
            }
        }

        private void lbFiles_KeyUp(object sender, KeyEventArgs e)
        {
            if (e.KeyCode == Keys.Escape)
            {
                lbFiles.SelectedIndices.Clear();
                e.Handled = true;
                return;
            }

            if (e.KeyCode == Keys.Delete)
            {
                btnRemove.PerformClick();
                e.Handled = true;
                return;
            }

            if (e.KeyCode == Keys.A && (ModifierKeys & Keys.Control) != 0)
            {
                lbFiles.BeginUpdate();
                for (int i = 0; i < lbFiles.Items.Count; i++)
                    lbFiles.SetSelected(i, true);
                lbFiles.EndUpdate();
                e.Handled = true;
                return;
            }
        }

        public void UpdateListBox()
        {
            lbFiles.Items.Clear();

            foreach (var file in Operation.Files)
            {
                lbFiles.Items.Add(file);
            }

            ButtonOkEnabled = btnRemove.Enabled = btnSort.Enabled = btnClear.Enabled = Operation.Files.Count > 0;
            lbCount.Text = $"Layers: {Operation.Files.Count}";
            UpdateResultText();
        }

        private void UpdateResultText()
        {
            if (Operation.Files.Count > 0)
            {
                UpdateOperation();
                uint modelTotalLayers = Operation.CalculateTotalLayers(Program.SlicerFile.LayerCount);
                string textFactor = "grow";
                if (modelTotalLayers < Program.SlicerFile.LayerCount)
                {
                    textFactor = "shrink";
                }
                else if (modelTotalLayers == Program.SlicerFile.LayerCount)
                {
                    textFactor = "keep";
                }
                lbResult.Text = 
                    $"{Operation.Files.Count} layers will be imported into model starting from layer {nmInsertAfterLayer.Value} {lbHeight.Text}.\n" +
                    $"Model will {textFactor} from layers {Program.SlicerFile.LayerCount} ({Program.SlicerFile.TotalHeight}mm) to {modelTotalLayers} ({Program.SlicerFile.GetHeightFromLayer(modelTotalLayers, false)}mm)";
            }
            else
            {
                lbResult.Text = string.Empty;
            }
        }

        private void lbFiles_MouseDoubleClick(object sender, MouseEventArgs e)
        {
            if (lbFiles.SelectedItem is StringTag file)
            {
                try
                {
                    using (Process.Start(file.TagString))
                    { }
                }
                catch (Exception exception)
                {
                    Debug.WriteLine(exception);
                }
                
                return;
            }
        }
    }
}