#ifndef CBMTRDHITRATEQA_H_
#define CBMTRDHITRATEQA_H_

#include "FairTask.h"
#include "CbmTrdDigi.h"

#include <map>
#include <list>
#include <vector>

class CbmTrdDigiPar;
class CbmTrdModule;
class CbmTrdRadiator;
class CbmTrdGeoHandler;

class TClonesArray;
class TCanvas;
class TH1F;
class TH2F;

typedef struct HitRateGeoPara
{
  Int_t moduleId;
  Int_t layerId;
  Int_t stationId;

  Double_t mPos[3];     // center of module (x,y,z)
  Double_t mSize[3];    // half size of pad plane in (x,y,z) from digi.par

  Double_t sSize[3][3]; // sector size
  Double_t pSize[3][3]; // pad size     // [sectors][dimensions]

  Int_t nSec;      // 3 - number of sectors in module
  Int_t rot_angle; // 0,1,2,3 - angle between long pad axis and x axis

  Int_t nCol;
  Int_t nRow;
  Int_t sCol[3];  // number of cols in sector
  Int_t sRow[3];  // number of rows in sector

  Double_t vOrigin[3]; // vector of pad 0,0
  Double_t vX[3];      // differential vector of pad 1,0 to pad 0,0
  Double_t vY[3];      // differential vector of pad 0,1 to pad 0,0
  Double_t vN[3];      // normal vector

  Double_t lambda;
  Double_t cosX;
  Double_t cosY;
  Int_t stepDirection[3];

} HitRateGeoPara;

class CbmTrdHitRateQa : public FairTask {

 public:

  /** Default constructor **/
  CbmTrdHitRateQa();

  /** Standard constructor **/
  CbmTrdHitRateQa(const char *name, const char *title="CBM Task");

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

  /** Initialisation **/
  virtual InitStatus ReInit();
  virtual InitStatus Init();
  virtual void SetParContainers();

  /** Executed task **/
  virtual void Exec(Option_t * option);

  /** Finish (called after each event) **/
  virtual void FinishEvent();
 
  /** Finish task (called after all event) **/
  virtual void FinishTask(){;}

  /*
    void AddHit(TVector3 &posHit, TVector3 &posHitErr,
    Int_t TrackID, Int_t PlaneID, Int_t ref, Double_t ELoss,
    Double_t ELossTR, Double_t ELossdEdX);
  */

  void Register();

 private:

  void HistoInit(TCanvas*& c1, TCanvas*& c2, TH2F*& Layer,TH1F*& HitPad, Double_t ZRangeL, Double_t ZRangeU, Double_t mm2bin);

  void GetModuleInformationFromDigiPar(HitRateGeoPara *GeoPara, Bool_t Fast, Bool_t Lines, Int_t VolumeID, TH2F* Layer, TCanvas* c1, TH1F* HitPad, TCanvas* c2, TH2F* Topview[3], TCanvas* c0, Double_t mm2bin);

  void GetModuleInformation();

  void Histo(HitRateGeoPara *GeoPara, Bool_t Fast, TH2F* Layer, TCanvas* c1, TH1F* HitPad, TCanvas* c2,  TH2F* Topview[3], TCanvas* c0, Double_t mm2bin);
 
  Double_t CalcHitRate(HitRateGeoPara *GeoPara, Double_t StartX, Double_t StopX, Int_t xSteps, Double_t StartY, Double_t StopY, Int_t ySteps, Double_t* Mpos, TH2F* Topview[3], TCanvas* c0);

  void DrawBorders(HitRateGeoPara *GeoPara, TH2F* Layer, TCanvas* c1);

  void DrawPads(HitRateGeoPara *GeoPara, TH2F* Layer, TCanvas* c1);

  void DrawDigi();

 
  Int_t Digicounter;

  TFile *tFile;

  Bool_t fDraw;

  Int_t   fPlane;  // DE
  Int_t   fStation;
  Int_t   fLayer;

