BmnMCTrackCreator.h

Go to the documentation of this file.

 
#ifndef BMNMCTRACKCREATOR_H_
#define BMNMCTRACKCREATOR_H_
 
#include "BmnMCTrack.h"
#include <map>
#include <iostream>
#include <assert.h>
#include "BmnGemStripStationSet.h"
#include "BmnSiliconStationSet.h"
#include "BmnCSCStationSet.h"
#include "BmnGemStripStationSet_RunSpring2017.h"
 
class TClonesArray;
class FairMCPoint;
class CbmStsPoint;
class CbmGeoStsPar;
class CbmStsDigiPar;
class CbmStsDigiScheme;
 
class BmnMCTrackCreator {
private:
    BmnMCTrackCreator(TString gem, TString sil, TString csc);
 
public:
    virtual ~BmnMCTrackCreator();
 
    static BmnMCTrackCreator* Instance(TString gem, TString sil, TString csc);
 
    void Create();
 
    Bool_t TrackExists(Int_t mcId) const {
        return (fBmnMCTracks.count(mcId) > 0) ? kTRUE : kFALSE;
        //        if (fBmnMCTracks.count(mcId) == 0) return kFALSE;
        //        Int_t NPointsSil = fBmnMCTracks.find(mcId)->second.GetNofPoints(kSILICON);
        //        Int_t NPointsGem = fBmnMCTracks.find(mcId)->second.GetNofPoints(kGEM);
        //        if (NPointsSil > fSilDetector->GetNStations()) return kFALSE;
        //        if (NPointsGem > fGemDetector->GetNStations()) return kFALSE;
        //        //if (NPointsSil + NPointsGem < 4) return kFALSE;
        //        return kTRUE;
 
    }
 
    const BmnMCTrack& GetTrack(int mcId) const {
        assert(TrackExists(mcId));
        return fBmnMCTracks.find(mcId)->second;
    }
 
    Int_t GetNofTracks() const {
        return fBmnMCTracks.size();
    }
 
private:
 
    void ReadDataBranches();
 
    void AddPoints(DetectorId detId, const TClonesArray* array);
 
    void AddRichHits();
 
    void AddRingParameters();
 
    void FairMCPointToBmnMCPoint(
            const FairMCPoint* fairPoint,
            BmnMCPoint* bmnPoint,
            int refId,
            int stationId);
 
    void FairMCPointCoordinatesAndMomentumToBmnMCPoint(// for DCH1, DCH2, TOF1 points
            const FairMCPoint* fairPoint,
            BmnMCPoint* bmnPoint);
 
    TClonesArray* fMCTracks;
    TClonesArray* fSilPoints;
    TClonesArray* fSsdPoints;
    TClonesArray* fGemPoints;
    TClonesArray* fTof400Points;
    TClonesArray* fTof700Points;
    TClonesArray* fCscPoints;
    TClonesArray* fDchPoints;
    
    Int_t fNSiliconStations;
    Int_t fNGemStations;
    Int_t fNCscStations;
 
    // Stores created BmnMCTrack objects.
    // std::map<MC track index, BmnMCTrack object>.
    std::map<Int_t, BmnMCTrack> fBmnMCTracks;
 
    BmnGemStripStationSet* fGemDetector;
    BmnSiliconStationSet* fSilDetector;
    BmnCSCStationSet* fCscDetector;
};
 
#endif /* BMNMCTRACKCREATOR_H_ */