CbmEventVertexDca.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dario Ramirez [committer] */
 
#ifndef CBMEVENTVERTEXDCA_H
#define CBMEVENTVERTEXDCA_H
 
#include "CbmDcaVertexFinder.h"
#include "CbmEvent.h"
#include "CbmGlobalTrack.h"
#include "CbmMCTrack.h"
#include "CbmStsAnaBase.h"
#include "CbmStsCluster.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
#include "CbmStsUtils.h"
#include "CbmTrack.h"
 
#include <FairTask.h>
 
#include <TClonesArray.h>
#include <TVector3.h>
 
#include <cstring>
#include <map>
#include <unordered_map>
#include <unordered_set>

class CbmEventVertexDca : public FairTask, public CbmStsAnaBase {
 public:
  CbmEventVertexDca()  = default;
  ~CbmEventVertexDca() = default;
 
  InitStatus Init();
  void Exec(Option_t*);
  void FinishTask();
 
  void SetVertexFinder(std::shared_ptr<CbmDcaVertexFinder>);
 
 private:
  std::shared_ptr<CbmDcaVertexFinder> fVertexFinder;
 
  bool input_has_mc{false};
 
  std::vector<CbmGlobalTrack*> fGlbTrks;
 
  TClonesArray* fMCTrkArray{nullptr};
  TClonesArray* fCbmEvtArray{nullptr};
  TClonesArray* fGlbTrkArray{nullptr};
  TClonesArray* fStsTrkArray{nullptr};
  TClonesArray* fRchTrkArray{nullptr};
  TClonesArray* fMchTrkArray{nullptr};
  TClonesArray* fTrdTrkArray{nullptr};
  TClonesArray* fTofTrkArray{nullptr};
  TClonesArray* fStsHitArray{nullptr};
  TClonesArray* fStsCluArray{nullptr};
 
  void BookHistograms();
 
  void CheckVertex();

  void ProcessEvent(CbmEvent*);

  void ProcessGlobalTrack(CbmGlobalTrack*);
 
  ClassDef(CbmEventVertexDca, 1);
};
#endif