testwwfit2.C

Go to the documentation of this file.

// testwwfit
// Test program to test fitter classes 
// J. Boehme, B. List 3.4.04


#include <iostream.h>              // - cout
using std::cout;
using std::endl;

#include "jbltools/kinfit/JetFitObject.h"
#include "jbltools/kinfit/PxConstraint.h"
#include "jbltools/kinfit/PyConstraint.h"
#include "jbltools/kinfit/MassConstraint.h"
#include "jbltools/kinfit/WWFitter.h"


int main(int argc, char** argv) {
 
  // W+W- -> 4 jets event
  
  // false: each W is constraint individually
  // true: equal mass constraint
  bool equal_mass_constraint = false;
   
  JetFitObject j1 (47., 0.84 , 0.64,  5.0, 0.1, 0.1, 0.);
  JetFitObject j2 (45., 2.30 , 2.50,  5.0, 0.1, 0.1, 0.);
  JetFitObject j3 (46., 0.996, 3.83,  5.0, 0.1, 0.1, 0.);
  JetFitObject j4 (42., 2.21 , 5.82,  5.0, 0.1, 0.1, 0.);
  
  cout << "start four-vector of jet 1: " << j1 << endl;
  cout << "start four-vector of jet 2: " << j2 << endl;
  cout << "start four-vector of jet 3: " << j3 << endl;
  cout << "start four-vector of jet 4: " << j4 << endl;
  
  PxConstraint pxc;
  pxc.addToFOList (j1);
  pxc.addToFOList (j2);
  pxc.addToFOList (j3);
  pxc.addToFOList (j4);
  
  PyConstraint pyc;
  pyc.addToFOList (j1);
  pyc.addToFOList (j2);
  pyc.addToFOList (j3);
  pyc.addToFOList (j4);
  
  MassConstraint w(0.);
  MassConstraint w1(80.4);
  MassConstraint w2(80.4);
 
  if (equal_mass_constraint) {
    
    w.addToFOList (j1, 1);
    w.addToFOList (j2, 1);
    w.addToFOList (j3, 2);
    w.addToFOList (j4, 2);
    cout << "start mass of W 1: " << w.getMass(1) << endl;
    cout << "start mass of W 2: " << w.getMass(2) << endl;
  
  }
  else {  
  
    w1.addToFOList (j1);
    w1.addToFOList (j2);
    cout << "start mass of W 1: " << w1.getMass() << endl;
  
    w2.addToFOList (j3);
    w2.addToFOList (j4);
    cout << "start mass of W 2: " << w2.getMass() << endl; 
     
  }
  
  WWFitter fitter;
  fitter.addFitObject (j1);
  fitter.addFitObject (j2);
  fitter.addFitObject (j3);
  fitter.addFitObject (j4);
  fitter.addConstraint (pxc);
  fitter.addConstraint (pyc);
  if (equal_mass_constraint) fitter.addConstraint (w);
  else {
    fitter.addConstraint (w1);
    fitter.addConstraint (w2);
  }
  
  double prob = fitter.fit();
  cout << "fit probability = " << prob << endl;
  cout << "final four-vector of jet 1: " << j1 << endl;
  cout << "final four-vector of jet 2: " << j2 << endl;
  cout << "final four-vector of jet 3: " << j3 << endl;
  cout << "final four-vector of jet 4: " << j4 << endl;
  if (equal_mass_constraint) {
    cout << "final mass of W 1: " << w.getMass(1) << endl;
    cout << "final mass of W 2: " << w.getMass(2) << endl;
  }  
  else {
    cout << "final mass of W 1: " << w1.getMass() << endl;
    cout << "final mass of W 2: " << w2.getMass() << endl;
  }
  
  return 0;
  
  
   
}

---