PndSttCellTrackFinderAnalysisTask.h

Go to the documentation of this file.

/*
 * PndSttCellTrackFinderAnalysisTask.h
 *
 *  Created on: Jun 26, 2013
 *      Author: schumann
 */

#ifndef PndSttCellTrackFinderAnalysisTask_H_
#define PndSttCellTrackFinderAnalysisTask_H_

#include "FairTask.h"
#include "TH1I.h"
#include "TCanvas.h"
#include "FairMultiLinkedData.h"

#include <map>
#include <vector>
#include <fstream>

class TClonesArray;
class PndMCMatch;
class PndGeoSttPar;
class PndMCResult;

class PndSttCellTrackFinderAnalysisTask: public FairTask {
public:
        PndSttCellTrackFinderAnalysisTask(): FairTask("Stt Cell TrackFinder Analysis Task"){ };
        virtual ~PndSttCellTrackFinderAnalysisTask(){};

        virtual void SetParContainers();
        virtual InitStatus Init();

        virtual void Exec(Option_t* opt);
        virtual void FinishEvent();
        virtual void Finish();

    void SetVerbose(Int_t verbose){ fVerbose = verbose;};

private:

    /*Method for testing functionality of Conformal Mapping (Circle Fit)*/
    void CheckConformalMap();

    /*Method verifies the result of the first step of the trackfinding-algorithm*/
    void CheckFirstTracklets();

    /*Method verifies the result of the second step of the trackfinding-algorithm*/
    void CheckTrackletCombinations();

    /*Method for drawing STTHits + Circles of Riemann Fit for the first step of trackfinding*/
    void DrawFirstRiemannPlots(int eventNumber);

    /*Method for drawing STTHits + Circles of Riemann Fit for the second step of trackfinding*/
    void DrawCombiRiemannPlots(int eventNumber);

    int GetNumLinksOfHits(FairMultiLinkedData& hitLinks, PndMCResult& sttHitsToMCTrack, PndMCResult& sttHitToPoint);

    void TestRecoQualityFirstStep();

    void TestRecoQualityCombi();

    void CountMaxHitsFirstStep();

    void CountMaxHitsCombi();

    std::map<int, bool> fFoundMC;

        PndMCMatch* fMCMatch;

        TClonesArray* fMCTrack;

        TClonesArray* fFirstTrackCand;
        TClonesArray* fFirstRiemannTrack;
        TClonesArray* fCombiTrackCand;
        TClonesArray* fCombiRiemannTrack;

        TClonesArray* fEventHeader;
        TClonesArray* fSTTHits;

        std::map<int, std::vector<int> > fMapTubeIDToHits;              // map<tube-id, vector<indices of SttHits in fSTTHits>>
        std::map<int, int> fMapHitIndexToTubeID;                                // map<index of SttHit in fSTTHits, tube-id>

        PndGeoSttPar *fSttParameters;                   // for filling fTubeArray
        TClonesArray *fTubeArray;                               // contains at index tube-id corresponding PndSttTube

        // histograms for first step of trackfinding
        TH1I* fHistoNumberOfLinks1;
        TH1I* fHistoNumberOfTracklets1;
        TH1I* fHistoNumberOfAssignedHits1;
        TH1I* fHistoNumberOfHits1;
        TH1I* fHistoNumberOfErrors1;
        TH1I* fHistoMaxHitsOfMCTrack1;
        TH1I* fHistoNumberOfMissingHits1;

        TH1I* fHistoQualityFirstStep;

        // histograms for second step of trackfinding
        TH1I* fHistoNumberOfLinks2;
        TH1I* fHistoNumberOfTracklets2;
        TH1I* fHistoNumberOfHits2;
        TH1I* fHistoNumberOfErrors2;
        TH1I* fHistoMaxHitsOfMCTrack2;
        TH1I* fHistoNumberOfMissingHits2;

        TH1I* fHistoQualityCombi;
        TH1I* fHistoNumberOfHitsMCTrack;

        TCanvas* fCanvas;

        ClassDef(PndSttCellTrackFinderAnalysisTask,1);


};

#endif /* PndSttCellTrackFinderAnalysisTask_H_ */