ET9_WebGL_Dev/CSharp/Game/BehaviorTree/BehaviorTreeFactory.cs

using System;
using System.Collections.Generic;

namespace BehaviorTree
{
    public class BehaviorTreeFactory
    {
        private readonly Dictionary<string, Func<Config, Node>> dictionary =
                new Dictionary<string, Func<Config, Node>>();

        public BehaviorTreeFactory()
        {
            this.dictionary.Add("selector", config => new Selector(config));
            this.dictionary.Add("sequence", config => new Sequence(config));
            this.dictionary.Add("not", config => new Not(config));
        }

        public void Register(string name, Func<Config, Node> action)
        {
            this.dictionary.Add(name, action);
        }

        private Node CreateNode(Config config)
        {
            if (!this.dictionary.ContainsKey(config.Name))
            {
                throw new KeyNotFoundException(string.Format("CreateNode cannot found: {0}",
                        config.Name));
            }
            return this.dictionary[config.Name](config);
        }

        private Node CreateTreeNode(Config config)
        {
            var node = this.CreateNode(config);
            if (config.SubConfigs == null)
            {
                return node;
            }

            foreach (var subConfig in config.SubConfigs)
            {
                var subNode = this.CreateTreeNode(subConfig);
                node.AddChild(subNode);
            }
            return node;
        }

        public BehaviorTree CreateTree(Config config)
        {
            var node = this.CreateTreeNode(config);
            return new BehaviorTree(node);
        }
    }
}