enable to fetch data from graph instance at runtime. (#130)

Author: quabug

Packages/builder.visual/Runtime/BehaviorTreeBuilder.cs

@@ -53,6 +53,8 @@ public void Execute<TCtx>(TCtx ctx) where TCtx : IGraphInstance
                 Blob = blob, Thread = Thread, AutoCreation = AutoCreation
             });
 
+            dstManager.AddComponentObject(entity, new GraphInstanceComponent {Value = instance});
+
 #if UNITY_EDITOR
             var debugName = ctx.ReadString(DebugName);
             if (!string.IsNullOrEmpty(debugName)) dstManager.SetName(entity, debugName);
@@ -102,7 +104,34 @@ public static IEnumerable<INodeDataBuilder> ToBuilderNode(this OutputTriggerMult
         }
 
         [Pure]
-        public static unsafe IVariableProperty<T> ToVariableProperty<T>(this InputDataPort port, [NotNull] GraphInstance instance, [NotNull] GraphDefinition definition) where T : unmanaged
+        public static IVariableProperty<T> ToVariablePropertyReadOnly<T>(this InputDataPort port, [NotNull] GraphInstance instance, [NotNull] GraphDefinition definition) where T : unmanaged
+        {
+            return ToVariableProperty(port, instance, definition, () => new GraphVariableProperty<T>(port));
+        }
+
+        [Pure]
+        public static IVariableProperty<T> ToVariablePropertyReadWrite<T>(this InputDataPort port, [NotNull] GraphInstance instance, [NotNull] GraphDefinition definition) where T : unmanaged
+        {
+            return ToVariableProperty(port, instance, definition, () => ToConstVariable<T>(instance, port));
+        }
+
+        [Pure]
+        private static unsafe IVariableProperty<T> ToConstVariable<T>([NotNull] GraphInstance instance, InputDataPort port) where T : unmanaged
+        {
+            T data;
+            void* ptr = &data;
+            var value = instance.ReadValue(port);
+            Value.SetPtrToValue(ptr, value.Type, value);
+            return new CustomVariableProperty<T> {CustomValue = data};
+        }
+
+        [Pure]
+        private static IVariableProperty<T> ToVariableProperty<T>(
+            InputDataPort port
+          , [NotNull] GraphInstance instance
+          , [NotNull] GraphDefinition definition
+          , [NotNull] Func<IVariableProperty<T>> createGraphVariable
+        ) where T : unmanaged
         {
             var inputPortIndex = (int)port.Port.Index;
             Assert.IsTrue(inputPortIndex < definition.PortInfoTable.Count);
@@ -118,11 +147,10 @@ public static unsafe IVariableProperty<T> ToVariableProperty<T>(this InputDataPo
             if (dataNode is IVisualVariablePropertyNode<T> genericPropertyNode)
                 return genericPropertyNode.GetVariableProperty(instance, definition);
 
-            T data;
-            void* ptr = &data;
-            var value = instance.ReadValue(port);
-            Value.SetPtrToValue(ptr, value.Type, value);
-            return new CustomVariableProperty<T> { CustomValue = data };
+            if (dataNode is IConstantNode)
+                return ToConstVariable<T>(instance, port);
+
+            return createGraphVariable();
         }
 
         // copy from `GraphInstance.GetComponentFieldDescription`

Packages/builder.visual/Runtime/GraphInstanceComponent.cs

@@ -0,0 +1,10 @@
+using Runtime;
+using Unity.Entities;
+
+namespace EntitiesBT.Builder.Visual
+{
+    public class GraphInstanceComponent : IComponentData
+    {
+        public GraphInstance Value;
+    }
+}

Packages/builder.visual/Runtime/GraphInstanceComponent.cs.meta

@@ -29,7 +29,7 @@ public INodeDataBuilder GetBuilder(GraphInstance instance, GraphDefinition defin
             void BuildImpl(BlobBuilder blobBuilder, ref TimerNode data, INodeDataBuilder self, ITreeNode<INodeDataBuilder>[] builders)
             {
                 data.BreakReturnState = @this.BreakReturnState;
-                @this.CountdownSeconds.ToVariableProperty<float>(instance, definition)
+                @this.CountdownSeconds.ToVariablePropertyReadWrite<float>(instance, definition)
                     .Allocate(ref blobBuilder, ref data.CountdownSeconds, self, builders)
                 ;
             }

Packages/builder.visual/Runtime/Node/VisualTimer.cs

@@ -0,0 +1,59 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using EntitiesBT.Core;
+using EntitiesBT.Entities;
+using EntitiesBT.Variable;
+using Runtime;
+using Unity.Entities;
+using UnityEngine.Scripting;
+
+namespace EntitiesBT.Builder.Visual
+{
+    public class GraphVariableProperty<T> : VariableProperty<T> where T : unmanaged
+    {
+        private readonly InputDataPort _port;
+        public override int VariablePropertyTypeId => ID;
+
+        public GraphVariableProperty(InputDataPort port)
+        {
+            _port = port;
+        }
+
+        protected override void AllocateData(ref BlobBuilder builder, ref BlobVariable<T> blobVariable, INodeDataBuilder self, ITreeNode<INodeDataBuilder>[] tree)
+        {
+            builder.Allocate(ref blobVariable, _port);
+        }
+
+        static GraphVariableProperty()
+        {
+            var type = typeof(GraphVariableProperty<T>);
+            VariableRegisters<T>.Register(ID, type.Getter("GetData"), null, GetComponentAccess);
+
+            IEnumerable<ComponentType> GetComponentAccess(ref BlobVariable<T> variable)
+            {
+                return ComponentType.ReadOnly<CurrentBehaviorTreeComponent>().Yield();
+            }
+        }
+
+        public static readonly int ID = new Guid("A3B60190-98AF-42DC-A723-35AD725172BB").GetHashCode();
+
+        [Preserve]
+        private static unsafe T GetData<TNodeBlob, TBlackboard>(ref BlobVariable<T> blobVariable, int index, ref TNodeBlob blob, ref TBlackboard bb)
+            where TNodeBlob : struct, INodeBlob
+            where TBlackboard : struct, IBlackboard
+        {
+            ref var port = ref blobVariable.Value<InputDataPort>();
+            var behaviorTree = bb.GetData<CurrentBehaviorTreeComponent>().RefValue.BehaviorTree;
+            // HACK: how to support multiple worlds?
+            var entityManager = World.DefaultGameObjectInjectionWorld.EntityManager;
+            var graphInstance = entityManager.GetComponentObject<GraphInstanceComponent>(behaviorTree).Value;
+
+            T data;
+            void* ptr = &data;
+            var value = graphInstance.ReadValue(port);
+            Value.SetPtrToValue(ptr, value.Type, value);
+            return data;
+        }
+    }
+}

Packages/builder.visual/Runtime/Variable/GraphVariableProperty.cs

@@ -1,7 +1,7 @@
 {
   "name": "com.quabug.entities-bt.builder.visual",
   "description": "Behavior Tree for Unity Entities - Visual Builder",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "unity": "2020.1",
   "displayName": "EntitiesBT - Visual Builder",
   "dependencies": {