Direct databindings - Added a large amount of modifiers

Direct databindings - Allow modifiers to have a different input parameter
Direct databindings - Categorised modifiers
Direct databindings - Added descriptions to modifiers

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/ConditionalDataBinding/ConditionalDataBindingModeView.xaml

@@ -1,13 +1,13 @@
 <UserControl x:Class="Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.ConditionalDataBinding.ConditionalDataBindingModeView"
              xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
              xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
-             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" 
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
              xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
              xmlns:materialDesign="http://materialdesigninxaml.net/winfx/xaml/themes"
              xmlns:s="https://github.com/canton7/Stylet"
              xmlns:dd="urn:gong-wpf-dragdrop" xmlns:Converters="clr-namespace:Artemis.UI.Converters"
              xmlns:utilities="clr-namespace:Artemis.UI.Utilities"
-             mc:Ignorable="d" 
+             mc:Ignorable="d"
              d:DesignHeight="450" d:DesignWidth="800">
     <UserControl.Resources>
         <ResourceDictionary>
@@ -78,7 +78,7 @@
             </ListBox.ItemTemplate>
             <ListBox.ItemsPanel>
                 <ItemsPanelTemplate>
-                    <VirtualizingStackPanel/>
+                    <VirtualizingStackPanel />
                 </ItemsPanelTemplate>
             </ListBox.ItemsPanel>
         </ListBox>

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/ConditionalDataBinding/ConditionalDataBindingModeViewModel.cs

@@ -11,8 +11,8 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.Conditio
 {
     public class ConditionalDataBindingModeViewModel<TLayerProperty, TProperty> : Screen, IDataBindingModeViewModel
     {
-        private readonly IProfileEditorService _profileEditorService;
         private readonly IDataBindingsVmFactory _dataBindingsVmFactory;
+        private readonly IProfileEditorService _profileEditorService;
         private bool _updating;
 
         public ConditionalDataBindingModeViewModel(ConditionalDataBinding<TLayerProperty, TProperty> conditionalDataBinding,
@@ -36,12 +36,25 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.Conditio
         {
             UpdateConditionViewModels();
         }
-        
+
         public object GetTestValue()
         {
             throw new NotSupportedException();
         }
 
+        #region IDisposable
+
+        public void Dispose()
+        {
+            ConditionalDataBinding.ConditionsUpdated -= ConditionalDataBindingOnConditionsUpdated;
+
+            foreach (var conditionViewModel in ConditionViewModels)
+                conditionViewModel.Dispose();
+            ConditionViewModels.Clear();
+        }
+
+        #endregion
+
         public void AddCondition(string type)
         {
             var condition = ConditionalDataBinding.AddCondition();
@@ -105,18 +118,5 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.Conditio
         {
             UpdateConditionViewModels();
         }
-
-        #region IDisposable
-
-        public void Dispose()
-        {
-            ConditionalDataBinding.ConditionsUpdated -= ConditionalDataBindingOnConditionsUpdated;
-
-            foreach (var conditionViewModel in ConditionViewModels)
-                conditionViewModel.Dispose();
-            ConditionViewModels.Clear();
-        }
-
-        #endregion
     }
 }

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/ConditionalDataBinding/DataBindingConditionViewModel.cs

@@ -32,6 +32,15 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.Conditio
             ValueViewModel.Value = DataBindingCondition.Value;
         }
 
+        public DataBindingCondition<TLayerProperty, TProperty> DataBindingCondition { get; }
+
+        public DataModelStaticViewModel ValueViewModel { get; set; }
+
+        public void Dispose()
+        {
+            ValueViewModel.Dispose();
+        }
+
         private void ActiveItemOnUpdated(object sender, EventArgs e)
         {
             if (!ActiveItem.GetChildren().Any())
@@ -43,14 +52,5 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.Conditio
             DataBindingCondition.Value = (TProperty) Convert.ChangeType(e.Value, typeof(TProperty));
             _profileEditorService.UpdateSelectedProfileElement();
         }
-
-        public DataBindingCondition<TLayerProperty, TProperty> DataBindingCondition { get; }
-
-        public DataModelStaticViewModel ValueViewModel { get; set; }
-
-        public void Dispose()
-        {
-            ValueViewModel.Dispose();
-        }
     }
 }

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DataBindingView.xaml.cs

