CbmStsFindTracks.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                   CbmStsFindTracks header file                -----
// -----                  Created 02/02/05  by V. Friese               -----
// -------------------------------------------------------------------------
 
/** CbmStsFindTracks
 *@author V.Friese <v.friese@gsi.de>
 **
 ** Task class for track finding in the STS.
 ** Input: TClonesArray of CbmStsHit
 ** Output: TClonesArray of CbmStsTrack
 **
 ** Uses as track finding algorithm classes derived from CbmStsTrackFinder.
 **/
 
#ifndef CBMSTSFINDTRACKS
#define CBMSTSFINDTRACKS 1
 
#include "BmnTask.h"
#include "CbmGeoStsPar.h"
#include "CbmStsDigiPar.h"
#include "CbmStsDigiScheme.h"
#include "CbmStsTrackFinder.h"
#include "CbmStsTrackFinderIdeal.h"
#include "FairField.h"
#include "TClonesArray.h"
#include "TStopwatch.h"
 
class CbmStsFindTracks : public BmnTask
{
  public:
    /** Default constructor **/
    CbmStsFindTracks();
    /** Standard constructor
     *@param iVerbose     Verbosity level
     *@param sFinderName  Name of STS track finder concrete class
     *@param useMvd       Include MVD hits in track finding
     *@param name         Task name
     **/
    CbmStsFindTracks(Int_t iVerbose,
                     const TString& sFinderName,
                     Bool_t useMvd = kFALSE,
                     const char* name = "STSFindTracks");
    /** Destructor **/
    virtual ~CbmStsFindTracks();
 
    /** Task execution **/
    virtual void Exec(Option_t* opt);
 
    /** Accessors **/
    CbmStsTrackFinder* GetFinder() { return fFinder; };
 
    /** Return if Mvd is used or not **/
    Bool_t MvdUsage() const { return fUseMvd; }
 
    /** Set concrete track finder **/
    void UseFinder(CbmStsTrackFinder* finder)
    {
        if (fFinder)
            delete fFinder;
        fFinder = finder;
    };
 
    virtual void OnlineWrite(const std::unique_ptr<TTree>& resultTree);
 
  private:
    Bool_t fUseMvd;                  // Inclusion of MVD hits
    CbmGeoStsPar* fGeoPar;           //! STS geometry parameters
    CbmStsDigiPar* fDigiPar;         //! STS digitisation parameters
    CbmStsDigiScheme* fDigiScheme;   //! STS digitisation scheme
    FairField* fField;               //! Magnetic field
    CbmStsTrackFinder* fFinder;      //! TrackFinder concrete class
    TClonesArray* fMvdHits;          //! Input array of MVD hits
    TClonesArray* fStsHits;          //! Input array of STS hits
    TClonesArray* fTracks;           //! Output array of CbmStsTracks
    TStopwatch fTimer;               // Timer
    Int_t fNEvents;                  /** Number of events with success **/
    Int_t fNEventsFailed;            /** Number of events with failure **/
    Double_t fTime;                  /** Total real time used for good events **/
    Double_t fNTracks;               /** Number of tracks created **/
    TString finderName;
 
    /** Get parameter containers **/
    virtual void SetParContainers();
 
    void InitFinder();
    /** Initialisation at beginning of each event **/
    virtual InitStatus Init();
 
    /** Finish at the end of each event **/
    virtual void Finish();
 
    void BuildDigitisationScheme();
 
    CbmStsFindTracks(const CbmStsFindTracks&);
    CbmStsFindTracks operator=(const CbmStsFindTracks&);
 
    ClassDef(CbmStsFindTracks, 1);
};
 
#endif