/*
 * CbmMvdClusteringNoCharges.h
 *
 *  Created on: Jun 7, 2012
 *      Author: kozlov
 */

#ifndef CBMMVDCLUSTERINGNOCHARGES_H_
#define CBMMVDCLUSTERINGNOCHARGES_H_

#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 TH1F;

class CbmMvdClusteringNoCharges: public FairTask
{
public:
   /**
    * \brief Constructor.
    */
	CbmMvdClusteringNoCharges();

   /**
    * \brief Destructor.
    */
   virtual ~CbmMvdClusteringNoCharges();

   /**
    * \brief Derived from FairTask.
    */
   virtual InitStatus Init();

   /**
    * \brief Derived from FairTask.
    */
   virtual void Exec(Option_t* opt);

   /**
    * \brief Derived from FairTask.
    */
   virtual void Finish();

   void FindClusters();

   void Step1_SLClustering();
   void Step1_SLRec(pair<Int_t, Int_t> it, Int_t cl);

   void CalculateAcuracy();

   //void FillHistogramsForRadius();

   //void TestStrangeClusters();

   void PrintClusters();


private:

   void ReadDataBranches();

   TClonesArray* fDigis;
   TClonesArray* fPoints;
   TClonesArray* fHits;
   TClonesArray* fCluster;
   TClonesArray* fMatches;

   TClonesArray* fMCTracks;

   map<pair<Int_t, Int_t>, Int_t> fDigiMap;
   map<pair<Int_t, Int_t>, Int_t>::iterator fDigiMapIt;

   map<pair<Int_t, Int_t>, Int_t> fClusterMap;
   map<pair<Int_t, Int_t>, Int_t>::iterator fClusterMapIt;

   map<pair<Int_t, Int_t>, Bool_t> fActiveMap;
   map<pair<Int_t, Int_t>, Bool_t>::iterator fActiveMapIt;

   map<pair<Int_t, Int_t>, Bool_t> fChargeMap1;
   map<pair<Int_t, Int_t>, Bool_t>::iterator fChargeMap1It;

   map<pair<Int_t, Int_t>, Bool_t> fChargeMap2;
   map<pair<Int_t, Int_t>, Bool_t>::iterator fChargeMap2It;

   /*map<pair<Int_t, Int_t>, Int_t> fChargeMap2;
   map<pair<Int_t, Int_t>, Int_t>::iterator fChargeMap2It;*/
   Int_t fNofDigis;
   Int_t fNofPoints;

   Double_t globalCTime;
   Double_t globalRTime;

   Int_t fNofPointsBySt[4];
   Int_t fNofDigisBySt[4];
   Int_t fNofClustersBySt[4];

   Bool_t fUseTrashold;
   Int_t fTrashold;

   static const Int_t nDigiByCl = 150;
   Int_t fNofClusters;
   struct Cluster{
	   Int_t nofCluster;
   	   Float_t xc;
   	   Float_t yc;
   	   Float_t sumClCharge;
   	   Int_t nofDidis;
   	   //Int_t digisInCluster[nDigiByCl];
   	   vector<Int_t> digisInCluster;
   	   Int_t nDigiMax;
   	   Int_t nMCPoint;
   Cluster() : nofCluster(0), xc(0.), yc(0.), sumClCharge(0.),
	     nofDidis(0), digisInCluster(), nDigiMax(0),
	     nMCPoint(0) {;}
   };
   Cluster* fClusters;
   vector<Cluster> fFinalClusters;

   map<Int_t, CbmMvdStation*> fStationMap;

   /*static const Int_t nRadiusSteps = 100;
   TH1F* fhErrPoint_Hit;
   TH1F* fhErrPoint_Pixel;
   TH1F* fhErrHit_Pixel;*/

   TH1F* fhErrorsHit_MCPoint;
   TH1F* fhErrorsHit_Pixel;
   TH1F* fhErrorsPixel_MCPoint;

   TH1F* fhDigisInCluster;
   TH1F* fhDigisInClusterPDG211;
   TH1F* fhDigisInClusterPDG11;
   TH1F* fhDigisInClusterPDG321;
   TH1F* fhDigisInClusterPDG2212;
   TH1F* fhDigisInClusterPDG3222;
   TH1F* fhDigisInClusterPDG3112;
   TH1F* fhDigisInClusterPDG13;
   TH1F* fhMaxDigiCharges;
   TH1F* fhMaxDigiChargesStation1;
   TH1F* fhMaxDigiChargesStation2;

   TH1F* fhResidual_X_St1;
   TH1F* fhResidual_Y_St1;
   TH1F* fhResidual_X_St2;
   TH1F* fhResidual_Y_St2;
   TH1F* fhResidual_X;
   TH1F* fhResidual_Y;
   TH1F* fhPull_X_St1;
   TH1F* fhPull_Y_St1;
   TH1F* fhPull_X_St2;
   TH1F* fhPull_Y_St2;

   Int_t fNofEvents;
   Float_t fEfficiency;
   Float_t fErrHit_Pixel;
   Float_t fErrHit_Point;
   Float_t fErrPoint_Pixel;
   Int_t fLooseCl;
   Int_t fLooseClustersBySt[4];
   Int_t fFakeCl;
   Int_t fFakeClustersBySt[4];

   Int_t GetMvdGeometry();
   void Register();

   void ChangeClusters(pair<Int_t, Int_t> h, Int_t nCl0, Int_t nCl1);

   CbmMvdClusteringNoCharges(const CbmMvdClusteringNoCharges&);
   CbmMvdClusteringNoCharges& operator=(const CbmMvdClusteringNoCharges&);
   
   ClassDef(CbmMvdClusteringNoCharges, 1);
};

#endif /* CBMMVDCLUSTERINGNOCHARGES_H_ */