AtSpaceChargeTask.h

Go to the documentation of this file.

/*****************************************************************/
/*    AtSpaceChargeTask: Simulates the effect of space charge on */
/*    the electrons as they're drifted to the pad plane. Based   */
/*    on code by Juan Zamora                                     */
/*    Author: Adam Anthony (NSCL)             */
/*    anthonya@frib.msu.edu                                      */
/*****************************************************************/
#ifndef ATSPACECHARGETASK_H
#define ATSPACECHARGETASK_H
 
#include <FairTask.h>
 
class TClonesArray;
class AtDigiPar;
class AtMap;
class AtMCPoint;
 
using AtMapPtr = std::shared_ptr<AtMap>;
 
class AtSpaceChargeTask : public FairTask {
 
protected:
   AtMapPtr fMap; 
 
   Int_t fEventID = 0; 
 
   Double_t fVelDrift = 0;    
   Double_t fDetPadPlane = 0; 
 
   AtDigiPar *fPar; 
 
   TClonesArray *fMCPointArray;          
   TClonesArray *fSpaceChargePointArray; 
   Bool_t fIsPersistent = kTRUE;         
 
   TString fInputBranchName = "AtTpcPoint";
   TString fOutputBranchName = "AtSpaceChargePoint";
 
   void getParameters();
   void processPoint(AtMCPoint *point);
 
public:
   AtSpaceChargeTask();
   AtSpaceChargeTask(const char *name);
   ~AtSpaceChargeTask();
 
   void SetPersistence(Bool_t val) { fIsPersistent = val; }
   void SetMap(AtMapPtr map) { fMap = map; };
   void SetInputBranchName(TString name) { fInputBranchName = name; }
   void SetOutputBranchName(TString name) { fOutputBranchName = name; }
 
   virtual InitStatus Init() override;        
   virtual void Exec(Option_t *opt) override; 
   virtual void SetParContainers() override;  
 
   ClassDefOverride(AtSpaceChargeTask, 1);
};
 
#endif //#ifndef ATSPACECHARGETASK_H