SQDigitizer.h

Go to the documentation of this file.

#ifndef SQDigitizer_H
#define SQDigitizer_H
 
#include <fun4all/SubsysReco.h>
#include <geom_svc/GeomSvc.h>
 
#include <vector>
#include <string>
#include <map>
 
#ifndef __CINT__
#include <boost/array.hpp>
#include <Geant4/G4ThreeVector.hh>
#endif
 
#include <TSpline.h>
 
class PHG4Hit;
class SQHit;
class SQHitVector;
class PHG4HitContainer;
 
class SQDigitizer: public SubsysReco
{
public:
  SQDigitizer(const std::string& name = "SQDigitizer", const int verbose = 0);
  virtual ~SQDigitizer();
 
#ifndef __CINT__
  int Init(PHCompositeNode* topNode);
#endif
 
  int InitRun(PHCompositeNode* topNode);
 
  int process_event(PHCompositeNode* topNode);
 
  void digitizePlane(const std::string& detName);
 
  void digitizeEMCal(const std::string& detName);
 
  int getTriggerLv(int detectorID) { return p_geomSvc->getTriggerLv(detectorID); }
 
  void registerEMCal(std::string ecalName, int ecalID = 100) { detIDByName[ecalName] = ecalID; }
 
  void set_enable_st1dc(const bool en)  { enableDC1 = en; }
  void set_enable_dphodo(const bool en) { enableDPHodo = en; }
 
  void set_digitize_secondaries(const bool val) { digitize_secondaries = val; }
 
private:
  GeomSvc* p_geomSvc;
 
  SQHitVector* digits;
 
  std::map<std::string, PHG4HitContainer*> hitContainerByName;
 
  std::map<std::string, int> detIDByName;
 
  bool enableDC1;
  bool enableDPHodo;
  bool digitize_secondaries;
};
 
#endif