gtsam/cpp/VectorConfig.h

112 lines
2.9 KiB
C
Raw Normal View History

2009-08-22 06:23:24 +08:00
/**
* @file VectorConfig.h
2009-08-22 06:23:24 +08:00
* @brief Factor Graph Configuration
* @author Carlos Nieto
* @author Christian Potthast
*/
// \callgraph
#pragma once
#include <map>
#include <boost/serialization/map.hpp>
#include "Testable.h"
#include "Vector.h"
2009-08-22 06:23:24 +08:00
namespace gtsam {
/** Factor Graph Configuration */
class VectorConfig : public Testable<VectorConfig> {
2009-08-22 06:23:24 +08:00
protected:
/** Map from string indices to values */
std::map<std::string, Vector> values;
public:
typedef std::map<std::string, Vector>::iterator iterator;
typedef std::map<std::string, Vector>::const_iterator const_iterator;
2009-10-22 22:42:19 +08:00
VectorConfig() {}
VectorConfig(const VectorConfig& cfg_in): values(cfg_in.values) {}
2009-08-22 06:23:24 +08:00
virtual ~VectorConfig() {}
2009-08-22 06:23:24 +08:00
/** return all the nodes in the graph **/
2009-12-12 05:38:08 +08:00
std::vector<std::string> get_names() const;
2009-08-22 06:23:24 +08:00
/** Insert a value into the configuration with a given index */
2009-12-12 05:38:08 +08:00
VectorConfig& insert(const std::string& name, const Vector& val);
2009-08-22 06:23:24 +08:00
/** Add to vector at position j */
2009-12-12 05:38:08 +08:00
void add(const std::string& j, const Vector& a);
2009-08-22 06:23:24 +08:00
/**
* Add a delta config, needed for use in NonlinearOptimizer
* For VectorConfig, this is just addition.
2009-08-22 06:23:24 +08:00
*/
2009-12-12 06:43:34 +08:00
VectorConfig exmap(const VectorConfig& delta) const;
/**
* Add a delta vector (not a config)
* Will use the ordering that map uses to loop over vectors
*/
VectorConfig exmap(const Vector& delta) const;
2009-08-29 04:06:29 +08:00
const_iterator begin() const {return values.begin();}
const_iterator end() const {return values.end();}
2009-08-22 06:23:24 +08:00
/** get a vector in the configuration by name */
const Vector& get(const std::string& name) const;
2009-08-22 06:23:24 +08:00
/** operator[] syntax for get */
inline const Vector& operator[](const std::string& name) const { return get(name); }
2009-08-22 06:23:24 +08:00
bool contains(const std::string& name) const {
const_iterator it = values.find(name);
2009-12-12 05:38:08 +08:00
return (it!=values.end());
2009-08-22 06:23:24 +08:00
}
2009-12-12 05:38:08 +08:00
/** Nr of vectors */
size_t size() const { return values.size();}
/** Total dimensionality */
size_t dim() const;
/** Scale */
VectorConfig scale(double s) const;
VectorConfig operator*(double s) const;
/** Add */
VectorConfig operator+(const VectorConfig &b) const;
/** Subtract */
VectorConfig operator-(const VectorConfig &b) const;
2009-08-22 06:23:24 +08:00
/** print */
void print(const std::string& name = "") const;
/** equals, for unit testing */
bool equals(const VectorConfig& expected, double tol=1e-9) const;
2009-08-22 06:23:24 +08:00
void clear() {values.clear();}
2009-12-12 05:38:08 +08:00
/** Dot product */
double dot(const VectorConfig& b) const;
2009-08-22 06:23:24 +08:00
private:
/** Serialization function */
friend class boost::serialization::access;
template<class Archive>
void serialize(Archive & ar, const unsigned int version)
{
ar & BOOST_SERIALIZATION_NVP(values);
}
2009-12-12 05:38:08 +08:00
}; // VectorConfig
/** Dot product */
double dot(const VectorConfig&, const VectorConfig&);
} // gtsam