// -------------------------------------------------------------------------
// -----              PndTrackCandInfoTask header file                 -----
// -------------------------------------------------------------------------

/** PndTrackCandInfoTask.h
 * @author A. Goerres <a.goerres@fz-juelich.de>
 * @date February 2015
 * @brief A task to count the number of hits in tracking detectors.
 */


#ifndef PndTrackCandInfoTask_H_
#define PndTrackCandInfoTask_H_

#include "TClonesArray.h"

#include "FairTask.h"
#include "PndTrackCandInfo.h"

class PndTrackCandInfoTask : public FairTask
{

public:

    /** Default constructor */
    PndTrackCandInfoTask();

    /** Destructor */
    virtual ~PndTrackCandInfoTask();


    /** Initialize the task */
    InitStatus Init();

    /** Execute the task */
    void Exec(Option_t * opt);

    /** Finish the task */
    void Finish();

    /** Enable storing to file */
    void SetPersistence(Bool_t pers = kTRUE) { fPersistence = pers; };

    /** Accessor functions to fTrackInputBranch */
    void SetTrackInputBranch(TString name) { fTrackInputBranch = name; };
    TString GetTrackInputBranch() { return fTrackInputBranch; };

    /** Accessor functions to fTrackInputBranchFts */
    void SetTrackOutputBranch(TString name) { fTrackInfoOutputBranch = name; };
    TString GetTrackOutputBranch() { return fTrackInfoOutputBranch; };


private:

    /** Branch names */
    TString fTrackInputBranch;
    TString fMvdPixelHitBranch;
    TString fMvdStripHitBranch;
    TString fSttHitBranch;
    TString fGemHitBranch;
    TString fFtsHitBranch;
    TString fTrackInfoOutputBranch;

    /** Array of tracks from ideal track finder */
    TClonesArray * fInputTrack;
    TClonesArray * fMCTrack;
    TClonesArray * fSelectedTrack;

    /** TClonesArrays of hits */
    TClonesArray * fMvdPixelHitArray;
    TClonesArray * fMvdStripHitArray;
    TClonesArray * fSttHitArray;
    TClonesArray * fGemHitArray;
    TClonesArray * fFtsHitArray;

    /** branch IDs for hits */
    Int_t fHitIDMvdPixel, fHitIDMvdStrip, fHitIDStt, fHitIDGem, fHitIDFts;

    /** Store output into file */
    Bool_t fPersistence;

    /** Hit count data container */
    TClonesArray * fHitCountsArray;


    ClassDef(PndTrackCandInfoTask,1);
};


#endif /* PndTrackCandInfoTask_H_ */