MainForm.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Windows.Forms;

namespace GCEd
{
	public partial class MainForm : Form
	{
		private ViewState viewState;
		private Settings settings;

		public MainForm()
		{
			InitializeComponent();
			DoubleBuffered = true;
			settings = new Settings();
			canvas.Settings = settings;
			viewState = new ViewState();
			canvas.ViewState = viewState;
			operationsList.ViewState = viewState;
			operationProperties.ViewState = viewState;
			lineEditor.ViewState = viewState;
			viewState.NewProgram();
			KeyPreview = true;
			Icon = System.Drawing.Icon.ExtractAssociatedIcon(Assembly.GetExecutingAssembly().Location);
		}

		private bool ConfirmAbandonDirty()
		{
			if (!viewState.IsDirty) return true;
			var result = MessageBox.Show(viewState.CurrentFile == null ? "Save current file?" : $"Save \"{viewState.CurrentFile}\"?", "GCEd", MessageBoxButtons.YesNoCancel, MessageBoxIcon.Question, MessageBoxDefaultButton.Button1);
			if (result == DialogResult.Cancel) return false;
			if (result == DialogResult.No) return true;
			if (viewState.CurrentFile != null)
			{
				viewState.SaveProgram(viewState.CurrentFile);
			}
			else
			{
				if (saveFileDialog.ShowDialog() != DialogResult.OK) return false;
				viewState.SaveProgram(saveFileDialog.FileName);
			}
			return true;
		}

		private void NewFile()
		{
			if (!ConfirmAbandonDirty()) return;
			viewState.SaveUndoState();
			Environment.CurrentDirectory = Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments);
			viewState.NewProgram();
			Text = "GCEd";
		}

		private void SaveFile()
		{
			if (viewState.CurrentFile == null) SaveFileAs();
			else viewState.SaveProgram(viewState.CurrentFile);
		}

		private void SaveFileAs()
		{
			if (saveFileDialog.ShowDialog() != DialogResult.OK) return;
			Environment.CurrentDirectory = Path.GetDirectoryName(saveFileDialog.FileName)!;
			viewState.SaveProgram(saveFileDialog.FileName);
			Text = "GCEd [" + saveFileDialog.FileName + "]";
		}

		private void OpenFile()
		{
			if (openFileDialog.ShowDialog() != DialogResult.OK) return;
			if (!ConfirmAbandonDirty()) return;
			viewState.SaveUndoState();
			Environment.CurrentDirectory = Path.GetDirectoryName(openFileDialog.FileName)!;
			viewState.LoadProgram(openFileDialog.FileName);
			Text = "GCEd [" + openFileDialog.FileName + "]";
		}

		private void SelectAll(bool skipRapid)
		{
			var selectedOperations = new List<GOperation>();
			foreach (var operation in viewState.Operations)
			{
				if (!operation.Line.IsVisible) continue;
				if (operation.Line.Instruction == GInstruction.G0 && skipRapid) continue;
				selectedOperations.Add(operation);
			}
			viewState.SetSelection(selectedOperations);
		}

		private void DeleteSelected()
		{
			viewState.SaveUndoState();
			viewState.DeleteOperations(viewState.SelectedOperations);
		}

		private void Undo()
		{
			viewState.Undo();
		}

		private void Redo()
		{
			viewState.Redo();
		}

		private void ConvertToAbsolute()
		{
			viewState.SaveUndoState();
			viewState.ConvertToAbsolute(viewState.SelectedOperations);
		}

		private void ConvertToRelative()
		{
			viewState.SaveUndoState();
			viewState.ConvertToRelative(viewState.SelectedOperations);
		}

		private void ConvertToOutline()
		{
			using var form = new OutlineParameters();
			if (form.ShowDialog() != DialogResult.OK) return;
			viewState.SaveUndoState();
			viewState.ConvertToOutline(viewState.SelectedOperations, form.Thickness, form.LeftFirst, form.SkipLeft, form.SkipRight, form.BothForward);
		}

