// -------------------------------------------------------------------------
// -----                  CbmLitTrdTrackFinderSts header file               -----
// -----                  Created 06/07/06  by A. Lebedev               -----
// -------------------------------------------------------------------------


/** CbmLitTrdTrackFinderSts.h
 *@author A.Lebedev <alebedev@jinr.ru>
 **
 **  STS -> TRD track-finder
 **/  

#ifndef CbmLitTrdTrackFinderStsHHH
#define CbmLitTrdTrackFinderStsHHH


#include "CbmLitTrdTrackFinder.h"

#include <set>



class TClonesArray;
class CbmTrdTrack;
class CbmTrdHit;




class CbmLitTrdTrackFinderSts : public CbmLitTrdTrackFinder {

public:

    // Default constructor
    CbmLitTrdTrackFinderSts();


    // Destructor
    ~CbmLitTrdTrackFinderSts();

    // Initialisation
    void Init();

    // Track finding algorithm
    Int_t DoFind(TClonesArray* hitArray,
                 TClonesArray* trackArray);

private:

    // Data
    TClonesArray* fArrayStsTrack;

    //vector with TRD tracks, which were created from STS tracks
    std::vector<CbmTrdTrack*> fTrdTracksS;

    // Set with STS track indexes
    std::set<Int_t> fStsIdSet;
    
    // Creates Trd tracks from Sts tracks
    void CreateTrdFromSts();

    // extrapolate STS tracks to first TRD layer
    void ExtrapolateToTrd();

    // Initiate tracks
    void InitTracks();

    // Copy to output TClonesArray
    void CopyToOutput(TClonesArray* trackArray);
    
    // Set iteration parameters
    void SetIterPar(Double_t SigmaCoef, Int_t beginStation, 
                    Int_t endStation, Int_t maxNofMissingHitsInStation);

public:


    ClassDef(CbmLitTrdTrackFinderSts, 1);

};


#endif