Fix to allow a correct use of the bounds and scales setter

roboptim · May 21, 2015 · e41e510 · e41e510
1 parent b3cc8e0
commit e41e510
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/include/roboptim/core/manifold-map/decorator/problem-factory.hh b/include/roboptim/core/manifold-map/decorator/problem-factory.hh
@@ -45,8 +45,8 @@ namespace roboptim
   template<class U>
   struct BoundsAndScalesSetter
   {
-    void setBounds(typename Function::intervals_t& bounds);
-    void setScales(typename U::scales_t& scales);
+    BoundsAndScalesSetter<U>& setBounds(typename Function::intervals_t& bounds);
+    BoundsAndScalesSetter<U>& setScales(typename U::scales_t& scales);
 
   private:
     std::pair<typename Function::intervals_t, typename U::scales_t>& bNSPair_;

diff --git a/include/roboptim/core/manifold-map/decorator/problem-factory.hxx b/include/roboptim/core/manifold-map/decorator/problem-factory.hxx
@@ -236,17 +236,21 @@ ProblemFactory<U>::ProblemFactory()
 // ---- //
 
 template<class U>
-void BoundsAndScalesSetter<U>::setBounds(typename Function::intervals_t& bounds)
+BoundsAndScalesSetter<U>& BoundsAndScalesSetter<U>::setBounds(typename Function::intervals_t& bounds)
 {
   this->bNSPair_.first.clear();
   this->bNSPair_.first.insert(this->bNSPair_.first.end(), bounds.begin(), bounds.end());
+
+  return *this;
 }
 
 template<class U>
-void BoundsAndScalesSetter<U>::setScales(typename U::scales_t& scales)
+BoundsAndScalesSetter<U>& BoundsAndScalesSetter<U>::setScales(typename U::scales_t& scales)
 {
   this->bNSPair_.second.clear();
   this->bNSPair_.second.insert(this->bNSPair_.second.end(), scales.begin(), scales.end());
+
+  return *this;
 }
 
 template<class U>