PndSttFindTracks.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                   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 = NULL, 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