// -------------------------------------------------------------------------
// -----                 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 "PndSdsPixel.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 "PndSdsStripClusterBuilder.h"

#include <string>
#include <vector>

class TClonesArray;
class PndSdsChargeWeightingAlgorithms;

class PndSdsStripClusterTask : public FairTask
{
 public:

    /** Default constructor **/


	PndSdsStripClusterTask(Int_t ClusterMod=0, Int_t RadChannel=2, Int_t RadTime=0);

	PndSdsStripClusterTask(Double_t chargecut, TString geofile, Int_t meanalgo=0, Int_t clustermod=0, Int_t RadChannel=2, Int_t RadTime=0);

    /** 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();
    virtual InitStatus Init();
    virtual InitStatus ReInit();

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


  protected:

    TVector2 CalcLineCross(TVector2 point1, TVector2 dir1, TVector2 point2, TVector2 dir2) const;
    void 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();

    Double_t fRadius;
    Int_t fFEcolumns;
    Int_t fFErows;
    TString fGeoFile;
    Double_t fChargeCut;
    Int_t fMeanAlgo;
	Int_t fRadChannel;
	Int_t fRadTime;

    PndSdsStripDigiPar* fDigiParRect;     //! Digitization Parameters (Barrel)
    PndSdsStripDigiPar* fDigiParTrap;     //! Digitization Parameters (Disks)
    PndSdsStripDigiPar* fCurrentDigiPar;  //! Digitization Parameters
    //! Calculator objects
    PndSdsCalcStrip* fStripCalcTopRect;
    PndSdsCalcStrip* fStripCalcBotRect;
    PndSdsCalcStrip* fStripCalcTopTrap;
    PndSdsCalcStrip* fStripCalcBotTrap;
    PndSdsCalcStrip* fStripCalcTop;
    PndSdsCalcStrip* fStripCalcBot;

    Int_t fClusterMod;					/// selected clusterfinder 1...simple 0...default

    PndSdsGeoHandling* fGeoH;      //! Geometry name handling
  //  PndSdsGeoPar* fGeoPar;
    PndSdsChargeWeightingAlgorithms* fChargeAlgos;
    PndSdsStripClusterBuilder* fClusterfinder;
//     TH1F* fHChgDiff;
//     TH1F* fHChgMC;
//     TH1F* fHChgFake;
//     TH1F* fHChgGhost;

  ClassDef(PndSdsStripClusterTask,1);

};

#endif /* SDSCLUSTERTASK_H */