toppair.C

Go to the documentation of this file.

// toppair
// Test program to test fitter classes:
// Generate Top pairs and let them decay
// B. List 13.4.04


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

#include "jbltools/kinfit/WWFitterFast.h"
#include "jbltools/kinfit/WWFitter.h"
#include "jbltools/kinfit/TopEvent.h"
#include "jbltools/kinfit/ftypes.h"
#include "jbltools/kinfit/hbook.h"



int main(int argc, char** argv) {
  int failed = 0;
  int total = 1000;
  WWFitter fitter;
  
  // ntuple variables
  FInteger IERR;
  FReal FITPROB, CHI2;
  FReal WSTARTMASS[2], TSTARTMASS[2];
  FReal WSTOPMASS[2], TSTOPMASS[2];
  FReal PXSTART, PYSTART;
  FReal PXSTOP, PYSTOP;
//  FReal J1[4], J2[4], J3[4], J4[4], J5[4], J6[4];
  FReal J1E[4], J2E[4], J3E[4], J4E[4], J5E[4], J6E[4];
  FReal J1TRUE[4], J2TRUE[4], J3TRUE[4], J4TRUE[4], J5TRUE[4], J6TRUE[4];
  FReal J1FIT[4], J2FIT[4], J3FIT[4], J4FIT[4], J5FIT[4], J6FIT[4];
  
  FInteger LREC=1024, ISTAT=0;
  hropen (21, "KINFIT", "toppair.hbook", "N", LREC, ISTAT);
  hbnt (999,"kinfit","");
  hbname (999, "FIT", IERR, "IERR:I");
  hbname (999, "FIT", FITPROB, "FITPROB:R");
  hbname (999, "FIT", CHI2, "CHI2:R");
  hbname (999, "FIT", WSTARTMASS[0], "WSTARTMASS(2):R");
  hbname (999, "FIT", WSTOPMASS[0], "WSTOPMASS(2):R");
  hbname (999, "FIT", TSTARTMASS[0], "TSTARTMASS(2):R");
  hbname (999, "FIT", TSTOPMASS[0], "TSTOPMASS(2):R");
  hbname (999, "FIT", PXSTART, "PXSTART:R");
  hbname (999, "FIT", PYSTART, "PYSTART:R");
  hbname (999, "FIT", PXSTOP, "PXSTOP:R");
  hbname (999, "FIT", PYSTOP, "PYSTOP:R");
  hbname (999, "FIT", J1TRUE[0], "J1TRUE(4):R");
  hbname (999, "FIT", J2TRUE[0], "J2TRUE(4):R");
  hbname (999, "FIT", J3TRUE[0], "J3TRUE(4):R");
  hbname (999, "FIT", J4TRUE[0], "J4TRUE(4):R");
  hbname (999, "FIT", J5TRUE[0], "J5TRUE(4):R");
  hbname (999, "FIT", J6TRUE[0], "J6TRUE(4):R");
  hbname (999, "FIT", J1FIT[0], "J1FIT(4):R");
  hbname (999, "FIT", J2FIT[0], "J2FIT(4):R");
  hbname (999, "FIT", J3FIT[0], "J3FIT(4):R");
  hbname (999, "FIT", J4FIT[0], "J4FIT(4):R");
  hbname (999, "FIT", J5FIT[0], "J5FIT(4):R");
  hbname (999, "FIT", J6FIT[0], "J6FIT(4):R");
  hbname (999, "FIT", J1E[0], "J1E(4):R");
  hbname (999, "FIT", J2E[0], "J2E(4):R");
  hbname (999, "FIT", J3E[0], "J3E(4):R");
  hbname (999, "FIT", J4E[0], "J4E(4):R");
  hbname (999, "FIT", J5E[0], "J5E(4):R");
  hbname (999, "FIT", J6E[0], "J6E(4):R");
  
  
  
  for (int i = 0; i < total; i++) {
    TopEvent evt;
    evt.genEvent();
    
    // start values
    WSTARTMASS[0] = evt.getW1Constraint().getMass();
    WSTARTMASS[1] = evt.getW2Constraint().getMass();
    TSTARTMASS[0] = evt.getTopConstraint().getMass(1);
    TSTARTMASS[1] = evt.getTopConstraint().getMass(2);
    PXSTART = evt.getPxConstraint().getValue();
    PYSTART = evt.getPyConstraint().getValue();
    
    if ((IERR = evt.fitEvent(fitter)) != 0) {
      failed++;
      cout << "IERR = " << IERR << endl;
    }
    FITPROB = fitter.getProbability();
    CHI2    = fitter.getChi2();
    
    // stop values
    WSTOPMASS[0] = evt.getW1Constraint().getMass();
    WSTOPMASS[1] = evt.getW2Constraint().getMass();
    TSTOPMASS[0] = evt.getTopConstraint().getMass(1);
    TSTOPMASS[1] = evt.getTopConstraint().getMass(2);
    PXSTOP = evt.getPxConstraint().getValue();
    PYSTOP = evt.getPyConstraint().getValue();
    
    // parameters & errors
    for (int i = 0; i < 3; ++i) {
      J1TRUE[i] = evt.getTrueFitObject(0)->getParam(i);
      J2TRUE[i] = evt.getTrueFitObject(1)->getParam(i);
      J3TRUE[i] = evt.getTrueFitObject(2)->getParam(i);
      J4TRUE[i] = evt.getTrueFitObject(3)->getParam(i);
      J5TRUE[i] = evt.getTrueFitObject(4)->getParam(i);
      J6TRUE[i] = evt.getTrueFitObject(5)->getParam(i);
      J1FIT[i] = evt.getSmearedFitObject(0)->getParam(i);
      J2FIT[i] = evt.getSmearedFitObject(1)->getParam(i);
      J3FIT[i] = evt.getSmearedFitObject(2)->getParam(i);
      J4FIT[i] = evt.getSmearedFitObject(3)->getParam(i);
      J5FIT[i] = evt.getSmearedFitObject(4)->getParam(i);
      J6FIT[i] = evt.getSmearedFitObject(5)->getParam(i);
      J1E[i] = evt.getSmearedFitObject(0)->getError(i);
      J2E[i] = evt.getSmearedFitObject(1)->getError(i);
      J3E[i] = evt.getSmearedFitObject(2)->getError(i);
      J4E[i] = evt.getSmearedFitObject(3)->getError(i);
      J5E[i] = evt.getSmearedFitObject(4)->getError(i);
      J6E[i] = evt.getSmearedFitObject(5)->getError(i);
    }
    
    hfnt(999);
  }
  
  std::cerr << "Failed fits: " << failed 
            << "(" << (100.0*failed)/total << "%)" << endl;
            
  hrout(999);
  hrend("KINFIT");          

  return 0;
}

---