PyConstraint.C

Go to the documentation of this file.

#include "jbltools/kinfit/PyConstraint.h"
#include "jbltools/kinfit/ParticleFitObject.h"

#include<iostream>
#include<cassert>

using std::cout;
using std::endl;

// constructor
PyConstraint::PyConstraint () {}

// destructor
PyConstraint::~PyConstraint () {}

// calulate current value of constraint function
double PyConstraint::getValue() const {
  double totpy = 0;
  for (unsigned int i = 0; i < fitobjects.size(); i++) {
    totpy += fitobjects[i]->getPy(); 
  }
  return totpy;
}

// calculate vector/array of derivatives of this contraint 
// w.r.t. to ALL parameters of all fitobjects
// here: d sum(py) /d par(i,j) 
//                      = d sum(py) /d py(i) * d py(i) /d par(i, j)
//                                      =  1 * d py(i) /d par(i, j)
void PyConstraint::getDerivatives(int idim, double der[]) const {
  for (unsigned int i = 0; i < fitobjects.size(); i++) {
    for (int ilocal = 0; ilocal < fitobjects[i]->getNPar(); ilocal++) {
      int iglobal = fitobjects[i]->getGlobalParNum (ilocal);
      if (!fitobjects[i]->isParamFixed(ilocal)) {
        assert (iglobal >= 0 && iglobal < idim);
        der[iglobal] = fitobjects[i]->getDPy(ilocal);
//       cout << "Py: der[" << iglobal << "] = " << der[iglobal] 
//            << " for jet " << i << " and ilocal = " << ilocal << endl;
      }
    }
  }
}
  
void PyConstraint::add1stDerivativesToMatrix(int idim, double *M) const {
  assert (M);
  int kglobal = getGlobalNum();
  assert (kglobal >= 0 && kglobal < idim);
  
  for (ConstFitObjectIterator i = fitobjects.begin(); i != fitobjects.end(); ++i) {
    const ParticleFitObject *fo = *i;
    assert (fo);
    for (int ilocal = 0; ilocal < fo->getNPar(); ++ilocal) {
      if (!fo->isParamFixed(ilocal)) {
        int iglobal = fo->getGlobalParNum (ilocal);
        assert (iglobal >= 0 && iglobal < idim);
        double d = fo->getDPy(ilocal);
        M[idim*iglobal+kglobal] += d;
        M[idim*kglobal+iglobal] += d;
      }
    }
  }
}
  
void PyConstraint::add2ndDerivativesToMatrix(int idim, double *M, double lambda) const {
  assert (M);
  for (ConstFitObjectIterator i = fitobjects.begin(); i != fitobjects.end(); ++i) {
    const ParticleFitObject *fo = *i;
    assert (fo);
    fo->addTo2ndDerivatives(M, idim, 0, lambda, 0, 0);
  }
}

  

  

---