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

#ifndef CBMTOFRECOMCQA_H
#define CBMTOFRECOMCQA_H 1


#include "FairTask.h"

#include "TString.h"

#include <map>
#include <set>

class TClonesArray;
class FairFileSource;
class CbmTofDigiTbPar;
class TH1F;
class TH2F;
class TH3F;
class TEfficiency;
class TFile;

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

	public:

		CbmTofRecoMCQa();

		virtual ~CbmTofRecoMCQa();

    virtual void Exec(Option_t *option);

//    virtual void FinishEvent();

    void SetAverageSpills(Bool_t bAverage) {fbAverageSpills = bAverage;}
    void SetLocalRadius(Double_t dRadius) {fdLocalRadius = dRadius;}
    void SetSpillLength(Double_t dLength) {fdSpillLength = dLength;}
    void SetSpillBreakLength(Double_t dLength) {fdSpillBreakLength = dLength;}
    void SetDataTimeOffset(Double_t dOffset) {fdDataTimeOffset = dOffset;}
    void SetNSerialSpills(Int_t iNSpills) {fiNSerialSpills = iNSpills;}
    void SetTimeAxisBinWidth(Double_t dBinWidth) {fdTimeAxisBinWidth = dBinWidth;}
    void SetHistogramFileName(const char* cFileName) {fHistogramFileName = TString(cFileName);}
    void SetFillLocalHistos(Bool_t bFill) {fbFillLocalHistos = bFill;}
    void SetHitCompatibilityMode(Bool_t bCompatibility = kTRUE) {fbHitCompatibilityMode = bCompatibility;}
    void SetAlternativeBranchNames(Bool_t bAlternative = kTRUE) {fbAlternativeBranchNames = bAlternative;}
    void SetConsiderPointClusters(Bool_t bPointClusters = kTRUE) {fbConsiderPointClusters = bPointClusters;}
    void SetRealClustering(Bool_t bRealClustering = kTRUE) {fbRealClustering = bRealClustering;}
    void SetConsiderRefTracks(Bool_t bRefTracks = kTRUE) {fbConsiderRefTracks = bRefTracks;}
    void SetChiSquareLimit(Double_t dLimit) {fdChiSquareLimit = dLimit;}

  protected:

    virtual InitStatus Init();

    virtual void SetParContainers();

    virtual void Finish();

	private:

    CbmTofRecoMCQa(const CbmTofRecoMCQa&);

    CbmTofRecoMCQa& operator=(const CbmTofRecoMCQa&);

    void CreateHistograms();
    void WriteHistograms();

    void CreateSigmaHistograms();
    Double_t GetAcrossPositionResidualSigma(Int_t iCounterAddress, Double_t dPointTime);
    Double_t GetAlongPositionResidualSigma(Int_t iCounterAddress, Double_t dPointTime);
    Double_t GetTimeResidualSigma(Int_t iCounterAddress, Double_t dPointTime);
    Double_t GetAcrossPositionResidualSigmaSPH(Int_t iCounterAddress, Double_t dPointTime);
    Double_t GetAlongPositionResidualSigmaSPH(Int_t iCounterAddress, Double_t dPointTime);
    Double_t GetTimeResidualSigmaSPH(Int_t iCounterAddress, Double_t dPointTime);

    FairFileSource* fFileSource; //!
    TClonesArray* fTofPointsTB;
		TClonesArray* fTofDigis;
		TClonesArray* fTofHits;
		TClonesArray* fTofHitDigiMatches;
    TClonesArray* fTofHitPointMatches;
    TClonesArray* fTofPointMatches;
    CbmTofDigiTbPar* fDigiTbParSet;
    Int_t fiNTimeAxisBins;
    Double_t fdTimeAxisBinWidth;
    Double_t fdTimeAxisRange;
    std::map<Int_t, TH2F*> fClusterSizes;
    std::map<Int_t, TH2F*> fStripToTSpectra;
    std::map<Int_t, TH2F*> fStripPositionSpectra;
    std::map<Int_t, TH2F*> fPointPositionSpectra;
    std::map<Int_t, TH2F*> fHitPositionSpectra;
    std::map<Int_t, TH3F*> fStripTimeResidualWalk;
    std::map<Int_t, TH1F*> fHitAcrossPositionResidual;
    std::map<Int_t, TH1F*> fHitAlongPositionResidual;
    std::map<Int_t, TH1F*> fHitTimeResidual;
    std::map<Int_t, TH2F*> fCentralClusterSizeEvolution;
    std::map<Int_t, TH2F*> fCentralTimeResidualEvolution;
    std::map<Int_t, TH1F*> fHitChiSquare;
    std::map<Int_t, TH1F*> fCorrectedHitChiSquare;
    std::map<Int_t, TH2F*> fClusterSizeEvolution;
    std::map<Int_t, TH2F*> fAcrossPositionResidualEvolution;
    std::map<Int_t, TH2F*> fAlongPositionResidualEvolution;
    std::map<Int_t, TH2F*> fTimeResidualEvolution;
    std::map<Int_t, TH2F*> fCentralStripToTEvolution;
    std::map<Int_t, TEfficiency*> fIntegratedEfficiency;
    std::map<Int_t, TEfficiency*> fChiSquareEfficiency;
    std::map<Int_t, TEfficiency*> fCorrectedChiSquareEfficiency;
    std::map<Int_t, TH2F*> fClusterSizesSPH;
    std::map<Int_t, TH2F*> fStripToTSpectraSPH;
    std::map<Int_t, TH2F*> fStripPositionSpectraSPH;
    std::map<Int_t, TH2F*> fHitPositionSpectraSPH;
    std::map<Int_t, TH3F*> fStripTimeResidualWalkSPH;
    std::map<Int_t, TH1F*> fHitAcrossPositionResidualSPH;
    std::map<Int_t, TH1F*> fHitAlongPositionResidualSPH;
    std::map<Int_t, TH1F*> fHitTimeResidualSPH;
    std::map<Int_t, TH1F*> fHitChiSquareSPH;
    std::map<Int_t, TH1F*> fCorrectedHitChiSquareSPH;
    std::map<Int_t, TH2F*> fClusterSizeEvolutionSPH;
    std::map<Int_t, TH2F*> fAcrossPositionResidualEvolutionSPH;
    std::map<Int_t, TH2F*> fAlongPositionResidualEvolutionSPH;
    std::map<Int_t, TH2F*> fTimeResidualEvolutionSPH;
    std::map<Int_t, TH2F*> fCentralStripToTEvolutionSPH;
    std::map<Int_t, TH1F*> fDistanceToLastPoint;
    std::map<Int_t, TH1F*> fDistanceToLocalLastPoint;
    std::map<Int_t, TH1F*> fDistanceToLastHitUnbiased;
    std::map<Int_t, TH1F*> fDistanceToLastHitBiased;
    std::map<Int_t, TEfficiency*> fEfficiencyToLastHitUnbiased;
    std::map<Int_t, TEfficiency*> fEfficiencyToLastHitBiased;
    std::map<Int_t, TH2F*> fClusterSizeToLastHitUnbiased;
    std::map<Int_t, TH2F*> fClusterSizeToLastHitBiased;
    std::map<Int_t, TH2F*> fTimeResidualToLastHitUnbiased;
    std::map<Int_t, TH2F*> fTimeResidualToLastHitBiased;
    std::map<Int_t, TEfficiency*> fLocalSurfaceFlux;
    std::map<Int_t, TEfficiency*> fChannelDigiRates;
    std::map<Int_t, TH1F*> fNHitsPerPoint;
    std::map<Int_t, TH1F*> fNPointsPerHit;
    std::map<Int_t, TH1F*> fAcrossPositionResidualEvolutionSigma;
    std::map<Int_t, TH1F*> fAlongPositionResidualEvolutionSigma;
    std::map<Int_t, TH1F*> fTimeResidualEvolutionSigma;
    std::map<Int_t, TH1F*> fAcrossPositionResidualEvolutionSigmaSPH;
    std::map<Int_t, TH1F*> fAlongPositionResidualEvolutionSigmaSPH;
    std::map<Int_t, TH1F*> fTimeResidualEvolutionSigmaSPH;
    TFile* fHistogramFile;
    TString fHistogramFileName;
    Int_t fiTotalNPoints;
    Int_t fiTotalNMergedPoints;
    Bool_t fbAverageSpills;
    Double_t fdSpillLength;
    Double_t fdSpillBreakLength;
    Double_t fdDataTimeOffset;
    Double_t fdLocalRadius;
    Bool_t fbFillLocalHistos;
    Int_t fiNSerialSpills;
    std::map<Int_t, Double_t> fdPreviousPointTime;
    std::set<std::pair<Int_t, Int_t> > fPointLinkNbSet;
    Bool_t fbHitCompatibilityMode;
    Bool_t fbAlternativeBranchNames;
    Bool_t fbConsiderPointClusters;
    Bool_t fbRealClustering;
    Bool_t fbConsiderRefTracks;
    std::set<std::tuple<Int_t, Int_t, Int_t> > fProcessedHits;
    std::set<std::tuple<Int_t, Int_t, Int_t> > fProcessedDigis;
    Double_t fdChiSquareLimit;

		ClassDef(CbmTofRecoMCQa, 0);

};

#endif