PndMvdSttGemRiemannTrackFinder.h

Go to the documentation of this file.

#ifndef PndMvdSttGemRiemannTrackFinder_H_
#define PndMvdSttGemRiemannTrackFinder_H_

#include "TClonesArray.h"
#include "PndRiemannTrackFinder.h"
#include "PndGeoHandling.h"
#include "PndSttStrawMap.h"

#include "TH2F.h"
#include <vector>

class PndSttHit;

class PndMvdSttGemRiemannTrackFinder : public  PndRiemannTrackFinder
{
public:
        PndMvdSttGemRiemannTrackFinder();
        virtual ~PndMvdSttGemRiemannTrackFinder();
  PndMvdSttGemRiemannTrackFinder(const PndMvdSttGemRiemannTrackFinder& ) = delete;
  PndMvdSttGemRiemannTrackFinder& operator=(const PndMvdSttGemRiemannTrackFinder& ) = delete;

        void FindTracks();                                                                              
        void AddHits(TClonesArray* hits, Int_t branchId);

        //void SetGeoH(PndGeoHandling geoH){fGeoH=geoH;};
        void SetCutDistH(TH2F* hist) {fCutDistH=hist;}
        void SetCutChi2H(TH2F* hist) {fCutChi2H=hist;}

        void SetSttTubeArray(TClonesArray* SttTubeArray){fStrawMap.GenerateStrawMap(SttTubeArray);}

private:
        PndGeoHandling* fGeoH;

        std::vector<std::set<Int_t> >  GetStartTracks();
        bool CheckSZ(PndRiemannTrack aTrack);                           
        bool CheckRiemannHit(PndRiemannTrack* track, PndRiemannHit* hit, FairHit* fairHit);
        bool CheckRiemannHitMvd(PndRiemannTrack* track, PndRiemannHit* hit, FairHit* fairHit);
        bool CheckRiemannHitGem(PndRiemannTrack* track, PndRiemannHit* hit, FairHit* fairHit);
        bool CheckRiemannHitStt(PndRiemannTrack* track, PndRiemannHit* hit, FairHit* fairHit);
        bool CheckRiemannHitSkewedStt(PndRiemannTrack* track, PndRiemannHit* hit, FairHit* fairHit);

//      bool CheckTrackStt(PndRiemannTrack* track);

        void AssignSttHits(PndRiemannTrack& actTrack, std::set<Int_t>& startTrack);
        void AssignSkewedSttHits(PndRiemannTrack& actTrack, std::set<Int_t>& startTrack);
        void AssignGemHits(PndRiemannTrack& actTrack, std::set<Int_t>& startTrack);

        bool CheckBoarderHitsStt(PndTrackCand* track);
        int GetStrawSector(PndRiemannTrack& track);

        int GetLayerGem(FairHit* hit);
        int GetLayerMvd(FairHit* hit);
        int GetLayer(TString identifier);
        void InitLayerMap();
        void InitLayerMapMvd();
        void InitLayerMapGem();

        TH2F* fCutDistH;  
        TH2F* fCutChi2H;

        TClonesArray* fSttHits;
        TClonesArray* fSkewedSttHits;
        TClonesArray* fGemHits;
        PndSttStrawMap fStrawMap;
        std::vector<std::vector<PndSttHit*> > fSttHitsInSectors;
        std::vector<std::vector<FairHit*> > fSttSkewedHitsInSectors;

        double fZClosePar;      

        std::vector< std::vector<int> >fLayers; 
        std::map<TString, int> fLayerMap; 
        int fLastLayerId; 
        /*       b - barrel layer;   d - disk layer
        0 - zero position
        1 - 1b
        2 - 1d
        3 - 2d
        4 - 2b
        5 - 3d
        6 - 4d
        7 - 3b
        8 - 5d
        9 - 4b
   10 - 6d
*/
        int fNLayers; 
        double GetMaxPlaneDist(double radius, double dip, bool sign); //getting cut distance
        double GetMaxSZChi2(double radius, double dip, bool sign);    //getting cut Chi2
        std::set<int> GetTooCloseHitsInLayer(int LayerNumber , int HitNumber ); // searching for too close hits to HitNumber-Hit in layer LayerNumber

public:
  ClassDef(PndMvdSttGemRiemannTrackFinder,1)
};

#endif /*PndMvdSttGemRiemannTrackFinder_H_*/