ProbabilitySelector.h

#pragma once
// ===========================================================================
// Copyright (c) 2015 Toolchefs Ltd. All rights reserved.
//
// Use of this software is subject to the terms of the Toolchefs license
// agreement provided at the time of installation or download, or which
// otherwise accompanies this software in either electronic or hard copy form.
// ===========================================================================
 
#include <Atoms/BehaviourTree/Composite.h>
#include <Atoms/BehaviourTree/BlackboardValue.h>
#include <vector>
 
namespace Atoms
{
    class ATOMS_EXPORT ProbabilitySelector : public Composite
    {
 
    public:
 
        struct ProbabilitySelectorData
        {
            float probability;
            float currentProbability;
            std::vector<int> failedIndeces;
            int index;
        };
 
        ProbabilitySelector();
 
        virtual ~ProbabilitySelector();
 
        virtual const char* typeName() const override;
 
        virtual void initialize(BehaviourTreeContext* context, State* data) override;
 
        virtual void update(BehaviourTreeContext* context, State* data) override;
 
        virtual void terminate(BehaviourTreeContext* context, State* data) override;
 
        virtual void releaseData(BehaviourTreeContext* context, State* data) override;
 
        virtual void onChildUpdated(BehaviourTreeContext* context, State* data, unsigned int childIndex, unsigned short childStatus) override;
 
        virtual void setAttributes(const AtomsCore::MapMetadata* attributes, Blackboard* blackboard) override;
 
        virtual void getAttributes(AtomsCore::MapMetadata* attributes, Blackboard* blackboard) override;
 
        virtual void getAttributeProperties(AtomsCore::MapMetadata* attributes) override;
 
        static Behaviour* creator();
 
        double getTotal(BehaviourTreeContext* context, State* data);
 
    private:
 
        BlackboardIntValue m_seed;
 
        BlackboardDoubleValue m_failChance;
 
        std::vector<BlackboardDoubleValue> m_weights;
 
        bool m_useAgentIdAsSeed;
 
        bool m_useAnimatedSeed;
    };
}