dynamics_matrix_descriptor.h

Go to the documentation of this file.

#ifndef DYNAMICS_MATRIX_DESCRIPTOR_H
#define DYNAMICS_MATRIX_DESCRIPTOR_H
 
#include "bitrl/bitrl_types.h"
 
#include <map>
#include <string>
 
namespace bitrl
{
namespace dynamics
{
 
class DynamicsMatrixDescriptor
{
 
  public:
    typedef DynMat<real_t> matrix_type;
    typedef DynVec<real_t> vector_type;
 
    typedef std::map<std::string, matrix_type>::iterator matrix_iterator;
    typedef std::map<std::string, matrix_type>::const_iterator const_matrix_iterator;
 
    typedef std::map<std::string, vector_type>::iterator vector_iterator;
    typedef std::map<std::string, vector_type>::const_iterator const_vector_iterator;
 
    DynamicsMatrixDescriptor();
 
    ~DynamicsMatrixDescriptor() {}
 
    matrix_iterator find_matrix(const std::string &name);
 
    const_matrix_iterator find_matrix(const std::string &name) const;
 
    vector_iterator find_vector(const std::string &name);
 
    const_vector_iterator find_vector(const std::string &name) const;
 
    matrix_type &get_matrix(const std::string &name);
    const matrix_type &get_matrix(const std::string &name) const;
    void set_matrix(const std::string &name, const matrix_type &mat)
    {
        matrices_.insert_or_assign(name, mat);
    }
    bool has_matrix(const std::string &name) const;
 
    vector_type &get_vector(const std::string &name);
    const vector_type &get_vector(const std::string &name) const;
    void set_vector(const std::string &name, const vector_type &vec)
    {
        vectors_.insert_or_assign(name, vec);
    }
 
    matrix_iterator matrix_begin() { return matrices_.begin(); }
    matrix_iterator matrix_end() { return matrices_.end(); }
    const_matrix_iterator matrix_begin() const { return matrices_.cbegin(); }
    const_matrix_iterator matrix_end() const { return matrices_.cend(); }
 
    vector_iterator vector_begin() { return vectors_.begin(); }
    vector_iterator vector_end() { return vectors_.end(); }
    const_vector_iterator vector_begin() const { return vectors_.cbegin(); }
    const_vector_iterator vector_end() const { return vectors_.cend(); }
 
  protected:
    std::map<std::string, matrix_type> matrices_;
    std::map<std::string, vector_type> vectors_;
};
 
} // namespace dynamics
} // namespace bitrl
 
#endif // DYNAMICS_MATRIX_DESCRIPTOR_H