Fix serialization of ISAM2 class

borglab · Jul 9, 2021 · df579ec · df579ec
1 parent cd3854a
commit df579ec
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/gtsam/inference/VariableIndex.h b/gtsam/inference/VariableIndex.h
@@ -182,6 +182,16 @@ class GTSAM_EXPORT VariableIndex {
     return item->second; 
   }
 
+private:
+  /** Serialization function */
+  friend class boost::serialization::access;
+  template<class ARCHIVE>
+  void serialize(ARCHIVE & ar, const unsigned int /*version*/) {
+    ar & BOOST_SERIALIZATION_NVP(index_);
+    ar & BOOST_SERIALIZATION_NVP(nFactors_);
+    ar & BOOST_SERIALIZATION_NVP(nEntries_);
+  }
+
   /// @}
 };
 

diff --git a/gtsam/nonlinear/ISAM2.h b/gtsam/nonlinear/ISAM2.h
@@ -315,6 +315,26 @@ class GTSAM_EXPORT ISAM2 : public BayesTree<ISAM2Clique> {
   void removeVariables(const KeySet& unusedKeys);
 
   void updateDelta(bool forceFullSolve = false) const;
+
+ private:
+  /** Serialization function */
+  friend class boost::serialization::access;
+  template<class ARCHIVE>
+  void serialize(ARCHIVE & ar, const unsigned int /*version*/) {
+      ar & boost::serialization::base_object<BayesTree<ISAM2Clique> >(*this);
+      ar & BOOST_SERIALIZATION_NVP(theta_);
+      ar & BOOST_SERIALIZATION_NVP(variableIndex_);
+      ar & BOOST_SERIALIZATION_NVP(delta_);
+      ar & BOOST_SERIALIZATION_NVP(deltaNewton_);
+      ar & BOOST_SERIALIZATION_NVP(RgProd_);
+      ar & BOOST_SERIALIZATION_NVP(deltaReplacedMask_);
+      ar & BOOST_SERIALIZATION_NVP(nonlinearFactors_);
+      ar & BOOST_SERIALIZATION_NVP(linearFactors_);
+      ar & BOOST_SERIALIZATION_NVP(doglegDelta_);
+      ar & BOOST_SERIALIZATION_NVP(fixedVariables_);
+      ar & BOOST_SERIALIZATION_NVP(update_count_);
+  }
+
 };  // ISAM2
 
 /// traits