CastorShowerLibrary.h

Go to the documentation of this file.

#ifndef SimG4CMS_CastorShowerLibrary_h
#define SimG4CMS_CastorShowerLibrary_h
// File: CastorShowerLibrary.h
// Description: Gets information from a shower library
//              Adapted from HFShowerLibrary class
//
//              Wagner Carvalho (wcarvalh@cern.ch)  
//

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "SimDataFormats/CaloHit/interface/CastorShowerLibraryInfo.h"
#include "SimDataFormats/CaloHit/interface/CastorShowerEvent.h"
#include "DetectorDescription/Core/interface/DDsvalues.h"

#include "G4ParticleTable.hh"
#include "G4ThreeVector.hh"
 
#include <string>
#include <memory>

class G4Step;
class TFile;
class TBranchObject;

class CastorShowerLibrary {
  
public:
  
  //Constructor and Destructor
  CastorShowerLibrary(const std::string & name, edm::ParameterSet const & p);
  ~CastorShowerLibrary();

public:

  CastorShowerEvent   getShowerHits(const G4Step*, bool&);
  int                 FindEnergyBin(double);
  int                 FindEtaBin(double);
  int                 FindPhiBin(double);

protected:

  void                initFile(edm::ParameterSet const & );
  void                getRecord(int, int);
  void                loadEventInfo(TBranchObject *);
  // if eta or phi is not given, take into account only the binning in energy
  void                select(int, double,double =0,double=9); 

private:

  TFile               *hf;                      
  TBranchObject       *evtInfo;                 // pointer to CastorShowerLibraryInfo-type branch 
  TBranchObject       *emBranch, *hadBranch;    // pointer to CastorShowerEvent-type branch

  // User defined classes in Root Dictionary 
  CastorShowerLibraryInfo  *eventInfo;
  CastorShowerEvent        *showerEvent;

  bool                verbose;
  unsigned int        nMomBin, totEvents, evtPerBin;
  
  std::vector<double> pmom;

  // new variables (bins in eta and phi)
  unsigned int        nBinsE, nBinsEta, nBinsPhi;
  unsigned int        nEvtPerBinE, nEvtPerBinEta, nEvtPerBinPhi;
  std::vector<double> SLenergies;
  std::vector<double> SLetas;
  std::vector<double> SLphis;
};
#endif