testwwfit3.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/NeutrinoFitObject.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- -> 2 jets, 1 lepton, 1 neutrino event
  
  // false: each W is constraint individually
  // true: equal mass constraint
  bool equal_mass_constraint = false;
   
  JetFitObject j1 (45.7, 0.837 , 0.647,  5.0, 0.1, 0.1, 0.);
  JetFitObject j2 (44.3, 2.293 , 2.501,  5.0, 0.1, 0.1, 0.);
  JetFitObject j3 (47.3,  0.998,  3.822,  5.0, 0.1, 0.5, 0.);
//  NeutrinoFitObject n4 (42.6, 2.216 , 5.819,  5.0, 0.1, 0.1, 0.);
//   NeutrinoFitObject n4 (35., 2. , 6.,  5.0, 0.1, 0.1, 0.);
   NeutrinoFitObject n4 (10., 1.5, 0.,  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: " << n4 << endl;
  
  PxConstraint pxc;
  pxc.addToFOList (j1);
  pxc.addToFOList (j2);
  pxc.addToFOList (j3);
  pxc.addToFOList (n4);
  cout << "start value of px constraint: " << pxc.getValue() << endl;
  
  PyConstraint pyc;
  pyc.addToFOList (j1);
  pyc.addToFOList (j2);
  pyc.addToFOList (j3);
  pyc.addToFOList (n4);
  cout << "start value of py constraint: " << pyc.getValue() << endl;
  
  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 (n4, 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;
//    cout << "start value of w1 constraint: " << w1.getValue() << endl;
  
    w2.addToFOList (j3);
    w2.addToFOList (n4);
    cout << "start mass of W 2: " << w2.getMass() << endl; 
    cout << "start value of w2 constraint: " << w2.getValue() << endl;
     
  }
  
  WWFitter fitter;
  fitter.addFitObject (j1);
  fitter.addFitObject (j2);
  fitter.addFitObject (j3);
  fitter.addFitObject (n4);
  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: " << n4 << endl;
  
  cout << "final value of px constraint: " << pxc.getValue() << endl;
  cout << "final value of py constraint: " << pyc.getValue() << endl;
  if (equal_mass_constraint) {
    cout << "final value of w  constraint: " << w.getValue() << endl;
    cout << "final mass of W 1: " << w.getMass(1) << endl;
    cout << "final mass of W 2: " << w.getMass(2) << endl;
  }  
  else {
    cout << "final value of w1 constraint: " << w1.getValue() << endl;
    cout << "final value of w2 constraint: " << w2.getValue() << endl;
    cout << "final mass of W 1: " << w1.getMass() << endl;
    cout << "final mass of W 2: " << w2.getMass() << endl;
  }
  
  return 0;
  
  
   
}

---