#include "CbmLitTrackFinderBase.h"

#include "CbmLitHit.h"
#include "CbmLitStripHit.h"
#include "CbmLitPixelHit.h"
#include "CbmLitTrack.h"
#include "CbmLitComparators.h"
#include "CbmLitMath.h"
#include "CbmLitTrackSelection.h"
#include "CbmLitTrackPropagator.h"

#include <iostream>
#include <cmath>
#include <algorithm>

CbmLitTrackFinderBase::CbmLitTrackFinderBase():
	fVerbose(1),
	fEventNo(0)
{
}

CbmLitTrackFinderBase::~CbmLitTrackFinderBase()
{
}

void CbmLitTrackFinderBase::ArrangeHits(
		HitPtrIterator itBegin,
		HitPtrIterator itEnd)
{
    for(HitPtrIterator it = itBegin; it != itEnd; it++) {
    	CbmLitHit* hit = *it;
    	if (fUsedHitsSet.find(hit->GetRefId()) != fUsedHitsSet.end()) continue;
     	fHitData.AddHit(hit->GetPlaneId(), hit);
    }

    if (fVerbose > 1) std::cout << fHitData.ToString();

    for (int i = 0; i < fLayout.GetNofStationGroups(); i++){
    	for (int j = 0; j < fLayout.GetNofStations(i); j++){
    		CbmLitStation station = fLayout.GetStation(i, j);
    		if (station.GetType() == kLITPIXELHIT) {
    			for (int k = 0; k < fLayout.GetNofSubstations(i, j); k++){
    				HitPtrIteratorPair hits = fHitData.GetHits(i, j, k);
    				std::sort(hits.first, hits.second, CompareHitPtrXULess());
//    				std::cout << "station group " << i << " station " << j
//    					<< " substation " << k << std::endl;
//    				for(HitPtrIterator it = hits.first; it != hits.second; it++)
//    					std::cout << (*it)->ToString();
    			}
    		} else {
//    			for (int k = 0; k < fLayout.GetNofSubstations(i, j); k++){
//					HitPtrIteratorPair hits = fHitData.GetHits(i, j, k);
////					std::sort(hits.first, hits.second, CompareHitPtrXULess());
//					std::cout << "station group " << i << " station " << j
//						<< " substation " << k << std::endl;
//					for(HitPtrIterator it = hits.first; it != hits.second; it++)
//						std::cout << (*it)->ToString();
//				}
    		}
    	}
    }
}

void CbmLitTrackFinderBase::RemoveHits(
		TrackPtrIterator itBegin,
		TrackPtrIterator itEnd)
{
	for(TrackPtrIterator it = itBegin; it != itEnd; it++) {
		CbmLitTrack* track = *it;
		if(track->GetQuality() == kLITBAD) continue;
		for (int hit = 0; hit < track->GetNofHits(); hit++)
			fUsedHitsSet.insert(track->GetHit(hit)->GetRefId());
	}
}

void CbmLitTrackFinderBase::CopyToOutput(
		TrackPtrIterator itBegin,
		TrackPtrIterator itEnd,
		TrackPtrVector& tracks)
{
	for(TrackPtrIterator it = itBegin; it != itEnd; it++) {
		CbmLitTrack* track = *it;
		if(track->GetQuality() == kLITBAD) continue;
		if (!track->CheckParams()) continue;
		fUsedSeedsSet.insert(track->GetPreviousTrackId());
		tracks.push_back(new CbmLitTrack(*track));
	}
}