PndMatchBarrelForward.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#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