		private void AddNewLine(bool before)
		{
			viewState.SaveUndoState();
			viewState.AppendNewLine(viewState.LastSelectedOperation, before);
			viewState.FocusLineEditor();
		}

		private void AddComment(bool before)
		{
			viewState.SaveUndoState();
			viewState.AppendNewLine(viewState.LastSelectedOperation, before, new GLine("; "));
			viewState.FocusLineEditor();
		}

		private void ToggleComment()
		{
			viewState.SaveUndoState();
			viewState.ToggleComment(viewState.SelectedOperations);
		}

		private void AddG0(bool before)
		{
			canvas.SuspendPanningToSelection();
			viewState.SaveUndoState();
			viewState.AppendNewLine(viewState.LastSelectedOperation, before, new GLine { Instruction = GInstruction.G0 });
			canvas.StartMouseEndMove();
			canvas.ResumePanningToSelection();
		}

		private void AddG1(bool before)
		{
			canvas.SuspendPanningToSelection();
			viewState.SaveUndoState();
			viewState.AppendNewLine(viewState.LastSelectedOperation, before, new GLine { Instruction = GInstruction.G1 });
			canvas.StartMouseEndMove();
			canvas.ResumePanningToSelection();
		}

		private void AddG2(bool before)
		{
			canvas.SuspendPanningToSelection();
			viewState.SaveUndoState();
			viewState.AppendNewLine(viewState.LastSelectedOperation, before, new GLine { Instruction = GInstruction.G2 });
			canvas.StartMouseEndMove();
			canvas.ResumePanningToSelection();
		}

		private void AddG3(bool before)
		{
			canvas.SuspendPanningToSelection();
			viewState.SaveUndoState();
			viewState.AppendNewLine(viewState.LastSelectedOperation, before, new GLine { Instruction = GInstruction.G3 });
			canvas.StartMouseEndMove();
			canvas.ResumePanningToSelection();
		}

		private void AddBackground()
		{
			using var openFileDialog = new OpenFileDialog();
			openFileDialog.Filter = "Images (*.bmp, *.jpg, *.png, *.gif)|*.bmp;*.jpg;*.jpeg;*.png;*.gif|All files|*.*";
			openFileDialog.RestoreDirectory = true;
			if (openFileDialog.ShowDialog() != DialogResult.OK) return;
			viewState.AppendNewLine(null, true, new GLine(";.background X0 Y0 I100 J100 S100 P\"" + Path.GetRelativePath(Path.GetDirectoryName(viewState.CurrentFile) ?? Environment.CurrentDirectory, openFileDialog.FileName) + "\""));
		}

		private void AddText()
		{
			using var textGenerator = new TextGenerator();
			if (textGenerator.ShowDialog() != DialogResult.OK || textGenerator.Program == null) return;
			var subProgram = textGenerator.Program;
			viewState.SaveUndoState();
			var baseOperation = viewState.LastSelectedOperation;
			if (baseOperation == null) baseOperation = viewState.Operations.LastOrDefault();
			if (baseOperation == null || !baseOperation.Absolute)
			{
				subProgram.Lines.AddFirst(new GLine { Instruction = GInstruction.G90 });
				subProgram.Lines.AddLast(new GLine { Instruction = GInstruction.G91 });
			}
			viewState.AppendProgram(viewState.LastSelectedOperation, subProgram);
		}

		private void Copy()
		{
			var code = String.Join(Environment.NewLine, viewState.SelectedOperations.Select(operation => operation.Line.ToString()));
			Clipboard.SetText(code);
		}

		private void Cut()
		{
			Copy();
			DeleteSelected();
		}

		private void Paste(bool before)
		{
			var code = Clipboard.GetText();
			if (String.IsNullOrWhiteSpace(code)) return;
			var subProgram = new GProgram();
			subProgram.Load(new StringReader(code));
			viewState.SaveUndoState();
			viewState.AppendProgram(viewState.LastSelectedOperation, subProgram, before);
		}

