aistudio-wpf-diagram/AIStudio.Wpf.DiagramDesigner.Demo/ViewModels/CustomNodeViewModel.cs

using System;
using System.Collections.Generic;
using System.Text;
using System.Windows;
using AIStudio.Wpf.DiagramDesigner.Models;

namespace AIStudio.Wpf.DiagramDesigner.Demo.ViewModels
{
    class CustomNodeViewModel : BaseViewModel
    {
        public CustomNodeViewModel()
        {
            Title = "Custom node";
            Info = "Creating your own custom design is very easy!";

            //_service.ColorViewModel.FillColor.Color = System.Windows.Media.Colors.Orange;

            DiagramViewModel = new DiagramViewModel();
            DiagramViewModel.CellHorizontalAlignment = CellHorizontalAlignment.Center;
            DiagramViewModel.CellVerticalAlignment = CellVerticalAlignment.Center;
            DiagramViewModel.PageSizeType = PageSizeType.Custom;
            DiagramViewModel.PageSize = new Size(double.NaN, double.NaN);

            DefaultDesignerItemViewModel node1 = new DefaultDesignerItemViewModel() { Left = 50, Top = 50, Text = "1" };
            DiagramViewModel.DirectAddItemCommand.Execute(node1);

            CustomDesignerItemViewModel node2 = new CustomDesignerItemViewModel() { Left = 300, Top = 300 };
            DiagramViewModel.DirectAddItemCommand.Execute(node2);

            CustomDesignerItemViewModel node3 = new CustomDesignerItemViewModel() { Left = 300, Top = 50 };
            DiagramViewModel.DirectAddItemCommand.Execute(node3);

            ConnectionViewModel connector1 = new ConnectionViewModel(node1.RightConnector, node2.TopConnector, DrawMode.ConnectingLineSmooth, RouterMode.RouterNormal);
            DiagramViewModel.DirectAddItemCommand.Execute(connector1);

            ConnectionViewModel connector2 = new ConnectionViewModel(node2.BottomConnector, node3.BottomConnector, DrawMode.ConnectingLineStraight, RouterMode.RouterOrthogonal);
            DiagramViewModel.DirectAddItemCommand.Execute(connector2);

            DiagramViewModel.ClearSelectedItemsCommand.Execute(null);
        }
    }

    public class CustomDesignerItemViewModel : DesignerItemViewModelBase
    {
        public CustomDesignerItemViewModel() : base()
        {

        }

        public CustomDesignerItemViewModel(IDiagramViewModel root, SelectableItemBase designer) : base(root, designer)
        {

        }

        public CustomDesignerItemViewModel(IDiagramViewModel root, SerializableItem serializableItem, string serializableType) : base(root, serializableItem, serializableType)
        {

        }

        public override SelectableItemBase GetSerializableObject()
        {
            return new DesignerItemBase(this, Answer);
        }

        protected override void Init()
        {
            base.Init();

            this.ItemWidth = 150;
            this.ItemHeight = 80;
            InitConnector();
        }

        protected override void InitConnector()
        {
            connectors.Add(new FullyCreatedConnectorInfo(this, ConnectorOrientation.Top));
            connectors.Add(new FullyCreatedConnectorInfo(this, ConnectorOrientation.Bottom));
        }

        protected override void LoadDesignerItemViewModel(IDiagramViewModel root, SelectableItemBase designerbase)
        {
            base.LoadDesignerItemViewModel(root, designerbase);

            if (designerbase is DesignerItemBase designer)
            {
                this.Answer = designer.Reserve;
            }
        }

        private string _answer;
        public string Answer
        {
            get
            {
                return _answer;
            }
            set
            {
                SetProperty(ref _answer, value);
            }
        }
    }
}