Prompt::VarHolder Class Reference

#include <VarHolder.h>

Collaboration diagram for Prompt::VarHolder:

Public Member Functions
	VarHolder ()
virtual	~VarHolder ()
bool	replaceVar (unsigned key, double value)
bool	addVar (unsigned key, double value)
	Adds a variable to the VarHolder.
bool	delVar (unsigned key)
double	getVar (const unsigned key) const
bool	getVar (const unsigned key, double &value) const
bool	getVar (const unsigned key, float &value) const
bool	hasKey (unsigned key) const
bool	hasVar (unsigned key) const
int	registerAllVars ()
bool	registerVar (Prompt::Def::Var var, const std::string &name)
Prompt::Def::Var	registerDynamicVar (const std::string &name)
std::string	convert2Str (const Prompt::Def::Var var) const
Prompt::Def::Var	convert2Var (const std::string &var)
Prompt::Def::Var	convert2Var (const uint32_t key)
void	getAllVarEnums ()
std::string	asStr (const uint32_t key, const double val)
std::string	asStr (const Prompt::Def::Var var)
std::vector< Prompt::Def::Var >	readVars (const std::string &config)
std::vector< Prompt::Def::Var >	readVectorVars (const std::vector< std::string > &keys)
virtual void	clearVars ()

Static Public Member Functions
static std::string	getObjectType ()

Private Attributes
VarMap	m_gPromptVars
VarEntryVec	m_fVars
std::vector< Prompt::Def::Var >	m_varEnums

Detailed Description

Definition at line 101 of file VarHolder.h.

Constructor & Destructor Documentation

VarHolder()

Prompt::VarHolder::VarHolder

(

)

Definition at line 16 of file VarHolder.cxx.

                           {
    registerAllVars();
}

~VarHolder()

virtual Prompt::VarHolder::~VarHolder ( )

inlinevirtual

Definition at line 105 of file VarHolder.h.

{}

Member Function Documentation

addVar()

bool Prompt::VarHolder::addVar ( unsigned key, double value )

inline

Adds a variable to the VarHolder.

This function adds a variable to the VarHolder using the specified key and value.

Parameters

key	The key of the variable.
value	The value of the variable.

Returns

True if the variable was successfully added, false otherwise.

Definition at line 200 of file VarHolder.h.

  {
    using namespace asg::msgUserCode;
 
    if(!hasKey(key)) {
      m_fVars.push_back(VarEntry(key, value));
      return true;
    }
 
    ANA_MSG_DEBUG(getObjectType() << "::addVar(" << key << ", " << value << ") - key already exists");
    return false;
  }

asStr() [1/2]

std::string Prompt::VarHolder::asStr

(

const Prompt::Def::Var

var

)

Definition at line 172 of file VarHolder.cxx.

{
  return convert2Str(var);
}

asStr() [2/2]

std::string Prompt::VarHolder::asStr	(	const uint32_t	key,
		const double	val )

Definition at line 178 of file VarHolder.cxx.

{
  std::stringstream s;
 
  const Prompt::Def::Var var = convert2Var(key);
 
  if(var != Prompt::Def::NONE) {
    s << asStr(var) << ": " << val;
  }
  else {
    s << var << ": " << val;
  }
 
  return s.str();
}

clearVars()

void Prompt::VarHolder::clearVars ( )

inlinevirtual

Definition at line 275 of file VarHolder.h.

  {
    m_fVars.clear();
  }

convert2Str()

std::string Prompt::VarHolder::convert2Str

(

const Prompt::Def::Var

var

)

const

Definition at line 77 of file VarHolder.cxx.

{
  const VarMap::const_iterator vit = m_gPromptVars.find(var);
 
  if(vit == m_gPromptVars.end()) {
    asg::msgUserCode::ANA_MSG_WARNING("Def::convert2Str - WARNING - unknown variable: " << var);
    return "UNKNOWN";
  }
 
  return vit->second;
}

convert2Var() [1/2]

Prompt::Def::Var Prompt::VarHolder::convert2Var

(

const std::string &

var

)

Definition at line 90 of file VarHolder.cxx.

{
  for (const VarMap::value_type &v: m_gPromptVars) {
    if(var == v.second) {
      return v.first;
    }
  }
 
  asg::msgUserCode::ANA_MSG_WARNING("Def::convert2Var - WARNING - unknown variable: " << var);
 
  return Prompt::Def::NONE;
}

convert2Var() [2/2]

Prompt::Def::Var Prompt::VarHolder::convert2Var

(

const uint32_t

key

)

Definition at line 104 of file VarHolder.cxx.