		public void ToggleGrid()
		{
			if (settings.ShowMinorGrid)
			{
				settings.ShowMinorGrid = false;
				settings.ShowMajorGrid = false;
				settings.ShowOriginGrid = false;
			}
			else if (!settings.ShowOriginGrid) settings.ShowOriginGrid = true;
			else if (!settings.ShowMajorGrid) settings.ShowMajorGrid = true;
			else settings.ShowMinorGrid = true;
			canvas.Invalidate();
		}

		public void ToggleSnapToGrid()
		{
			settings.SnapToGrid = !settings.SnapToGrid;
		}

		public void ToggleSnapToItems()
		{
			settings.SnapToItems = !settings.SnapToItems;
		}

		public void ToggleSnapToAxes()
		{
			settings.SnapToAxes = !settings.SnapToAxes;
		}

		public void ToggleFPS()
		{
			settings.ShowFPS = !settings.ShowFPS;
			canvas.Invalidate();
		}

		public void ToggleCoords()
		{
			settings.ShowCursorCoords = !settings.ShowCursorCoords;
			if (!settings.ShowCursorCoords) settings.ShowItemCoords = !settings.ShowItemCoords;
			canvas.Invalidate();
		}

		private void About()
		{
			using var about = new About();
			about.ShowDialog();
		}

