PndTrackFunctor.h

Go to the documentation of this file.

/*
 * PndTrackFunctor.h
 *
 *  Created on: 02.02.2017
 *      Author: Stockmanns
 */

#ifndef PNDTOOLS_TRACKINGQA_PNDTRACKFUNCTOR_H_
#define PNDTOOLS_TRACKINGQA_PNDTRACKFUNCTOR_H_

#include "FairMultiLinkedData.h"
#include "FairRootManager.h"

#include <functional>

class PndTrackFunctor : public std::binary_function<FairMultiLinkedData* , Bool_t, Bool_t>
{
  public :
    virtual Bool_t operator() (FairMultiLinkedData* a, Bool_t primary) {return Call(a, primary);};
    static PndTrackFunctor* make_PndTrackFunctor(std::string functorName);
    virtual Bool_t Call(FairMultiLinkedData* a, Bool_t primary) = 0;
    virtual void Print() = 0;

    virtual ~PndTrackFunctor() {};

};

class StandardTrackFunctor : public PndTrackFunctor
{
        Bool_t Call(FairMultiLinkedData* a, Bool_t ){ // primary //[R.K.03/2017] unused variable(s)
                FairRootManager* ioman = FairRootManager::Instance();
                Bool_t possibleTrack = kFALSE;
                possibleTrack = (possibleTrack | ((a->GetLinksWithType(ioman->GetBranchId("MVDHitsPixel")).GetNLinks() +
                                                                                 a->GetLinksWithType(ioman->GetBranchId("MVDHitsStrip")).GetNLinks()) > 3));

                possibleTrack = (possibleTrack | ((a->GetLinksWithType(ioman->GetBranchId("MVDHitsPixel")).GetNLinks() +
                                                                                 a->GetLinksWithType(ioman->GetBranchId("MVDHitsStrip")).GetNLinks() +
                                                                                 a->GetLinksWithType(ioman->GetBranchId("STTHit")).GetNLinks() +
                                                                                 a->GetLinksWithType(ioman->GetBranchId("GEMHit")).GetNLinks()) > 5));

                return possibleTrack;
        }

        void Print(){
                std::cout << "StandardTrackFunctor: > 3 Hits in MVD or > 5 Hits in (MVD+Stt+GEM)" << std::endl;
        }
};

class OnlySttFunctor : public PndTrackFunctor
{
        Bool_t Call(FairMultiLinkedData* a, Bool_t ){ // primary //[R.K.03/2017] unused variable(s)
                FairRootManager* ioman = FairRootManager::Instance();
                Bool_t possibleTrack = kFALSE;

                possibleTrack = (possibleTrack | (a->GetLinksWithType(ioman->GetBranchId("STTHit")).GetNLinks() > 5));

                return possibleTrack;
        }
        void Print(){
                std::cout << "OnlySttFunctor: > 5 Hits in Stt" << std::endl;
        }
};

class RiemannMvdSttGemFunctor : public PndTrackFunctor
{
        Bool_t Call(FairMultiLinkedData* a, Bool_t ){ // primary //[R.K.03/2017] unused variable(s)
                FairRootManager* ioman = FairRootManager::Instance();
                Bool_t possibleTrack = kFALSE;
                Bool_t mvdHits =  ((a->GetLinksWithType(ioman->GetBranchId("MVDHitsPixel")).GetNLinks() +
                                                        a->GetLinksWithType(ioman->GetBranchId("MVDHitsStrip")).GetNLinks()) > 2);

                if (mvdHits){
                        possibleTrack = (a->GetLinksWithType(ioman->GetBranchId("STTHit")).GetNLinks() > 1)
                                                        | (a->GetLinksWithType(ioman->GetBranchId("GEMHit")).GetNLinks() > 1);
                }
                return possibleTrack;
        }
        void Print(){
                std::cout << "RiemannMvdSttGemFunctor: > 2 Hits in MVD and >0 Hits in (Stt+Gem)" << std::endl;
        }

};

class CircleHoughTrackFunctor : public PndTrackFunctor
{
        Bool_t Call(FairMultiLinkedData* a, Bool_t primary){
                if (primary == kFALSE) return kFALSE;
                FairRootManager* ioman = FairRootManager::Instance();
                Bool_t possibleTrack = kFALSE;

                Int_t nHitsMvdPixel = a->GetLinksWithType(ioman->GetBranchId("MVDHitsPixel")).GetNLinks();
                Int_t nHitsMvdStrip = a->GetLinksWithType(ioman->GetBranchId("MVDHitsStrip")).GetNLinks();
                if (nHitsMvdPixel + nHitsMvdStrip > 2) {  // First requirement: more than two MVD hits
                        //Int_t nHitsStt = a->GetLinksWithType(ioman->GetBranchId("STTHit")).GetNLinks();
                        //Int_t nHitsGem = a->GetLinksWithType(ioman->GetBranchId("GEMHit")).GetNLinks();
                        possibleTrack = kTRUE;
                        //possibleTrack = (nHitsMvdPixel + nHitsMvdStrip + nHitsStt + nHitsGem > 6);  // Second requirement: More than six hits total
                }
                return possibleTrack;
        }
        void Print(){
                std::cout << "CircleHoughTrackFunctor: >= 3 Hits in MVD and primary track" << std::endl;
        }

};

class FtsTrackFunctor : public PndTrackFunctor
{
        Bool_t Call(FairMultiLinkedData* a, Bool_t ) { // primary //[R.K.03/2017] unused variable(s)
                FairRootManager* ioman = FairRootManager::Instance();
                Bool_t possibleTrack = kFALSE;

                possibleTrack = (possibleTrack | (a->GetLinksWithType(ioman->GetBranchId("FTSHit")).GetNLinks() > 5));

                return possibleTrack;
        }
        void Print() {
                std::cout << "FTSTrackFunctor: > 5 Hits in Fts" << std::endl;
        }
};

class NoFtsTrackFunctor : public PndTrackFunctor
{
        Bool_t Call(FairMultiLinkedData* a, Bool_t primary) {
                FairRootManager* ioman = FairRootManager::Instance();
                Bool_t possibleTrack = kFALSE;

                possibleTrack = (possibleTrack | !(a->GetLinksWithType(ioman->GetBranchId("FTSHit")).GetNLinks() > 0));
                if (possibleTrack == kFALSE) return kFALSE;
                StandardTrackFunctor standard;
                possibleTrack = kFALSE;
                possibleTrack = (possibleTrack | standard(a, primary));

                return possibleTrack;
        }
        void Print() {
                std::cout << "noFTSTrackFunctor: no Hits in Fts + standardFunctor!" << std::endl;
        }
};

class AllTracksFunctor : public PndTrackFunctor
{
        Bool_t Call(FairMultiLinkedData* , Bool_t ) { //a  primary//[R.K.03/2017] unused variable(s)
                return kTRUE;
        }
        void Print() {
                std::cout << "AllTracksFunctor: take everything" << std::endl;
        }
};



#endif /* PNDTOOLS_TRACKINGQA_PNDTRACKFUNCTOR_H_ */