BmnGemStripDigitizer.h

Go to the documentation of this file.

#ifndef BMNGEMSTRIPDIGITIZER_H
#define BMNGEMSTRIPDIGITIZER_H 1
 
#include "BmnGemStripConfiguration.h"
#include "BmnGemStripDigit.h"
#include "BmnGemStripHit.h"
#include "BmnGemStripStationSet.h"
#include "BmnGemStripTransform.h"
#include "FairField.h"
#include "FairMCPoint.h"
#include "FairRunSim.h"
#include "FairTask.h"
#include "Rtypes.h"
#include "TClonesArray.h"
 
#include <iostream>
 
using namespace std;
 
class BmnGemStripDigitizer : public FairTask
{
  public:
    BmnGemStripDigitizer();
 
    BmnGemStripDigitizer(UInt_t run_period);
 
    virtual ~BmnGemStripDigitizer();
 
    virtual InitStatus Init();
 
    virtual void Exec(Option_t* opt);
 
    virtual void Finish();
 
    void ProcessMCPoints();
 
    // Setters
 
    void SetOnlyPrimary(Bool_t opt = kFALSE) { fOnlyPrimary = opt; }
 
    void SetStripMatching(Bool_t opt = kTRUE) { fStripMatching = opt; }
 
    void SetUseRealEffects(Bool_t opt = kTRUE) { fUseRealEffects = opt; }
 
    void SetCurrentConfig(BmnGemStripConfiguration::GEM_CONFIG config) { fCurrentConfig = config; }
 
    void SetXMLConfig(TString xml_file)
    {
        fCurrentConfig = BmnGemStripConfiguration::None;
        XMLConfigFile = xml_file;
    }
 
    void SetNoise(Double_t sigma, Double_t thresh)
    {
        fSigma = sigma;
        fThresh = thresh;
    }   // AZ-240823 - set noise sigma and threshold
    TClonesArray* GetMCPoints() { return fBmnGemStripPointsArray; }   // AZ-110124
    TClonesArray* GetMCTracks() { return fMCTracksArray; }            // AZ-110124
 
  private:
    TString fInputBranchName;
    TString fOutputDigitsBranchName;
    TString fOutputDigitMatchesBranchName;
 
    /** Input array of Gem Points **/
    TClonesArray* fBmnGemStripPointsArray;
 
    /** Input array of MC Tracks **/
    TClonesArray* fMCTracksArray;
 
    /** Output array of Gem Digits **/
    TClonesArray* fBmnGemStripDigitsArray;
 
    /** Output array of GEM Digit Matches **/
    TClonesArray* fBmnGemStripDigitMatchesArray;
 
    Bool_t fOnlyPrimary;
    Bool_t fStripMatching;
    Bool_t fUseRealEffects;
 
    FairField* fField;
 
    BmnGemStripConfiguration::GEM_CONFIG fCurrentConfig;
 
    TString XMLConfigFile;
 
    BmnGemStripStationSet* StationSet;          // Entire GEM detector
    BmnGemStripTransform* GlobalModTransfSet;   // Transformations for detector modules
 
    Double_t fSigma;    // AZ-240823
    Double_t fThresh;   // AZ-240823
 
    ClassDef(BmnGemStripDigitizer, 1);
};
 
#endif