AnimeFunsion/modular-avatar
1
0
Fork 0
mirror of https://github.com/bdunderscore/modular-avatar.git synced 2025-04-15 17:09:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add GetChildren method to MenuTree

Browse Source
This commit is contained in:
raiti-chan 2022-12-12 18:19:50 +09:00
parent 0bfb94fac3
commit 937b2a17b3
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
Packages/nadena.dev.modular-avatar/Editor/MenuTree.cs
View File

@ -1,22 +1,21 @@
using System.Collections.Generic; using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq; using System.Linq;
using NUnit.Framework;
using UnityEngine; using UnityEngine;
using VRC.SDK3.Avatars.Components; using VRC.SDK3.Avatars.Components;
using VRC.SDK3.Avatars.ScriptableObjects; using VRC.SDK3.Avatars.ScriptableObjects;
using static VRC.SDK3.Avatars.ScriptableObjects.VRCExpressionsMenu;
using static VRC.SDK3.Avatars.ScriptableObjects.VRCExpressionsMenu.Control; using static VRC.SDK3.Avatars.ScriptableObjects.VRCExpressionsMenu.Control;
// ReSharper disable once CheckNamespace // ReSharper disable once CheckNamespace
namespace nadena.dev.modular_avatar.core.editor { namespace nadena.dev.modular_avatar.core.editor {
public class MenuTree { public class MenuTree {
private VRCAvatarDescriptor _descriptor;
private readonly VRCExpressionsMenu _rootMenu; private readonly VRCExpressionsMenu _rootMenu;
private readonly HashSet<VRCExpressionsMenu> _included; private readonly HashSet<VRCExpressionsMenu> _included;
private readonly Dictionary<VRCExpressionsMenu, List<VRCExpressionsMenu>> _childrenMap; private readonly Dictionary<VRCExpressionsMenu, List<VRCExpressionsMenu>> _childrenMap;
private readonly Dictionary<VRCExpressionsMenu, ImmutableArray<VRCExpressionsMenu>> _flashedChildrenMap;
public MenuTree(VRCAvatarDescriptor descriptor) { public MenuTree(VRCAvatarDescriptor descriptor) {
this._descriptor = descriptor;
this._rootMenu = descriptor.expressionsMenu; this._rootMenu = descriptor.expressionsMenu;
this._included = new HashSet<VRCExpressionsMenu>(); this._included = new HashSet<VRCExpressionsMenu>();
this._childrenMap = new Dictionary<VRCExpressionsMenu, List<VRCExpressionsMenu>>(); this._childrenMap = new Dictionary<VRCExpressionsMenu, List<VRCExpressionsMenu>>();
@ -40,6 +39,12 @@ namespace nadena.dev.modular_avatar.core.editor {
this.MappingMenu(installer.menuToAppend, parent); this.MappingMenu(installer.menuToAppend, parent);
} }
public IEnumerable<VRCExpressionsMenu> GetChildren(VRCExpressionsMenu parent) {
// TODO: ライブラリとするのであれば、複製したリスト or ImmutableArray,を返すのが好ましい
if (parent == null) parent = this._rootMenu;
return this._childrenMap.TryGetValue(parent, out List<VRCExpressionsMenu> children) ? children : Enumerable.Empty<VRCExpressionsMenu>();
}
private void MappingMenu(VRCExpressionsMenu root, VRCExpressionsMenu customParent = null) { private void MappingMenu(VRCExpressionsMenu root, VRCExpressionsMenu customParent = null) {
Queue<VRCExpressionsMenu> queue = new Queue<VRCExpressionsMenu>(); Queue<VRCExpressionsMenu> queue = new Queue<VRCExpressionsMenu>();
queue.Enqueue(root); queue.Enqueue(root);