MagnetFieldModule.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 <AtomsUtils/Mesh.h>
 
 
namespace AtomsModules
{
    class MagnetFieldModule : public Atoms::BehaviourModule
    {
    public:
 
        struct DebugData
        {
            AtomsCore::Vector3 pos;
            AtomsCore::Vector3 dir;
            AtomsCore::Vector3 sourcePos;
            double force;
            bool affected;
            bool valid;
        };
 
        MagnetFieldModule();
 
        ~MagnetFieldModule();
 
        void initSimulation(Atoms::AgentGroup* agentGroup);
 
        void initFrame(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;
 
        AtomsUtils::Mesh m_mesh;
 
        AtomsCore::Matrix m_meshMatrix;
 
        AtomsCore::Matrix m_meshInverseMatrix;
 
        bool m_activeDraw;
 
        bool m_first;
 
    };
}