Changeset 1232 for trunk/src/STFitter.cpp

Timestamp:

09/01/06 13:14:11 (18 years ago)

Author:

mar637

Message:

Merge from Release2.1.0b tag

File:

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

trunk/src/STFitter.cpp

@@ -64 +64 @@
   funcs_.resize(0,True);
   parameters_.resize();
+  fixedpar_.resize();
   error_.resize();
   thefit_.resize();
@@ -205 +206 @@
     if (parameters_.nelements() > 0 && tmppar.nelements() != parameters_.nelements())
         throw (AipsError("Number of parameters inconsistent with function."));
-    if (parameters_.nelements() == 0)
+    if (parameters_.nelements() == 0) {
         parameters_.resize(tmppar.nelements());
-        fixedpar_.resize(tmppar.nelements());
-        fixedpar_ = False;
+        if (tmppar.nelements() != fixedpar_.nelements()) {
+            fixedpar_.resize(tmppar.nelements());
+            fixedpar_ = False;
+        }
+    }
     if (dynamic_cast<Gaussian1D<Float>* >(funcs_[0]) != 0) {
         uInt count = 0;
@@ -224 +228 @@
         }
     }
+    // reset
+    if (params.size() == 0) {
+        parameters_.resize();
+        fixedpar_.resize();
+    }
     return true;
 }
@@ -229 +238 @@
 bool Fitter::setFixedParameters(std::vector<bool> fixed)
 {
-    Vector<Bool> tmp(fixed);
     if (funcs_.nelements() == 0)
         throw (AipsError("Function not yet set."));
-    if (fixedpar_.nelements() > 0 && tmp.nelements() != fixedpar_.nelements())
+    if (fixedpar_.nelements() > 0 && fixed.size() != fixedpar_.nelements())
         throw (AipsError("Number of mask elements inconsistent with function."));
+    if (fixedpar_.nelements() == 0) {
+        fixedpar_.resize(parameters_.nelements());
+        fixedpar_ = False;
+    }
     if (dynamic_cast<Gaussian1D<Float>* >(funcs_[0]) != 0) {
         uInt count = 0;
         for (uInt j=0; j < funcs_.nelements(); ++j) {
             for (uInt i=0; i < funcs_[j]->nparameters(); ++i) {
-                funcs_[j]->mask(i) = !tmp[count];
-                fixedpar_[count] = !tmp[count];
+                funcs_[j]->mask(i) = !fixed[count];
+                fixedpar_[count] = fixed[count];
                 ++count;
             }
@@ -245 +257 @@
     } else if (dynamic_cast<Polynomial<Float>* >(funcs_[0]) != 0) {
         for (uInt i=0; i < funcs_[0]->nparameters(); ++i) {
-            fixedpar_[i] = tmp[i];
-            funcs_[0]->mask(i) =  tmp[i];
-        }
-    }
-    //fixedpar_ = !tmpmsk;
+            fixedpar_[i] = fixed[i];
+            funcs_[0]->mask(i) =  !fixed[i];
+        }
+    }
     return true;
 }
@@ -263 +274 @@
   Vector<Bool> out(parameters_.nelements());
   if (fixedpar_.nelements() == 0) {
-    out = False;
+    return std::vector<bool>();
     //throw (AipsError("No parameter mask set."));
   } else {
@@ -312 +323 @@
   residual_ =  y_;
   fitter.residual(residual_,x_);
-
   // use fitter.residual(model=True) to get the model
   thefit_.resize(x_.nelements());