TreeBuilder.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <string>
 
#include "Core/BehaviorTree.h"
#include "Core/Node.h"
#include "Network/Schema/generated/ember_protocol_generated.h"
 
namespace Ember {
namespace Network {
 
class TreeBuilder {
  public:
    struct BuildResult {
        bool success = false;
        std::shared_ptr<EmberCore::BehaviorTree> tree;
        int node_count = 0;
        std::string error;
    };
 
    BuildResult Build(const Protocol::TreeInit *tree_init);
 
  private:
    std::unique_ptr<EmberCore::Node> BuildNode(const Protocol::NodeDefinition *def);
 
    void BuildBlackboards(EmberCore::BehaviorTree *tree,
                          const flatbuffers::Vector<flatbuffers::Offset<Protocol::BlackboardDefinition>> *blackboards);
 
    EmberCore::Node::Type ConvertNodeType(Protocol::NodeType fb_type);
 
    int m_nodeCount = 0;
};
 
} // namespace Network
} // namespace Ember