Reconstruction/tauRecTools/tauRecTools/HelperFunctions.h

Go to the documentation of this file.

#/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_HELPERFUNCTIONS_H
#define TAURECTOOLS_HELPERFUNCTIONS_H
 
#include "xAODTracking/VertexContainer.h"
#include "xAODJet/Jet.h"
#include "xAODTau/TauJet.h"
#include "xAODCaloEvent/CaloCluster.h"
#include "xAODPFlow/PFO.h"
#include "xAODPFlow/FlowElement.h"
#include "AsgMessaging/MessageCheck.h"
 
#include "MVAUtils/BDT.h"
#include "TLorentzVector.h"
#include "TString.h"
 
#include <vector>
#include <map>
 
 
namespace tauRecTools
{
  ANA_MSG_HEADER(msgHelperFunction)
 
   
  const xAOD::Vertex* getJetVertex(const xAOD::Jet& jet);
  
  TLorentzVector getTauAxis(const xAOD::TauJet& tau, bool doVertexCorrection = true);
 
  TLorentzVector GetConstituentP4(const xAOD::JetConstituent& constituent);

  bool doPi0andShots(const xAOD::TauJet& tau);
 
  xAOD::TauTrack::TrackFlagType isolateClassifiedBits(xAOD::TauTrack::TrackFlagType flag);
  bool sortTracks(const ElementLink<xAOD::TauTrackContainer> &l1, const ElementLink<xAOD::TauTrackContainer> &l2);
 
  std::unique_ptr<MVAUtils::BDT> configureMVABDT( std::map<TString, float*> &availableVars, const TString& weightFile);
  // initialise the BDT and return the list of input variable names
  std::unique_ptr<MVAUtils::BDT> configureMVABDT(std::vector<TString>& variableNames, const TString& weightFile);
 
  std::vector<TString> parseString(const TString& str, const TString& delim=",");
  std::vector<TString> parseStringMVAUtilsBDT(const TString& str, const TString& delim=",");
}
 
#endif // TAURECTOOLS_HELPERFUNCTIONS_H