PndTrackingQATask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 PndTrackingQATask header file             -----
// -----                  Created 18/01/10  by T.Stockmanns             -----
// -------------------------------------------------------------------------


#ifndef PndTrackingQATask_H
#define PndTrackingQATask_H


// framework includes
#include "FairTask.h"
#include "PndDetectorList.h"
#include "PndTrack.h"
#include "PndTrackCand.h"
#include "PndTrackingQualityMCInfo.h"
#include "PndGeoSttPar.h"
#include "TH2.h"
#include "THStack.h"
#include "RhoHistogram/RhoTuple.h"
#include "PndTrackFunctor.h"


#include <vector>
#include <map>

class TClonesArray;

class PndTrackingQATask : public FairTask
{
 public:

        PndTrackingQATask(TString trackBranchName, TString idealBranchName, Bool_t pndTrackData = kTRUE);

  virtual ~PndTrackingQATask();


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



  virtual void Exec(Option_t* opt);

  virtual void Finish();

  void SetIdealTrackBranchName(TString name){fIdealTrackBranchName = name;};

  void SetTrackBranchName(TString name, Bool_t pndTrackData = kTRUE) {fPndTrackOrTrackCand = pndTrackData; fTrackBranchName = name;} //alternatively PndTrackCandData can be analysed. Therefore pndTrackData has to be set to kFALSE

  void AddHitsBranchName(TString name){ fBranchNames.push_back(name);}

  void SetOutputMCInfoBranchName(TString name) { fMCInfoBranchName = name; }
  void SetOutputRecoInfoBranchName(TString name) { fRecoInfoBranchName = name; }

  void SetFunctorName(TString name){ fPossibleTrackFunctorName = name; }
  void SetFunctor();

 private:
//
//  virtual void FillMapTrackQualifikation();
  const TString fHypoName[5]={"Electron","Muon","Pion","Kaon","Proton"};  


  virtual void FillQualyHisto(std::map<Int_t, Int_t> trackQualifikation, Int_t nGhosts);
  virtual void FillMCStatus(std::map<Int_t, Int_t> trackMCStatus);
  virtual void FillEfficiencies(std::map<Int_t, std::map<TString, std::pair<Double_t, Int_t > > > efficiencies);
  virtual void MapToHist(std::map<Int_t, Double_t>, TH1*);

  virtual void SetQualyHisto(TH1* histo, Bool_t relative, Int_t base=1);

  virtual Int_t GetSumOfAllValidMCHits(FairMultiLinkedData* trackData);


  void AssociateRecoTracksToMCTracks();


  PndTrackingQualityMCInfo GetMCInfoFromIdealTrack(PndTrack *idealtrack);
  Int_t GetMCInfoIdFromIdealTrackId(int idealtrackid) { return fMCInfoIdIdealId[idealtrackid]; }

  void InitializeHistograms();
  void LabelQualyHistogram(TH1 *);
  void ColorHistogram();
  std::vector<TString> fBranchNames;                     
  TString fMCInfoBranchName, fRecoInfoBranchName;        

  std::map<TString, FairMultiLinkedData> fMapLinkData;   
//  std::map<TString, Double_t> fPossibleTrackParameter;
  std::map<TString, TH2*> fMapEfficiencies;              

  std::map< int, int > fMCInfoIdIdealId; 

  Int_t fNGhosts;

  TClonesArray* fTrack;
  TClonesArray* fMCTrack;
  TClonesArray* fTrackCand;
  //  TClonesArray* fIdealTrackCand;
  TClonesArray *fSttHitArray;
  TClonesArray* fMCTrackInfo;
  TClonesArray* fRecoTrackInfo;
  TClonesArray* fIdealTrack;
  TClonesArray* fSttTubeArray;

  FairRootManager* ioman;

  std::map<Int_t, Int_t> fMapTrackQualifikation; 

  TString fTrackBranchName;
  TString fIdealTrackBranchName;
  Bool_t fPndTrackOrTrackCand; //kTRUE if track and kFALSE if track cand
  PndGeoSttPar *fSttParameters;

  TString fPossibleTrackFunctorName;

  PndTrackFunctor* fPossibleTrackFunctor;


  RhoTuple * fTuple;
  TH1* fIdealTracksPerEvent;
  TH1* fIdealPHisto;
  TH1* fIdealPtHisto;
  TH1* fIdealPlHisto;
  TH1* fIdealPrimTracksPerEvent;
  TH1* fIdealSecTracksPerEvent;
  TH1* fPHisto;
  TH1* fPRelHisto;
  TH1* fPtHisto;
  TH1* fPtRelHisto;
  TH1* fPlHisto;
  TH1* fPlRelHisto;
  TH1* fQualyHisto;
  THStack * fQualyStack;
  TH1 * fQualyHisto_mc;
  TH1 * fQualyHisto_neg;
  TH1 * fQualyHisto_pos;
  TH1 * fQualyHisto_all;
  TH1* fQualyHisto_rel_all;
  TH1* fQualyHisto_rel_possible;


  Int_t fEventNr;

  void Register();

  void Reset();


  ClassDef(PndTrackingQATask,1);

};

#endif