EtaEnergyShowerLib.h

Go to the documentation of this file.

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

 
  class EtaEnergyShowerLib : public IShowerLib {
 
  public:

    static IShowerLib* readFromROOTFile(TFile* source);

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

    virtual ~EtaEnergyShowerLib()
          {
              for ( auto& eta : m_libData) {  // eta bins
                  for (auto& ene : eta.second )  {  // energy map
                      for ( auto& spot : ene.second ) {
                          delete  spot;
                      }
                      ene.second.clear();
                  }
                  eta.second.clear();
              }
              m_libData.clear();
          }

    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 "Eta Energy ShowerLib"; }
 
  protected:
 
    virtual const std::string printParameters() const {return "";}
 
  private:
 
    EtaEnergyShowerLib(): m_mineta(0), m_maxeta(0), m_onlyPositive(false) {}

    bool read(TTree* source);
    bool write(TTree* dest) const;
    bool readStructure(std::vector<float>& structure);
 
    bool checkEtaAndStuff(const G4Track* track) const;
 
    float m_mineta, m_maxeta;
 
    typedef std::map<float,Shower> etabin;
    typedef std::map<float,etabin> library;
 
    library m_libData;
    bool m_onlyPositive;
 
  };
 
} // namespace ShowerLib
 
#endif // SHOWER_LIB_ETA_SHOWER_LIB_H