{
  getAllVarEnums();
 
  //
  // Find matching enum by value
  //
  for(Prompt::Def::Var var: m_varEnums) {
    if(static_cast<uint32_t>(var) == key) {
      return var;
    }
  }
 
  asg::msgUserCode::ANA_MSG_WARNING("Def::convert2Var - WARNING - unknown key: " << key);
 
  return Prompt::Def::NONE;
}

delVar()

bool Prompt::VarHolder::delVar ( unsigned key )

inline

Definition at line 213 of file VarHolder.h.

  {
    VarEntryVec::iterator vit = m_fVars.begin();
    while(vit != m_fVars.end()) {
      if(vit->getKey() == key) {
        vit = m_fVars.erase(vit);
      }
      else {
        ++vit;
      }
    }
 
    return false;
  }

getAllVarEnums()

void Prompt::VarHolder::getAllVarEnums

(

)

Definition at line 67 of file VarHolder.cxx.

{
  if(m_varEnums.empty()) {
    for(const Prompt::VarMap::value_type &v: m_gPromptVars) {
      m_varEnums.push_back(v.first);
    }
  }
}

getObjectType()

std::string Prompt::VarHolder::getObjectType ( )

inlinestatic

Definition at line 107 of file VarHolder.h.

{ return "VarHolder"; }

getVar() [1/3]

double Prompt::VarHolder::getVar ( const unsigned key ) const

inline

Definition at line 265 of file VarHolder.h.

  {
    //
    // Find and return, if exists, value stored at key
    //
    double val = -1.0e9;
    getVar(key, val);
    return val;
  }

getVar() [2/3]

bool Prompt::VarHolder::getVar ( const unsigned key, double & value ) const

inline

Definition at line 251 of file VarHolder.h.

  {
    //
    // Read variable
    //
    const VarEntryVec::const_iterator ivar = std::find(m_fVars.begin(), m_fVars.end(), key);
    if(ivar != m_fVars.end()) {
      value = ivar->getData();
      return true;
    }
 
    return false;
  }

getVar() [3/3]

bool Prompt::VarHolder::getVar ( const unsigned key, float & value ) const

inline

Definition at line 237 of file VarHolder.h.

  {
    //
    // Read variable
    //
    const VarEntryVec::const_iterator ivar = std::find(m_fVars.begin(), m_fVars.end(), key);
    if(ivar != m_fVars.end()) {
      value = ivar->getData();
      return true;
    }
 
    return false;
  }

hasKey()

bool Prompt::VarHolder::hasKey ( unsigned key ) const

inline

Definition at line 228 of file VarHolder.h.

  {
    return std::find(m_fVars.begin(), m_fVars.end(), key) != m_fVars.end();
  }

hasVar()

bool Prompt::VarHolder::hasVar ( unsigned key ) const

inline

Definition at line 232 of file VarHolder.h.

  {
    return std::find(m_fVars.begin(), m_fVars.end(), key) != m_fVars.end();
  }

readVars()

std::vector< Prompt::Def::Var > Prompt::VarHolder::readVars

(

const std::string &

config

)

Definition at line 123 of file VarHolder.cxx.

{
  //
  // Read vector of variable names and convert to Var enums
  //
  std::vector<std::string> keys;
  Def::StringTok(keys, config, ", ");
 
  std::vector<Prompt::Def::Var> vars;
 
  for(const std::string &key: keys) {
    const Prompt::Def::Var var = convert2Var(key);
    if(var != Prompt::Def::NONE) {
      vars.push_back(var);
    }
    else {
      asg::msgUserCode::ANA_MSG_WARNING("Prompt::Def::readVars - unknown variable name: " << key);
    }
  }
 
  return vars;
}

readVectorVars()

std::vector< Prompt::Def::Var > Prompt::VarHolder::readVectorVars

(

const std::vector< std::string > &

keys

)

Definition at line 149 of file VarHolder.cxx.

{
  //
  // Read vector of variable names and convert to Var enums
  //
  std::vector<Prompt::Def::Var> vars;
 
  for(const std::string &key: keys) {
    const Prompt::Def::Var var = convert2Var(key);
    if(var != Prompt::Def::NONE) {
      vars.push_back(var);
    }
    else {
      asg::msgUserCode::ANA_MSG_WARNING("Prompt::Def::readVars - unknown variable name: " << key);
    }
  }
 
  return vars;
}

registerAllVars()

int Prompt::VarHolder::registerAllVars

(

)

Definition at line 228 of file VarHolder.cxx.

