DirectionField.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 <AtomsModules/Globals.h>
#include <Atoms/BehaviourModule.h>
#include <AtomsUtils/Mesh.h>
#include <AtomsUtils/KdTreePoint.h>
 
namespace AtomsModules
{
    class ATOMSMODULES_EXPORT DirectionFieldModule : public Atoms::BehaviourModule
    {
    public:
 
        DirectionFieldModule();
 
        virtual ~DirectionFieldModule();
 
        void initSimulation(Atoms::AgentGroup* agentGroup);
 
        void initFrame(const std::vector<Atoms::Agent*>& agents, Atoms::AgentGroup* agentGroup);
 
        void draw(Atoms::DrawContext* context, const std::vector<Atoms::Agent*>& agents, Atoms::AgentGroup* agentGroup = nullptr);
 
        static Atoms::BehaviourModule* creator(const std::string& parameter);
 
        inline const std::vector<AtomsCore::Vector3>& samples() const { return m_samples; }
 
        inline const AtomsCore::MapMetadata& tangentSamples() const { return m_tangentSamples; }
 
        inline void setTangentSamples(const AtomsCore::MapMetadata& tSamples) { m_tangentSamples = tSamples; }
 
        inline AtomsPtr<AtomsUtils::Mesh> mesh() { return m_mesh; }
 
    private:
 
        std::vector<AtomsCore::Vector3> m_samples;
 
        std::vector<AtomsCore::Vector3> m_samplePoints;
 
        std::vector<AtomsCore::Vector3> m_debugDirectionPoints;
 
        std::vector<AtomsCore::Vector3> m_debugInfluencesPoints;
 
        std::vector<std::pair<int,std::string>> m_samplesKeys;
 
        AtomsCore::MapMetadata m_tangentSamples;
 
        AtomsUtils::KdTreePoint m_tree;
 
        AtomsPtr<AtomsUtils::Mesh> m_mesh;
 
        bool m_debugActive;
    };
}