/**
 * @file
 * @author Christian Simon <csimon@physi.uni-heidelberg.de>
 * @since 2018-08-02
 */

#ifndef CBMTOFFINDPOINTTYPE_H
#define CBMTOFFINDPOINTTYPE_H 1

#include "CbmTofPointCluster.h"

#include "FairTask.h"

#include <map>
#include <tuple>

class CbmMCDataArray;
class TClonesArray;
class CbmTofDigiTbPar;

/**
 * @brief ...
 * @author Christian Simon <csimon@physi.uni-heidelberg.de>
 * @since 2018-08-02
 * @version 1.0
 * @details ...
 */
class CbmTofFindPointType : public FairTask
{
  public:

    CbmTofFindPointType();

    virtual ~CbmTofFindPointType();

    virtual void Exec(Option_t *option);

//    virtual void FinishEvent();

    void SetAlternativeBranchNames(Bool_t bAlternative = kTRUE) {fbAlternativeBranchNames = bAlternative;}
    void SetFileIndex(Int_t iIndex) {fiFileIndex = iIndex;}
    void SetConsiderPointClusters(Bool_t bPointClusters = kTRUE) {fbConsiderPointClusters = bPointClusters;}
    void SetConsiderRefTracks(Bool_t bRefTracks = kTRUE) {fbConsiderRefTracks = bRefTracks;}

  protected:

    virtual InitStatus Init();

    virtual void SetParContainers();

//    virtual void Finish();

  private:

    CbmTofFindPointType(const CbmTofFindPointType&);

    CbmTofFindPointType& operator=(const CbmTofFindPointType&);

    CbmMCDataArray* fMCTracks;
    TClonesArray* fTofPointsTB;
    CbmTofDigiTbPar* fDigiTbParSet;
    std::map<Int_t, std::map<std::tuple<Int_t, Int_t, Int_t>, CbmTofPointCluster>> fPointClusterMap;
    std::map<std::tuple<Int_t, Int_t, Int_t>, Int_t> fReferencePointMap;
    Bool_t fbAlternativeBranchNames;
    Int_t fiFileIndex;
    Bool_t fbConsiderPointClusters;
    Bool_t fbConsiderRefTracks;

    ClassDef(CbmTofFindPointType, 0)
};

#endif