TestShowerLib.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
 
#ifndef SHOWER_LIB_TEST_SHOWER_LIB_H
#define SHOWER_LIB_TEST_SHOWER_LIB_H
 
// STL includes
#include <list>
 
// local includes
#include "LArG4ShowerLib/IShowerLib.h"
 
// G4 forward declarations
class G4Track;
// forward declarations in namespace
#include "AtlasHepMC/SimpleVector.h"
// Namespace for the ShowerLib related classes
namespace ShowerLib {

 
  class TestShowerLib : public IShowerLib {
 
  public:

    static IShowerLib* readFromROOTFile(TFile* source);

    static IShowerLib* createEmptyLib(const std::string& inputFile);

    virtual ~TestShowerLib();

    virtual std::vector<EnergySpot>* getShower(const G4Track* track, ShowerLibStatistics* stats, int randomShift) const;
    virtual double getContainmentZ(const G4Track* track) const;
    virtual double getContainmentR(const G4Track* track) const;
    virtual bool storeShower(HepMC::ConstGenParticlePtr genParticle,const Shower* shower);
    virtual bool writeToROOT(TFile* dest);
 
    virtual ShowerLibStatistics* createStatistics() const;
 
    inline virtual const std::string getName() const { return "Test Energy ShowerLib"; }
 
  protected:
 
    virtual const std::string printParameters() const {return "";}
 
  private:
 
    TestShowerLib() {}
 
    struct genInfo {
                std::unique_ptr<HepMC::FourVector> vertex;
                std::unique_ptr<HepMC::FourVector> momentum;
    };
 
    typedef std::pair<genInfo,Shower> storedShower;
    typedef std::list<storedShower> library;

    bool read(TTree* source);
    bool write(TTree* dest) const;
 
    library m_libData;
  };
 
} // namespace ShowerLib
 
#endif // SHOWER_LIB_ETA_SHOWER_LIB_H