PndSttCellTrackFinder.cxx

Go to the documentation of this file.

/*
 * PndSttCellTrackFinder.cxx
 *
 *  Created on: May 20, 2014
 *      Author: schumann
 */

#include "PndSttCellTrackFinder.h"
#include "TString.h"
#include "PndSttHit.h"
#include "FairRootManager.h"
#include "PndSttSkewedHit.h"
#include "PndSttCellTrackletGenerator.h"
#include "TTimeStamp.h"

using namespace std;

ClassImp(PndSttCellTrackFinder);

void PndSttCellTrackFinder::AddHits(TClonesArray* hits, TString branchName) {

        //std::cout << "hits added!  branch name: " << branchName << "#################################################################"<<std::endl;
        //std::cout << "size of tconesarray: " << hits->GetEntriesFast() << "#################################################################"<<std::endl;
        fTrackFinderData->AddHits(hits, branchName);

}

void PndSttCellTrackFinder::FindTracks() {

        // init TrackFinderData
        vector<Double_t> timesGenerateNeighorhoodData;

        //if(fRunTimeBased){ // Bug?????????????? J.R. 01/06-2018
        fTrackFinderData->SetRunTimeBased(fRunTimeBased);
        //}
        fTrackFinderData->SetClusterTime(fClusterTime);
        //store timestamps for runtime analysis

        timesGenerateNeighorhoodData.push_back(TTimeStamp());
        fTrackFinderData->GenerateNeighborhoodData();
        timesGenerateNeighorhoodData.push_back(TTimeStamp());

        //fTrackFinderData->PrintInfo();


        fTrackletGenerator = new PndSttCellTrackletGenerator(fTrackFinderData);

        fTrackletGenerator->SetCalcFirstTrackletInf(fCalcFirstTrackletInf);
        fTrackletGenerator->SetVerbose(fVerbose);
        fTrackletGenerator->SetBz(fBz);

        //calculation on GPU is only possible without multiple STTHits and a maximum number of MAX_THREADS_PER_BLOCK
        if (fUseGPU && (fTrackFinderData->GetAllowDoubleHits() == kFALSE)
                        && (fTrackFinderData->GetNumHitsWithoutDouble()
                                        < MAX_THREADS_PER_BLOCK)) {
                fTrackletGenerator->SetUseGPU(fUseGPU);
                fTrackletGenerator->SetDevTubeNeighboringsPointer(
                                fDev_tubeNeighborings);
        }

        fTrackFinderData->SetClusterTime(fClusterTime);
        fTrackFinderData->SetRunTimeBased(fRunTimeBased);

        if (fCalcWithCorrectedIsochrones) {
                fHitCorrector = new PndSttHitCorrector(fTrackFinderData);
                fHitCorrector->CorrectHits();
                map<int, FairHit*> correctedIsochrones =
                                fHitCorrector->GetCorrectedHits();

                //Get tracks with corrected hits
                fTrackletGenerator->SetCorrectedHits(correctedIsochrones);
        }

        fTrackletGenerator->FindTracks();
        //fTrackletGenerator->PrintInfo();

        StoreTrackData();

        // cpu time measurement
        fTimeStampsTrackletGen.push_back(fTrackletGenerator->GetTimeStamps());
        fTimeStampsGenerateNeighborhoodData.push_back(timesGenerateNeighorhoodData);
}

;