#include <BDTHelper.h>

Inheritance diagram for tauRecTools::BDTHelper:

Collaboration diagram for tauRecTools::BDTHelper:

Public Member Functions
	BDTHelper ()
	~BDTHelper ()
StatusCode	initialize (const TString &weightFileName)
float	getGradBoostMVA (const std::map< TString, float > &availableVariables) const
float	getResponse (const std::map< TString, float * > &availableVariables) const
MVAUtils::BDT *	getBDT () const
void	setLevel (MSG::Level lvl)
	Change the current logging level.
Functions providing the same interface as AthMessaging
bool	msgLvl (const MSG::Level lvl) const
	Test the output level of the object.
MsgStream &	msg () const
	The standard message stream.
MsgStream &	msg (const MSG::Level lvl) const
	The standard message stream.

Private Member Functions
std::vector< TString >	parseString (const TString &str, const TString &delim=",") const
std::vector< float >	getInputVariables (const std::map< TString, float > &availableVariables) const
std::vector< float >	getInputVariables (const std::map< TString, float * > &availableVariables) const
void	initMessaging () const
	Initialize our message level and MessageSvc.

Private Attributes
std::unique_ptr< MVAUtils::BDT >	m_BDT
std::vector< TString >	m_inputVariableNames
std::string	m_nm
	Message source name.
boost::thread_specific_ptr< MsgStream >	m_msg_tls
	MsgStream instance (a std::cout like with print-out levels)
std::atomic< IMessageSvc * >	m_imsg { nullptr }
	MessageSvc pointer.
std::atomic< MSG::Level >	m_lvl { MSG::NIL }
	Current logging level.
std::atomic_flag m_initialized	ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
	Messaging initialized (initMessaging)

Detailed Description

Definition at line 16 of file BDTHelper.h.

Constructor & Destructor Documentation

◆ BDTHelper()

tauRecTools::BDTHelper::BDTHelper ( )

Definition at line 14 of file BDTHelper.cxx.

                     : 
  asg::AsgMessaging("BDTHelper"),
  m_BDT(nullptr) {
}

◆ ~BDTHelper()

tauRecTools::BDTHelper::~BDTHelper ( )

Definition at line 20 of file BDTHelper.cxx.

20 {

21}

Member Function Documentation

◆ getBDT()

MVAUtils::BDT * tauRecTools::BDTHelper::getBDT ( ) const

inline

Definition at line 27 of file BDTHelper.h.

27{ return m_BDT.get(); }

◆ getGradBoostMVA()

float tauRecTools::BDTHelper::getGradBoostMVA ( const std::map< TString, float > & availableVariables ) const

Definition at line 113 of file BDTHelper.cxx.

                                                                                       {
  std::vector<float> values = getInputVariables(availableVariables);
 
  if (values.size() < m_inputVariableNames.size()) {
    ATH_MSG_ERROR("There are missing variables when calculating the BDT score, will return -999");
    return  -999;
  }
  else {  
    return  m_BDT->GetGradBoostMVA(values);
  }
}

◆ getInputVariables() [1/2]

std::vector< float > tauRecTools::BDTHelper::getInputVariables ( const std::map< TString, float * > & availableVariables ) const

private

Definition at line 96 of file BDTHelper.cxx.

                                                                                                     {  
  std::vector<float> values;
 
  // sort the input variables by the order in varList (from BDT)
  for (const TString& name : m_inputVariableNames) {
    std::map<TString, float*>::const_iterator itr = availableVariables.find(name);
    if(itr==availableVariables.end()) {
      ATH_MSG_ERROR(name << " not available");
    }
    else {
      values.push_back(*itr->second);
    }
  }
 
  return values;
}

◆ getInputVariables() [2/2]

std::vector< float > tauRecTools::BDTHelper::getInputVariables ( const std::map< TString, float > & availableVariables ) const

private

Definition at line 79 of file BDTHelper.cxx.

                                                                                                    {  
  std::vector<float> values;
 
  // sort the input variables by the order in varList (from BDT)
  for (const TString& name : m_inputVariableNames) {
    std::map<TString, float>::const_iterator itr = availableVariables.find(name);
    if(itr==availableVariables.end()) {
      ATH_MSG_ERROR(name << " not available");
    }
    else {
      values.push_back(itr->second);
    }
  }
 
  return values;
}

◆ getResponse()

float tauRecTools::BDTHelper::getResponse ( const std::map< TString, float * > & availableVariables ) const

Definition at line 126 of file BDTHelper.cxx.

                                                                                    {
  std::vector<float> values = getInputVariables(availableVariables);
 
  if (values.size() < m_inputVariableNames.size()) {
    ATH_MSG_ERROR("There are missing variables when calculating the BDT score, will return -999");
    return -999;
  }
  else {
    return  m_BDT->GetResponse(values);
  }
}

◆ initialize()

StatusCode tauRecTools::BDTHelper::initialize ( const TString & weightFileName )

Definition at line 23 of file BDTHelper.cxx.

                                                              {
  
  std::unique_ptr<TFile> file(TFile::Open(weightFileName));
  if (!file) {
    ATH_MSG_ERROR("Cannot find input BDT file: " << weightFileName);
    return StatusCode::FAILURE;
  }
  ATH_MSG_INFO( "Open file: " << weightFileName);
 
  TTree* tree = dynamic_cast<TTree*> (file->Get("BDT"));
  if (!tree) {
    ATH_MSG_ERROR("Cannot find input BDT tree");
    return StatusCode::FAILURE;
  }
  m_BDT = std::make_unique<MVAUtils::BDT>(tree);
 
  TNamed* varList = dynamic_cast<TNamed*> (file->Get("varList"));
  if (!varList) {
    ATH_MSG_ERROR("No variable list in file: " << weightFileName);
    return StatusCode::FAILURE;
  }
  TString names = varList->GetTitle();
  delete varList;
 
  // abtain the list of input variables  
  m_inputVariableNames = parseString(names);
 
  file->Close();
 
  return StatusCode::SUCCESS;
}

