testwwfit4.C

Go to the documentation of this file.

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

// now with jet permutations


#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"
#include "jbltools/kinfit/FourJetPairing.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;
  
  enum {NJETS = 4};
   
  // original fit objects - save for next permutation
  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.);

  // these get changed by the fit -> reset after each permutation!
  JetFitObject fitjets[NJETS] = {j1, j2, j3, j4};
  
  // these point allways to the fitjets array, which gets reset.
  JetFitObject *jets[NJETS];
  for (int i = 0; i < NJETS; ++i) jets[i] = &fitjets[i];
  
  FourJetPairing pairing (jets);
  JetFitObject *permutedjets[NJETS];
  
  for (int iperm = 0; iperm < pairing.getNPerm(); iperm++) {
  
    cout << "=================================================" << endl;
    cout << "             iperm = " << iperm << endl; 
    
    // important: (re-)set fitjets array!
    fitjets[0] = j1;
    fitjets[1] = j2;
    fitjets[2] = j3;
    fitjets[3] = j4;
    
    pairing.nextPermutation (permutedjets);
  
    for (int i = 0; i < NJETS; ++i) 
       cout << "start four-vector of jet " << i << ": " << *(permutedjets[i]) << endl;
  
    PxConstraint pxc;
    for (int i = 0; i < NJETS; ++i) 
       pxc.addToFOList (*(permutedjets[i]));
  
    PyConstraint pyc;
    for (int i = 0; i < NJETS; ++i) 
       pyc.addToFOList (*(permutedjets[i]));
  
    MassConstraint w(0.);
    MassConstraint w1(80.4);
    MassConstraint w2(80.4);
 
    if (equal_mass_constraint) {
      
      w.addToFOList (*(permutedjets[0]), 1);
      w.addToFOList (*(permutedjets[1]), 1);
      w.addToFOList (*(permutedjets[2]), 2);
      w.addToFOList (*(permutedjets[3]), 2);
      cout << "start mass of W 1: " << w.getMass(1) << endl;
      cout << "start mass of W 2: " << w.getMass(2) << endl;
  
    }
    else {  
  
      w1.addToFOList (*(permutedjets[0]));
      w1.addToFOList (*(permutedjets[1]));
      cout << "start mass of W 1: " << w1.getMass() << endl;
  
      w2.addToFOList (*(permutedjets[2]));
      w2.addToFOList (*(permutedjets[3]));
      cout << "start mass of W 2: " << w2.getMass() << endl; 
       
    }
  
    WWFitter fitter;
    for (int i = 0; i < NJETS; ++i) 
       fitter.addFitObject (*(permutedjets[i]));
    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;
    for (int i = 0; i < NJETS; ++i) 
       cout << "final four-vector of jet " << i << ": " << *(permutedjets[i]) << 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;
  
  
   
}

---