FCALDistEnergyShowerLib.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef SHOWER_LIB_FCALDIST_ENERGY_SHOWER_LIB_H
#define SHOWER_LIB_FCALDIST_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 FCALDistEnergyShowerLib : public IShowerLib {
 
  public:
 
    static IShowerLib* readFromROOTFile(TFile* source);
 
    static IShowerLib* createEmptyLib(const std::string& inputFile);
 
    virtual ~FCALDistEnergyShowerLib()
          {
              for ( auto& dist : m_libData) {  // dist bins
                  for (auto& ene : dist.second )  {  // energy map
                      for ( auto& spot : ene.second ) {
                          delete  spot;
                      }
                      ene.second.clear();
                  }
                  dist.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 Energy ShowerLib"; }
 
  protected:
 
    virtual const std::string printParameters() const;
 
  private:
 
    FCALDistEnergyShowerLib():m_xrodcent(0.0),m_yrodcent(0.0),m_step(0.0) {}
 
    bool read(TTree* source);
    bool write(TTree* dest) const;
    bool readStructure(std::vector<float>& structure);
 
    float distance(double x, double y) const;
 
    typedef std::map<float,Shower> distbin;
    typedef std::map<float,distbin> library;
 
    library m_libData;
 
    //distance calculator parameters
    double m_xrodcent;
    double m_yrodcent;
    double m_step;
 
  };
 
} // namespace ShowerLib
 
#endif // SHOWER_LIB_FCALDIST_SHOWER_LIB_H