From e401fdf964b93a599265a337d7502bd91e287e8c Mon Sep 17 00:00:00 2001
From: Robert <mail@rbeekman.nl>
Date: Sat, 13 Aug 2022 17:09:53 +0200
Subject: [PATCH] Editor - Autofit surface on open

Data model picker - Expand on double click
---
 .../DataModelPicker/DataModelPicker.cs        | 15 ++++-
 .../VisualEditor/VisualEditorView.axaml       |  2 +-
 .../VisualEditor/VisualEditorView.axaml.cs    | 60 +++++++++++++++++++
 .../VisualEditor/VisualEditorViewModel.cs     |  7 +++
 .../Screens/EnumEqualsNodeCustomView.axaml    | 10 ++--
 5 files changed, 86 insertions(+), 8 deletions(-)

diff --git a/src/Artemis.UI.Shared/Controls/DataModelPicker/DataModelPicker.cs b/src/Artemis.UI.Shared/Controls/DataModelPicker/DataModelPicker.cs
index bb343bec1..621a96161 100644
--- a/src/Artemis.UI.Shared/Controls/DataModelPicker/DataModelPicker.cs
+++ b/src/Artemis.UI.Shared/Controls/DataModelPicker/DataModelPicker.cs
@@ -13,6 +13,8 @@ using Avalonia;
 using Avalonia.Controls;
 using Avalonia.Controls.Primitives;
 using Avalonia.Data;
+using Avalonia.Interactivity;
+using Avalonia.LogicalTree;
 using Avalonia.Threading;
 using Material.Icons.Avalonia;
 using ReactiveUI;
@@ -129,7 +131,7 @@ public class DataModelPicker : TemplatedControl
         get => GetValue(DataModelViewModelProperty);
         private set => SetValue(DataModelViewModelProperty, value);
     }
-    
+
     /// <summary>
     ///     A list of types to filter the selectable paths on.
     /// </summary>
@@ -246,6 +248,13 @@ public class DataModelPicker : TemplatedControl
             DataModelPath = new DataModelPath(dataModelEvent.DataModelPath);
     }
 
