/*
 * 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){
		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)" << std::endl;
	}
};

class OnlySttFunctor : public PndTrackFunctor
{
	Bool_t Call(FairMultiLinkedData* a, Bool_t primary){
		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){
		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) {
		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 = (possibleTrack | standard(a, primary));

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

class AllTracksFunctor : public PndTrackFunctor
{
	Bool_t Call(FairMultiLinkedData* a, Bool_t primary) {
		return kTRUE;
	}
	void Print() {
		std::cout << "AllTracksFunctor: take everything" << std::endl;
	}
};



#endif /* PNDTOOLS_TRACKINGQA_PNDTRACKFUNCTOR_H_ */