ToolBox.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_TOOLBOX_H
#define ASSOCIATIONUTILS_TOOLBOX_H
 
// Framework includes
#include "AsgMessaging/AsgMessaging.h"
#include "AsgTools/AnaToolHandle.h"
 
// Local includes
#include "AssociationUtils/IOverlapTool.h"
#include "AssociationUtils/IOverlapRemovalTool.h"
 
// STL includes
#include <string>
 
// Forward declarations
class INamedInterface;
namespace asg {
  class AsgTool;
}
 
namespace ADD_CMODE(ORUtils)
{

  class ToolBox : public asg::AsgMessaging
  {
 
    public:

      using parentType_t = INamedInterface;
      using MasterHandle_t = asg::AnaToolHandle<IOverlapRemovalTool>;
      using OverlapHandle_t = asg::AnaToolHandle<IOverlapTool>;
 

      ToolBox(const std::string& name = "ORToolBox",
              parentType_t* theParent = nullptr);

      StatusCode initialize();

      std::vector<OverlapHandle_t*> getOverlapTools();

      template<class T> StatusCode
      setGlobalProperty(const std::string& property, const T& value);

      MasterHandle_t masterTool;

      OverlapHandle_t muPFJetORT;
      OverlapHandle_t eleEleORT;
      OverlapHandle_t eleMuORT;
      OverlapHandle_t eleJetORT;
      OverlapHandle_t muJetORT;
      OverlapHandle_t tauEleORT;
      OverlapHandle_t tauMuORT;
      OverlapHandle_t tauJetORT;
      OverlapHandle_t phoEleORT;
      OverlapHandle_t phoMuORT;
      OverlapHandle_t phoJetORT;
      OverlapHandle_t eleFatJetORT;
      OverlapHandle_t jetFatJetORT;

      parentType_t* parent() { return m_parent; }
 
    private:

      StatusCode initTool(OverlapHandle_t& handle, const std::string& key);

      StatusCode initMaster();

      parentType_t* m_parent;
 
  };
 
 
  //---------------------------------------------------------------------------
  // Implementation method to set a property on all tools.
  //---------------------------------------------------------------------------
  template<class T> StatusCode
  ToolBox::setGlobalProperty(const std::string& property, const T& value)
  {
    // Handle the overlap tools
    for(auto handlePtr : getOverlapTools()) {
      ATH_CHECK( handlePtr->setProperty(property, value) );
    }
    // Handle the master tool
    if(!masterTool.empty()) ATH_CHECK( masterTool.setProperty(property, value) );
    return StatusCode::SUCCESS;
  }
 
} // namespace ADD_CMODE(ORUtils)
 
#endif