// -------------------------------------------------------------------------
// -----                 PndSdsStripClusterTask header file             -----
// -------------------------------------------------------------------------



#ifndef PNDSDSSTRIPCLUSTERTASK_H
#define PNDSDSSTRIPCLUSTERTASK_H

#include "FairTask.h"
//#include "PndSdsGeoPar.h"
#include "PndSdsStripDigiPar.h"
#include "PndSdsHit.h"
#include "PndSdsMCPoint.h"
// #include "PndSdsCluster.h"
#include "FairGeoVector.h"
#include "FairGeoTransform.h"
#include "TVector3.h"
#include "TRandom.h"
#include "TGeoMatrix.h"
#include "TGeoBBox.h"
#include "TH2F.h"
#include "PndSdsHybridHitProducer.h"
#include "PndSdsStripHitProducer.h"
#include "PndSdsStripClusterer.h"
#include "PndDetectorList.h" 
#include <string>
#include <vector>

class TClonesArray;
class PndSdsChargeWeightingAlgorithms;

class PndSdsStripClusterTask : public FairTask
{
 public:

    /** Default constructor **/

        PndSdsStripClusterTask(); 

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

   /** pure virtual method SetBranchNames
   **
   ** called by Init()
   ** function to set individual branch names
   **/
   virtual void SetBranchNames()=0;


    /** Virtual method Init **/
    virtual void SetParContainers()=0;
    virtual InitStatus Init();
    virtual InitStatus ReInit();

    /** Virtual method Exec **/
    virtual void Exec(Option_t* opt);
    /** Virtual method Finish **/
    virtual void Finish();
 
    void SetPersistance(Bool_t p = kTRUE) {fPersistance=p;};
    Bool_t GetPersistance() {return fPersistance;};
  
  protected:
  
    Bool_t fPersistance; // switch to turn on/off storing the arrays to a file
  
    void SetCalculators(); 
    TVector2 CalcLineCross(TVector2 point1, TVector2 dir1, TVector2 point2, TVector2 dir2) const;
    void SetCurrentCalculators(PndSdsStripDigiPar* digipar); 
    Bool_t SelectSensorParams(TString detname);
    void CalcMeanCharge(std::vector<Int_t> &onecluster, Double_t &meanstrip, Double_t &meanerr, Double_t &charge);
    Bool_t Backmap( TVector2 meantopPoint, Double_t toperr , TVector2 meanbotPoint, Double_t boterr,
    	         	TVector3 &hitpos, TVector3 &hiterr, TString &detname);
    TClonesArray* fDigiArray;  // Input array of PndSdsDigis
    TClonesArray* fClusterArray; // Output array of PndSdsClusters
    TClonesArray* fHitArray;  // Output array of PndSdsHits

    TString fBranchName;
    TString fClustBranchName;
    TString fHitBranchName;
    TString fFolderName;

    void Register();
    void Reset();
    void ProduceHits();
    void ResetClusterFinders(); 
    
    Int_t fFEcolumns;
    Int_t fFErows;
    Double_t fChargeCut;
    Int_t fMeanAlgo;
    Int_t fRadChannel;
    Int_t fRadTime;

    //! Digitization Parameters 
    TList* fDigiParameterList; 
    PndSdsStripDigiPar* fCurrentDigiPar; 
    //! Calculator objects 
    std::map<const char*,PndSdsCalcStrip*> fStripCalcTop; 
    std::map<const char*,PndSdsCalcStrip*> fStripCalcBot; 
    PndSdsCalcStrip* fCurrentStripCalcTop; 
    PndSdsCalcStrip* fCurrentStripCalcBot; 

	  
    PndGeoHandling* fGeoH;      //! Geometry name handling
    PndSdsChargeWeightingAlgorithms* fChargeAlgos;
    PndSdsStripClusterer* fCurrentClusterfinder;
    std::map<const char*,PndSdsStripClusterer*> fClusterFinderList; 

    ClassDef(PndSdsStripClusterTask,1);

};

#endif /* SDSCLUSTERTASK_H */