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);

  bool realize(SQHit& dHit);

  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; }

private:
  GeomSvc* p_geomSvc;

  SQHitVector* digits;

  std::map<std::string, PHG4HitContainer*> hitContainerByName;

  std::map<std::string, int> detIDByName;

  bool enableDC1;
  bool enableDPHodo;
};

#endif