		protected override void OnKeyDown(KeyEventArgs e)
		{
			e.Handled = true;

			var editorFocused = false;
			IContainerControl? container = this;
			while (container != null)
			{
				editorFocused = container.ActiveControl is TextBox;
				container = container.ActiveControl as IContainerControl;
			}
			if (e.KeyCode == Keys.N && e.Modifiers == Keys.Control) NewFile();
			else if (e.KeyCode == Keys.O && e.Modifiers == Keys.Control) OpenFile();
			else if (e.KeyCode == Keys.S && e.Modifiers == Keys.Control) SaveFile();
			else if (e.KeyCode == Keys.S && e.Modifiers == (Keys.Control | Keys.Shift)) SaveFileAs();
			else if (e.KeyCode == Keys.Menu) { }
			else if (editorFocused) { e.Handled = false; }
			else if (e.KeyCode == Keys.D0 && e.Modifiers == Keys.None) AddG0(false);
			else if (e.KeyCode == Keys.D0 && e.Modifiers == Keys.Shift) AddG0(true);
			else if (e.KeyCode == Keys.D1 && e.Modifiers == Keys.None) AddG1(false);
			else if (e.KeyCode == Keys.D1 && e.Modifiers == Keys.Shift) AddG1(true);
			else if (e.KeyCode == Keys.D2 && e.Modifiers == Keys.None) AddG2(false);
			else if (e.KeyCode == Keys.D2 && e.Modifiers == Keys.Shift) AddG2(true);
			else if (e.KeyCode == Keys.D3 && e.Modifiers == Keys.None) AddG3(false);
			else if (e.KeyCode == Keys.D3 && e.Modifiers == Keys.Shift) AddG3(true);
			else if (e.KeyCode == Keys.A && e.Modifiers == Keys.None) ConvertToAbsolute();
			else if (e.KeyCode == Keys.A && e.Modifiers == Keys.Shift) ConvertToRelative();
			else if (e.KeyCode == Keys.A && e.Modifiers == Keys.Control) SelectAll(false);
			else if (e.KeyCode == Keys.A && e.Modifiers == (Keys.Control | Keys.Shift)) SelectAll(true);
			else if (e.KeyCode == Keys.A && e.Modifiers == Keys.Alt) ToggleSnapToAxes();
			else if (e.KeyCode == Keys.B && e.Modifiers == Keys.Control) AddBackground();
			else if (e.KeyCode == Keys.C && e.Modifiers == Keys.None) ToggleCoords();
			else if (e.KeyCode == Keys.C && e.Modifiers == Keys.Control) Copy();
			else if (e.KeyCode == Keys.E && e.Modifiers == Keys.None) canvas.StartMouseEndMove();
			else if (e.KeyCode == Keys.E && e.Modifiers == Keys.Shift) canvas.StartMouseOffsetMove();
			else if (e.KeyCode == Keys.F && e.Modifiers == Keys.None) ToggleFPS();
			else if (e.KeyCode == Keys.G && e.Modifiers == Keys.None) ToggleGrid();
			else if (e.KeyCode == Keys.G && e.Modifiers == Keys.Alt) ToggleSnapToGrid();
			else if (e.KeyCode == Keys.H && e.Modifiers == Keys.None) canvas.PanZoomViewToFit();
			else if (e.KeyCode == Keys.H && e.Modifiers == Keys.Shift) canvas.PanViewToSelection();
			else if (e.KeyCode == Keys.I && e.Modifiers == Keys.None) AddNewLine(false);
			else if (e.KeyCode == Keys.I && e.Modifiers == Keys.Shift) AddNewLine(true);
			else if (e.KeyCode == Keys.I && e.Modifiers == Keys.Alt) ToggleSnapToItems();
			else if (e.KeyCode == Keys.O && e.Modifiers == Keys.None) ConvertToOutline();
			else if (e.KeyCode == Keys.P && e.Modifiers == Keys.None) canvas.StartMousePan();
			else if (e.KeyCode == Keys.R && e.Modifiers == Keys.None) canvas.StartMouseRotate();
			else if (e.KeyCode == Keys.S && e.Modifiers == Keys.None) canvas.StartMouseScale();
			else if (e.KeyCode == Keys.T && e.Modifiers == Keys.None) canvas.StartMouseTranslate();
			else if (e.KeyCode == Keys.T && e.Modifiers == Keys.Control) AddText();
			else if (e.KeyCode == Keys.V && e.Modifiers == Keys.Control) Paste(false);
			else if (e.KeyCode == Keys.V && e.Modifiers == (Keys.Control | Keys.Shift)) Paste(true);
			else if (e.KeyCode == Keys.X && e.Modifiers == Keys.Control) Cut();
			else if (e.KeyCode == Keys.Y && e.Modifiers == Keys.Control) Redo();
			else if (e.KeyCode == Keys.Z && e.Modifiers == Keys.Control) Undo();
			else if (e.KeyCode == Keys.Delete && e.Modifiers == Keys.None) DeleteSelected();
			else if (e.KeyCode == Keys.Escape) canvas.Abort();
			else if (e.KeyCode == Keys.OemSemicolon && e.Modifiers == Keys.None) AddComment(false);
			else if (e.KeyCode == Keys.OemSemicolon && e.Modifiers == Keys.Shift) AddComment(true);
			else if (e.KeyCode == Keys.OemQuestion && e.Modifiers == Keys.None) ToggleComment();
			else if (e.KeyCode == Keys.Oemplus && e.Modifiers == Keys.None) canvas.Zoom(1.1f);
			else if (e.KeyCode == Keys.OemMinus && e.Modifiers == Keys.None) canvas.Zoom(1 / 1.1f);
			else { e.Handled = false; }

			if (e.Handled && e.Alt) { e.SuppressKeyPress = true; }

			base.OnKeyDown(e);
		}

