Add abort type to behaviour tree builder

ormesam · Nov 16, 2021 · aa50710 · aa50710
1 parent f3b851f
commit aa50710
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Assets/Splatter/Scripts/Runtime/AI/BehaviourTree/BehaviourTreeBuilder.cs b/Assets/Splatter/Scripts/Runtime/AI/BehaviourTree/BehaviourTreeBuilder.cs
@@ -14,14 +14,14 @@ public BehaviourTreeBuilder(BehaviourTree tree) {
             stack = new Stack<Node>();
         }
 
-        public BehaviourTreeBuilder Sequence(bool resetIfInterrupted) {
-            AddNode(new Sequencer(tree, resetIfInterrupted));
+        public BehaviourTreeBuilder Sequence(bool resetIfInterrupted = false, AbortType abortType = AbortType.None, Func<bool> abortCondition = null) {
+            AddNode(new Sequencer(tree, resetIfInterrupted, abortType, abortCondition));
 
             return this;
         }
 
-        public BehaviourTreeBuilder Selector() {
-            AddNode(new Selector(tree));
+        public BehaviourTreeBuilder Selector(AbortType abortType = AbortType.None, Func<bool> abortCondition = null) {
+            AddNode(new Selector(tree, abortType, abortCondition));
 
             return this;
         }