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_ */