@@ -1,5 +1,4 @@
-using System.Windows;
-using System.Windows.Controls;
+using System.Windows.Controls;
 
 namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings
 {

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DataBindingViewModel.cs

@@ -12,16 +12,16 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings
     public class DataBindingViewModel<TLayerProperty, TProperty> : Conductor<IDataBindingModeViewModel>, IDataBindingViewModel
     {
         private readonly IDataBindingsVmFactory _dataBindingsVmFactory;
-        private readonly IProfileEditorService _profileEditorService;
         private readonly IDataModelUIService _dataModelUIService;
+        private readonly IProfileEditorService _profileEditorService;
         private DataBinding<TLayerProperty, TProperty> _dataBinding;
         private int _easingTime;
+        private bool _isDataBindingEnabled;
         private bool _isEasingTimeEnabled;
         private DataBindingModeType _selectedDataBindingMode;
         private TimelineEasingViewModel _selectedEasingViewModel;
 
         private bool _updating;
-        private bool _isDataBindingEnabled;
 
         public DataBindingViewModel(DataBindingRegistration<TLayerProperty, TProperty> registration,
             IProfileEditorService profileEditorService,
@@ -219,9 +219,7 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings
                 TestResultValue.UpdateValue(Registration.DataBinding != null ? Registration.DataBinding.GetValue(currentValue) : default);
             }
             else
-            {
                 TestResultValue.UpdateValue(Registration.DataBinding != null ? Registration.DataBinding.GetValue(default) : default);
-            }
         }
 
         private void EnableDataBinding()

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DataBindingsViewModel.cs

@@ -26,6 +26,12 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings
             set => SetAndNotify(ref _selectedItemIndex, value);
         }
 
+        protected override void OnClose()
+        {
+            _profileEditorService.SelectedDataBindingChanged -= ProfileEditorServiceOnSelectedDataBindingChanged;
+            base.OnClose();
+        }
+
         private void CreateDataBindingViewModels()
         {
             Items.Clear();
@@ -44,12 +50,6 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings
             SelectedItemIndex = 0;
         }
 
-        protected override void OnClose()
-        {
-            _profileEditorService.SelectedDataBindingChanged -= ProfileEditorServiceOnSelectedDataBindingChanged;
-            base.OnClose();
-        }
-
         private void ProfileEditorServiceOnSelectedDataBindingChanged(object? sender, EventArgs e)
         {
             CreateDataBindingViewModels();

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/DataBindingModifierView.xaml

@@ -8,9 +8,9 @@
              xmlns:converters="clr-namespace:Artemis.UI.Converters"
              xmlns:utilities="clr-namespace:Artemis.UI.Utilities"
              xmlns:local="clr-namespace:Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDataBinding"
+             xmlns:modifierTypes="clr-namespace:Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDataBinding.ModifierTypes"
              mc:Ignorable="d"
-             d:DesignHeight="450" d:DesignWidth="800"
-             d:DataContext="{d:DesignInstance local:DataBindingModifierViewModel}">
+             d:DesignHeight="450" d:DesignWidth="800">
     <UserControl.Resources>
         <ResourceDictionary>
             <ResourceDictionary.MergedDictionaries>
@@ -56,17 +56,33 @@
                 BorderBrush="#7B7B7B"
                 Click="PropertyButton_OnClick">
             <Button.ContextMenu>
-                <ContextMenu ItemsSource="{Binding ModifierTypes}">
+                <ContextMenu ItemsSource="{Binding ModifierTypeViewModels.Children}">
                     <ContextMenu.ItemTemplate>
                         <DataTemplate>
-                            <StackPanel Orientation="Horizontal">
-                                <materialDesign:PackIcon Kind="{Binding Icon}" VerticalAlignment="Center" Margin="0 0 15 0" />
-                                <TextBlock Text="{Binding Description}" VerticalAlignment="Center" />
+                            <StackPanel>
+                                <StackPanel.Resources>
+                                    <DataTemplate DataType="{x:Type modifierTypes:ModifierTypeCategoryViewModel}">
+                                        <TextBlock Text="{Binding Category}" />
+                                    </DataTemplate>
+                                    <DataTemplate DataType="{x:Type modifierTypes:ModifierTypeViewModel}">
+                                        <StackPanel Orientation="Horizontal">
+                                            <materialDesign:PackIcon Kind="{Binding ModifierType.Icon}" 
+                                                                     VerticalAlignment="Center" 
+                                                                     Margin="0 0 15 0" 
+                                                                     Visibility="{Binding ModifierType.Icon, Converter={StaticResource NullToVisibilityConverter}}"/>
+                                            <TextBlock Text="{Binding ModifierType.Name}" 
+                                                       VerticalAlignment="Center" 
+                                                       ToolTip="{Binding ModifierType.Description}"/>
+                                        </StackPanel>
+                                    </DataTemplate>
+                                </StackPanel.Resources>
+                                <ContentPresenter Content="{Binding}" />
                             </StackPanel>
                         </DataTemplate>
                     </ContextMenu.ItemTemplate>
                     <ContextMenu.ItemContainerStyle>
                         <Style TargetType="{x:Type MenuItem}" BasedOn="{StaticResource MaterialDesignMenuItem}">
+                            <Setter Property="ItemsSource" Value="{Binding Children}" />
                             <Setter Property="Command" Value="{Binding Data.SelectModifierTypeCommand, Source={StaticResource DataContextProxy}}" />
                             <Setter Property="CommandParameter" Value="{Binding}" />
                             <Setter Property="CommandTarget" Value="{Binding}" />
@@ -75,14 +91,14 @@
                 </ContextMenu>
             </Button.ContextMenu>
             <Grid>
-                <TextBlock Text="{Binding SelectedModifierType.Description}"
+                <TextBlock Text="{Binding SelectedModifierType.Name}"
                            Visibility="{Binding SelectedModifierType, Converter={StaticResource NullToVisibilityConverter}}" />
                 <TextBlock FontStyle="Italic"
                            Visibility="{Binding SelectedModifierType, Converter={StaticResource NullToVisibilityConverter}, ConverterParameter=Inverted}">
                     « Select a modifier »
                 </TextBlock>
             </Grid>
-            
+
         </Button>
 
         <ContentControl Grid.Column="3" s:View.Model="{Binding DynamicSelectionViewModel}" VerticalContentAlignment="Stretch" HorizontalContentAlignment="Stretch" IsTabStop="False" />

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/DataBindingModifierView.xaml.cs

@@ -4,7 +4,7 @@ using System.Windows.Controls;
 namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDataBinding
 {
     /// <summary>
-    /// Interaction logic for DataBindingModifierView.xaml
+    ///     Interaction logic for DataBindingModifierView.xaml
     /// </summary>
     public partial class DataBindingModifierView : UserControl
     {
@@ -23,4 +23,4 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
             }
         }
     }
