CbmFindPrimaryVertex.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 CbmFindPrimaryVertex header file              -----
// -----                  Created 28/11/05  by V. Friese               -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMFINDPRIMARYVERTEX_H
#define CBMFINDPRIMARYVERTEX_H 1
 
 
#include "FairTask.h"
 
class TClonesArray;
class CbmPrimaryVertexFinder;
class CbmVertex;
 
 
 
class CbmFindPrimaryVertex : public FairTask
{
 
 public:
 
  CbmFindPrimaryVertex();
 
 
  CbmFindPrimaryVertex(CbmPrimaryVertexFinder* pvFinder);
  
 
  CbmFindPrimaryVertex(const char* name, const char* title,
           CbmPrimaryVertexFinder* pvFinder);
 
 
  virtual ~CbmFindPrimaryVertex();
 
 
  virtual InitStatus Init();
 
 
  virtual void Exec(Option_t* opt);
 
 
  virtual void Finish();
 
  void SetTrackBranch(TString trBranch = "StsTrack") { fTrBranch = trBranch; } //AZ - set track branch name 
 
 private:
 
  CbmPrimaryVertexFinder* fFinder;
  TClonesArray*           fTracks;
  CbmVertex*              fPrimVert;
  TString                 fTrBranch;     //AZ - Track branch name
 
  CbmFindPrimaryVertex(const CbmFindPrimaryVertex&) = delete;
  CbmFindPrimaryVertex& operator=(const CbmFindPrimaryVertex&) = delete;
 
  ClassDef(CbmFindPrimaryVertex,1);
};
 
#endif