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