BmnCSCDigitizer.h

Go to the documentation of this file.

#ifndef BMNCSCDIGITIZER_H
#define BMNCSCDIGITIZER_H 1
 
#include "BmnCSCConfiguration.h"
#include "BmnCSCDigit.h"
#include "BmnCSCHit.h"
#include "BmnCSCStationSet.h"
#include "BmnCSCTransform.h"
#include "FairMCPoint.h"
#include "FairTask.h"
#include "Rtypes.h"
#include "TClonesArray.h"
 
#include <iostream>
 
using namespace std;
 
class BmnCSCDigitizer : public FairTask
{
  public:
    BmnCSCDigitizer();
 
    BmnCSCDigitizer(UInt_t run_period);
 
    virtual ~BmnCSCDigitizer();
 
    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 SetCurrentConfig(BmnCSCConfiguration::CSC_CONFIG config) { fCurrentConfig = config; }
 
    void SetXMLConfig(TString xml_file)
    {
        fCurrentConfig = BmnCSCConfiguration::None;
        XMLConfigFile = xml_file;
    }
 
  private:
    TString fInputBranchName;
    TString fOutputDigitsBranchName;
    TString fOutputDigitMatchesBranchName;
 
    TClonesArray* fBmnCSCPointsArray;
 
    TClonesArray* fMCTracksArray;
 
    TClonesArray* fBmnCSCDigitsArray;
 
    TClonesArray* fBmnCSCDigitMatchesArray;
 
    Bool_t fOnlyPrimary;
    Bool_t fStripMatching;
 
    BmnCSCConfiguration::CSC_CONFIG fCurrentConfig;
    TString XMLConfigFile;
    BmnCSCStationSet* StationSet;   // Entire CSC detector
    BmnCSCTransform* TransfSet;     // Transformations for each module of the detector
 
    ClassDef(BmnCSCDigitizer, 1);
};
 
#endif