#ifndef MuonDigiAlg_H
#define MuonDigiAlg_H
#include "k4FWCore/DataHandle.h"
#include "GaudiKernel/NTuple.h"
#include "GaudiAlg/GaudiAlgorithm.h"
#include "edm4hep/SimTrackerHitCollection.h"
#include "edm4hep/TrackerHitCollection.h"
#include "edm4hep/MCRecoTrackerAssociationCollection.h"
#include <DDRec/DetectorData.h>
#include "DetInterface/IGeomSvc.h"
#include "k4FWCore/DataHandle.h"
#include "GaudiAlg/GaudiAlgorithm.h"
#include "edm4hep/MutableCaloHitContribution.h"
#include "edm4hep/MutableSimCalorimeterHit.h"
#include "edm4hep/SimCalorimeterHit.h"
#include "edm4hep/CalorimeterHit.h"
#include "edm4hep/CalorimeterHitCollection.h"
#include "edm4hep/SimCalorimeterHitCollection.h"
#include "edm4hep/MCRecoCaloAssociationCollection.h"
#include "edm4hep/MCRecoCaloParticleAssociationCollection.h"
#include <DDRec/DetectorData.h>
#include <DDRec/CellIDPositionConverter.h>
#include <DD4hep/Segmentations.h>
#include "DetInterface/IGeomSvc.h"
#include "TVector3.h"
#include "TRandom3.h"
#include "TFile.h"
#include "TString.h"
#include "TH3.h"
#include "TH1.h"
#include <cstdlib>
#include "time.h"
#include <TTimeStamp.h>
#include <ctime>
#include <iostream>
#include <TMath.h>
#define PI 3.141592653
class MuonDigiAlg : public GaudiAlgorithm
{
public:
MuonDigiAlg(const std::string& name, ISvcLocator* svcLoc);
virtual StatusCode initialize() ;
virtual StatusCode execute() ;
virtual StatusCode finalize() ;
void MuonHandle(int _i); // 2->Barrel 0->Endcap
void Clear();
void GetSimHit(edm4hep::SimTrackerHit _simhit, dd4hep::DDSegmentation::BitFieldCoder* _m_decoder, int _i, int _message[6], double & _Edep, unsigned long long & _cellid, double & _xydist, edm4hep::Vector3d & _pos, dd4hep::Position & _ddpos, std::array<unsigned long long, 2> & _key, double & _mcpz);
double Gethit_sipm_length_Barrel(dd4hep::Position _ddpos, edm4hep::Vector3d _pos, int _message[6]);
double Gethit_sipm_length_Endcap(dd4hep::Position _ddpos, edm4hep::Vector3d _pos, int _message[6]);
double EdeptoADC(double _Edep, double _hit_sipm_length);
void SaveData_mapcell(std::array<unsigned long long, 2> _key, double _Edep, int _message[6], edm4hep::Vector3d _pos);
void Cut3();
void Find_anotherlayer(int _i, std::array<unsigned long long, 2> _key1, std::array<unsigned long long, 2> _key2, double _ddposi, int & _anotherlayer_cell_num);
void Save_trkhit(edm4hep::TrackerHitCollection* _trkhitVec, std::array<unsigned long long, 2> _key, int _pdgid, edm4hep::Vector3d _pos);
void Save_onelayer_signal(edm4hep::TrackerHitCollection* _trkhitVec);
protected:
TTree* m_tree;
TFile* m_wfile;
SmartIF<IGeomSvc> m_geosvc;
dd4hep::DDSegmentation::BitFieldCoder* m_decoder_barrel;
dd4hep::DDSegmentation::BitFieldCoder* m_decoder_endcap;
dd4hep::rec::CellIDPositionConverter* m_cellIDConverter;
dd4hep::Detector* m_dd4hep;
std::map<std::array<unsigned long long, 2>, double> map_cell_edep;
std::map<std::array<unsigned long long, 2>, double> map_cell_adc;
std::map<std::array<unsigned long long, 2>, int> map_cell_layer;
std::map<std::array<unsigned long long, 2>, int> map_cell_slayer;
std::map<std::array<unsigned long long, 2>, int> map_cell_strip;
std::map<std::array<unsigned long long, 2>, int> map_cell_fe;
std::map<std::array<unsigned long long, 2>, int> map_cell_env;
std::map<std::array<unsigned long long, 2>, int> map_cell_pdgid;
std::map<std::array<unsigned long long, 2>, edm4hep::Vector3d> map_cell_pos;
std::map<int, double> map_muonhit;
std::map<int, int> map_pdgid;
#define MAX_SIZE 10000
int n_hit, n_cell;
unsigned long long hit_cellid[MAX_SIZE];
double hit_posx[MAX_SIZE];
double hit_posy[MAX_SIZE];
double hit_posz[MAX_SIZE];
double hit_edep[MAX_SIZE];
int hit_layer[MAX_SIZE];
int hit_slayer[MAX_SIZE];
int hit_strip[MAX_SIZE];
int hit_fe[MAX_SIZE];
int hit_env[MAX_SIZE];
unsigned long long cell_cellid[MAX_SIZE];
double cell_edep[MAX_SIZE];
double cell_adc[MAX_SIZE];
double cell_posx[MAX_SIZE];
double cell_posy[MAX_SIZE];
double cell_posz[MAX_SIZE];
int cell_layer[MAX_SIZE];
int cell_slayer[MAX_SIZE];
int cell_strip[MAX_SIZE];
int cell_fe[MAX_SIZE];
int cell_env[MAX_SIZE];
int cell_superlayernumber;
double cell_pt;
Gaudi::Property<int> _writeNtuple{this, "WriteNtuple", 1, "Write ntuple"};
Gaudi::Property<std::string> _filename{this, "OutFileName", "testout.root", "Output file name"};
Gaudi::Property<double> m_hitEff{ this, "SiPMEff", 1, "Efficiency of a single hit on a Strip" };
Gaudi::Property<double> m_EdepMin{ this, "EdepMin", 0.0001, "Minimum Edep of a mip" };
Gaudi::Property<double> m_hit_Edep_min{ this, "HitEdepMin", 0.000001, "Minimum Edep of a mip"};
Gaudi::Property<double> m_hit_Edep_max{ this, "HitEdepMax", 0.1, "Maximum Edep of a mip"};
// number of strips parallel to beam direction
// in each slayer, each strip width=4cm, so
int strip_length[6] = {26, 38, 50, 62, 74, 86}; //for barrel
double endcap_strip_length[193] = {2.12, 2.12, 2.12, 2.13, 2.14, 2.15, 2.16, 2.17, 2.19, 2.22,
2.24, 2.28, 2.32, 2.37, 2.45, 2.65, 2.64, 2.63, 2.62, 2.61,
2.60, 2.59, 2.57, 2.56, 2.54, 2.53, 2.51, 2.50, 2.48, 2.46,
2.44, 2.42, 2.40, 2.38, 2.36, 2.33, 2.31, 2.28, 2.26, 2.23,
2.20, 2.17, 2.14, 2.11, 2.07, 2.04, 2.00, 1.97, 1.93, 1.89,
1.84, 1.80, 1.75, 1.70, 1.65, 1.60, 1.54, 1.48, 1.42, 1.35,
1.28, 1.20, 1.12, 1.02, 0.92, 0.80, 0.65, 0.46, 2.20, 2.20,
2.20, 2.20, 2.20, 2.20, 2.20, 2.21, 2.21, 2.21, 2.21, 2.22,
2.22, 2.22, 2.23, 2.23, 2.23, 2.24, 2.24, 2.25, 2.25, 2.26,
2.26, 2.27, 2.28, 2.28, 2.29, 2.30, 2.31, 2.32, 2.32, 2.33,
2.34, 2.35, 2.36, 2.38, 2.39, 2.40, 2.41, 2.43, 2.44, 2.45,
2.47, 2.49, 2.50, 2.52, 2.54, 2.56, 2.58, 2.60, 2.62, 2.65,
2.67, 2.70, 2.73, 2.76, 2.79, 2.82, 2.86, 2.90, 2.94, 2.99,
3.04, 3.10, 3.16, 3.23, 3.31, 3.41, 3.53, 3.70, 4.14, 4.12,
4.09, 4.06, 4.03, 4.00, 3.97, 3.94, 3.91, 3.87, 3.84, 3.81,
3.77, 3.74, 3.70, 3.67, 3.63, 3.59, 3.55, 3.51, 3.47, 3.43,
3.38, 3.34, 3.30, 3.25, 3.20, 3.15, 3.10, 3.05, 3.00, 2.95,
2.89, 2.83, 2.77, 2.71, 2.65, 2.58, 2.52, 2.45, 2.37, 2.30,
2.22, 2.14, 2.05, 1.96, 1.86, 1.76, 1.65, 1.53, 1.40, 1.25,
1.09, 0.89, 0.63};
// Input collections
DataHandle<edm4hep::SimTrackerHitCollection> m_inputMuonBarrel{"MuonBarrelHitsCollection", Gaudi::DataHandle::Reader, this};
DataHandle<edm4hep::SimTrackerHitCollection> m_inputMuonEndcap{"MuonEndcapHitsCollection", Gaudi::DataHandle::Reader, this};
// Output collections
DataHandle<edm4hep::TrackerHitCollection> m_outputMuonBarrel{"MuonBarrelTrackerHits", Gaudi::DataHandle::Writer, this};
DataHandle<edm4hep::TrackerHitCollection> m_outputMuonEndcap{"MuonEndcapTrackerHits", Gaudi::DataHandle::Writer, this};
//DataHandle<edm4hep::MCRecoTrackerAssociationCollection> m_assMuonBarrel{"MuonBarrelTrackerHitAssociationCollection", Gaudi::DataHandle::Writer, this};
SmartIF<IRndmGenSvc> m_randSvc;
edm4hep::TrackerHitCollection* trkhitVec;
int m_nEvt=0;
};
#endif