AgentSeparationModule.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/BehaviourModule.h>
#include <AtomsCore/Metadata/Vector3Metadata.h>
 
namespace AtomsModules
{
    class AgentSeparationModule : public Atoms::BehaviourModule
    {
    public:
 
        struct DebugData
        {
            AtomsCore::Vector3 pos;
            std::vector<int> targetAgentsIds;
            std::vector<AtomsCore::Vector3> targetAgentsPos;
            bool valid;
        };
 
 
        AgentSeparationModule();
 
        ~AgentSeparationModule();
 
        void initFrame(const std::vector<Atoms::Agent*>& agents, Atoms::AgentGroup* agentGroup = nullptr);
 
        void endFrame(const std::vector<Atoms::Agent*>& agents, Atoms::AgentGroup* agentGroup = nullptr);
 
        void preDraw(Atoms::DrawContext* context, const std::vector<Atoms::Agent*>& agents, Atoms::AgentGroup* agentGroup = nullptr);
 
        void draw(Atoms::DrawContext* context, const std::vector<Atoms::Agent*>& agents, Atoms::AgentGroup* agentGroup = nullptr);
 
        static Atoms::BehaviourModule* creator(const std::string& parameter);
 
    private:
 
        std::vector<DebugData> m_debugData;
 
        bool m_activeDraw;
    };
}