  Int_t   fCol_mean,fCol_in,fCol_out; //Calculated pixel column were the hit is in
  Int_t   fRow_mean,fRow_in,fRow_out; //Calculated pixel row were the hit is in
  Int_t   fModuleID;//Unique number for detector module
  Int_t   fMCindex;// index to MCPoint
  // --------------->[cm]<---------------
  Double_t local_meanLL[3];
  Double_t local_meanC[3];
  Double_t global_meanLL[3];//[cm]
  Double_t global_meanC[3];
  Double_t local_inLL[3];
  Double_t local_inC[3];
  Double_t global_inLL[3];//[cm]
  Double_t global_inC[3];

  Double_t local_outLL[3];
  Double_t local_outC[3];
  Double_t global_outLL[3];//[cm]
  Double_t global_outC[3];
  // --------------->[mm]<---------------

  Float_t fx_in, fx_out, fy_in, fy_out, fz_in, fz_out, fx_mean, fy_mean, fz_mean;
  Int_t fSector;
  //Int_t tempNosectors =  (fModuleInfo->GetNoSectors);
  static const Int_t accuracy = 1;// '1/accuracy' integration step width [mm]
  static const Int_t fPadNrX = 7;//7; // has to be odd
  static const Int_t fPadNrY = 5;//5; // has to be odd

  /*
    static const Bool_t Histo = true;
    //static const Bool_t Histo = false;
    static const Bool_t TEST = true;
    //static const Bool_t TEST = false;
    //static const Bool_t Sector = true;
    static const Bool_t Sector = false;
  */
  
  Double_t padsize[3];
  Double_t modulesize[3];

  Float_t fELoss;//energy loss from MCPoint 
  Float_t fELossdEdX;
  Float_t fELossTR;
  Float_t fPosXLL, fPosYLL;//Hit position in chamber coordinates origin 'L'ower 'L'eft corner
 
  Float_t fPadPosxLL, fPadPosyLL;//Hit position in pad coordinates (not rotated) origin 'L'ower 'L'eft corner
  Float_t fPadPosxC, fPadPosyC;//Hit position in pad coordinates (not rotated) origin pad 'C'enter
  Float_t fDeltax, fDeltay;



  Float_t fPadCharge[fPadNrY][fPadNrX]; //Charge on 3 adjacent pads calculated by using the Mathieson formula

  Float_t fPRFHitPositionLL, fPRFHitPositionC;

  Float_t fEfficiency; // Digi production efficiency (0-100%)
  //    Double_t fthreshold; //pixel threshold in electrons
  //    Double_t fnoise; //pixel noise in electrons

  TClonesArray *fTrdPoints; //! Trd MC points
  TClonesArray *fDigiCollection; //! TRD digis
  TClonesArray *fDigiMatchCollection; //! Corresponding MCPoints to TRD digis
  TClonesArray *fMCStacks;  //! MC Track information

  CbmTrdDigiPar  *fDigiPar;
  CbmTrdModule   *fModuleInfo;
  //CbmTrdRadiator *fRadiators; 

  CbmTrdGeoHandler* fGeoHandler; //!
    
  /**  map to store digis for pair of x,y position in module **/
  // map<pair<ModuleID,pair<x,y>>, CbmTrdDigi*>
  std::map<std::pair< Int_t, std::pair< Int_t, Int_t > >, CbmTrdDigi* > fDigiMap; 
  /**  iterator over map to store digis for pair of x,y position in module **/
  std::map<std::pair< Int_t, std::pair< Int_t, Int_t > >, CbmTrdDigi* >::iterator fDigiMapIt; //! iterator over array above

  CbmTrdHitRateQa(const CbmTrdHitRateQa&);
  CbmTrdHitRateQa& operator=(const CbmTrdHitRateQa&);

  ClassDef(CbmTrdHitRateQa,2)

};

#endif // CBMTRDHITRATEQA_H