diff --git a/gtsam/base/MatrixSerialization.h b/gtsam/base/MatrixSerialization.h index 11b6a417a..43c97097d 100644 --- a/gtsam/base/MatrixSerialization.h +++ b/gtsam/base/MatrixSerialization.h @@ -24,6 +24,7 @@ #include +#include #include #include #include @@ -87,6 +88,45 @@ void serialize(Archive& ar, gtsam::Matrix& m, const unsigned int version) { split_free(ar, m, version); } +/******************************************************************************/ +/// Customized functions for serializing Eigen::SparseVector +template +void save(Archive& ar, const Eigen::SparseVector<_Scalar, _Options, _Index>& m, + const unsigned int /*version*/) { + _Index size = m.size(); + + std::vector> data; + for (typename Eigen::SparseVector<_Scalar, _Options, _Index>::InnerIterator + it(m); + it; ++it) + data.push_back({it.index(), it.value()}); + + ar << BOOST_SERIALIZATION_NVP(size); + ar << BOOST_SERIALIZATION_NVP(data); +} + +template +void load(Archive& ar, Eigen::SparseVector<_Scalar, _Options, _Index>& m, + const unsigned int /*version*/) { + _Index size; + ar >> BOOST_SERIALIZATION_NVP(size); + m.resize(size); + + std::vector> data; + ar >> BOOST_SERIALIZATION_NVP(data); + + for (auto&& d : data) { + m.coeffRef(d.first) = d.second; + } +} + +template +void serialize(Archive& ar, Eigen::SparseVector<_Scalar, _Options, _Index>& m, + const unsigned int version) { + split_free(ar, m, version); +} +/******************************************************************************/ + } // namespace serialization } // namespace boost #endif