Changeset 2666 for trunk/src/STFitter.cpp

Timestamp:

10/15/12 15:52:38 (12 years ago)

Author:

Malte Marquarding

Message:

Ticket #173: added setting of constraints on the fitter.

File:

• trunk/src/STFitter.cpp (modified) (6 diffs)

trunk/src/STFitter.cpp

@@ -80 +80 @@
   y_.resize();
   m_.resize();
+  constraints_.clear();
 }
 
@@ -294 +295 @@
 }
 
+void Fitter::addConstraint(const std::vector<float>& constraint)
+{
+  if (funcs_.nelements() == 0)
+    throw (AipsError("Function not yet set."));
+  constraints_.push_back(constraint);
+  
+}
+
+void Fitter::applyConstraints(GenericL2Fit<Float>& fitter)
+{
+  std::vector<std::vector<float> >::const_iterator it;
+  for (it = constraints_.begin(); it != constraints_.end(); ++it) {
+    Vector<Float> tmp(*it);
+    fitter.addConstraint(tmp(Slice(0,tmp.nelements()-1)),
+                         tmp(tmp.nelements()-1));
+  }
+}
+
 bool Fitter::setFixedParameters(std::vector<bool> fixed)
 {
@@ -377 +396 @@
   // Convergence criterium
   fitter.setCriteria(0.001);
+  applyConstraints(fitter);
 
   // Fit
@@ -397 +417 @@
   chisquared_ = fitter.getChi2();
 
-//   residual_.resize();
-//   residual_ =  y_;
-//   fitter.residual(residual_,x_);
   // use fitter.residual(model=True) to get the model
   thefit_.resize(x_.nelements());
   fitter.residual(thefit_,x_,True);
-  // residual = data - model
   residual_.resize(x_.nelements()); 
   residual_ = y_ - thefit_ ;
@@ -419 +435 @@
 
   fitter.setFunction(func);
-  //fitter.setMaxIter(50+n*10);
-  // Convergence criterium
-  //fitter.setCriteria(0.001);
-
-  // Fit
-//   Vector<Float> sigma(x_.nelements());
-//   sigma = 1.0;
+  applyConstraints(fitter);
 
   parameters_.resize();
-//   parameters_ = fitter.fit(x_, y_, sigma, &m_);
   parameters_ = fitter.fit(x_, y_, &m_);
   std::vector<float> ps;
@@ -439 +448 @@
   chisquared_ = fitter.getChi2();
 
-//   residual_.resize();
-//   residual_ =  y_;
-//   fitter.residual(residual_,x_);
-  // use fitter.residual(model=True) to get the model
   thefit_.resize(x_.nelements());
   fitter.residual(thefit_,x_,True);
-  // residual = data - model
   residual_.resize(x_.nelements()); 
   residual_ = y_ - thefit_ ;