void TTbar_plus_X_analyser::fillCommonTreesNoBSelection(const EventPtr event, const unsigned int selectionCriteria, std::string folder ) {
SelectionCriteria::selection selection = SelectionCriteria::selection(selectionCriteria);
// Jets
const JetCollection jets(event->CleanedJets());
// B Jets
const JetCollection bJets(event->CleanedBJets());
// Lepton
const LeptonPointer signalLepton = event->getSignalLepton( selection );
treeMan_->setCurrentFolder(folder);
treeMan_->Fill("EventWeight", event->weight());
treeMan_->Fill("PUWeight", event->PileUpWeight());
treeMan_->Fill("PUWeight_up", event->PileUpWeight(1));
treeMan_->Fill("PUWeight_down", event->PileUpWeight(-1));
treeMan_->Fill("NJets",Event::NJets(jets));
treeMan_->Fill("NBJets",Event::NJets(bJets));
treeMan_->Fill("NVertices", event->Vertices().size());
treeMan_->Fill("BJetWeight",event->BJetWeight());
treeMan_->Fill("BJetUpWeight",event->BJetUpWeight());
treeMan_->Fill("BJetDownWeight",event->BJetDownWeight());
treeMan_->Fill("LightJetUpWeight",event->LightJetUpWeight());
treeMan_->Fill("LightJetDownWeight",event->LightJetDownWeight());
if (selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsReference)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIso03DeltaBeta());
}
else if (selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsReference)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIso04DeltaBeta());
}
fillLeptonEfficiencyCorrectionBranches( event, selectionCriteria, signalLepton );
}
void TTbar_plus_X_analyser::fillCommonTreesNoBSelection(const EventPtr event, const unsigned int selectionCriteria, std::string folder ) {
SelectionCriteria::selection selection = SelectionCriteria::selection(selectionCriteria);
// Jets
const JetCollection jets(event->CleanedJets());
// B Jets
const JetCollection bJets(event->CleanedBJets());
// Lepton
const LeptonPointer signalLepton = event->getSignalLepton( selection );
// MET
const METPointer MET_original(event->MET((METAlgorithm::value) 0));
treeMan_->setCurrentFolder(folder);
treeMan_->Fill("EventWeight", event->weight());
treeMan_->Fill("PUWeight", event->PileUpWeight());
treeMan_->Fill("PUWeight_up", event->PileUpWeight(1));
treeMan_->Fill("PUWeight_down", event->PileUpWeight(-1));
treeMan_->Fill("M3",Event::M3(jets));
if ( Event::NJets(bJets) > 0 ) {
treeMan_->Fill("M_bl",Event::M_bl(bJets, signalLepton));
treeMan_->Fill("angle_bl",Event::angle_bl(bJets, signalLepton));
}
treeMan_->Fill("HT",Event::HT(jets));
treeMan_->Fill("MET",MET_original->et());
treeMan_->Fill("MET_phi",MET_original->phi());
treeMan_->Fill("ST",Event::ST(jets, signalLepton, MET_original));
treeMan_->Fill("WPT",Event::WPT(signalLepton, MET_original));
treeMan_->Fill("MT",Event::MT(signalLepton, MET_original));
treeMan_->Fill("NJets",Event::NJets(jets));
treeMan_->Fill("NBJets",Event::NJets(bJets));
treeMan_->Fill("NVertices", event->Vertices().size());
treeMan_->Fill("BJetWeight",event->BJetWeight());
treeMan_->Fill("BJetEfficiencyCorrectionWeight",event->BJetEfficiencyCorrectionWeight());
treeMan_->Fill("BJetAlternativeWeight",event->BJetAlternativeWeight());
treeMan_->Fill("BJetUpWeight",event->BJetUpWeight());
treeMan_->Fill("BJetDownWeight",event->BJetDownWeight());
treeMan_->Fill("LightJetUpWeight",event->LightJetUpWeight());
treeMan_->Fill("LightJetDownWeight",event->LightJetDownWeight());
if (selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsReference)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIso03DeltaBeta());
}
else if (selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsReference)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIso04DeltaBeta());
}
for (unsigned int index = 0; index < jets.size(); ++index) {
treeMan_->Fill("jet_csv", jets.at(index)->getBTagDiscriminator(BtagAlgorithm::CombinedSecondaryVertexV2) );
}
fillLeptonEfficiencyCorrectionBranches( event, selectionCriteria, signalLepton );
}
void TTbar_plus_X_analyser::fillCommonTrees(const EventPtr event, const unsigned int selectionCriteria, std::string folder ) {
SelectionCriteria::selection selection = SelectionCriteria::selection(selectionCriteria);
// Jets
const JetCollection jets(event->CleanedJets());
// B Jets
unsigned int numberOfBjets = event->getNBJets( selection );
const JetCollection bJets(event->CleanedBJets());
// Lepton
const LeptonPointer signalLepton = event->getSignalLepton( selection );
// MET
const METPointer MET_original(event->MET((METAlgorithm::value) 0));
treeMan_->setCurrentFolder(folder);
treeMan_->Fill("EventWeight", event->weight());
treeMan_->Fill("PUWeight", event->PileUpWeight());
treeMan_->Fill("PUWeight_up", event->PileUpWeight(1));
treeMan_->Fill("PUWeight_down", event->PileUpWeight(-1));
treeMan_->Fill("lepton_eta",signalLepton->eta());
treeMan_->Fill("lepton_pt",signalLepton->pt());
treeMan_->Fill("lepton_charge",signalLepton->charge());
if (selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsReference) ||
selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsQCDNonIsolated) ||
selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsQCDConversion)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIsoWithEA());
treeMan_->Fill("lepton_hltECALisolation", signalLepton->hltECALIso());
treeMan_->Fill("lepton_hltHCALisolation", signalLepton->hltHCALIso());
treeMan_->Fill("lepton_hltTrackerisolation", signalLepton->hltTrackerIso());
if ( signalLepton->pt() > 0 ) {
treeMan_->Fill("lepton_hltTrackerisolation_overPt", signalLepton->hltTrackerIso() / signalLepton->pt() );
}
}
else if (selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsReference) ||
selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsQCDNonIsolated1p5to3) ||
selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsQCDNonIsolated3toInf)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIso04DeltaBeta());
}
treeMan_->Fill("M3",Event::M3(jets));
if ( numberOfBjets > 0 ) {
treeMan_->Fill("M_bl",Event::M_bl(bJets, signalLepton));
treeMan_->Fill("angle_bl",Event::angle_bl(bJets, signalLepton));
// if ( numberOfBjets >= 2 ) {
// unsigned int highestCSVJetIndex = 0;
// unsigned int secondHighestCSVJetIndex = 0;
// Event::getTopTwoCSVJets( bJets, highestCSVJetIndex, secondHighestCSVJetIndex );
// JetPointer highestCSVJet(bJets.at(highestCSVJetIndex));
// JetPointer secondHighestCSVJet(bJets.at(secondHighestCSVJetIndex));
// treeMan_->Fill("deltaPhi_bb", fabs( Event::deltaPhi_bb(highestCSVJet, secondHighestCSVJet) ) ) ;
// treeMan_->Fill("deltaEta_bb", fabs( Event::deltaEta_bb(highestCSVJet, secondHighestCSVJet) ) ) ;
// treeMan_->Fill("angle_bb", Event::angle_bb(highestCSVJet, secondHighestCSVJet));
// }
}
for (unsigned int index = 0; index < jets.size(); ++index) {
treeMan_->Fill("jet_pt", jets.at(index)->pt() );
treeMan_->Fill("jet_eta", jets.at(index)->eta() );
treeMan_->Fill("jet_csv", jets.at(index)->getBTagDiscriminator(BtagAlgorithm::CombinedSecondaryVertexV2) );
}
treeMan_->Fill("HT",Event::HT(jets));
treeMan_->Fill("MET",MET_original->et());
treeMan_->Fill("MET_phi",MET_original->phi());
treeMan_->Fill("ST",Event::ST(jets, signalLepton, MET_original));
treeMan_->Fill("WPT",Event::WPT(signalLepton, MET_original));
treeMan_->Fill("MT",Event::MT(signalLepton, MET_original));
treeMan_->Fill("NJets",Event::NJets(jets));
treeMan_->Fill("NBJets",Event::NJets(bJets));
treeMan_->Fill("NVertices", event->Vertices().size());
treeMan_->Fill("BJetWeight",event->BJetWeight());
treeMan_->Fill("BJetEfficiencyCorrectionWeight",event->BJetEfficiencyCorrectionWeight());
treeMan_->Fill("BJetAlternativeWeight",event->BJetAlternativeWeight());
treeMan_->Fill("BJetUpWeight",event->BJetUpWeight());
treeMan_->Fill("BJetDownWeight",event->BJetDownWeight());
treeMan_->Fill("LightJetUpWeight",event->LightJetUpWeight());
treeMan_->Fill("LightJetDownWeight",event->LightJetDownWeight());
treeMan_->Fill("tau1", event->getTau1());
treeMan_->Fill("tau2", event->getTau2());
treeMan_->Fill("tau3", event->getTau3());
treeMan_->Fill("tau4", event->getTau4());
treeMan_->Fill("tau5", event->getTau5());
treeMan_->Fill("tau6", event->getTau6());
// MET Uncertainties
for ( unsigned int unc_i = 0; unc_i < MET_original->getAllMETUncertainties().size(); ++unc_i ) {
METPointer METForUnc_i = MET_original->getMETForUncertainty( unc_i );
treeMan_->Fill("MET_METUncertainties",METForUnc_i->et());
treeMan_->Fill("ST_METUncertainties",Event::ST(jets, signalLepton, METForUnc_i));
treeMan_->Fill("WPT_METUncertainties",Event::WPT(signalLepton, METForUnc_i));
}
double topPtWeight = 1.;
if ( ( event->isSemiLeptonicElectron() || event->isSemiLeptonicMuon() ) ) {
const TTGenInfoPointer ttGen( event->TTGenInfo() );
const ParticlePointer leptonicTop = ttGen->getLeptonicTop();
const ParticlePointer hadronicTop = ttGen->getHadronicTop();
if ( leptonicTop != 0 && hadronicTop != 0 ) {
// Store info on leptonic top
treeMan_->Fill("lepTopPt_parton", leptonicTop->pt() );
treeMan_->Fill("hadTopPt_parton", hadronicTop->pt() );
topPtWeight = exp( 0.0615 - 0.0005 * leptonicTop->pt() ) * exp( 0.0615 - 0.0005 * hadronicTop->pt() );
}
}
treeMan_->Fill("topPtWeight",topPtWeight);
fillLeptonEfficiencyCorrectionBranches( event, selectionCriteria, signalLepton );
wAnalyser_->analyseHadronicW( event, jets, bJets, folder );
}
void TTbar_plus_X_analyser::fillCommonTrees(const EventPtr event, const unsigned int selectionCriteria, std::string folder ) {
SelectionCriteria::selection selection = SelectionCriteria::selection(selectionCriteria);
// Jets
const JetCollection jets(event->CleanedJets());
// B Jets
unsigned int numberOfBjets = event->getNBJets( selection );
const JetCollection bJets(event->CleanedBJets());
// Lepton
const LeptonPointer signalLepton = event->getSignalLepton( selection );
// MET
const METPointer MET_original(event->MET((METAlgorithm::value) 0));
treeMan_->setCurrentFolder(folder);
treeMan_->Fill("EventWeight", event->weight());
treeMan_->Fill("PUWeight", event->PileUpWeight());
treeMan_->Fill("PUWeight_up", event->PileUpWeight(1));
treeMan_->Fill("PUWeight_down", event->PileUpWeight(-1));
treeMan_->Fill("lepton_eta",signalLepton->eta());
treeMan_->Fill("lepton_pt",signalLepton->pt());
treeMan_->Fill("lepton_charge",signalLepton->charge());
if (selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsReference) ||
selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsQCDNonIsolated) ||
selection == SelectionCriteria::selection(SelectionCriteria::ElectronPlusJetsQCDConversion)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIsoWithEA());
}
else if (selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsReference) ||
selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsQCDNonIsolated1p5to3) ||
selection == SelectionCriteria::selection(SelectionCriteria::MuonPlusJetsQCDNonIsolated3toInf)){
treeMan_->Fill("lepton_isolation", signalLepton->PFRelIso04DeltaBeta());
}
treeMan_->Fill("M3",Event::M3(jets));
if ( numberOfBjets > 0 ) {
treeMan_->Fill("M_bl",Event::M_bl(bJets, signalLepton));
treeMan_->Fill("angle_bl",Event::angle_bl(bJets, signalLepton));
}
for (unsigned int index = 0; index < jets.size(); ++index) {
treeMan_->Fill("jet_pt", jets.at(index)->pt() );
treeMan_->Fill("jet_eta", jets.at(index)->eta() );
}
treeMan_->Fill("HT",Event::HT(jets));
treeMan_->Fill("MET",MET_original->et());
treeMan_->Fill("ST",Event::ST(jets, signalLepton, MET_original));
treeMan_->Fill("WPT",Event::WPT(signalLepton, MET_original));
treeMan_->Fill("MT",Event::MT(signalLepton, MET_original));
treeMan_->Fill("NJets",Event::NJets(jets));
treeMan_->Fill("NBJets",Event::NJets(bJets));
treeMan_->Fill("NVertices", event->Vertices().size());
treeMan_->Fill("BJetWeight",event->BJetWeight());
treeMan_->Fill("BJetUpWeight",event->BJetUpWeight());
treeMan_->Fill("BJetDownWeight",event->BJetDownWeight());
treeMan_->Fill("LightJetUpWeight",event->LightJetUpWeight());
treeMan_->Fill("LightJetDownWeight",event->LightJetDownWeight());
// MET Uncertainties
for ( unsigned int unc_i = 0; unc_i < MET_original->getAllMETUncertainties().size(); ++unc_i ) {
METPointer METForUnc_i = MET_original->getMETForUncertainty( unc_i );
treeMan_->Fill("MET_METUncertainties",METForUnc_i->et());
treeMan_->Fill("ST_METUncertainties",Event::ST(jets, signalLepton, METForUnc_i));
treeMan_->Fill("WPT_METUncertainties",Event::WPT(signalLepton, METForUnc_i));
}
fillLeptonEfficiencyCorrectionBranches( event, selectionCriteria, signalLepton );
}
BAT::TtbarHypothesis HitFitAnalyser::analyseAndReturn(const EventPtr event, const JetCollection jets, const JetCollection bjets, const LeptonPointer selectedLepton ) {
weight_ = event->weight() * prescale_ * scale_;
treeMan_->setCurrentFolder(histogramFolder_);
treeMan_->Fill("EventWeight", weight_ );
// const JetCollection jets(event->getCleanedJets( SelectionCriteria::ElectronPlusJetsReference ));
// const JetCollection bJets(event->getCleanedBJets( SelectionCriteria::ElectronPlusJetsReference ));
// const LeptonPointer selectedLepton = event->getSignalLepton( SelectionCriteria::ElectronPlusJetsReference );
string metPrefix = METAlgorithm::names.at(0);
const METPointer met(event->MET((METAlgorithm::value) 0));
// Get cleaned jets that aren't b tagged
JetCollection leadingLightJets;
JetCollection leadingBJets;
unsigned int maxNJet = std::min(5, int(jets.size()));
for ( unsigned int jetIndex=0; jetIndex < maxNJet; ++jetIndex ) {
bool isBJet = false;
JetPointer thisJet = jets[jetIndex];
for ( unsigned int bJetIndex=0; bJetIndex < bjets.size(); ++bJetIndex ) {
JetPointer thisBJet = bjets[bJetIndex];
if ( thisJet == thisBJet ) {
isBJet = true;
break;
}
}
if ( !isBJet ) leadingLightJets.push_back( thisJet );
else leadingBJets.push_back( thisJet );
}
//set MC matching flag
if (event->getDataType() == DataType::TTJets_amcatnloFXFX)
do_MC_matching = true;
else
do_MC_matching = false;
//prepare the jets collection
// Copy jets into an array
JetCollection jetCopy;
for (JetCollection::const_iterator j = leadingLightJets.begin(); j != leadingLightJets.end(); ++j) {
jetCopy.push_back(*j);
}
JetCollection bJetCopy;
for (JetCollection::const_iterator j = leadingBJets.begin(); j != leadingBJets.end(); ++j) {
bJetCopy.push_back(*j);
}
std::sort(jetCopy.begin(), jetCopy.end(), jetPtComp);
std::sort(bJetCopy.begin(), bJetCopy.end(), jetPtComp);
jetsForFitting.clear();
unsigned numJetsToFit = jetCopy.size();
if (jetCopy.size() >= 2) {
if (numJetsToFit > 5)
numJetsToFit = 5;
jetsForFitting.insert(jetsForFitting.begin(), jetCopy.begin(), jetCopy.begin() + numJetsToFit);
}
bJetsForFitting.clear();
numJetsToFit = bJetCopy.size();
if (bJetCopy.size() >= 2) {
if (numJetsToFit > 5)
numJetsToFit = 5;
bJetsForFitting.insert(bJetsForFitting.begin(), bJetCopy.begin(), bJetCopy.begin() + numJetsToFit);
}
BatHitFit hhFitter(electronTranslator_, muonTranslator_, jetTranslator_, metTranslator_, hitfitDefault_,
hitfitLepWMass_, hitfitHadWMass_, hitfitTopMass_);
// Clear the internal state
hhFitter.clear();
// Add lepton into hitfit
if ( isElectronChannel_ ) {
const ElectronPointer signalElectron(boost::static_pointer_cast<Electron>(selectedLepton));
//TODO: fix the fitter to accept lepton class OR particle!!
hhFitter.AddLepton(*signalElectron);
}
else {
const MuonPointer signalMuon(boost::static_pointer_cast<Muon>(selectedLepton));
//TODO: fix the fitter to accept lepton class OR particle!!
hhFitter.AddLepton(*signalMuon);
}
// Add jets into HitFit
// Also check if jets matched to ttbar partons are in the jets passed to the fit
bool quarkInCollection = false, quarkBarInCollection = false, lebBInCollection = false, hadBInCollection = false;
for (size_t jet = 0; jet != jetsForFitting.size(); ++jet) {
hhFitter.AddJet(*jetsForFitting.at(jet));
// cout << "Adding light jet with csv : " << jetsForFitting.at(jet)->getBTagDiscriminator(BAT::BtagAlgorithm::value::CombinedSecondaryVertexV2) << endl;
if ( jetsForFitting.at(jet)->ttbar_decay_parton() ) {
int partonPdg = jetsForFitting.at(jet)->ttbar_decay_parton();
if ( partonPdg == 3 ) quarkInCollection = true;
else if ( partonPdg == 4 ) quarkBarInCollection = true;
else if ( partonPdg == 5 ) lebBInCollection = true;
else if ( partonPdg == 6 ) hadBInCollection = true;
}
}
for (size_t jet = 0; jet != bJetsForFitting.size(); ++jet) {
hhFitter.AddBJet(*bJetsForFitting.at(jet));
// cout << "Adding b jet with csv : " << bJetsForFitting.at(jet)->getBTagDiscriminator(BAT::BtagAlgorithm::value::CombinedSecondaryVertexV2) << endl;
if ( bJetsForFitting.at(jet)->ttbar_decay_parton() ) {
int partonPdg = bJetsForFitting.at(jet)->ttbar_decay_parton();
if ( partonPdg == 3 ) quarkInCollection = true;
else if ( partonPdg == 4 ) quarkBarInCollection = true;
else if ( partonPdg == 5 ) lebBInCollection = true;
else if ( partonPdg == 6 ) hadBInCollection = true;
}
}
// Check if jets matched to ttbar partons are in the jets passed to the fit
if (do_MC_matching) {
if ( quarkInCollection && quarkBarInCollection && lebBInCollection && hadBInCollection ) {
allTTBarJetsPassedToFit_ = true;
}
int lastTTBarJetPosition = positionOfLastTTBarJet( jets );
if ( lastTTBarJetPosition != -1 ) {
allTTBarJetsPassSelection_ = true;
treeMan_->Fill("PositionOfLastTTbarJet", lastTTBarJetPosition + 1 );
}
else treeMan_->Fill("PositionOfLastTTbarJet", lastTTBarJetPosition );
}
// Add missing transverse energy into HitFit
hhFitter.SetMet(*met);
// Container for input of all jet permutation of the event
std::vector<hitfit::Lepjets_Event> hitfitEventsInput;
// Container for results of the fit for all jet permutation of the event
std::vector<hitfit::Fit_Result> hitfitResult;
//
// R U N H I T F I T
//
// Run the kinematic fit and get how many permutations is possible
// in the fit
size_t nHitFit = hhFitter.FitAllPermutation();
// // Get the number of jets
// nHitFitJet = hhFitter.GetEvent().njets();
// Get the input events for all permutations
hitfitEventsInput = hhFitter.GetUnfittedEvent();
// Get the fit results for all permutations
hitfitResult = hhFitter.GetFitAllPermutation();
double bestChi2 = 999.;
unsigned bestX2pos = nHitFit + 1;
// // Loop over all permutations and extract the information
for (size_t fit = 0; fit != nHitFit; ++fit) {
// Get the event after the fit
hitfit::Fit_Result fitResult = hitfitResult[fit];
if (hitfitResult[fit].chisq() > 0.0) {
treeMan_->Fill("FitChiSquaredAllSolutions",fitResult.chisq());
}
// Is this the permutation with smallest chi2?
if (fitResult.chisq() > 0.0 && fitResult.chisq() < bestChi2) {
bestChi2 = fitResult.chisq();
bestX2pos = fit;
}
}
//
// END PART WHICH EXTRACTS INFORMATION FROM HITFIT
//
if (bestX2pos < nHitFit + 1) {
treeMan_->Fill("FitChiSquaredBestSolutions",hitfitResult[bestX2pos].chisq());
treeMan_->Fill("FitChiSquaredProbabilityBestSolutions",TMath::Prob(hitfitResult[bestX2pos].chisq(),1));
//pass hitfit event into BAT format
lepton_charge = selectedLepton->charge();
BAT::TtbarHypothesis newHyp = BatEvent(hitfitResult[bestX2pos].ev(), event, "SolutionCategory");
treeMan_->Fill("FittedLeptonicTopPtBestSolution", newHyp.leptonicTop->pt());
treeMan_->Fill("FittedHadronicTopPtBestSolution", newHyp.hadronicTop->pt());
treeMan_->Fill("FittedLeptonicTopRapidityBestSolution", newHyp.leptonicTop->rapidity());
treeMan_->Fill("FittedHadronicTopRapidityBestSolution", newHyp.hadronicTop->rapidity());
treeMan_->Fill("FittedTTbarMassBestSolution", newHyp.resonance->mass());
treeMan_->Fill("FittedTTbarPtBestSolution", newHyp.resonance->pt());
treeMan_->Fill("FittedTTbarRapidityBestSolution", newHyp.resonance->rapidity());
// Get the event before the fit
hitfit::Lepjets_Event unfittedEvent = hitfitEventsInput[bestX2pos];
// Now need to perform second kinematic fit
hitfit::Fit_Result secondFitResult = performSecondKinematicFit( unfittedEvent, event);
BAT::TtbarHypothesis newHyp_afterSecondFit = BatEvent(secondFitResult.ev(), event, "SolutionCategory_second");
treeMan_->Fill("FitChiSquaredBestSolutions_second",secondFitResult.chisq());
treeMan_->Fill("FitChiSquaredProbabilityBestSolutions_second",TMath::Prob(secondFitResult.chisq(),2));
treeMan_->Fill("FittedLeptonicTopPtBestSolution_second", newHyp_afterSecondFit.leptonicTop->pt());
treeMan_->Fill("FittedHadronicTopPtBestSolution_second", newHyp_afterSecondFit.hadronicTop->pt());
treeMan_->Fill("FittedLeptonicTopRapidityBestSolution_second", newHyp_afterSecondFit.leptonicTop->rapidity());
treeMan_->Fill("FittedHadronicTopRapidityBestSolution_second", newHyp_afterSecondFit.hadronicTop->rapidity());
treeMan_->Fill("FittedTTbarMassBestSolution_second", newHyp_afterSecondFit.resonance->mass());
treeMan_->Fill("FittedTTbarPtBestSolution_second", newHyp_afterSecondFit.resonance->pt());
treeMan_->Fill("FittedTTbarRapidityBestSolution_second", newHyp_afterSecondFit.resonance->rapidity());
return newHyp_afterSecondFit;
} else {
// cout << "No HitFit solution found for this event" << endl;
treeMan_->Fill("SolutionCategory", 0 );
treeMan_->Fill("SolutionCategory_second", 0);
}
return BAT::TtbarHypothesis();
}
void UnfoldingRecoAnalyser::analyse(const EventPtr event) {
weight_ = event->weight();
treeMan_->setCurrentFolder(histogramFolder_);
treeMan_->Fill("EventWeight", event->weight());
double centralLHEWeight = event->centralLHEWeight();
for ( unsigned int unc_i = 0; unc_i < event->generatorSystematicWeights().size(); ++unc_i ) {
double weight = event->generatorSystematicWeights().at( unc_i ) / centralLHEWeight;
int weightID = event->generatorSystematicWeightsID().at( unc_i );
// cout << to_string(unc_i) << " " << weightID << " " << weight/centralLHEWeight << endl;
// string uncString = to_string(unc_i);
if ( weightID >= 1001 && weightID <= 1009 ) {
treeMan_->Fill( "muFmuRWeight_" + to_string(weightID-1001), weight);
}
else if ( weightID >= 2001 && weightID <= 2100 ) {
treeMan_->Fill( "pdfWeight_" + to_string(weightID-2001), weight);
}
else if ( weightID == 2101 || weightID == 2102 ) {
treeMan_->Fill( "alphaSWeight_" + to_string(weightID-2101), weight);
}
}
treeMan_->setCurrentFolder(histogramFolder_);
int selectionCriteria = -1;
if ( event->PassesElectronTriggerAndSelection() ) selectionCriteria = SelectionCriteria::ElectronPlusJetsReference;
else if ( event->PassesMuonTriggerAndSelection() ) selectionCriteria = SelectionCriteria::MuonPlusJetsReference;
if ( event->PassesElectronTriggerAndSelection() && event->PassesMuonTriggerAndSelection() ) {
cout << "-------------> This should not happen <-------------" << endl;
}
// Store selection flags first
treeMan_->Fill("passSelection",selectionCriteria);
if ( !( event->PassesMuonTriggerAndSelection() || event->PassesElectronTriggerAndSelection() )) return;
const JetCollection jets(event->CleanedJets());
const JetCollection bjets(event->CleanedBJets());
const LeptonPointer signalLepton = event->getSignalLepton( selectionCriteria );
METAlgorithm::value metType = (METAlgorithm::value) 0;
const METPointer met(event->MET(metType));
treeMan_->Fill("MET",met->et());
treeMan_->Fill("HT",Event::HT(jets));
treeMan_->Fill("ST",Event::ST(jets,signalLepton,met));
treeMan_->Fill("WPT",Event::WPT(signalLepton,met));
treeMan_->Fill("MT",Event::MT(signalLepton,met));
treeMan_->Fill("leptonPt", signalLepton->pt() );
treeMan_->Fill("leptonEta", signalLepton->eta() );
treeMan_->Fill("leptonCharge", signalLepton->charge() );
// Get lepton scale factor
if ( event->PassesMuonTriggerAndSelection() ) {
const MuonPointer signalMuon(boost::static_pointer_cast<Muon>(signalLepton));
double efficiencyCorrection = signalMuon->getEfficiencyCorrection( 0 );
treeMan_->Fill("LeptonEfficiencyCorrection", efficiencyCorrection);
treeMan_->Fill("LeptonEfficiencyCorrectionUp", signalMuon->getEfficiencyCorrection( 1 ));
treeMan_->Fill("LeptonEfficiencyCorrectionDown", signalMuon->getEfficiencyCorrection( -1 ));
}
else if ( event->PassesElectronTriggerAndSelection() ) {
const ElectronPointer signalElectron(boost::static_pointer_cast<Electron>(signalLepton));
double efficiencyCorrection = signalElectron->getEfficiencyCorrection( 0 );
treeMan_->Fill("LeptonEfficiencyCorrection", efficiencyCorrection);
treeMan_->Fill("LeptonEfficiencyCorrectionUp", signalElectron->getEfficiencyCorrection( 1 ));
treeMan_->Fill("LeptonEfficiencyCorrectionDown", signalElectron->getEfficiencyCorrection( -1 ));
}
treeMan_->Fill("PUWeight", event->PileUpWeight() );
treeMan_->Fill("PUWeight_up", event->PileUpWeight(1) );
treeMan_->Fill("PUWeight_down", event->PileUpWeight(-1) );
treeMan_->Fill("BJetWeight",event->BJetWeight());
treeMan_->Fill("LightJetUpWeight",event->LightJetUpWeight());
treeMan_->Fill("LightJetDownWeight",event->LightJetDownWeight());
treeMan_->Fill("BJetUpWeight",event->BJetUpWeight());
treeMan_->Fill("BJetDownWeight",event->BJetDownWeight());
// MET Uncertainties
for ( unsigned int unc_i = 0; unc_i < met->getAllMETUncertainties().size(); ++unc_i ) {
METPointer METForUnc_i = met->getMETForUncertainty( unc_i );
treeMan_->Fill("MET_METUncertainties",METForUnc_i->et());
treeMan_->Fill("ST_METUncertainties",Event::ST(jets, signalLepton, METForUnc_i));
treeMan_->Fill("WPT_METUncertainties",Event::WPT(signalLepton, METForUnc_i));
}
for (unsigned int index = 0; index < bjets.size(); ++index) {
treeMan_->Fill("bPt", bjets.at(index)->pt() );
treeMan_->Fill("bEta", bjets.at(index)->eta() );
}
for (unsigned int index = 0; index < jets.size(); ++index) {
treeMan_->Fill("jetPt", jets.at(index)->pt() );
treeMan_->Fill("jetEta", jets.at(index)->eta() );
treeMan_->Fill("jetParton", jets.at(index)->ttbar_decay_parton() );
}
BAT::TtbarHypothesis topHypothesis = event->ttbarHypothesis();
if ( topHypothesis.isValid() && topHypothesis.isPhysical() ) {
treeMan_->Fill("lepTopPt",topHypothesis.leptonicTop->pt());
treeMan_->Fill("hadTopPt",topHypothesis.hadronicTop->pt());
treeMan_->Fill("lepTopRap",topHypothesis.leptonicTop->rapidity());
treeMan_->Fill("hadTopRap",topHypothesis.hadronicTop->rapidity());
treeMan_->Fill("ttbarPt",topHypothesis.resonance->pt());
treeMan_->Fill("ttbarM",topHypothesis.resonance->mass());
treeMan_->Fill("ttbarRap",topHypothesis.resonance->rapidity());
}
unsigned int numberOfJets(event->NJets(jets));
unsigned int numberOfBJets(event->NJets(bjets));
treeMan_->Fill("NJets", numberOfJets );
treeMan_->Fill("NBJets", numberOfBJets );
}
