78 lines
3.0 KiB
C#
78 lines
3.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Sandbox.Game.Entities;
|
|
using Sandbox.Game.Entities.Character;
|
|
using Torch.Server.ViewModels.Entities;
|
|
using VRage.Game.ModAPI;
|
|
using VRage.ModAPI;
|
|
|
|
namespace Torch.Server.ViewModels
|
|
{
|
|
public class EntityTreeViewModel : ViewModel
|
|
{
|
|
public MTObservableCollection<GridViewModel> Grids { get; set; } = new MTObservableCollection<GridViewModel>();
|
|
public MTObservableCollection<CharacterViewModel> Characters { get; set; } = new MTObservableCollection<CharacterViewModel>();
|
|
public MTObservableCollection<EntityViewModel> FloatingObjects { get; set; } = new MTObservableCollection<EntityViewModel>();
|
|
public MTObservableCollection<VoxelMapViewModel> VoxelMaps { get; set; } = new MTObservableCollection<VoxelMapViewModel>();
|
|
|
|
private EntityViewModel _currentEntity;
|
|
|
|
public EntityViewModel CurrentEntity
|
|
{
|
|
get => _currentEntity;
|
|
set { _currentEntity = value; OnPropertyChanged(); }
|
|
}
|
|
|
|
public EntityTreeViewModel()
|
|
{
|
|
MyEntities.OnEntityAdd += MyEntities_OnEntityAdd;
|
|
MyEntities.OnEntityRemove += MyEntities_OnEntityRemove;
|
|
}
|
|
|
|
private void MyEntities_OnEntityRemove(VRage.Game.Entity.MyEntity obj)
|
|
{
|
|
switch (obj)
|
|
{
|
|
case MyCubeGrid grid:
|
|
Grids.RemoveWhere(m => m.Id == grid.EntityId);
|
|
break;
|
|
case MyCharacter character:
|
|
Characters.RemoveWhere(m => m.Id == character.EntityId);
|
|
break;
|
|
case MyFloatingObject floating:
|
|
FloatingObjects.RemoveWhere(m => m.Id == floating.EntityId);
|
|
break;
|
|
case MyVoxelBase voxel:
|
|
VoxelMaps.RemoveWhere(m => m.Id == voxel.EntityId);
|
|
break;
|
|
}
|
|
}
|
|
|
|
private void MyEntities_OnEntityAdd(VRage.Game.Entity.MyEntity obj)
|
|
{
|
|
switch (obj)
|
|
{
|
|
case MyCubeGrid grid:
|
|
if (Grids.All(g => g.Entity.EntityId != obj.EntityId))
|
|
Grids.Add(new GridViewModel(grid, this));
|
|
break;
|
|
case MyCharacter character:
|
|
if (Characters.All(g => g.Entity.EntityId != obj.EntityId))
|
|
Characters.Add(new CharacterViewModel(character, this));
|
|
break;
|
|
case MyFloatingObject floating:
|
|
if (FloatingObjects.All(g => g.Entity.EntityId != obj.EntityId))
|
|
FloatingObjects.Add(new FloatingObjectViewModel(floating, this));
|
|
break;
|
|
case MyVoxelBase voxel:
|
|
if (VoxelMaps.All(g => g.Entity.EntityId != obj.EntityId))
|
|
VoxelMaps.Add(new VoxelMapViewModel(voxel, this));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|