// -------------------------------------------------------------------------
// -----                   R3BCaloHitFinder source file                -----
// -----                  Created 27/08/10  by H.Alvarez               -----
// -----                Last modification 06/12/11 by H.Alvarez        -----
// -----                                  15/09/11 by Enrico Fiori     -----
// -----                                  05/07/12 by P. Cabanelas     -----
// -------------------------------------------------------------------------

/**  R3BCaloHitFinder.h
 **/


#ifndef R3BCALOHITFINDER_H
#define R3BCALOHITFINDER_H

#include "FairTask.h"
#include "R3BCaloHit.h"
#include "R3BCaloHitSim.h"

class TClonesArray;

class R3BCaloHitFinder : public FairTask
{

  public:

    /** Default constructor **/
    R3BCaloHitFinder();

    /** Destructor **/
    ~R3BCaloHitFinder();

    /** Virtual method Exec **/
    virtual void Exec(Option_t* opt);

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

    /** Public method  GetAngles **/
    void GetAngles(Int_t iD, Double_t* polar, Double_t* azimuthal);

    /** Public method  GetAngles **/
    void GetAngles(Int_t iD, Double_t* polar, Double_t* azimuthal, Double_t* rho);

    /** Public method SelectGeometryVersion
     **
     ** Defines the geometry
     *@param version  Integer parameter used to select the geometry:
     **     0- CALIFA 5.0, including BARREL and ENDCAP.
     **     1- CALIFA 7.05, only BARREL
     ** 2- CALIFA 7.07, only BARREL
     ** 3- CALIFA 7.09, only BARREL 
     ** 4- CALIFA 7.17, only ENDCAP (in CsI[Tl])
     ** 5- CALIFA 7.07+7.17,
     ** 6- CALIFA 7.09+7.17,
     ** 10- CALIFA 8.11, only BARREL
     ** ...
     **/
    void SelectGeometryVersion(Int_t version);

    /** Public method SetExperimentalResolution
     **
     ** Defines the experimental resolution of the crystals. The parameter sets the value
     ** at 1 MeV, and it scales according to and 1/sqrt(E) law.
     *@param crystalRes  Double parameter used to set the experimental resolution @ 1 MeV
     **/
    void SetExperimentalResolution(Double_t crystalRes);

    /** Public method SetDetectionThreshold
     **
     ** Defines the minimum energy requested in a crystal to be considered in a calorimeter Hit
     *@param thresholdEne  Double parameter used to set the threshold
     **/
    void SetDetectionThreshold(Double_t thresholdEne);

    /** Public method SetAngularWindow
     **
     ** Sets the angular window open around the crystal with the largest energy
     ** to search for additional crystal hits and addback to the same cal hit
     **/
    void SetAngularWindow(Double_t deltaPolar, Double_t deltaAzimuthal, Double_t DeltaAngleClust=0.);

    /** Public method SetClusteringAlgorithm
     **
     ** Select the clustering algorithm to be used and set the parameter to be used by one of them
     //  1  ->  square window
     //  2  ->  round window
     //  3  ->  advanced round window with opening proportional to the
     //         energy of the hit, need ParCluster1
     //  4  ->  advanced round window with opening proportional to the
     //         energy of the two hit, need ParCluster1 NOT IMPLEMENTED YET!
     **
     **/
    void SetClusteringAlgorithm(Int_t ClusteringAlgorithmSelector, Double_t ParCluster1) ;





  protected:


    /** Virtual method Init **/
    virtual InitStatus Init();

    /** Virtual method ReInit **/
    virtual InitStatus ReInit();

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

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


    TClonesArray* fCrystalHitCA;
    TClonesArray* fCaloHitCA;

    // Selecting the geometry of the CALIFA calorimeter
    Int_t fGeometryVersion;
    // Minimum energy requested in a crystal to be considered in a calorimeter Hit
    Double_t fThreshold;
    // Experimental resolution @ 1 MeV
    Double_t fCrystalResolution;
    // Angular window (polar)
    Double_t fDeltaPolar;
    // Angular window (azimuthal)
    Double_t fDeltaAzimuthal;
    // Angular opening used for the clustering condition
    Double_t fDeltaAngleClust;
    // Clustering algorithm selector
    Int_t fClusteringAlgorithmSelector;
    // Clustering parameter 1
    Double_t fParCluster1;
    // Flag for simulation
    Bool_t kSimulation;

    // Parameter class
    //R3BCaloHitFinderPar* fCaloHitFinderPar;

  private:

    /** Private method ExpResSmearing
    **
    ** Smears the energy according to some Experimental Resolution distribution
     ** Very simple preliminary scheme where the Experimental Resolution
     ** is introduced as a gaus random distribution with a width given by the
     ** parameter fCrystalResolution(in % @ MeV). Scales according to 1/sqrt(E)
     **/
    Double_t ExpResSmearing(Double_t inputEnergy);


    /** Private method AddHit
    **
    ** Adds a CaloHit to the HitCollection
    **/
    R3BCaloHit* AddHit(UInt_t Nbcrystals,Double_t ene,Double_t pAngle,Double_t aAngle);

    /** Private method AddHitSim
    **
    ** Adds a CaloHitSim to the HitSimCollection
    **/
    R3BCaloHitSim* AddHitSim(UInt_t Nbcrystals,Double_t ene,Double_t pAngle,Double_t aAngle, Double_t einc);


    ClassDef(R3BCaloHitFinder,1);

};


#endif