◆ initMessaging()

void AthMessaging::initMessaging ( ) const

privateinherited

Initialize our message level and MessageSvc.

This method should only be called once.

Definition at line 39 of file AthMessaging.cxx.

{
  m_imsg = Athena::getMessageSvc();
  // If user did not set an explicit level, set a default
  if (m_lvl == MSG::NIL) {
    m_lvl = m_imsg ?
      static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
      MSG::INFO;
  }
}

◆ msg() [1/2]

MsgStream & asg::AsgMessaging::msg ( ) const

inherited

The standard message stream.

Returns: A reference to the default message stream of this object.

Definition at line 49 of file AsgMessaging.cxx.

                                      {
#ifndef XAOD_STANDALONE
      return ::AthMessaging::msg();
#else // not XAOD_STANDALONE
      return m_msg;
#endif // not XAOD_STANDALONE
   }

◆ msg() [2/2]

MsgStream & asg::AsgMessaging::msg ( const MSG::Level lvl ) const

inherited

The standard message stream.

Parameters

lvl	The message level to set the stream to

Returns: A reference to the default message stream, set to level "lvl"

Definition at line 57 of file AsgMessaging.cxx.

                                                          {
#ifndef XAOD_STANDALONE
      return ::AthMessaging::msg( lvl );
#else // not XAOD_STANDALONE
      m_msg << lvl;
      return m_msg;
#endif // not XAOD_STANDALONE
   }

◆ msgLvl()

bool asg::AsgMessaging::msgLvl ( const MSG::Level lvl ) const

inherited

Test the output level of the object.

Parameters

lvl	The message level to test against

Returns: boolean Indicting if messages at given level will be printed; true If messages at level "lvl" will be printed

Definition at line 41 of file AsgMessaging.cxx.

                                                       {
#ifndef XAOD_STANDALONE
      return ::AthMessaging::msgLvl( lvl );
#else // not XAOD_STANDALONE
      return m_msg.msgLevel( lvl );
#endif // not XAOD_STANDALONE
   }

◆ parseString()

std::vector< TString > tauRecTools::BDTHelper::parseString	(	const TString &	str,
		const TString &	delim = "," ) const

private

Definition at line 55 of file BDTHelper.cxx.

                                                                                         {
  std::vector<TString> parsedString;
 
  TObjArray* objList = str.Tokenize(delim);
  size_t arraySize = objList->GetEntries();
  
  // split the string with ",", and put them into a vector
  for(size_t i = 0; i < arraySize; ++i) {
    if (auto *str = dynamic_cast<TObjString*> (objList->At(i))) {
      TString var = str->String();
      var.ReplaceAll(" ", "");
      if(var.Contains(":=")) {
        var=var(var.Index(":=")+2, var.Length()-var.Index(":=")-2);
      }
      if(0==var.Length()) continue;
      parsedString.push_back(var);
    }
  }
 
  delete objList;
 
  return parsedString;
}

◆ setLevel()

void AthMessaging::setLevel ( MSG::Level lvl )

inherited

Change the current logging level.

Use this rather than msg().setLevel() for proper operation with MT.

Definition at line 28 of file AthMessaging.cxx.

{
  m_lvl = lvl;
}

Member Data Documentation

◆ ATLAS_THREAD_SAFE

std::atomic_flag m_initialized AthMessaging::ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT

mutableprivateinherited

Messaging initialized (initMessaging)

Definition at line 141 of file AthMessaging.h.

◆ m_BDT

std::unique_ptr<MVAUtils::BDT> tauRecTools::BDTHelper::m_BDT

private

Definition at line 36 of file BDTHelper.h.

◆ m_imsg

std::atomic<IMessageSvc*> AthMessaging::m_imsg { nullptr }

mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

135{ nullptr };

◆ m_inputVariableNames

std::vector<TString> tauRecTools::BDTHelper::m_inputVariableNames

private

Definition at line 37 of file BDTHelper.h.

◆ m_lvl

std::atomic<MSG::Level> AthMessaging::m_lvl { MSG::NIL }

mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

138{ MSG::NIL };

◆ m_msg_tls

boost::thread_specific_ptr<MsgStream> AthMessaging::m_msg_tls

mutableprivateinherited

MsgStream instance (a std::cout like with print-out levels)

Definition at line 132 of file AthMessaging.h.

◆ m_nm

std::string AthMessaging::m_nm

privateinherited

Message source name.

Definition at line 129 of file AthMessaging.h.

The documentation for this class was generated from the following files:

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ BDTHelper()

◆ ~BDTHelper()

Member Function Documentation

◆ getBDT()

◆ getGradBoostMVA()

◆ getInputVariables() [1/2]

◆ getInputVariables() [2/2]

◆ getResponse()

◆ initialize()

◆ initMessaging()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ parseString()

◆ setLevel()

Member Data Documentation

◆ ATLAS_THREAD_SAFE

◆ m_BDT

◆ m_imsg

◆ m_inputVariableNames

◆ m_lvl

◆ m_msg_tls

◆ m_nm