IShowerLib.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef SHOWER_LIB_ISHOWER_LIB_H
#define SHOWER_LIB_ISHOWER_LIB_H
 
// STL includes
#include <string>
#include <vector>
#include "LArG4Code/EnergySpot.h"
 
#include "LArG4ShowerLib/Shower.h"
#include "LArG4ShowerLib/ShowerLibStatistics.h"
 
// forward declarations
class G4Track;
class TFile;
class TTree;
 
#include "AtlasHepMC/GenParticle_fwd.h"
 
// Namespace for the ShowerLib related classes
namespace ShowerLib {
 

 
  class IShowerLib {
 
  public:

    virtual ~IShowerLib() {}

    virtual std::vector<EnergySpot>* getShower(const G4Track* track, ShowerLibStatistics* stats, int randomShift) const = 0;
    virtual double getContainmentZ(const G4Track* track) const = 0;
    virtual double getContainmentR(const G4Track* track) const = 0;
    virtual bool storeShower(HepMC::ConstGenParticlePtr genParticle,const Shower* shower) = 0;
    virtual bool writeToROOT(TFile* dest) = 0;
 
    virtual ShowerLibStatistics* createStatistics() const = 0;
    const std::string statistics() const;

    inline virtual const std::string detector() const;
//    //! set detector tag
//    inline virtual void detector(const std::string& det);
    inline virtual int particle_id() const;
//    //! set particle tag
//    inline virtual void particle_id(const int particle_id);
    inline virtual const std::string release() const;
    inline virtual void release(const std::string& release);
    inline virtual const std::string geometry() const;
    inline virtual void geometry(const std::string& geometry);
    inline virtual const std::string geantVersion() const;
    inline virtual void geantVersion(const std::string& version);
    inline virtual const std::string physicsList() const;
    inline virtual void physicsList(const std::string& list);
    inline virtual const std::string comment() const;
    inline virtual void comment(const std::string& comment);
    inline virtual void addComment(const std::string& comment);
 
    inline virtual const std::string getName() const { return "IShowerLib"; }
 
  protected:

    IShowerLib() : m_detector("NotSet"), m_particle(0), m_release("NotSet"), m_geometry("NotSet"), m_geantVersion("NotSet"), m_physicsList("NotSet"), m_filled(false) {}
    bool readMeta(TTree* source);
    bool writeMeta(TTree* dest) const;
    
    virtual const std::string printParameters() const = 0;
 
    std::string m_detector;                         
    int m_particle;                                 
    std::string m_release;                          
    std::string m_geometry;                         
    std::string m_geantVersion;                     
    std::string m_physicsList;                      
    std::string m_comment;                          
 
    bool m_filled;                        
 
  };
 
 
  // inline functions
 
  const std::string IShowerLib::detector() const { return m_detector; }
  //void IShowerLib::detector(const std::string& det) { m_detector = det; }
 
  int IShowerLib::particle_id() const { return m_particle; }
  //void IShowerLib::particle_id(const int particle) { m_particle = particle; }
 
  const std::string IShowerLib::release() const { return m_release; }
  void IShowerLib::release(const std::string& release) { m_release = release; }
 
  const std::string IShowerLib::geometry() const { return m_geometry; }
  void IShowerLib::geometry(const std::string& geometry) { m_geometry = geometry; }
 
  const std::string IShowerLib::geantVersion() const { return m_geantVersion; }
  void IShowerLib::geantVersion(const std::string& version) { m_geantVersion = version; }
 
  const std::string IShowerLib::physicsList() const { return m_physicsList; }
  void IShowerLib::physicsList(const std::string& list) { m_physicsList = list; }
 
  const std::string IShowerLib::comment() const { return m_comment; }
  void IShowerLib::comment(const std::string& comment) { m_comment = comment; }
  void IShowerLib::addComment(const std::string& comment) { m_comment += comment; }
 
} // namespace ShowerLib
 
#endif // SHOWER_LIB_ISHOWER_LIB_H