76 lines
3.0 KiB
C#
76 lines
3.0 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Reflection;
|
||
using UnityEngine;
|
||
|
||
namespace BaseGames.AI
|
||
{
|
||
/// <summary>
|
||
/// 反射收集所有 [AiDefinition] 的 AiScript 子类，按 id 提供共享 AiGraph（flyweight）。
|
||
/// 同一 id 全实例共享同一 AiScript 实例 → 同一张不可变 AiGraph。
|
||
/// </summary>
|
||
public static class AiDefinitionRegistry
|
||
{
|
||
static Dictionary<string, AiScript> _byId;
|
||
|
||
// 项目已关闭 Domain Reload：静态缓存需在进入播放时重置，保证按最新类型重建。
|
||
[RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.BeforeSceneLoad)]
|
||
static void ResetOnPlay() => _byId = null;
|
||
|
||
static void EnsureBuilt()
|
||
{
|
||
if (_byId != null) return;
|
||
_byId = new Dictionary<string, AiScript>();
|
||
foreach (var asm in AppDomain.CurrentDomain.GetAssemblies())
|
||
{
|
||
Type[] types;
|
||
try { types = asm.GetTypes(); }
|
||
catch (ReflectionTypeLoadException e) { types = e.Types; }
|
||
if (types == null) continue;
|
||
for (int i = 0; i < types.Length; i++)
|
||
{
|
||
var t = types[i];
|
||
if (t == null || t.IsAbstract || !typeof(AiScript).IsAssignableFrom(t)) continue;
|
||
var attr = t.GetCustomAttribute<AiDefinitionAttribute>();
|
||
if (attr == null) continue;
|
||
if (_byId.ContainsKey(attr.Id))
|
||
throw new InvalidOperationException(
|
||
$"AiDefinitionRegistry: 重复的 AI 定义 id '{attr.Id}'（{t.FullName}）。");
|
||
_byId[attr.Id] = (AiScript)Activator.CreateInstance(t);
|
||
}
|
||
}
|
||
}
|
||
|
||
public static bool Has(string id)
|
||
{
|
||
EnsureBuilt();
|
||
return _byId.ContainsKey(id);
|
||
}
|
||
|
||
/// <summary>取 id 对应的共享 AiGraph；不存在则显式抛错（根因暴露，不做兜底）。</summary>
|
||
public static AiGraph GetGraph(string id)
|
||
{
|
||
EnsureBuilt();
|
||
if (!_byId.TryGetValue(id, out var script))
|
||
throw new InvalidOperationException(
|
||
$"AiDefinitionRegistry: 未找到 AI 定义 id '{id}'。请确认存在 [AiDefinition(\"{id}\")] 的 AiScript 子类。");
|
||
return script.GetOrBuildGraph();
|
||
}
|
||
|
||
/// <summary>取 id 对应的定义；不存在则显式抛错（根因暴露，不做兜底）。</summary>
|
||
public static IAiDefinition GetDefinition(string id)
|
||
{
|
||
EnsureBuilt();
|
||
if (!_byId.TryGetValue(id, out var script))
|
||
throw new InvalidOperationException(
|
||
$"AiDefinitionRegistry: 未找到 AI 定义 id '{id}'。请确认存在 [AiDefinition(\"{id}\")] 的 AiScript 子类。");
|
||
return script;
|
||
}
|
||
|
||
public static IEnumerable<string> Ids
|
||
{
|
||
get { EnsureBuilt(); return _byId.Keys; }
|
||
}
|
||
}
|
||
}
|