Code (C++)

protected override FrameworkElement CreateRenderer(bool activeView, bool lightweight)
{
   if (Template?.LoadContent() is FrameworkElement visual)
   {
      visual.DataContext = DataContext;

      switch (DataContext)
      {
         case IedsContainerViewModel vm:
            vm.BorderColor = BorderColor(this);
            break;
         case EquipmentContainerViewModel equipmentContainerViewModel:
            equipmentContainerViewModel.BorderColor = BorderColor(this);
            break;
      }

      return visual;
   }

   var renderer = new NodeRenderer { Node = this };
   return renderer;
}