-}
+}

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/DataBindingModifierViewModel.cs

@@ -1,7 +1,9 @@
 using System;
+using System.Linq;
 using Artemis.Core;
 using Artemis.Core.Services;
 using Artemis.UI.Exceptions;
+using Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDataBinding.ModifierTypes;
 using Artemis.UI.Shared;
 using Artemis.UI.Shared.Input;
 using Artemis.UI.Shared.Services;
@@ -15,6 +17,7 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
         private readonly IDataModelUIService _dataModelUIService;
         private readonly IProfileEditorService _profileEditorService;
         private DataModelDynamicViewModel _dynamicSelectionViewModel;
+        private ModifierTypeCategoryViewModel _modifierTypeViewModels;
         private DataBindingModifierType _selectedModifierType;
         private DataModelStaticViewModel _staticInputViewModel;
 
@@ -31,8 +34,6 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
             ShowDataModelValues = settingsService.GetSetting<bool>("ProfileEditor.ShowDataModelValues");
 
             Modifier = modifier;
-            ModifierTypes = new BindableCollection<DataBindingModifierType>();
-
             SelectModifierTypeCommand = new DelegateCommand(ExecuteSelectModifierTypeCommand);
 
             Update();
@@ -41,7 +42,12 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
         public DelegateCommand SelectModifierTypeCommand { get; }
         public PluginSetting<bool> ShowDataModelValues { get; }
         public DataBindingModifier<TLayerProperty, TProperty> Modifier { get; }
-        public BindableCollection<DataBindingModifierType> ModifierTypes { get; }
+
+        public ModifierTypeCategoryViewModel ModifierTypeViewModels
+        {
+            get => _modifierTypeViewModels;
+            set => SetAndNotify(ref _modifierTypeViewModels, value);
+        }
 
         public DataBindingModifierType SelectedModifierType
         {
@@ -61,6 +67,21 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
             private set => SetAndNotify(ref _staticInputViewModel, value);
         }
 
