// -------------------------------------------------------------------------
// -----                    CbmMvdFindHits header file                 -----
// -----                   Created 08/11/06  by C.Dritsa               -----
// -------------------------------------------------------------------------



#ifndef CBMMVDFINDHITS_H
#define CBMMVDFINDHITS_H 1

#include "FairTask.h"
#include "CbmMvdPoint.h"
#include "CbmMvdDigi.h"

#include "TArrayS.h"
#include "TObjArray.h"
#include "TRefArray.h"
#include "TCanvas.h"


#include "TRandom3.h"
#include "TStopwatch.h"
#include "TString.h"
#include "TMath.h"
#include "TH1.h"
#include "TH2.h"
#include "TH1F.h"
#include <vector>
#include <list>
#include <map>
#include <utility>

class TClonesArray;
class TRandom3;
class CbmMvdGeoPar;
class CbmMvdPileupManager;
class CbmMvdStation;

class CbmMvdFindHits : public FairTask
{

public:

    /** Default constructor **/
    CbmMvdFindHits();


    /** Standard constructor
     *@param name  Task name
     *@param mode  0 = no fake digis generation, 1 = generate fake digis
     **/
    CbmMvdFindHits(const char* name,
		   Int_t mode = 0, Int_t iVerbose = 1);



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


    /** Task execution **/
    virtual void Exec(Option_t* opt);


    /** Accessors **/
    //Double_t GetSigmaX()        const { return fSigmaX;     };

    /** Modifiers **/
    void SetSigmaNoise(Double_t sigmaNoise, Bool_t addNoise)  { fSigmaNoise = sigmaNoise; fAddNoise=addNoise; }
    void SetSeedThreshold(Double_t seedCharge)       { fSeedThreshold  = seedCharge; }
    void SetNeighbourThreshold(Double_t neighCharge) { fNeighThreshold = neighCharge;}
    /**

        proba(%) to have 1 fake hit  | Number of sigmas (of noise distr)
       ------------------------------|-----------------------------------
                            31.7     |   1
                            4.5      |   2
                            0.27     |   3
                            6.3e-3   |   4
                            5.7e-5   |   5
			    2.0e-7   |   6

    */


    void SetAdcDynamic( Int_t adcDynamic ) { fAdcDynamic = adcDynamic; };
    void SetAdcOffset(Int_t adcOffset )    { fAdcOffset  = adcOffset;  };
    void SetAdcBits(Int_t adcBits)         { fAdcBits    = adcBits;    };

    /**Detector Spatial resolution.
    Correlated with number of adc bits*/
    void SetHitPosErrX( Double_t errorX ) { errorX = fHitPosErrX; }
    void SetHitPosErrY( Double_t errorY ) { errorY = fHitPosErrY; }
    void SetHitPosErrZ( Double_t errorZ ) { errorZ = fHitPosErrZ; }
    void ShowDebugHistograms(){fShowDebugHistos=kTRUE;}


    //protected:
public:

    // ----------   Protected data members  ------------------------------------

    Int_t fAdcDynamic;
    Int_t fAdcOffset;
    Int_t fAdcBits;

    TClonesArray* fDigis;
    TClonesArray* fHits;
    TClonesArray* fClusters;
    TClonesArray* fMatches;

    TObjArray* fPixelChargeHistos;
    TObjArray* fTotalChargeInNpixelsArray;
    
    // Debug Histograms
    TH1F* fResolutionHistoX;
    TH1F* fResolutionHistoY;
    TH1F* fResolutionHistoCleanX;
    TH1F* fResolutionHistoCleanY;
    TH1F* fResolutionHistoMergedX;
    TH1F* fResolutionHistoMergedY;
    TH2F* fBadHitHisto;

    map<pair<Int_t, Int_t>, Int_t> fDigiMap;
    map<pair<Int_t, Int_t>, Int_t>::iterator fDigiMapIt;
    
    
    TH2F* h;
    TH2F* h3;
    TH1F* h1;
    TH1F* h2;
    TH1F* Qseed;

    TCanvas* c1;

private:

    Int_t fNEvent;
    Int_t fMode;
    Int_t fCounter;
    Double_t fSigmaNoise;
    Double_t fSeedThreshold;
    Double_t fNeighThreshold;
    Bool_t fShowDebugHistos;

    Double_t fLayerRadius;
    Double_t fLayerRadiusInner;
    Double_t fLayerPosZ;
    Double_t fHitPosX;
    Double_t fHitPosY;
    Double_t fHitPosZ;
    Double_t fHitPosErrX;
    Double_t fHitPosErrY;
    Double_t fHitPosErrZ;

    TString  fBranchName;

    Bool_t fAddNoise;

    /** Map of MC Volume Id to MvdStation **/
    std::map<Int_t, CbmMvdStation*> fStationMap;                  //!




    // -----   Private methods   ---------------------------------------------

    /** Register the output arrays to the IOManager **/
    void Register();

    /** Clear the arrays **/
    void Reset();

    /** Virtual method Finish **/
    virtual void Finish();

    /** Intialisation **/
    virtual InitStatus Init();


    /** Reinitialisation **/
    virtual InitStatus ReInit();


    /** Get MVD geometry parameters from database
     **@value Number of MVD stations
     **/
    Int_t GetMvdGeometry();



    void SetMvdGeometry(Int_t detId);
    void AddNoiseToDigis(CbmMvdDigi* digi);
    void GenerateFakeDigis(CbmMvdStation* station, Double_t pixelSizeX, Double_t pixelSizeY);
    void CheckForNeighbours(vector<CbmMvdDigi*>* clusterArray, Int_t clusterDigi, TArrayS* pixelUsed);
    void CreateHit(vector<CbmMvdDigi*>* clusterArray, CbmMvdStation* station , TVector3& pos, TVector3 &dpos);



    ClassDef(CbmMvdFindHits,1);

};


#endif