		private void newToolStripMenuItem_Click(object sender, EventArgs e) => NewFile();
		private void openToolStripMenuItem_Click(object sender, EventArgs e) => OpenFile();
		private void saveToolStripMenuItem_Click(object sender, EventArgs e) => SaveFile();
		private void saveAsToolStripMenuItem_Click(object sender, EventArgs e) => SaveFileAs();
		private void exitToolStripMenuItem_Click(object sender, EventArgs e) => Close();
		private void undoToolStripMenuItem_Click(object sender, EventArgs e) => Undo();
		private void redoToolStripMenuItem_Click(object sender, EventArgs e) => Redo();
		private void cutToolStripMenuItem_Click(object sender, EventArgs e) => Cut();
		private void copyToolStripMenuItem_Click(object sender, EventArgs e) => Copy();
		private void pasteToolStripMenuItem_Click(object sender, EventArgs e) => Paste(false);
		private void deleteToolStripMenuItem_Click(object sender, EventArgs e) => DeleteSelected();
		private void selectAllToolStripMenuItem_Click(object sender, EventArgs e) => SelectAll(false);
		private void convertToAbsoluteToolStripMenuItem_Click(object sender, EventArgs e) => ConvertToAbsolute();
		private void convertToRelativeToolStripMenuItem_Click(object sender, EventArgs e) => ConvertToRelative();
		private void moveEndpointToolStripMenuItem_Click(object sender, EventArgs e) => canvas.StartMouseEndMove();
		private void moveOffsetToolStripMenuItem_Click(object sender, EventArgs e) => canvas.StartMouseOffsetMove();
		private void showGridToolStripMenuItem_Click(object sender, EventArgs e) => ToggleGrid();
		private void snapToGridToolStripMenuItem_Click(object sender, EventArgs e) => ToggleSnapToGrid();
		private void snapToItemsToolStripMenuItem_Click(object sender, EventArgs e) => ToggleSnapToItems();
		private void showCoordinatesToolStripMenuItem_Click(object sender, EventArgs e) => ToggleCoords();
		private void showPerformanceStatsToolStripMenuItem_Click(object sender, EventArgs e) => ToggleFPS();
		private void newLineAfterCurrentToolStripMenuItem_Click(object sender, EventArgs e) => AddNewLine(false);
		private void newLineToolStripMenuItem_Click(object sender, EventArgs e) => AddNewLine(true);
		private void rapidToolStripMenuItem_Click(object sender, EventArgs e) => AddG0(false);
		private void lineToolStripMenuItem_Click(object sender, EventArgs e) => AddG1(false);
		private void clockwiseArcToolStripMenuItem_Click(object sender, EventArgs e) => AddG2(false);
		private void counterclockwiseArcToolStripMenuItem_Click(object sender, EventArgs e) => AddG3(false);
		private void commentToolStripMenuItem_Click(object sender, EventArgs e) => AddComment(false);
		private void commentBeforeCurrentToolStripMenuItem_Click(object sender, EventArgs e) => AddComment(true);
		private void backgroundImageToolStripMenuItem_Click(object sender, EventArgs e) => AddBackground();
		private void textShapeToolStripMenuItem_Click(object sender, EventArgs e) => AddText();
		private void aboutToolStripMenuItem_Click(object sender, EventArgs e) => About();
		private void panzoomToFitToolStripMenuItem_Click(object sender, EventArgs e) => canvas.PanZoomViewToFit();
		private void panViewToSelectionToolStripMenuItem_Click(object sender, EventArgs e) => canvas.PanViewToSelection();
		private void translateToolStripMenuItem_Click(object sender, EventArgs e) => canvas.StartMouseTranslate();
		private void snapToAxesToolStripMenuItem_Click(object sender, EventArgs e) => ToggleSnapToAxes();
		private void convertToOutlineToolStripMenuItem_Click(object sender, EventArgs e) => ConvertToOutline();
		private void rotateToolStripMenuItem_Click(object sender, EventArgs e) => canvas.StartMouseRotate();
		private void commentUncommentLinesToolStripMenuItem_Click(object sender, EventArgs e) => ToggleComment();
		private void zoomInToolStripMenuItem_Click(object sender, EventArgs e) => canvas.Zoom(1.5f);
		private void zoomOutToolStripMenuItem_Click(object sender, EventArgs e) => canvas.Zoom(1 / 1.5f);
		private void panViewToolStripMenuItem_Click(object sender, EventArgs e) => canvas.StartMousePan();
	}
}