ParticleSelectionAlg.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// ParticleSelectionAlg.h
// Header file for class ParticleSelectionAlg
// Author: Karsten Koeneke <karsten.koeneke@cern.ch>
#ifndef EVENTUTILS_PARTICLESELECTIONALG_H
#define EVENTUTILS_PARTICLESELECTIONALG_H 1
 
// FrameWork includes
#include "GaudiKernel/ToolHandle.h"
// #include "GaudiKernel/ServiceHandle.h"
// #include "AthenaBaseComps/AthAlgorithm.h"
#include "AthAnalysisBaseComps/AthAnalysisAlgorithm.h"
#include "xAODBase/IParticleContainer.h"
#include "PATCore/IAsgSelectionTool.h"
 
// STL includes
#include <string>
#include <vector>
 
#include "ExpressionEvaluation/ExpressionParserUser.h"
 
 
class ParticleSelectionAlg
  : public ExpressionParserUser<::AthAnalysisAlgorithm>
{
 
  // Public methods:
 public:
 
  // Copy constructor:
 
  ParticleSelectionAlg( const std::string& name, ISvcLocator* pSvcLocator );
 
  virtual ~ParticleSelectionAlg();
 
 
  virtual StatusCode  initialize() override;
 
  virtual StatusCode  start() override;
 
  virtual StatusCode  execute() override;
 
  virtual StatusCode  finalize() override;
 
 private:
  template<class CONT, class AUXCONT>
  StatusCode selectParticles(const xAOD::IParticleContainer* inContainer,
                             const std::vector<int>& resultVec) const;
 
 
  // Private data:
 private:
  ToolHandleArray<IAsgSelectionTool> m_selTools;
 
  StringProperty m_evtInfoName;
 
  StringProperty m_inCollKey;
 
  StringProperty m_outCollKey;
 
  BooleanProperty m_writeSplitAux;
 
  StringProperty m_outOwnPolicyName;
 
  StringProperty m_selection;
 
  bool m_doCutFlow;
 
  StringProperty m_cutBKCName;
 
 
 
 
  unsigned long m_nEventsProcessed;
 
  SG::OwnershipPolicy m_outOwnPolicy;
 
  enum contType_t {
    UNKNOWN,
    PHOTON,
    ELECTRON,
    MUON,
    TAU,
    JET,
    PARITCLEFLOW,
    NEUTRALPARTICLE,
    TRACKPARTICLE,
    TRUTHPARTICLE,
    COMPOSITEPARTICLE,
    PARTICLE,
    CALOCLUSTER
  };
 
  contType_t m_contType;
 
  std::size_t m_cutBKStartIdx;
 
  std::vector<std::size_t> m_selToolIdxOffset;
 
  std::size_t m_idxSelParster;
 
 
};
 
// Include the templated code here. This must be done from this header file.
#include "ParticleSelectionAlg.icc"
 
 
#endif //> !EVENTUTILS_PARTICLESELECTIONALG_H