WriteThinnedData.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
// WriteThinnedData.h 
// Header file for class WriteThinnedData
// Author: S.Binet<binet@cern.ch>
#ifndef ATHEXTHINNING_WRITETHINNEDDATA_H 
#define ATHEXTHINNING_WRITETHINNEDDATA_H 

 
// STL includes
#include <string>
 
 
// FrameWork includes
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/ReadHandleKeyArray.h"
#include "StoreGate/ThinningHandleKey.h"
 
// DataModel includes
#include "AthContainers/DataVector.h"
 
// AthExThinning includes
#include "AthExThinning/AthExIParticles.h"
#include "AthExThinning/AthExElephantino.h"
#include "AthExThinning/AthExDecay.h"
 
// Forward declaration
class AthExParticles;
//class AthExIParticles;
namespace SG {
  class ThinningDecisionBase;
}
 
namespace AthExThinning {
 
class WriteThinnedData : public ::AthAlgorithm
{ 
public: 
  WriteThinnedData( const std::string& name, ISvcLocator* pSvcLocator );

  virtual ~WriteThinnedData(); 
 
  // Assignment operator: 
  //WriteThinnedData &operator=(const WriteThinnedData &alg); 
 
  // Athena algorithm's Hooks
  virtual StatusCode  initialize() override;
  virtual StatusCode  execute(const EventContext& ctx) override;
  virtual StatusCode  finalize() override;
 
 
protected: 

  WriteThinnedData();

  StatusCode test( const EventContext& ctx,
                   int testNum, const std::string& testName );

  StatusCode doThinningTest1( const EventContext& ctx,
                              const SG::ThinningHandleKey<AthExParticles>& particlesKey ) const;

  StatusCode doThinningTest2( const EventContext& ctx,
                              const SG::ThinningHandleKey<AthExParticles>& particlesKey ) const;

  StatusCode doThinningTest3( const EventContext& ctx,
                              const SG::ThinningHandleKey<AthExIParticles>& iparticlesKey ) const;
 
 
protected: 
  // Containers
  
  StringProperty m_particlesName;
 
  SG::ThinningHandleKey<AthExParticles> m_particlesKey1
  { this, "ParticlesKey1", "", "" };
  SG::ThinningHandleKey<AthExParticles> m_particlesKey2
  { this, "ParticlesKey2", "", "" };
 
  SG::ReadHandleKeyArray<AthExIParticles> m_iparticlesKeys
  { this, "IParticlesKeys", {}, "" };
 
  SG::ThinningHandleKey<AthExIParticles> m_iparticlesKey3
  { this, "IParticlesKey3", "", "" };

  StringProperty m_decayName
  { this, "Decay", "TwoBodyDecay", "Input location of Decay" };
 
  SG::ReadHandleKeyArray<AthExDecay> m_decayKeys
  { this, "DecayKeys", {}, "" };

  StringProperty m_elephantinoName
  { this, "Elephantino", "PinkElephantino", "Input location of Elephantino" };
 
  SG::ReadHandleKeyArray<AthExElephantino> m_elephantinoKeys
  { this, "ElephantinoKeys", {}, "" };

  BooleanArrayProperty m_filter;
 
}; 


 
} //> end namespace AthExThinning
 
#endif //> ATHEXTHINNING_WRITETHINNEDDATA_H