+        public void Dispose()
+        {
+            if (DynamicSelectionViewModel != null)
+            {
+                DynamicSelectionViewModel.Dispose();
+                DynamicSelectionViewModel.PropertySelected -= ParameterSelectionViewModelOnPropertySelected;
+            }
+
+            if (StaticInputViewModel != null)
+            {
+                StaticInputViewModel.Dispose();
+                StaticInputViewModel.ValueUpdated -= StaticInputViewModelOnValueUpdated;
+            }
+        }
+
         public void Delete()
         {
             Modifier.DirectDataBinding.RemoveModifier(Modifier);
@@ -70,7 +91,10 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
         public void SwapType()
         {
             if (Modifier.ParameterType == ProfileRightSideType.Dynamic)
-                Modifier.UpdateParameter(Modifier.DirectDataBinding.GetSourceType().GetDefault());
+            {
+                var sourceType = Modifier.DirectDataBinding.GetSourceType();
+                Modifier.UpdateParameter((Modifier.ModifierType.ParameterType ?? sourceType).GetDefault());
+            }
             else
                 Modifier.UpdateParameter(null, null);
 
@@ -114,24 +138,30 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
                 if (DynamicSelectionViewModel != null)
                 {
                     DynamicSelectionViewModel.PropertySelected += ParameterSelectionViewModelOnPropertySelected;
-                    DynamicSelectionViewModel.FilterTypes = new[] {sourceType};
+                    DynamicSelectionViewModel.FilterTypes = new[] {Modifier.ModifierType.ParameterType ?? sourceType};
                 }
             }
             else
             {
                 DynamicSelectionViewModel = null;
-                if (Modifier.ModifierType.PreferredParameterType != null && sourceType.IsCastableFrom(Modifier.ModifierType.PreferredParameterType))
-                    StaticInputViewModel = _dataModelUIService.GetStaticInputViewModel(Modifier.ModifierType.PreferredParameterType);
-                else
-                    StaticInputViewModel = _dataModelUIService.GetStaticInputViewModel(sourceType);
-
+                StaticInputViewModel = _dataModelUIService.GetStaticInputViewModel(Modifier.ModifierType.ParameterType ?? sourceType);
                 if (StaticInputViewModel != null)
                     StaticInputViewModel.ValueUpdated += StaticInputViewModelOnValueUpdated;
             }
 
             // Modifier type
-            ModifierTypes.Clear();
-            ModifierTypes.AddRange(_dataBindingService.GetCompatibleModifierTypes(sourceType));
+            var root = new ModifierTypeCategoryViewModel(null, null);
+            var modifierTypes = _dataBindingService.GetCompatibleModifierTypes(sourceType).GroupBy(t => t.Category);
+            foreach (var dataBindingModifierTypes in modifierTypes)
+            {
+                var viewModels = dataBindingModifierTypes.Select(t => new ModifierTypeViewModel(t));
+                if (dataBindingModifierTypes.Key == null)
+                    root.Children.AddRange(viewModels);
+                else
+                    root.Children.Add(new ModifierTypeCategoryViewModel(dataBindingModifierTypes.Key, viewModels));
+            }
+
+            ModifierTypeViewModels = root;
             SelectedModifierType = Modifier.ModifierType;
 
             // Parameter
@@ -143,27 +173,13 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
 
         private void ExecuteSelectModifierTypeCommand(object context)
         {
-            if (!(context is DataBindingModifierType dataBindingModifierType))
+            if (!(context is ModifierTypeViewModel modifierTypeViewModel))
                 return;
 
-            Modifier.UpdateModifierType(dataBindingModifierType);
+            Modifier.UpdateModifierType(modifierTypeViewModel.ModifierType);
             _profileEditorService.UpdateSelectedProfileElement();
 
             Update();
         }
-
-        public void Dispose()
-        {
-            if (DynamicSelectionViewModel != null)
-            {
-                DynamicSelectionViewModel.Dispose();
-                DynamicSelectionViewModel.PropertySelected -= ParameterSelectionViewModelOnPropertySelected;
-            }
-
-            if (StaticInputViewModel != null)
-            {
-                StaticInputViewModel.Dispose();
-            }
-        }
     }
 }

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/DirectDataBindingModeView.xaml

@@ -19,8 +19,8 @@
         </Grid.RowDefinitions>
 
         <ContentControl Grid.Row="0"
-                        s:View.Model="{Binding TargetSelectionViewModel}" 
-                        VerticalContentAlignment="Stretch" 
+                        s:View.Model="{Binding TargetSelectionViewModel}"
+                        VerticalContentAlignment="Stretch"
                         HorizontalContentAlignment="Stretch"
                         IsTabStop="False" />
 

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/DirectDataBindingModeViewModel.cs

@@ -16,8 +16,8 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
         private readonly IDataBindingsVmFactory _dataBindingsVmFactory;
         private readonly IDataModelUIService _dataModelUIService;
         private readonly IProfileEditorService _profileEditorService;
-        private DataModelDynamicViewModel _targetSelectionViewModel;
         private bool _canAddModifier;
