/** CbmMuchDigitizeGem.cxx *@author Vikas Singhal *@since 01.10.16 *@version 2.0 *@author Evgeny Kryshen *@since 01.05.11 *@version 2.0 *@author Mikhail Ryzhinskiy *@since 19.03.07 *@version 1.0 ** ** CBM task class for digitizing MUCH for both Event by event and Time based mode. ** Task level RECO ** Produces objects of type CbmMuchDigi out of CbmMuchPoint. **/ // Includes from MUCH #include "CbmMuchDigitizeGem.h" #include "CbmMuchPoint.h" #include "CbmMuchSector.h" #include "CbmMuchStation.h" #include "CbmMuchModuleGem.h" #include "CbmMuchModuleGemRadial.h" #include "CbmMuchModuleGemRectangular.h" #include "CbmMuchPad.h" #include "CbmMuchPadRadial.h" #include "CbmMuchPadRectangular.h" #include "CbmMuchSectorRadial.h" #include "CbmMuchSectorRectangular.h" #include "CbmMuchDigi.h" #include "CbmMuchReadoutBuffer.h" // Includes from base #include "FairRootManager.h" #include "FairEventHeader.h" #include "FairMCEventHeader.h" #include "FairMCPoint.h" #include "FairRunAna.h" #include "FairRunSim.h" // Includes from Cbm #include "CbmMCTrack.h" #include "CbmMCEpoch.h" //#include "CbmMCBuffer.h" #include "CbmDaqBuffer.h" // Includes from ROOT #include "TObjArray.h" #include "TDatabasePDG.h" #include "TFile.h" #include "TRandom.h" #include "TChain.h" #include #include using std::map; // ------------------------------------------------------------------------- CbmMuchDigitizeGem::CbmMuchDigitizeGem(const char* digiFileName) : FairTask("MuchDigitizeGem",1), fgDeltaResponse(), fAlgorithm(1), fGeoScheme(CbmMuchGeoScheme::Instance()), fDigiFile(digiFileName), fPoints(NULL), fMCTracks(NULL), fDigis(NULL), fDigiMatches(NULL), fNFailed(0), fNOutside(0), fNMulti(0), fNADCChannels(256), fQMax(500000), fQThreshold(10000), fMeanNoise(1500), fSpotRadius(0.05), fMeanGasGain(1e4), fDTime(3), fDeadPadsFrac(0), fTimer(), fDaq(0), fMcChain(NULL), fDeadTime(400), fDriftVelocity(100), fPeakingTime(20), fRemainderTime(40), fTimeBinWidth(1), fNTimeBins(200), fNdigis(0), fTOT(0), fTotalDriftTime(0.4/fDriftVelocity*10000), // 40 ns fSigma(), fMPV(), fIsLight(1) // fIsLight = 1 (default) Store Light CbmMuchDigiMatch in output branch, fIsLight = 0 Create Heavy CbmMuchDigiMatch with fSignalShape info. { fSigma[0] = new TF1("sigma_e","pol6",-5,10); fSigma[0]->SetParameters(sigma_e); fSigma[1] = new TF1("sigma_mu","pol6",-5,10); fSigma[1]->SetParameters(sigma_mu); fSigma[2] = new TF1("sigma_p","pol6",-5,10); fSigma[2]->SetParameters(sigma_p); fMPV[0] = new TF1("mpv_e","pol6",-5,10); fMPV[0]->SetParameters(mpv_e); fMPV[1] = new TF1("mpv_mu","pol6",-5,10); fMPV[1]->SetParameters(mpv_mu); fMPV[2] = new TF1("mpv_p","pol6",-5,10); fMPV[2]->SetParameters(mpv_p); } // ------------------------------------------------------------------------- // ----- Destructor ---------------------------------------------------- CbmMuchDigitizeGem::~CbmMuchDigitizeGem() { if (fDigis) { fDigis->Delete(); delete fDigis; } if (fDigiMatches) { fDigiMatches->Delete(); delete fDigiMatches; } if (fSigma) { // fSigma->Delete(); delete fSigma[0]; delete fSigma[1]; delete fSigma[2]; //delete[] fSigma; } if (fMPV) { // fMPV->Delete(); delete fMPV[0]; delete fMPV[1]; delete fMPV[2]; //delete[] fMPV; } } // ------------------------------------------------------------------------- // ----- Private method Init ------------------------------------------- InitStatus CbmMuchDigitizeGem::Init() { FairRootManager* ioman = FairRootManager::Instance(); if (!ioman) Fatal("Init", "No FairRootManager"); // Initialize GeoScheme TFile* oldfile=gFile; TFile* file=new TFile(fDigiFile); TObjArray* stations = (TObjArray*) file->Get("stations"); file->Close(); file->Delete(); gFile=oldfile; fGeoScheme->Init(stations); // Determine drift volume width Double_t driftVolumeWidth = 0.4; // cm - default for (Int_t i=0;iGetNStations();i++){ CbmMuchStation* station = fGeoScheme->GetStation(i); if (station->GetNLayers()<=0) continue; CbmMuchLayerSide* side = station->GetLayer(0)->GetSide(0); if (side->GetNModules()<=0) continue; CbmMuchModule* module = side->GetModule(0); if (module->GetDetectorType()!=1 && module->GetDetectorType()!=3) continue; driftVolumeWidth = module->GetSize().Z(); break; } fTotalDriftTime = driftVolumeWidth/fDriftVelocity*10000; // [ns]; //Reading MC point as Event by event for time based digi generation also. // Get input array of MuchPoints fPoints = (TClonesArray*) ioman->GetObject("MuchPoint"); // Get input array of MC tracks fMCTracks = (TClonesArray*) ioman->GetObject("MCTrack"); //For event by event mode output will be stored via ioman if(!fDaq){ // Register output array MuchDigi fDigis = new TClonesArray("CbmMuchDigi", 1000); ioman->Register("MuchDigi", "Digital response in MUCH", fDigis, kTRUE); // Register output array MuchDigiMatches fDigiMatches = new TClonesArray("CbmMuchDigiMatch", 1000); ioman->Register("MuchDigiMatch", "Digi Match in MUCH", fDigiMatches, kTRUE); } else{ FairTask* daq = FairRun::Instance()->GetTask("Daq"); if ( daq ) { LOG(INFO) << GetName() << ": Using stream mode." << FairLogger::endl; //fMode = 0; } //? stream mode if (!CbmDaqBuffer::Instance() ) { fLogger->Fatal(MESSAGE_ORIGIN, "No CbmDaqBuffer present for building TimeSlice!"); return kFATAL; } // fMCTracks = (TClonesArray*) ioman->GetObject("MCTrack"); } //fgDeltaResponse is used in the CbmMuchSignal for analysing the Signal Shape, //it is generated once in the digitizer and then be used by each CbmMuchSignal. //For reducing the time therefore generated once in the CbmMuchDigitize Gem and //not generated in the CbmMuchSignal // Set response on delta function Int_t nShapeTimeBins=Int_t(gkResponsePeriod/gkResponseBin); fgDeltaResponse.Set(nShapeTimeBins); for (Int_t i=0;iGetInTree()->GetBranch("MCTrack")->GetReadEntry(); LOG(INFO) << "Event Number is "<< currentEvent << FairLogger::endl; fTimer.Start(); fNdigis = 0; Int_t nPoints=0; //Storing digi->Time < EventStartTime ReadAndRegister(); // cout<< "Exec Started and came out from first ReadAndRegister"<< endl; // --- Loop over all MuchPoints and execute the ExecPoint method for both Event by event and Time Based Digitization for (Int_t iPoint=0; iPointGetEntriesFast(); iPoint++) { const CbmMuchPoint* point = (const CbmMuchPoint*) fPoints->At(iPoint); nPoints++; ExecPoint(point, iPoint); } // MuchPoint loop // Add remaining digis //For event by event mode read all the data from CbmMuchReadoutBuffer and register in the Output. //Read the Signal, convert into Digi And Register into output ReadAndRegister(); fTimer.Stop(); gLogger->Info(MESSAGE_ORIGIN,"MuchDigitizeGem: %5.2f s, %i points, %i digis",fTimer.RealTime(),nPoints,fNdigis); // revert branch to "current event" FairRootManager::Instance()->GetInTree()->GetBranch("MCTrack")->GetEntry(currentEvent); } // ------------------------------------------------------------------------- //Read all the Signal from CbmMuchReadoutBuffer, convert the analog signal into the digital response and register Output according to event by event mode and Time based mode. void CbmMuchDigitizeGem::ReadAndRegister(){ std::vector SignalList; Double_t eventTime = -1.; if(fDaq){ eventTime = FairRun::Instance()->GetEventHeader()->GetEventTime(); } Int_t ReadOutSignal = CbmMuchReadoutBuffer::Instance()->ReadOutData(eventTime,SignalList); LOG(INFO)<< "Number of digi's read out from Buffer "<< ReadOutSignal << FairLogger::endl; for (std::vector::iterator LoopOver= SignalList.begin(); LoopOver != SignalList.end(); LoopOver++) { CbmMuchDigi* digi = ConvertSignalToDigi(*LoopOver); //assert(digi); if (!digi){ gLogger->Debug1(MESSAGE_ORIGIN,"Digi not created as signal is below threshold."); } else { gLogger->Debug1(MESSAGE_ORIGIN,"New digi: sector=%i channel=%i",CbmMuchAddress::GetSectorIndex(digi->GetAddress()),CbmMuchAddress::GetChannelIndex(digi->GetAddress())); if(fDaq) //Buffer these digi's in CbmDaqBuffer for Timeslice building. CbmDaqBuffer::Instance()->InsertData(digi); else{ new ((*fDigis)[fDigis->GetEntriesFast()]) CbmMuchDigi(digi); //Above syntax is equivalent as below:- //Int_t nDigis = fDigis->GetEntriesFast(); //CbmMuchDigi* digi = new ((*fDigis)[nDigis]) CbmMuchDigi(digi) new ((*fDigiMatches)[fDigiMatches->GetEntriesFast()]) CbmMuchDigiMatch((CbmMuchDigiMatch*)digi->GetMatch()); // Match object will be deleted in the digi destructor. delete digi; } fNdigis++; } } } //Convert Signal into the Digi with appropriate methods. CbmMuchDigi* CbmMuchDigitizeGem::ConvertSignalToDigi(CbmMuchSignal* signal){ Int_t TimeStamp = signal->GetTimeStamp(fQThreshold); if (TimeStamp < 0) return (NULL);//entire signal is below threshold, no digi generation. CbmMuchDigi *digi = new CbmMuchDigi(); digi->SetAddress(signal->GetAddress()); //Charge in number of electrons, need to be converted in ADC value digi->SetAdc((signal->GetCharge())*fNADCChannels/fQMax);//Charge should be computed as per Electronics Response. digi->SetTime(TimeStamp); digi->SetMatch(signal->GetMatch()); // digi->SetPileUp(); // digi->SetDiffEvent(); return(digi); } // ------------------------------------------------------------------------- void CbmMuchDigitizeGem::Finish(){ //if (fDaq) Exec(""); //Store all the remaining digi's in the buffer into the CbmDaqBuffer if (fDaq) ReadAndRegister(); } // ------------------------------------------------------------------------- // ------- Private method ExecAdvanced ------------------------------------- Bool_t CbmMuchDigitizeGem::ExecPoint(const CbmMuchPoint* point, Int_t iPoint) { TVector3 v1,v2,dv; point->PositionIn(v1); point->PositionOut(v2); dv = v2-v1; Int_t detectorId = point->GetDetectorID(); CbmMuchModule* module = fGeoScheme->GetModuleByDetId(detectorId); if (fAlgorithm==0){ // Simple digitization TVector3 v = 0.5*(v1+v2); CbmMuchPad* pad = 0; if (module->GetDetectorType()==1){ CbmMuchModuleGemRectangular* module1 = (CbmMuchModuleGemRectangular*) module; pad = module1->GetPad(v[0],v[1]); if (pad) printf("x0=%f,y0=%f\n",pad->GetX(),pad->GetY()); } else if (module->GetDetectorType()==3){ CbmMuchModuleGemRadial* module3 = (CbmMuchModuleGemRadial*) module; pad = module3->GetPad(v[0],v[1]); } if (!pad) return kFALSE; AddCharge(pad,fQMax,iPoint,point->GetTime(),0); return kTRUE; } // Start of advanced digitization Int_t nElectrons = Int_t(GetNPrimaryElectronsPerCm(point)*dv.Mag()); if (nElectrons<0) return kFALSE; Double_t time = point->GetTime(); if (module->GetDetectorType()==1) { CbmMuchModuleGemRectangular* module1 = (CbmMuchModuleGemRectangular*) module; map firedSectors; for (Int_t i=0;iRndm(); Double_t driftTime = (1-aL)*fTotalDriftTime; TVector3 ve = v1 + dv*aL; UInt_t ne = GasGain(); Double_t x = ve.X(); Double_t y = ve.Y(); Double_t x1 = x-fSpotRadius; Double_t x2 = x+fSpotRadius; Double_t y1 = y-fSpotRadius; Double_t y2 = y+fSpotRadius; Double_t s = 4*fSpotRadius*fSpotRadius; firedSectors[module1->GetSector(x1,y1)]=0; firedSectors[module1->GetSector(x1,y2)]=0; firedSectors[module1->GetSector(x2,y1)]=0; firedSectors[module1->GetSector(x2,y2)]=0; for (map::iterator it = firedSectors.begin(); it!= firedSectors.end(); it++) { CbmMuchSector* sector = (*it).first; if (!sector) continue; for (Int_t iPad=0;iPadGetNChannels();iPad++){ CbmMuchPad* pad = sector->GetPadByChannelIndex(iPad); Double_t xp0 = pad->GetX(); Double_t xpd = pad->GetDx()/2.; Double_t xp1 = xp0-xpd; Double_t xp2 = xp0+xpd; if (x1>xp2 || x2GetY(); Double_t ypd = pad->GetDy()/2.; Double_t yp1 = yp0-ypd; Double_t yp2 = yp0+ypd; if (y1>yp2 || y2xp1 ? (x2yp1 ? (y2GetDetectorType()==3) { CbmMuchModuleGemRadial* module3 = (CbmMuchModuleGemRadial*) module; CbmMuchSectorRadial* sFirst = (CbmMuchSectorRadial*) module3->GetSectorByIndex(0); //First sector CbmMuchSectorRadial* sLast = (CbmMuchSectorRadial*) module3->GetSectorByIndex(module3->GetNSectors()-1); //Last sector Double_t rMin = sFirst->GetR1(); //Mimimum radius of the Sector Double_t rMax = sLast->GetR2(); //Maximum radius of the Sector for (Int_t i=0;iRndm(); Double_t driftTime = -1; while(driftTime < 0) driftTime = (1-aL)*fTotalDriftTime + gRandom->Gaus(0, fDTime); //Finding drifttime with random factor of Detector Time Resolution TVector3 ve = v1 + dv*aL; UInt_t ne = GasGain(); //Number of secondary electrons Double_t r = ve.Perp(); // Double_t phi = ve.Phi(); Double_t r1 = r-fSpotRadius; Double_t r2 = r+fSpotRadius; Double_t phi1 = phi-fSpotRadius/r; Double_t phi2 = phi+fSpotRadius/r; if (r1rMin) {//Adding charge to the pad which is on Lower Boundary AddCharge(sFirst,UInt_t(ne*(r2-rMin)/(r2-r1)),iPoint,time,driftTime,phi1,phi2); continue; } if (r1rMax) {//Adding charge to the pad which is on Upper Boundary AddCharge(sLast,UInt_t(ne*(rMax-r1)/(r2-r1)),iPoint,time,driftTime,phi1,phi2); continue; } CbmMuchSectorRadial* s1 = module3->GetSectorByRadius(r1); CbmMuchSectorRadial* s2 = module3->GetSectorByRadius(r2); if (s1==s2) AddCharge(s1,ne,iPoint,time,driftTime,phi1,phi2); else {//Adding praportionate charge to both the pad AddCharge(s1,UInt_t(ne*(s1->GetR2()-r1)/(r2-r1)),iPoint,time,driftTime,phi1,phi2); AddCharge(s2,UInt_t(ne*(r2-s2->GetR1())/(r2-r1)),iPoint,time,driftTime,phi1,phi2); } } } return kTRUE; } // ------------------------------------------------------------------------- // ------------------------------------------------------------------------- Int_t CbmMuchDigitizeGem::GasGain() { Double_t gasGain = -fMeanGasGain * TMath::Log(1 - gRandom->Rndm()); if (gasGain < 0.) gasGain = 1e6; return (Int_t) gasGain; } // ------------------------------------------------------------------------- // ------------------------------------------------------------------------- Double_t CbmMuchDigitizeGem::Sigma_n_e(Double_t Tkin, Double_t mass) { Double_t logT; if (mass < 0.1) { logT = log(Tkin * 0.511 / mass); if (logT > 9.21034) logT = 9.21034; if (logT < min_logT_e) logT = min_logT_e; return fSigma[0]->Eval(logT); } else if (mass >= 0.1 && mass < 0.2) { logT = log(Tkin * 105.658 / mass); if (logT > 9.21034) logT = 9.21034; if (logT < min_logT_mu) logT = min_logT_mu; return fSigma[1]->Eval(logT); } else { logT = log(Tkin * 938.272 / mass); if (logT > 9.21034) logT = 9.21034; if (logT < min_logT_p) logT = min_logT_p; return fSigma[2]->Eval(logT); } } // ------------------------------------------------------------------------- // ------------------------------------------------------------------------- Double_t CbmMuchDigitizeGem::MPV_n_e(Double_t Tkin, Double_t mass) { Double_t logT; if (mass < 0.1) { logT = log(Tkin * 0.511 / mass); if (logT > 9.21034) logT = 9.21034; if (logT < min_logT_e) logT = min_logT_e; return fMPV[0]->Eval(logT); } else if (mass >= 0.1 && mass < 0.2) { logT = log(Tkin * 105.658 / mass); if (logT > 9.21034) logT = 9.21034; if (logT < min_logT_mu) logT = min_logT_mu; return fMPV[1]->Eval(logT); } else { logT = log(Tkin * 938.272 / mass); if (logT > 9.21034) logT = 9.21034; if (logT < min_logT_p) logT = min_logT_p; return fMPV[2]->Eval(logT); } } // ------------------------------------------------------------------------- // ------------------------------------------------------------------------- Double_t CbmMuchDigitizeGem::GetNPrimaryElectronsPerCm(const CbmMuchPoint* point){ Int_t trackId = point->GetTrackID(); Int_t eventId = point->GetEventID(); if (trackId < 0) return -1; /* Commented out on request of A. Senger from 22.01.2014 if (fDaq && eventId!=FairRootManager::Instance()->GetInTree()->GetBranch("MCTrack")->GetReadEntry()) FairRootManager::Instance()->GetInTree()->GetBranch("MCTrack")->GetEntry(eventId); */ CbmMCTrack* mcTrack = (CbmMCTrack*) fMCTracks->At(trackId); if (!mcTrack) return -1; Int_t pdgCode = mcTrack->GetPdgCode(); TParticlePDG *particle = TDatabasePDG::Instance()->GetParticle(pdgCode); // Assign proton hypothesis for unknown particles if (!particle) particle = TDatabasePDG::Instance()->GetParticle(2212); if (TMath::Abs(particle->Charge()) < 0.1) return -1; Double_t m = particle->Mass(); TLorentzVector p; p.SetXYZM(point->GetPx(),point->GetPy(),point->GetPz(),m); Double_t Tkin = p.E()-m; // kinetic energy of the particle Double_t sigma = CbmMuchDigitizeGem::Sigma_n_e(Tkin,m); // sigma for Landau distribution Double_t mpv = CbmMuchDigitizeGem::MPV_n_e(Tkin,m); // most probable value for Landau distr. Double_t n = gRandom->Landau(mpv, sigma); while (n > 5e4) n = gRandom->Landau(mpv, sigma); // restrict Landau tail to increase performance return m<0.1 ? n/l_e : n/l_not_e; } // ------------------------------------------------------------------------- // ------------------------------------------------------------------------- Bool_t CbmMuchDigitizeGem::AddCharge(CbmMuchSectorRadial* s,UInt_t ne, Int_t iPoint, Double_t time, Double_t driftTime, Double_t phi1, Double_t phi2){ CbmMuchPadRadial* pad1 = s->GetPadByPhi(phi1); CbmMuchPadRadial* pad2 = s->GetPadByPhi(phi2); if (pad1==pad2) AddCharge(pad1,ne,iPoint,time,driftTime); else { Double_t phi = pad1 ? pad1->GetPhi2() : pad2 ? pad2->GetPhi1() : 0; UInt_t pad1_ne = UInt_t(ne*(phi-phi1)/(phi2-phi1)); AddCharge(pad1,pad1_ne ,iPoint,time,driftTime); AddCharge(pad2,ne-pad1_ne,iPoint,time,driftTime); } return kFALSE; } // ------------------------------------------------------------------------- // ------------------------------------------------------------------------- void CbmMuchDigitizeGem::AddCharge(CbmMuchPad* pad, UInt_t charge, Int_t iPoint, Double_t time, Double_t driftTime){ if (!pad) return; Double_t eventTime = 0.; if (fDaq) eventTime = FairRun::Instance()->GetEventHeader()->GetEventTime(); //LOG(DEBUG) << GetName() << ": Readout time is " << readoutTime << " ns"<< FairLogger::endl; UInt_t AbsTime = eventTime + time + driftTime; //Creating a new Signal, it will be deleted by CbmReadoutBuffer() CbmMuchSignal* signal = new CbmMuchSignal(pad->GetAddress()); signal->SetTimeStart(AbsTime); signal->SetTimeStop(AbsTime+fDeadTime); signal->MakeSignalShape(charge,fgDeltaResponse); signal->AddNoise(fMeanNoise); UInt_t address = pad->GetAddress(); Int_t inputNr = 0; Int_t eventNr = 0; GetEventInfo(inputNr, eventNr, eventTime); LOG(DEBUG) << GetName() << ": Processing event " << eventNr << " from input " << inputNr << " at t = " << eventTime << " ns with " << fPoints->GetEntriesFast() << " MuchPoints " << FairLogger::endl; //match->AddCharge(iPoint,charge,time+driftTime,fgDeltaResponse,time,eventNr,inputNr); CbmLink link(charge,iPoint,eventNr,inputNr); //std::cout<<"Before AddLink"<< endl; (signal->GetMatch())->AddLink(link); //std::cout<<"After AddLink"<< endl; //Adding all these temporary signal into the CbmMuchReadoutBuffer CbmMuchReadoutBuffer::Instance()->Fill(address, signal); LOG(DEBUG4)<<" Registered the CbmMuchSignal into the CbmMuchReadoutBuffer "<GetEventHeader(); assert ( event ); inputNr = event->GetInputFileId(); eventNr = event->GetMCEntryNumber(); eventTime = event->GetEventTime(); } // --- In a FairRunSim, the input number and event time are always zero; // --- only the event number is retrieved. else { if ( ! FairRunSim::Instance() ) LOG(FATAL) << GetName() << ": neither SIM nor ANA run." << FairLogger::endl; FairMCEventHeader* event = FairRunSim::Instance()->GetMCEventHeader(); assert ( event ); inputNr = 0; eventNr = event->GetEventID(); eventTime = 0.; } } // ------------------------------------------------------------------------- ClassImp(CbmMuchDigitizeGem)