CalcTopPartonHistory.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
 */
 
// $Id: CalcTopPartonHistory.h 800464 2017-03-13 18:06:24Z tpelzer $
#ifndef ANALYSISTOP_TOPPARTONS_CALCTOPPARTONHISTORY_H
#define ANALYSISTOP_TOPPARTONS_CALCTOPPARTONHISTORY_H
 
// system include(s):
#include <memory>
#include <vector>
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AthContainers/DataVector.h"
#include "AthContainers/AuxElement.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "TopPartons/PartonHistory.h"
 
// forward declaration(s):
namespace top {
  class TopConfig;
}
 
namespace top {
  class CalcTopPartonHistory: public asg::AsgTool {
  public:
    explicit CalcTopPartonHistory(const std::string& name);
    virtual ~CalcTopPartonHistory() {};
 
    CalcTopPartonHistory(const CalcTopPartonHistory& rhs) = delete;
    CalcTopPartonHistory(CalcTopPartonHistory&& rhs) = delete;
    CalcTopPartonHistory& operator = (const CalcTopPartonHistory& rhs) = delete;
 
    bool topAfterFSR_SC(const xAOD::TruthParticleContainer* truthParticles, int start,
                        TLorentzVector& top_afterFSR_SC_p4);
 
    bool topWb(const xAOD::TruthParticleContainer* truthParticles, int start, TLorentzVector& t_beforeFSR_p4,
               TLorentzVector& t_afterFSR_p4, TLorentzVector& W_p4, TLorentzVector& b_p4, TLorentzVector& Wdecay1_p4,
               int& Wdecay1_pdgId, TLorentzVector& Wdecay2_p4, int& Wdecay2_pdgId);
    bool topWq(const xAOD::TruthParticleContainer* truthParticles, int start, TLorentzVector& t_beforeFSR_p4,
               TLorentzVector& t_afterFSR_p4, TLorentzVector& W_p4, TLorentzVector& q_p4, int& q_pdgId,
               TLorentzVector& Wdecay1_p4, int& Wdecay1_pdgId, TLorentzVector& Wdecay2_p4, int& Wdecay2_pdgId);
 
    bool b(const xAOD::TruthParticleContainer* truthParticles, TLorentzVector& b_beforeFSR, TLorentzVector& b_afterFSR);
 
    bool Wlv(const xAOD::TruthParticleContainer* truthParticles, TLorentzVector& W_p4, TLorentzVector& Wdecay1_p4,
             int& Wdecay1_pdgId, TLorentzVector& Wdecay2_p4, int& Wdecay2_pdgId);
 
    bool Wt_W(const xAOD::TruthParticleContainer* truthParticles, TLorentzVector& W_p4, int& W_pdgId,
              TLorentzVector& Wdecay1_p4, int& Wdecay1_pdgId, TLorentzVector& Wdecay2_p4, int& Wdecay2_pdgId);
 
    bool Wt_b(const xAOD::TruthParticleContainer* truthParticles, TLorentzVector& b_beforeFSR,
              TLorentzVector& b_afterFSR, int& b_pdgId);
 
    bool topPhWb(const xAOD::TruthParticleContainer* truthParticles, int topId, TLorentzVector& t_beforeFSR_p4,
                 TLorentzVector& t_afterFSR_p4, TLorentzVector& Ph_p4, TLorentzVector& W_p4, TLorentzVector& b_p4,
                 TLorentzVector& Wdecay1_p4, int& Wdecay1_pdgId, TLorentzVector& Wdecay2_p4, int& Wdecay2_pdgId,
                 bool& has_ph, int& BranchType, int& IniPartonType, bool& missingTop);
 
    virtual StatusCode execute();
  protected:
    std::shared_ptr<top::TopConfig> m_config;
 
    void fillEtaBranch(xAOD::PartonHistory* partonHistory, std:: string branchName, TLorentzVector& tlv);
    
    StatusCode buildContainerFromMultipleCollections(const std::vector<std::string> &collections, const std::string& out_contName);
    
    StatusCode linkBosonCollections();
    
    const xAOD::TruthParticle* getTruthParticleLinkedFromDecoration(const xAOD::TruthParticle* part, const std::string &decorationName);
    
  private:
    StatusCode decorateCollectionWithLinksToAnotherCollection(const std::string &collectionToDecorate, const std::string &collectionToLink, const std::string &nameOfDecoration);
    
  };
}
 
#endif