+        private DataModelDynamicViewModel _targetSelectionViewModel;
         private bool _updating;
 
         public DirectDataBindingModeViewModel(DirectDataBinding<TLayerProperty, TProperty> directDataBinding,
@@ -38,7 +38,6 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
 
         public DirectDataBinding<TLayerProperty, TProperty> DirectDataBinding { get; }
         public BindableCollection<DataBindingModifierViewModel<TLayerProperty, TProperty>> ModifierViewModels { get; }
-        public bool SupportsTestValue => true;
 
         public bool CanAddModifier
         {
@@ -52,6 +51,8 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
             private set => SetAndNotify(ref _targetSelectionViewModel, value);
         }
 
+        public bool SupportsTestValue => true;
+
         public void Update()
         {
             TargetSelectionViewModel.PopulateSelectedPropertyViewModel(DirectDataBinding.SourceDataModel, DirectDataBinding.SourcePropertyPath);
@@ -66,6 +67,21 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
             return TargetSelectionViewModel.SelectedPropertyViewModel?.GetCurrentValue();
         }
 
+        #region IDisposable
+
+        public void Dispose()
+        {
+            TargetSelectionViewModel.PropertySelected -= TargetSelectionViewModelOnPropertySelected;
+            TargetSelectionViewModel.Dispose();
+            DirectDataBinding.ModifiersUpdated -= DirectDataBindingOnModifiersUpdated;
+
+            foreach (var dataBindingModifierViewModel in ModifierViewModels)
+                dataBindingModifierViewModel.Dispose();
+            ModifierViewModels.Clear();
+        }
+
+        #endregion
+
         private void Initialize()
         {
             DirectDataBinding.ModifiersUpdated += DirectDataBindingOnModifiersUpdated;
@@ -110,7 +126,7 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
             DirectDataBinding.AddModifier(ProfileRightSideType.Dynamic);
             _profileEditorService.UpdateSelectedProfileElement();
         }
-        
+
         private void UpdateModifierViewModels()
         {
             _updating = true;
@@ -142,20 +158,5 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDa
         }
 
         #endregion
-
-        #region IDisposable
-
-        public void Dispose()
-        {
-            TargetSelectionViewModel.PropertySelected -= TargetSelectionViewModelOnPropertySelected;
-            TargetSelectionViewModel.Dispose();
-            DirectDataBinding.ModifiersUpdated -= DirectDataBindingOnModifiersUpdated;
-
-            foreach (var dataBindingModifierViewModel in ModifierViewModels)
-                dataBindingModifierViewModel.Dispose();
-            ModifierViewModels.Clear();
-        }
-
-        #endregion
     }
 }

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/ModifierTypes/IModifierTypeViewModel.cs

@@ -0,0 +1,6 @@
+namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDataBinding.ModifierTypes
+{
+    public interface IModifierTypeViewModel
+    {
+    }
+}

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/ModifierTypes/ModifierTypeCategoryViewModel.cs

@@ -0,0 +1,19 @@
+using System.Collections.Generic;
+using Stylet;
+
+namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDataBinding.ModifierTypes
+{
+    public class ModifierTypeCategoryViewModel : IModifierTypeViewModel
+    {
+        public ModifierTypeCategoryViewModel(string category, IEnumerable<IModifierTypeViewModel> children)
+        {
+            Category = category;
+            Children = children == null
+                ? new BindableCollection<IModifierTypeViewModel>()
+                : new BindableCollection<IModifierTypeViewModel>(children);
+        }
+
+        public string Category { get; set; }
+        public BindableCollection<IModifierTypeViewModel> Children { get; set; }
+    }
+}

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/DirectDataBinding/ModifierTypes/ModifierTypeViewModel.cs

@@ -0,0 +1,14 @@
+using Artemis.Core;
+
+namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings.DirectDataBinding.ModifierTypes
+{
+    public class ModifierTypeViewModel : IModifierTypeViewModel
+    {
+        public ModifierTypeViewModel(DataBindingModifierType modifierType)
+        {
+            ModifierType = modifierType;
+        }
+
+        public DataBindingModifierType ModifierType { get; set; }
+    }
+}

src/Artemis.UI/Screens/ProfileEditor/LayerProperties/DataBindings/IDataBindingModeViewModel.cs

@@ -5,8 +5,8 @@ namespace Artemis.UI.Screens.ProfileEditor.LayerProperties.DataBindings
 {
     public interface IDataBindingModeViewModel : IScreen, IDisposable
     {
-        void Update();
         bool SupportsTestValue { get; }
+        void Update();
         object GetTestValue();
     }
 }