PndSttFindTracks.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".              *
//****************************************************************************

// -------------------------------------------------------------------------
// -----                   PndSttFindTracks header file                -----
// -----                  Created 29/03/06  by V. Friese               -----
// -------------------------------------------------------------------------

#ifndef PNDSTTFINDTRACKS
#define PNDSTTFINDTRACKS 1

#include <PndPersistencyTask.h>
#include "PndGeoSttPar.h"

#include <string>
#include <vector>

class PndSttTrackFinder;
class TClonesArray;

class PndSttFindTracks : public PndPersistencyTask {

 public:
  PndSttFindTracks();

  PndSttFindTracks(PndSttTrackFinder *finder, Int_t verbose = 1);

  PndSttFindTracks(const char *name, const char *title = "PndPersistencyTask", PndSttTrackFinder *finder = nullptr, Int_t verbose = 1);

  virtual ~PndSttFindTracks();

  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  virtual void Finish();

  void SetParContainers();

  PndSttTrackFinder *GetFinder() { return fFinder; };
  Int_t GetNofTracks() { return fNofTracks; };

  void UseFinder(PndSttTrackFinder *finder) { fFinder = finder; };

  void AddHitCollectionName(char *hitCollectionName, char *pointCollectionName);

  void SetPersistence(Bool_t persistence) { SetPersistency(persistence); }

  void SetHelixHitProduction(Bool_t hhprod = kTRUE) { fHelixHitProduction = hhprod; }

 private:
  void AddAllCollections();
  void AddHitCollection(char const *collectionName, char const *pointCollectionName);

  PndSttTrackFinder *fFinder;    // Pointer to TrackFinder concrete class
  TClonesArray *fTrackCandArray; // Output array of PndTrackCand
  TClonesArray *fTrackArray;     // Output array of PndTrack

  TClonesArray *fHelixHitArray; // Output array of PndSttHelixHit

  Int_t fNofTracks; // Number of tracks created
  Int_t fVerbose;   // Verbosity level

  std::vector<std::string> fHitCollectionNames;
  std::vector<std::string> fPointCollectionNames;

  Bool_t fCollectionsComplete;

  Bool_t fHelixHitProduction; 

  Int_t fEventCounter; 
 //  CHECK added
  TClonesArray *fTubeArray;

  PndGeoSttPar *fSttParameters; //  CHECK added

  ClassDef(PndSttFindTracks, 1);
};

#endif