PndTrackingQualityBarrelTaskNewLinks.h

Go to the documentation of this file.

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


#ifndef PndTrackingQualityBarrelTaskNewLinks_H
#define PndTrackingQualityBarrelTaskNewLinks_H


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


#include <vector>
#include <map>

class TClonesArray;
class PndTrack;

class PndTrackingQualityBarrelTaskNewLinks : public FairTask
{
 public:

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

  virtual ~PndTrackingQualityBarrelTaskNewLinks();


  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; }

 private:
//
//  virtual void FillMapTrackQualifikation();


  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 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;

  RhoTuple * fTuple;

  TH1* fPHisto;
  TH1* fPRelHisto;
  TH1* fPtHisto;
  TH1* fPtRelHisto;
  TH1* fQualyHisto;
  THStack * fQualyStack;
  TH1 * fQualyHisto_mc;
  TH1 * fQualyHisto_neg;
  TH1 * fQualyHisto_pos;
  TH1 * fQualyHisto_all;

  Int_t fEventNr;

  void Register();

  void Reset();


  ClassDef(PndTrackingQualityBarrelTaskNewLinks,1);

};

#endif