PndSdsStripHitProducer.h

Go to the documentation of this file.

#ifndef PNDSDSSTRIPHITPRODUCER_H
#define PNDSDSSTRIPHITPRODUCER_H

#include "PndSdsTask.h"
#include "PndSdsStripDigiPar.h"
#include "PndSdsMCPoint.h"
#include "PndSdsStrip.h"
#include "PndSdsDigiStrip.h"
#include "PndDetectorList.h"
#include "FairGeoVector.h"
#include "FairGeoTransform.h"
#include "FairMCEventHeader.h"
#include "TVector3.h"
#include "TRandom.h"
#include "TGeoMatrix.h"
#include "TGeoBBox.h"

#include "PndGeoHandling.h"
#include "PndSdsCalcStrip.h"
#include "PndSdsChargeConversion.h"
#include "PndSdsDigiStripWriteoutBuffer.h"

#include <string>
#include <vector>
#include <map>

class TClonesArray;


class PndSdsStripHitProducer : public PndSdsTask {
 public:
  PndSdsStripHitProducer();

  PndSdsStripHitProducer(PndSdsStripHitProducer &other)
    : PndSdsTask(), fPointArray(other.fPointArray), fStripArray(other.fStripArray), fDataBuffer(other.fDataBuffer), fDigiParameterList(other.fDigiParameterList),
      fChargeDigiParameterList(other.fChargeDigiParameterList), fCurrentDigiPar(other.fCurrentDigiPar), fCurrentChargeConverter(other.fCurrentChargeConverter),
      fStripCalcTop(other.fStripCalcTop), fStripCalcBot(other.fStripCalcBot), fChargeConverter(other.fChargeConverter), fCurrentStripCalcTop(other.fCurrentStripCalcTop),
      fCurrentStripCalcBot(other.fCurrentStripCalcBot), fMcEventHeader(other.fMcEventHeader), fGeoH(other.fGeoH), fOverrideParams(other.fOverrideParams),
      fTimeOrderedDigi(other.fTimeOrderedDigi), fEventNr(other.fEventNr){};
  PndSdsStripHitProducer(const char *name);

  virtual ~PndSdsStripHitProducer();

  virtual void SetParContainers();
  virtual InitStatus Init();
  virtual InitStatus ReInit();

  virtual void SetBranchNames() = 0;
  virtual void SetMCPointType(){};

  virtual void Exec(Option_t *opt);
  virtual void FinishEvent();
  virtual void FinishTask();
  void AddDigi(Int_t &iStrip, Int_t iPoint, Int_t detID, Int_t sensorID, Int_t fe, Int_t chan, Double_t charge);

  void RunTimeBased() { fTimeOrderedDigi = kTRUE; }

  PndSdsStripHitProducer &operator=(PndSdsStripHitProducer &other)
  {
    if (this != &other) // protect against invalid self-assignment
    {
      fPointArray = other.fPointArray;
      fStripArray = other.fStripArray;
      fDataBuffer = other.fDataBuffer;
      fDigiParameterList = other.fDigiParameterList;
      fChargeDigiParameterList = other.fChargeDigiParameterList;
      fCurrentDigiPar = other.fCurrentDigiPar;
      fCurrentChargeConverter = other.fCurrentChargeConverter;
      fStripCalcTop = other.fStripCalcTop;
      fStripCalcBot = other.fStripCalcBot;
      fChargeConverter = other.fChargeConverter;
      fCurrentStripCalcTop = other.fCurrentStripCalcTop;
      fCurrentStripCalcBot = other.fCurrentStripCalcBot;
      fMcEventHeader = other.fMcEventHeader;
      fGeoH = other.fGeoH;
      fOverrideParams = other.fOverrideParams;
      fTimeOrderedDigi = other.fTimeOrderedDigi;
      fEventNr = other.fEventNr;
    }
    return *this;
  }

 protected:
  TClonesArray *fPointArray;

  TClonesArray *fStripArray;

  FairWriteoutBuffer *fDataBuffer;

  TList *fDigiParameterList;
  TList *fChargeDigiParameterList;
  PndSdsStripDigiPar *fCurrentDigiPar;
  PndSdsChargeConversion *fCurrentChargeConverter;

  std::map<const char *, PndSdsCalcStrip *> fStripCalcTop;
  std::map<const char *, PndSdsCalcStrip *> fStripCalcBot;
  std::map<const char *, PndSdsChargeConversion *> fChargeConverter;
  PndSdsCalcStrip *fCurrentStripCalcTop;
  PndSdsCalcStrip *fCurrentStripCalcBot;

  void Register();
  void Reset();
  void ProduceHits();
  virtual void SetCalculators();
  Bool_t SelectSensorParams(Int_t sensorID);

  Double_t DigitizeTime(Double_t time, Double_t charge);

  FairMCEventHeader *fMcEventHeader; 

  PndGeoHandling *fGeoH;   
  Bool_t fOverrideParams;  
  Bool_t fTimeOrderedDigi; 

  Int_t fEventNr; 

  ClassDef(PndSdsStripHitProducer, 5);
};

#endif