/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/IsolationSelection/IsolationSelection/IsolationSelectionTool.h

// Dear emacs, this is -*- c++ -*-
// $Id: IsolationSelectionTool.h 704447 2015-10-29 12:00:39Z jdevivi $
#ifndef ISOLATIONSELECTION_ISOLATIONSELECTIONTOOL_H
#define ISOLATIONSELECTION_ISOLATIONSELECTIONTOOL_H

// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"

// Local include(s):
#include "IsolationSelection/IIsolationSelectionTool.h"
#include "IsolationSelection/IsolationWP.h"

// Forward declaration(s):
class TFile;
class TH1F;

class Interp3D;

namespace CP 
{

  
  class IsolationSelectionTool :  public virtual CP::IIsolationSelectionTool,
                                  public asg::AsgTool {

    ASG_TOOL_CLASS( IsolationSelectionTool , CP::IIsolationSelectionTool )

    public:
      IsolationSelectionTool( const std::string& name ); 
      virtual ~IsolationSelectionTool();
      
      virtual StatusCode initialize();
      virtual StatusCode finalize();   
      
      enum IsoWPType{Efficiency, Cut};
      virtual const Root::TAccept& accept( const xAOD::Photon& x   ) const;
      virtual const Root::TAccept& accept( const xAOD::Electron& x ) const;
      virtual const Root::TAccept& accept( const xAOD::Muon& x     ) const;
      virtual const Root::TAccept& accept( const strObj& x     ) const;
      virtual const Root::TAccept& accept(const xAOD::IParticle& x) const; // for tracks, and others?
      
      virtual const Root::TAccept& getPhotonTAccept()   const;
      virtual const Root::TAccept& getElectronTAccept() const;
      virtual const Root::TAccept& getMuonTAccept()     const;
      virtual const Root::TAccept& getObjTAccept()     const;
     
      StatusCode addWP(std::string WP, xAOD::Type::ObjectType type);
      StatusCode addWP(IsolationWP* wp, xAOD::Type::ObjectType type);
      StatusCode addMuonWP(std::string wpname);
      StatusCode addPhotonWP(std::string wpname);
      StatusCode addElectronWP(std::string wpname);
      StatusCode addUserDefinedWP(std::string WPname, xAOD::Type::ObjectType ObjType, std::vector< std::pair<xAOD::Iso::IsolationType, std::string> >& cuts, std::string key="", IsoWPType type=Efficiency);
      StatusCode setIParticleCutsFrom(xAOD::Type::ObjectType ObjType);
      void addCutToWP(IsolationWP* wp, std::string key, const xAOD::Iso::IsolationType t, const std::string expression);
      std::vector< IsolationWP* >& getMuonWPs(){return m_muWPs;}
      std::vector< IsolationWP* >& getElectronWPs(){return m_elWPs;}
      std::vector< IsolationWP* >& getPhotonWPs(){return m_phWPs;}
      std::vector< IsolationWP* >& getObjWPs(){return m_objWPs;}

      // Clearing, for very special use
      void clearPhotonWPs()   { m_phWPs.clear();}
      void clearElectronWPs() { m_elWPs.clear();}
      void clearMuonWPs()     { m_muWPs.clear();}
    
    private:
      // Stupid ROOT
      TDirectory* getTemporaryDirectory(void) const;
      
      std::string m_muWPname;
      std::string m_elWPname;
      std::string m_phWPname;
      std::string m_muWPKey;
      std::string m_elWPKey;
      std::string m_phWPKey;

      std::string m_calibFileName;
      std::shared_ptr<TFile> m_calibFile;

      std::vector< IsolationWP* > m_muWPs;
      std::vector< IsolationWP* > m_elWPs;
      std::vector< IsolationWP* > m_phWPs;
      std::vector< IsolationWP* > m_objWPs;

      mutable Root::TAccept m_photonAccept;
      mutable Root::TAccept m_electronAccept;
      mutable Root::TAccept m_muonAccept;
      mutable Root::TAccept m_objAccept;

      std::vector< IsolationWP* >* m_iparWPs;
      mutable Root::TAccept* m_iparAccept;

      // for cut interpolation
      bool m_doInterpM, m_doInterpE;
      Interp3D *m_Interp;

      // Doc
      std::string m_TwikiLoc;

  };
} // namespace

#endif

---