ProbabilityNode.h

Go to the documentation of this file.

#include "mtrand.h"

namespace BehaviorTree
{

struct WeightedBehaviorTreeNode {
  float weight;
  BehaviorTreeNode* node;

  WeightedBehaviorTreeNode(float weight_, BehaviorTreeNode* node_) : weight(weight_), node(node_) {}

  bool isNull() const { return (node == NULL); }

  static const WeightedBehaviorTreeNode NULL_WEIGHTED_NODE;
};


class ProbabilityNode:public BehaviorTreeInternalNode
{
public:
  ProbabilityNode();
  virtual ~ProbabilityNode();

  virtual BehaviorTreeNode* setChildren(BehaviorTreeNode* node, ...);

  virtual BehaviorTreeNode* setWeightedChildren(WeightedBehaviorTreeNode weightedNode, ...);

#if is_computer()
  virtual ProbabilityNode* addChild(BehaviorTreeNode* node, float weight);
#endif

  void init(void* object);
    BEHAVIOR_STATUS execute(void* object);

private:
    MTRand_closed random;
    float totalSum;
    BehaviorTreeNode* currentNode;
    float* weighting;
};

}