state.hh

Go to the documentation of this file.

#ifndef UTOPIA_MODELS_SIMPLEFLOCKING_STATE_HH
#define UTOPIA_MODELS_SIMPLEFLOCKING_STATE_HH
 
#include <cmath>
#include <random>
 
#include <utopia/core/types.hh>
 
#include "utils.hh"
 
 
namespace Utopia::Models::SimpleFlocking {
 
class AgentState {
    double speed;
 
 
    double orientation;
 
    SpaceVecType<2> displacement;
 
public:
    AgentState()
    :
        speed(0.)
    ,   orientation(0.)
    ,   displacement({0., 0.})
    {}
 
    template<typename RNGType>
    AgentState(const Config& cfg, const std::shared_ptr<RNGType>& rng)
    :
        speed(get_as<double>("speed", cfg, 0.))
    ,   orientation(random_angle(rng))
    ,   displacement({0., 0.})
    {
        update_displacement();
    }
 
    // .. Getters .............................................................
 
    auto get_speed () const {
        return speed;
    }
 
 
    auto get_orientation () const {
        return orientation;
    }
 
    const auto& get_displacement () const {
        return displacement;
    }
 
    // .. Setters .............................................................
 
    auto set_speed (double new_speed) {
        speed = new_speed;
        update_displacement();
    }
 
 
    auto set_orientation (double new_orientation) {
        orientation = constrain_angle(new_orientation);
        update_displacement();
    }
 
protected:
    // .. Helpers .............................................................
 
    void update_displacement () {
        displacement[0] = speed * std::cos(orientation);
        displacement[1] = speed * std::sin(orientation);
    }
};
 
 
} // namespace Utopia::Models::SimpleFlocking
 
#endif // UTOPIA_MODELS_SIMPLEFLOCKING_STATE_HH