Builder is now extended instead of being inherited for extension

Author: ashblue

Assets/FluidBehaviorTree/Scripts/BehaviorTree/BehaviorTreeBuilder.cs

@@ -0,0 +1,142 @@
+using System;
+using System.Collections.Generic;
+using Adnc.FluidBT.Decorators;
+using Adnc.FluidBT.TaskParents;
+using Adnc.FluidBT.TaskParents.Composites;
+using Adnc.FluidBT.Tasks;
+using Adnc.FluidBT.Tasks.Actions;
+using UnityEngine;
+
+namespace Adnc.FluidBT.Trees {
+    public class BehaviorTreeBuilder {
+        private readonly BehaviorTree _tree;
+        private readonly List<ITaskParent> _pointers = new List<ITaskParent>();
+
+        private ITaskParent PointerCurrent {
+            get {
+                if (_pointers.Count == 0) return null;
+                return _pointers[_pointers.Count - 1];
+            }
+        }
+
+        public BehaviorTreeBuilder (GameObject owner) {
+            _tree = new BehaviorTree(owner);
+            _pointers.Add(_tree.Root);
+        }
+
+        public BehaviorTreeBuilder ParentTask<P> (string name) where P : ITaskParent, new() {
+            var parent = new P { Name = name };
+            AddNode(parent);
+            _pointers.Add(parent);
+
+            return this;
+        }
+
+        public BehaviorTreeBuilder Decorator (string name, Func<ITask, TaskStatus> logic) {
+            var decorator = new DecoratorGeneric {
+                updateLogic = logic,
+                Name = name
+            };
+                        
+            AddNode(decorator);
+            _pointers.Add(decorator);
+            
+            return this;
+        }
+        
+        public BehaviorTreeBuilder Decorator (Func<ITask, TaskStatus> logic) {
+            return Decorator("decorator", logic);
+        }
+
+        public BehaviorTreeBuilder Inverter (string name = "inverter") {
+            return ParentTask<Inverter>(name);
+        }
+        
+        public BehaviorTreeBuilder ReturnSuccess (string name = "return success") {
+            return ParentTask<ReturnSuccess>(name);
+        }
+        
+        public BehaviorTreeBuilder ReturnFailure (string name = "return failure") {
+            return ParentTask<ReturnFailure>(name);
+        }
+        
+        public BehaviorTreeBuilder Sequence (string name = "sequence") {
+            return ParentTask<Sequence>(name);
+        }
+
+        public BehaviorTreeBuilder Selector (string name = "selector") {
+            return ParentTask<Selector>(name);
+        }
+        
+        public BehaviorTreeBuilder Parallel (string name = "parallel") {
+            return ParentTask<Parallel>(name);
+        }
+
+        public BehaviorTreeBuilder Do (string name, Func<TaskStatus> action) {
+            return AddNode(new ActionGeneric {
+                Name = name,
+                updateLogic = action
+            });
+        }
+        
+        public BehaviorTreeBuilder Do (Func<TaskStatus> action) {
+            return Do("action", action);
+        }
+
+        public BehaviorTreeBuilder Condition (string name, Func<bool> action) {
+            return AddNode(new ConditionGeneric {
+                Name = name,
+                updateLogic = action
+            });
+        }
+        
+        public BehaviorTreeBuilder Condition (Func<bool> action) {
+            return Condition("condition", action);
+        }
+        
+        public BehaviorTreeBuilder RandomChance (string name, int chance, int outOf, int seed = 0) {
+            return AddNode(new RandomChance {
+                Name = name,
+                chance = chance,
+                outOf = outOf,
+                seed = seed
+            });
+        }
+
+        public BehaviorTreeBuilder RandomChance (int chance, int outOf, int seed = 0) {
+            return RandomChance("random chance", chance, outOf, seed);
+        }
+
+        public BehaviorTreeBuilder Wait (string name, int turns = 1) {
+            return AddNode(new Wait {
+                Name = name,
+                turns = turns
+            });
+        }
+
+        public BehaviorTreeBuilder Wait (int turns = 1) {
+            return Wait("wait", turns);
+        }
+        
+        public BehaviorTreeBuilder AddNode (ITask node) {
+            _tree.AddNode(PointerCurrent, node);
+            return this;
+        }
+
+        public BehaviorTreeBuilder Splice (BehaviorTree tree) {
+            _tree.Splice(PointerCurrent, tree);
+
+            return this;
+        }
+
+        public BehaviorTreeBuilder End () {
+            _pointers.RemoveAt(_pointers.Count - 1);
+            
+            return this;
+        }
+
+        public BehaviorTree Build () {
+            return _tree;
+        }
+    }
+}

Assets/FluidBehaviorTree/Scripts/BehaviorTree/BehaviorTreeBuilderBase.cs

@@ -0,0 +1,42 @@
+using System;
+using Adnc.FluidBT.Tasks;
+using Adnc.FluidBT.Tasks.Actions;
+using NUnit.Framework;
+
+namespace Adnc.FluidBT.Trees.Testing {
+    public static class BehaviorTreeExtensionActionExamples {
+        public static BehaviorTreeBuilder ExampleAction (this BehaviorTreeBuilder builder, string name, Action callback) {
+            builder.AddNode(new BehaviorTreeBuilderExtensionActionTest.ExtensionAction {
+                Name = name,
+                callback = callback,
+            });
+            
+            return builder;
+        }
+    }
+    
+    public class BehaviorTreeBuilderExtensionActionTest {
+        public class ExtensionAction : ActionBase {
+            public Action callback;
+        
+            protected override TaskStatus OnUpdate () {
+                callback();
+                return TaskStatus.Success;
+            }
+        }
+        
+        [Test]
+        public void It_should_run_the_custom_action () {
+            var result = false;
+            var tree = new BehaviorTreeBuilder(null)
+                .Sequence()
+                    .ExampleAction("test", () => result = true)
+                .End()
+                .Build();
+
+            tree.Tick();
+            
+            Assert.IsTrue(result);
+        }
+    }
+}