+    private void DataModelTreeViewOnDoubleTapped(object? sender, RoutedEventArgs e)
+    {
+        TreeViewItem? treeViewItem = (e.Source as ILogical)?.FindLogicalAncestorOfType<TreeViewItem>();
+        if (treeViewItem != null)
+            treeViewItem.IsExpanded = !treeViewItem.IsExpanded;
+    }
+
     private void UpdateCurrentPath(bool selectCurrentPath)
     {
         if (DataModelPath == null)
@@ -278,6 +287,8 @@ public class DataModelPicker : TemplatedControl
     {
         if (_dataModelTreeView != null)
             _dataModelTreeView.SelectionChanged -= DataModelTreeViewOnSelectionChanged;
+        if (_dataModelTreeView != null)
+            _dataModelTreeView.DoubleTapped -= DataModelTreeViewOnDoubleTapped;
 
         _currentPathIcon = e.NameScope.Find<MaterialIcon>("CurrentPathIcon");
         _currentPathDisplay = e.NameScope.Find<TextBlock>("CurrentPathDisplay");
@@ -286,6 +297,8 @@ public class DataModelPicker : TemplatedControl
 
         if (_dataModelTreeView != null)
             _dataModelTreeView.SelectionChanged += DataModelTreeViewOnSelectionChanged;
+        if (_dataModelTreeView != null)
+            _dataModelTreeView.DoubleTapped += DataModelTreeViewOnDoubleTapped;
     }
 
     #region Overrides of Visual
diff --git a/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml b/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml
index 2c88bddea..7b79b8273 100644
--- a/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml
+++ b/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml
@@ -38,7 +38,7 @@
                 </Grid.Transitions>
 
                 <!-- The bottom layer consists of devices -->
-                <ItemsControl Items="{CompiledBinding Devices}" ClipToBounds="False">
+                <ItemsControl Name="DevicesContainer" Items="{CompiledBinding Devices}" ClipToBounds="False">
                     <ItemsControl.Styles>
                         <Style Selector="ContentPresenter">
                             <Setter Property="Canvas.Left" Value="{Binding X}" />
diff --git a/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml.cs b/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml.cs
index fc52e71f0..704ad7ab6 100644
--- a/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml.cs
+++ b/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorView.axaml.cs
@@ -1,15 +1,22 @@
+using System;
+using System.Linq;
+using System.Reactive.Disposables;
 using Avalonia;
 using Avalonia.Controls;
 using Avalonia.Controls.PanAndZoom;
+using Avalonia.Input;
 using Avalonia.Markup.Xaml;
 using Avalonia.Media;
 using Avalonia.ReactiveUI;
+using Avalonia.Threading;
+using ReactiveUI;
 
 namespace Artemis.UI.Screens.ProfileEditor.VisualEditor;
 
 public class VisualEditorView : ReactiveUserControl<VisualEditorViewModel>
 {
     private readonly ZoomBorder _zoomBorder;
+    private bool _movedByUser;
 
     public VisualEditorView()
     {
@@ -17,7 +24,37 @@ public class VisualEditorView : ReactiveUserControl<VisualEditorViewModel>
 
         _zoomBorder = this.Find<ZoomBorder>("ZoomBorder");
         _zoomBorder.PropertyChanged += ZoomBorderOnPropertyChanged;
+        _zoomBorder.PointerMoved += ZoomBorderOnPointerMoved;
+        _zoomBorder.PointerWheelChanged += ZoomBorderOnPointerWheelChanged;
         UpdateZoomBorderBackground();
+
+        this.WhenActivated(d =>
+        {
+            ViewModel!.AutoFitRequested += ViewModelOnAutoFitRequested;
+            Disposable.Create(() => ViewModel.AutoFitRequested -= ViewModelOnAutoFitRequested).DisposeWith(d);
+        });
+
+        this.WhenAnyValue(v => v.Bounds).Subscribe(_ =>
+        {
+            if (!_movedByUser)
+                AutoFit(true);
+        });
+    }
+
+    private void ZoomBorderOnPointerWheelChanged(object? sender, PointerWheelEventArgs e)
+    {
+        _movedByUser = true;
+    }
+
+    private void ZoomBorderOnPointerMoved(object? sender, PointerEventArgs e)
+    {
+        if (e.GetCurrentPoint(_zoomBorder).Properties.IsMiddleButtonPressed)
+            _movedByUser = true;
+    }
+
+    private void ViewModelOnAutoFitRequested(object? sender, EventArgs e)
+    {
+        Dispatcher.UIThread.Post(() => AutoFit(false));
     }
 
     private void ZoomBorderOnPropertyChanged(object? sender, AvaloniaPropertyChangedEventArgs e)
@@ -41,4 +78,27 @@ public class VisualEditorView : ReactiveUserControl<VisualEditorViewModel>
     {
         UpdateZoomBorderBackground();
     }
+
+    private void AutoFit(bool skipTransitions)
+    {
+        if (ViewModel == null)
+            return;
+
+        double left = ViewModel.Devices.Select(d => d.Rectangle.Left).Min();
+        double top = ViewModel.Devices.Select(d => d.Rectangle.Top).Min();
+        double bottom = ViewModel.Devices.Select(d => d.Rectangle.Bottom).Max();
+        double right = ViewModel.Devices.Select(d => d.Rectangle.Right).Max();
+
+        // Add a 10 pixel margin around the rect
+        Rect scriptRect = new(new Point(left - 10, top - 10), new Point(right + 10, bottom + 10));
+
+        // The scale depends on the available space
+        double scale = Math.Min(3, Math.Min(Bounds.Width / scriptRect.Width, Bounds.Height / scriptRect.Height));
+
+        // Pan and zoom to make the script fit
+        _zoomBorder.Zoom(scale, 0, 0, skipTransitions);
+        _zoomBorder.Pan(Bounds.Center.X - scriptRect.Center.X * scale, Bounds.Center.Y - scriptRect.Center.Y * scale, skipTransitions);
+
+        _movedByUser = false;
+    }
 }
\ No newline at end of file
diff --git a/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorViewModel.cs b/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorViewModel.cs
index bf6747f8f..aff8d0b2d 100644
--- a/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorViewModel.cs
+++ b/src/Artemis.UI/Screens/ProfileEditor/Panels/VisualEditor/VisualEditorViewModel.cs
@@ -113,4 +113,11 @@ public class VisualEditorViewModel : ActivatableViewModelBase
         visualizerViewModels.Add(_vmFactory.LayerShapeVisualizerViewModel(layer));
         visualizerViewModels.Add(_vmFactory.LayerVisualizerViewModel(layer));
     }
+    
+    public void RequestAutoFit()
+    {
+        AutoFitRequested?.Invoke(this, EventArgs.Empty);
+    }
+
+    public event EventHandler? AutoFitRequested;
 }
\ No newline at end of file
diff --git a/src/Artemis.VisualScripting/Nodes/Operators/Screens/EnumEqualsNodeCustomView.axaml b/src/Artemis.VisualScripting/Nodes/Operators/Screens/EnumEqualsNodeCustomView.axaml
index d0b607bb5..0b19e7bd5 100644
--- a/src/Artemis.VisualScripting/Nodes/Operators/Screens/EnumEqualsNodeCustomView.axaml
+++ b/src/Artemis.VisualScripting/Nodes/Operators/Screens/EnumEqualsNodeCustomView.axaml
@@ -2,13 +2,11 @@
              xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
              xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
              xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
-             xmlns:screens="clr-namespace:Artemis.VisualScripting.Nodes.Operators.Screens"
              mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450"
-             x:Class="Artemis.VisualScripting.Nodes.Operators.Screens.EnumEqualsNodeCustomView"
-             x:DataType="screens:EnumEqualsNodeCustomViewModel">
-    <ComboBox IsEnabled="{CompiledBinding EnumValues.Count}"
-              Items="{CompiledBinding EnumValues}"
-              SelectedItem="{CompiledBinding CurrentValue}"
+             x:Class="Artemis.VisualScripting.Nodes.Operators.Screens.EnumEqualsNodeCustomView">
+    <ComboBox IsEnabled="{Binding EnumValues.Count}"
+              Items="{Binding EnumValues}"
+              SelectedItem="{Binding CurrentValue}"
               VirtualizationMode="Simple"
               PlaceholderText="Select a value"
               Classes="condensed"