#include "CbmLitTrackSelectionMomentum.h"

#include "CbmLitTrack.h"

#include <cmath>
#include <iostream>

CbmLitTrackSelectionMomentum::CbmLitTrackSelectionMomentum():
	fMinMomentum(0.)
{
}

CbmLitTrackSelectionMomentum::~CbmLitTrackSelectionMomentum()
{
}

LitStatus CbmLitTrackSelectionMomentum::Initialize()
{
	return kLITSUCCESS;
}

LitStatus CbmLitTrackSelectionMomentum::Finalize()
{
	return kLITSUCCESS;
}

LitStatus CbmLitTrackSelectionMomentum::DoSelect(
		TrackIterator itBegin,
		TrackIterator itEnd)
{
	if (itBegin == itEnd) return kLITSUCCESS;

	for (TrackIterator iTrack = itBegin; iTrack != itEnd; iTrack++) {
		if ((*iTrack)->GetQuality() == kLITBAD) continue;
		if ((*iTrack)->GetParamLast()->GetQp() == 0.) {
			(*iTrack)->SetQuality(kLITBAD);
			continue;
		}
		Double_t mom = std::abs(1. / (*iTrack)->GetParamLast()->GetQp());
		if (mom < fMinMomentum) (*iTrack)->SetQuality(kLITBAD);		
	}	
	return kLITSUCCESS;
}

LitStatus CbmLitTrackSelectionMomentum::DoSelect(
		TrackVector& tracks)
{
	return DoSelect(tracks.begin(), tracks.end());
}

ClassImp(CbmLitTrackSelectionMomentum)