{
  using namespace Prompt::Def;
 
  int result = 0;
 
  result += registerVar( NONE,            "NONE");
  result += registerVar( TrackJetNTrack,  "TrackJetNTrack");
  result += registerVar( DRlj,            "DRlj");
  result += registerVar( PtFrac,          "PtFrac");
  result += registerVar( PtRel,           "PtRel");
 
  // track VarHolder
  result += registerVar( LepTrackDR,            "LepTrackDR");
  result += registerVar( Pt,                    "Pt");
  result += registerVar( AbsEta,                "AbsEta");
  result += registerVar( NumberOfPIXHits,       "NumberOfPIXHits");
  result += registerVar( NumberOfSCTHits,       "NumberOfSCTHits");
  result += registerVar( NumberOfSiHits,        "NumberOfSiHits");
  result += registerVar( NumberOfSharedSiHits,  "NumberOfSharedSiHits");
  result += registerVar( NumberOfSiHoles,       "NumberOfSiHoles");
  result += registerVar( NumberOfPixelHoles,    "NumberOfPixelHoles");
  result += registerVar( TrackJetDR,            "TrackJetDR");
  result += registerVar( TrackPtOverTrackJetPt, "TrackPtOverTrackJetPt");
  result += registerVar( Z0Sin,                 "Z0Sin");
  result += registerVar( D0Sig,                 "D0Sig");
 
  // PromptLeptonImproved
  result += registerVar( MVAXBin,                                                 "MVAXBin");
  result += registerVar( RawPt,                                                   "RawPt");
  result += registerVar( PromptLeptonRNN_prompt,                                  "PromptLeptonRNN_prompt");
  result += registerVar( CaloClusterERel,                                         "CaloClusterERel");
  result += registerVar( Topoetcone30rel,                                         "topoetcone30rel");
  result += registerVar( Ptvarcone30rel,                                          "ptvarcone30rel");
  result += registerVar( Ptvarcone30_TightTTVA_pt500rel,                          "ptvarcone30_TightTTVA_pt500rel");
  result += registerVar( CaloClusterSumEtRel,                                     "CaloClusterSumEtRel");
  result += registerVar( CandVertex_normDistToPriVtxLongitudinalBest,             "CandVertex_normDistToPriVtxLongitudinalBest");
  result += registerVar( CandVertex_normDistToPriVtxLongitudinalBest_ThetaCutVtx, "CandVertex_normDistToPriVtxLongitudinalBest_ThetaCutVtx");
  result += registerVar( CandVertex_NPassVtx,                                     "CandVertex_NPassVtx");
 
  return result;
}

registerDynamicVar()

Prompt::Def::Var Prompt::VarHolder::registerDynamicVar

(

const std::string &

name

)

Definition at line 41 of file VarHolder.cxx.

{
  unsigned last_key = Prompt::Def::NONE;
 
  for(const Prompt::VarMap::value_type &v: m_gPromptVars) {
    if(name == v.second) {
      asg::msgUserCode::ANA_MSG_WARNING("Def::registerDynamicVar - variable with this name already exists: " << name);
 
      return v.first;
    }
 
    last_key = std::max<unsigned>(last_key, v.first);
  }
 
  const Def::Var new_key = static_cast<Def::Var>(last_key+1);
 
 
  if(!registerVar(new_key, name)) {
    asg::msgUserCode::ANA_MSG_WARNING("Def::registerDynamicVar - WARNING - failed to register variable name=" << name);
    return Prompt::Def::NONE;
  }
 
  return new_key;
}

registerVar()

bool Prompt::VarHolder::registerVar	(	Prompt::Def::Var	var,
		const std::string &	name )

Definition at line 21 of file VarHolder.cxx.

{
  for(const VarMap::value_type &v: m_gPromptVars) {
    if(name == v.second) {
      asg::msgUserCode::ANA_MSG_WARNING("Def::registerVar - WARNING - variable with this name already exists: var=" << var << ", name=" << name);
      return false;
    }
  }
 
  std::pair<VarMap::iterator, bool> vit = m_gPromptVars.insert(VarMap::value_type(var, name));
 
  if(!vit.second) {
    asg::msgUserCode::ANA_MSG_WARNING("Def::registerVar - WARNING - variable with this key already exists: var=" << var << ", name=" << name);
    return false;
  }
 
  return true;
}

replaceVar()

bool Prompt::VarHolder::replaceVar ( unsigned key, double value )

inline

Definition at line 178 of file VarHolder.h.

  {
    if(!hasKey(key)) {
      m_fVars.push_back(VarEntry(key, value));
      return true;
    }
    else{
      delVar(key);
      addVar(key, value);
    }
    return false;
  }

Member Data Documentation

m_fVars

VarEntryVec Prompt::VarHolder::m_fVars

private

Definition at line 143 of file VarHolder.h.

m_gPromptVars

VarMap Prompt::VarHolder::m_gPromptVars

private

Definition at line 142 of file VarHolder.h.

m_varEnums

std::vector<Prompt::Def::Var> Prompt::VarHolder::m_varEnums

private

Definition at line 145 of file VarHolder.h.

---

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

• VarHolder.h
• VarHolder.cxx