PndMatchBarrelForward.h

Go to the documentation of this file.

#ifndef MATCH_BARREL_FORWARD_H
#define MATCH_BARREL_FORWARD_H

#include "FairRootManager.h"
#include "FairLogger.h"

#include "TString.h"
#include "TClonesArray.h"

#include "PndPersistencyTask.h"
#include "PndTrack.h"

class PndMatchBarrelForward : public PndPersistencyTask {
public:
  virtual void Exec(Option_t *option);
  virtual InitStatus Init();

  PndMatchBarrelForward(){};                                               
  virtual ~PndMatchBarrelForward(){};

  void SetBarrelTrackBranch(TString branch) { fTrackBranchBarrel = branch; };
  void SetForwardTrackBranch(TString branch) { fTrackBranchForward = branch; };

  Bool_t DoTracksMatch(PndTrack *BarrelTrack, PndTrack *ForwardTrack);

  void CombineTracks(PndTrack *BarrelTrack, PndTrack *ForwardTrack);

  void AddTrack(PndTrack *Track, PndTrack::TrackType type);

  Bool_t FetchTracks();

  void MatchTracks();

private:
  Int_t fTrackCounter = 0;
  Int_t fEvent = 0;

  TString fTrackBranchBarrel = "";             
  TString fTrackBranchForward = ""; 

  TClonesArray *fTracksCombined = nullptr;
  TClonesArray *fTracksBarrel = nullptr;
  TClonesArray *fTracksForward = nullptr;

  FairRootManager *fManager = FairRootManager::Instance();

  ClassDef(PndMatchBarrelForward, 1);
};

#endif