FCALDistEtaEnergyShowerLib.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef SHOWER_LIB_FCALDIST_ETA_ENERGY_SHOWER_LIB_H
#define SHOWER_LIB_FCALDIST_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 {
 
  // forward declarations in namespace

 
  class FCALDistEtaEnergyShowerLib : public IShowerLib {
 
  public:

    static IShowerLib* readFromROOTFile(TFile* source);

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

    virtual ~FCALDistEtaEnergyShowerLib()
          {
              for ( auto& eta : m_libData) {  // dist bins
                  for ( auto& dist : eta.second ) { // eta bins
                      for (auto& ene : dist.second )  { // energy map
                          for ( auto& spot : ene.second ) {
                              delete  spot;
                          }
                          ene.second.clear();
                      }
                      dist.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 "FCALDist Eta Energy ShowerLib"; }
 
  protected:
 
    virtual const std::string printParameters() const;
 
  private:
 
    FCALDistEtaEnergyShowerLib():m_xrodcent(0.0)
                                ,m_yrodcent(0.0)
                                ,m_step(0.0)
#ifndef __FSLIB_NO_BACKWARD_COMPAT__
                                ,m_compat(false)
#endif
    {}

    bool read(TTree* source);
    bool write(TTree* dest) const;
    bool readStructure(std::map<float, std::vector<float> >& structure);
 
    float distance(double x, double y) const;
 
    typedef std::map<float,Shower> distbin;
    typedef std::map<float,distbin> etabin;
    typedef std::map<float,etabin> library;
 
    library m_libData;
 
    //distance calculator parameters
    double m_xrodcent;
    double m_yrodcent;
    double m_step;
 
#ifndef __FSLIB_NO_BACKWARD_COMPAT__
    bool m_compat;
#endif
 
  };
 
} // namespace ShowerLib
 
#endif // SHOWER_LIB_FCALDIST_SHOWER_LIB_H