BmnCSCHitMaker.h

Go to the documentation of this file.

#ifndef BMNCSCHITMAKER_H
#define BMNCSCHITMAKER_H 1
 
#include "BmnCSCConfiguration.h"
#include "BmnCSCDigit.h"
#include "BmnCSCHit.h"
#include "BmnCSCStationSet.h"
#include "BmnCSCTransform.h"
#include "BmnTask.h"
#include "FairField.h"
#include "FairMCPoint.h"
#include "TClonesArray.h"
#include "TRegexp.h"
#include "TString.h"
 
#include <fstream>
#include <iostream>
#include <sstream>
 
using namespace std;
 
class BmnCSCHitMaker : public BmnTask
{
  public:
    BmnCSCHitMaker();
    BmnCSCHitMaker(Int_t, Int_t, Bool_t, TString alignFile = "default", Bool_t isSrc = kFALSE);
    virtual ~BmnCSCHitMaker();
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
    void ProcessDigits();
 
    void SetHitMatching(Bool_t opt = kTRUE) { fHitMatching = opt; }
 
    void SetCurrentConfig(BmnCSCConfiguration::CSC_CONFIG config)
    {
        fCurrentConfig = config;
        configSetExplicitly = true;
    }
 
    void SetXMLConfig(TString xml_file)
    {
        fCurrentConfig = BmnCSCConfiguration::None;
        XMLConfigFile = xml_file;
        configSetExplicitly = true;
    }
 
    virtual void OnlineWrite(const std::unique_ptr<TTree>& resultTree);
 
  private:
    void initCurrentConfig();
 
    TString fInputPointsBranchName;
    TString fInputDigitsBranchName;
    TString fInputDigitMatchesBranchName;
 
    TString fOutputHitsBranchName;
 
    TClonesArray* fBmnCSCPointsArray;         
    TClonesArray* fBmnCSCDigitsArray;         
    TClonesArray* fBmnCSCDigitMatchesArray;   
 
    TClonesArray* fBmnCSCHitsArray;   
    TClonesArray* fBmnCSCUpperClustersArray;   
    TClonesArray* fBmnCSCLowerClustersArray;   
 
    Bool_t fHitMatching;
    Bool_t fIsExp;   // Specify type of input data (MC or real data)
    Bool_t fIsSrc;   // Specify type of setup (SRC or BM@N)
 
    BmnCSCConfiguration::CSC_CONFIG fCurrentConfig;
    bool configSetExplicitly = false;
 
    TString XMLConfigFile;
 
    BmnCSCStationSet* StationSet;   
 
    BmnCSCTransform* TransfSet;   
 
    FairField* fField;   
 
    TString fBmnEvQualityBranchName;
    TClonesArray* fBmnEvQuality;   
 
    void LoadDetectorConfiguration();
 
    ClassDef(BmnCSCHitMaker, 1);
};
 
#endif