Meta - Replace WPF projects with Avalonia projects

src/Artemis.UI/Screens/ProfileEditor/Panels/Properties/PropertyGroupViewModel.cs

@@ -0,0 +1,140 @@
+using System;
+using System.Collections.Generic;
+using System.Collections.ObjectModel;
+using System.Linq;
+using System.Reflection;
+using Artemis.Core;
+using Artemis.Core.LayerBrushes;
+using Artemis.Core.LayerEffects;
+using Artemis.UI.Ninject.Factories;
+using Artemis.UI.Screens.ProfileEditor.Properties.Timeline;
+using Artemis.UI.Screens.ProfileEditor.Properties.Timeline.Keyframes;
+using Artemis.UI.Screens.ProfileEditor.Properties.Tree;
+using Artemis.UI.Shared;
+using Artemis.UI.Shared.Services.PropertyInput;
+
+namespace Artemis.UI.Screens.ProfileEditor.Properties;
+
+public class PropertyGroupViewModel : ViewModelBase, IDisposable
+{
+    private readonly ILayerPropertyVmFactory _layerPropertyVmFactory;
+    private readonly IPropertyInputService _propertyInputService;
+    private bool _hasChildren;
+    private bool _isExpanded;
+    private bool _isVisible;
+
+    public PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup, ILayerPropertyVmFactory layerPropertyVmFactory, IPropertyInputService propertyInputService)
+    {
+        _layerPropertyVmFactory = layerPropertyVmFactory;
+        _propertyInputService = propertyInputService;
+        Children = new ObservableCollection<ViewModelBase>();
+        LayerPropertyGroup = layerPropertyGroup;
+        TreeGroupViewModel = layerPropertyVmFactory.TreeGroupViewModel(this);
+        TimelineGroupViewModel = layerPropertyVmFactory.TimelineGroupViewModel(this);
+
+        LayerPropertyGroup.VisibilityChanged += LayerPropertyGroupOnVisibilityChanged;
+        _isVisible = !LayerPropertyGroup.IsHidden;
+
+        PopulateChildren();
+    }
+
+    public PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup, ILayerPropertyVmFactory layerPropertyVmFactory, IPropertyInputService propertyInputService,
+        BaseLayerBrush layerBrush)
+        : this(layerPropertyGroup, layerPropertyVmFactory, propertyInputService)
+    {
+        LayerBrush = layerBrush;
+    }
+
+    public PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup, ILayerPropertyVmFactory layerPropertyVmFactory, IPropertyInputService propertyInputService,
+        BaseLayerEffect layerEffect)
+        : this(layerPropertyGroup, layerPropertyVmFactory, propertyInputService)
+    {
+        LayerEffect = layerEffect;
+    }
+
+    public ObservableCollection<ViewModelBase> Children { get; }
+    public LayerPropertyGroup LayerPropertyGroup { get; }
+    public BaseLayerBrush? LayerBrush { get; }
+    public BaseLayerEffect? LayerEffect { get; }
+
+    public TreeGroupViewModel TreeGroupViewModel { get; }
+    public TimelineGroupViewModel TimelineGroupViewModel { get; }
+
+    public bool IsVisible
+    {
+        get => _isVisible;
+        set => RaiseAndSetIfChanged(ref _isVisible, value);
+    }
+
+    public bool IsExpanded
+    {
+        get => _isExpanded;
+        set => RaiseAndSetIfChanged(ref _isExpanded, value);
+    }
+
+    public bool HasChildren
+    {
+        get => _hasChildren;
+        set => RaiseAndSetIfChanged(ref _hasChildren, value);
+    }
+
+    public List<ITimelineKeyframeViewModel> GetAllKeyframeViewModels(bool expandedOnly)
+    {
+        List<ITimelineKeyframeViewModel> result = new();
+        if (expandedOnly && !IsExpanded)
+            return result;
+
+        foreach (ViewModelBase child in Children)
+        {
+            if (child is PropertyViewModel profileElementPropertyViewModel)
+                result.AddRange(profileElementPropertyViewModel.TimelinePropertyViewModel.GetAllKeyframeViewModels());
+            else if (child is PropertyGroupViewModel profileElementPropertyGroupViewModel)
+                result.AddRange(profileElementPropertyGroupViewModel.GetAllKeyframeViewModels(expandedOnly));
+        }
+
+        return result;
+    }
+
+    private void PopulateChildren()
+    {
+        // Get all properties and property groups and create VMs for them
+        // The group has methods for getting this without reflection but then we lose the order of the properties as they are defined on the group
+        foreach (PropertyInfo propertyInfo in LayerPropertyGroup.GetType().GetProperties())
+        {
+            if (Attribute.IsDefined(propertyInfo, typeof(LayerPropertyIgnoreAttribute)))
+                continue;
+
+            if (typeof(ILayerProperty).IsAssignableFrom(propertyInfo.PropertyType))
+            {
+                ILayerProperty? value = (ILayerProperty?) propertyInfo.GetValue(LayerPropertyGroup);
+                // Ensure a supported input VM was found, otherwise don't add it
+                if (value != null && _propertyInputService.CanCreatePropertyInputViewModel(value))
+                    Children.Add(_layerPropertyVmFactory.PropertyViewModel(value));
+            }
+            else if (typeof(LayerPropertyGroup).IsAssignableFrom(propertyInfo.PropertyType))
+            {
+                LayerPropertyGroup? value = (LayerPropertyGroup?) propertyInfo.GetValue(LayerPropertyGroup);
+                if (value != null)
+                    Children.Add(_layerPropertyVmFactory.PropertyGroupViewModel(value));
+            }
+        }
+
+        HasChildren = Children.Any(i => i is PropertyViewModel {IsVisible: true} || i is PropertyGroupViewModel {IsVisible: true});
+    }
+
+    private void LayerPropertyGroupOnVisibilityChanged(object? sender, EventArgs e)
+    {
+        IsVisible = !LayerPropertyGroup.IsHidden;
+    }
+
+    /// <inheritdoc />
+    public void Dispose()
+    {
+        LayerPropertyGroup.VisibilityChanged -= LayerPropertyGroupOnVisibilityChanged;
+        foreach (ViewModelBase viewModelBase in Children)
+        {
+            if (viewModelBase is IDisposable disposable)
+                disposable.Dispose();
+        }
+    }
+}