Public Member Functions
	__init__ (self, fout, makeClusterScifi)

	digitize (self)

	digitizeScifi (self)

	digitizeMuFilter (self)

	clusterScifi (self)

	finish (self)

Detailed Description

Definition at line 10 of file SndlhcDigi.py.

Constructor & Destructor Documentation

◆ init()

SndlhcDigi.SndlhcDigi.__init__	(	self,
		fout,
		makeClusterScifi
	)

Definition at line 12 of file SndlhcDigi.py.

    def __init__(self,fout, makeClusterScifi):
 
        self.iEvent = 0
 
        outdir=os.getcwd()
        outfile=outdir+"/"+fout
        self.fn = ROOT.TFile(fout,'update')
        self.sTree = self.fn.Get("cbmsim")
 
        # event header
        self.header  = ROOT.SNDLHCEventHeader()
        self.eventHeader  = self.sTree.Branch("EventHeader.",self.header,32000,1)
        #scifi
        self.digiScifi   = ROOT.TClonesArray("sndScifiHit")
        self.digiScifiBranch   = self.sTree.Branch("Digi_ScifiHits",self.digiScifi,32000,1)
        self.digiScifi2MCPoints =  ROOT.TClonesArray("Hit2MCPoints")
        self.digiScifi2MCPoints.BypassStreamer(ROOT.kTRUE)
        self.digiScifi2MCPointsBranch   = self.sTree.Branch("Digi_ScifiHits2MCPoints",self.digiScifi2MCPoints,32000,1)
        lsOfGlobals  = ROOT.gROOT.GetListOfGlobals()
        self.scifiDet = lsOfGlobals.FindObject('Scifi')
        self.mufiDet = lsOfGlobals.FindObject('MuFilter')
        if not self.scifiDet or not self.mufiDet: 
              exit('detector(s) not found, stop')
# make sipm to fibre mapping
        self.fibresSiPM = SciFiMapping.getFibre2SiPMCPP(self.scifiDet)
        self.siPMFibres = SciFiMapping.getSiPM2FibreCPP(self.scifiDet)
 
        #scifi clusters
        self.makeClusterScifi = makeClusterScifi
        if self.makeClusterScifi:
           self.clusScifi   = ROOT.TClonesArray("sndCluster")
           self.clusScifiBranch    = self.sTree.Branch("Cluster_Scifi",self.clusScifi,32000,1)
 
        #muonFilter
        self.digiMuFilter   = ROOT.TClonesArray("MuFilterHit")
        self.digiMuFilterBranch   = self.sTree.Branch("Digi_MuFilterHits",self.digiMuFilter,32000,1)
        self.digiMuFilter2MCPoints =  ROOT.TClonesArray("Hit2MCPoints")
        self.digiMuFilter2MCPoints.BypassStreamer(ROOT.kTRUE)
        self.digiMuFilter2MCPointsBranch   = self.sTree.Branch("Digi_MuFilterHits2MCPoints",self.digiMuFilter2MCPoints,32000,1)
 

Member Function Documentation

◆ clusterScifi()

SndlhcDigi.SndlhcDigi.clusterScifi ( self )

Definition at line 147 of file SndlhcDigi.py.

    def clusterScifi(self):
        index = 0
        hitDict = {}
        for k in range(self.sTree.Digi_ScifiHits.GetEntries()):
            d = self.sTree.Digi_ScifiHits[k]
            if not d.isValid(): continue 
            hitDict[d.GetDetectorID()] = k
        hitList = list(hitDict.keys())
        if len(hitList)>0:
              hitList.sort()
              tmp = [ hitList[0] ]
              cprev = hitList[0]
              ncl = 0
              last = len(hitList)-1
              hitlist = ROOT.std.vector("sndScifiHit*")()
              for i in range(len(hitList)):
                   if i==0 and len(hitList)>1: continue
                   c=hitList[i]
                   if (c-cprev)==1: 
                        tmp.append(c)
                   if (c-cprev)!=1 or c==hitList[last]:
                        first = tmp[0]
                        N = len(tmp)
                        hitlist.clear()
                        for aHit in tmp: hitlist.push_back( self.sTree.Digi_ScifiHits[hitDict[aHit]],)
                        aCluster = ROOT.sndCluster(first,N,hitlist,self.scifiDet)
                        if self.clusScifi.GetSize() == index: self.clusScifi.Expand(index+10)
                        aCluster_TCA = self.clusScifi.ConstructedAt(index)
                        ROOT.std.swap(aCluster, aCluster_TCA)
                        index+=1
                        if c!=hitList[last]:
                             ncl+=1
                             tmp = [c]
                   cprev = c
 
 

◆ digitize()

SndlhcDigi.SndlhcDigi.digitize ( self )

Definition at line 52 of file SndlhcDigi.py.

    def digitize(self):
 
        self.header.SetRunId( self.sTree.MCEventHeader.GetRunID() )
        self.header.SetEventNumber( self.sTree.MCEventHeader.GetEventID() )  # counts from 1
        self.header.SetBunchType(101);
        self.eventHeader.Fill()
        self.digiScifi.Clear('C')
        self.digiScifi2MCPoints.Clear('C')
        self.digitizeScifi()
        self.digiScifiBranch.Fill()
        self.digiScifi2MCPointsBranch.Fill()
        if self.makeClusterScifi:
           self.clusScifi.Clear('C')
           self.clusterScifi()
           self.clusScifiBranch.Fill()
 
        self.digiMuFilter.Clear('C')
        self.digiMuFilter2MCPoints.Clear('C')
        self.digitizeMuFilter()
        self.digiMuFilterBranch.Fill()
        self.digiMuFilter2MCPointsBranch.Fill()
 

◆ digitizeMuFilter()

SndlhcDigi.SndlhcDigi.digitizeMuFilter ( self )

Definition at line 114 of file SndlhcDigi.py.

    def digitizeMuFilter(self):
        hitContainer = {}
        mcLinks = ROOT.Hit2MCPoints()
        mcPoints = {}
        norm = {}
        k=-1
        for p in self.sTree.MuFilterPoint:
            k+=1
            # collect all hits in same detector element
            detID = p.GetDetectorID()
            if not detID in hitContainer: 
                  hitContainer[detID]=[]
                  mcPoints[detID] = {}
                  norm[detID] = 0
            hitContainer[detID].append(p)
            mcPoints[detID][k]=p.GetEnergyLoss()
            norm[detID]+= p.GetEnergyLoss()
        index = 0
        for detID in hitContainer:
            allPoints = ROOT.std.vector('MuFilterPoint*')()
            for p in hitContainer[detID]:
                 allPoints.push_back(p)
            aHit = ROOT.MuFilterHit(detID,allPoints)
            
            if self.digiMuFilter.GetSize() == index: self.digiMuFilter.Expand(index+100)
            aHit_TCA = self.digiMuFilter.ConstructedAt(index)
            ROOT.std.swap(aHit, aHit_TCA)
            index+=1
            for k in mcPoints[detID]:
                mcLinks.Add(detID,k, mcPoints[detID][k]/norm[detID])
        mcLinks_TCA = self.digiMuFilter2MCPoints.ConstructedAt(0)
        ROOT.std.swap(mcLinks, mcLinks_TCA)
 

◆ digitizeScifi()

SndlhcDigi.SndlhcDigi.digitizeScifi ( self )

Definition at line 74 of file SndlhcDigi.py.

    def digitizeScifi(self):
        hitContainer = {}
        mcLinks = ROOT.Hit2MCPoints()
        mcPoints = {}
        norm = {}
        k=-1
        for p in self.sTree.ScifiPoint:
            k+=1
            # collect all hits in same SiPM channel
            detID = p.GetDetectorID()
            locFibreID = detID%100000
            if not locFibreID in self.siPMFibres: continue # fibres in dead areas
            for sipmChan in self.siPMFibres[locFibreID]:
                  globsipmChan = int(detID/100000)*100000+sipmChan
                  if not globsipmChan in hitContainer: 
                       hitContainer[globsipmChan]=[]
                       mcPoints[globsipmChan] = {}
                       norm[globsipmChan] = 0
                  w = self.siPMFibres[locFibreID][sipmChan]['weight']
                  hitContainer[globsipmChan].append([p,w])
                  dE = p.GetEnergyLoss()*w
                  mcPoints[globsipmChan][k]=dE
                  norm[globsipmChan]+= dE
        index = 0
        for detID in hitContainer:
               allPoints = ROOT.std.vector('ScifiPoint*')()
               allWeights = ROOT.std.vector('Float_t')()
               for p in hitContainer[detID]:
                    allPoints.push_back(p[0])
                    allWeights.push_back(p[1])
               aHit = ROOT.sndScifiHit(detID,allPoints,allWeights)
               if self.digiScifi.GetSize() == index: self.digiScifi.Expand(index+100)
               aHit_TCA = self.digiScifi.ConstructedAt(index)
               ROOT.std.swap(aHit, aHit_TCA)
               index+=1
               for k in mcPoints[detID]:
                    mcLinks.Add(detID,k, mcPoints[detID][k]/norm[detID])
        mcLinks_TCA = self.digiScifi2MCPoints.ConstructedAt(0)
        ROOT.std.swap(mcLinks, mcLinks_TCA)
 

◆ finish()

SndlhcDigi.SndlhcDigi.finish ( self )

Definition at line 183 of file SndlhcDigi.py.

    def finish(self):
        print('finished writing tree')
        self.sTree.Write()

Member Data Documentation

◆ clusScifi

SndlhcDigi.SndlhcDigi.clusScifi

Definition at line 42 of file SndlhcDigi.py.

◆ clusScifiBranch

SndlhcDigi.SndlhcDigi.clusScifiBranch

Definition at line 43 of file SndlhcDigi.py.

◆ digiMuFilter

SndlhcDigi.SndlhcDigi.digiMuFilter

Definition at line 46 of file SndlhcDigi.py.

◆ digiMuFilter2MCPoints

SndlhcDigi.SndlhcDigi.digiMuFilter2MCPoints

Definition at line 48 of file SndlhcDigi.py.

◆ digiMuFilter2MCPointsBranch

SndlhcDigi.SndlhcDigi.digiMuFilter2MCPointsBranch

Definition at line 50 of file SndlhcDigi.py.

◆ digiMuFilterBranch

SndlhcDigi.SndlhcDigi.digiMuFilterBranch

Definition at line 47 of file SndlhcDigi.py.

◆ digiScifi

SndlhcDigi.SndlhcDigi.digiScifi

Definition at line 25 of file SndlhcDigi.py.

◆ digiScifi2MCPoints

SndlhcDigi.SndlhcDigi.digiScifi2MCPoints

Definition at line 27 of file SndlhcDigi.py.

◆ digiScifi2MCPointsBranch

SndlhcDigi.SndlhcDigi.digiScifi2MCPointsBranch

Definition at line 29 of file SndlhcDigi.py.

◆ digiScifiBranch

SndlhcDigi.SndlhcDigi.digiScifiBranch

Definition at line 26 of file SndlhcDigi.py.

◆ eventHeader

SndlhcDigi.SndlhcDigi.eventHeader

Definition at line 23 of file SndlhcDigi.py.

◆ fibresSiPM

SndlhcDigi.SndlhcDigi.fibresSiPM

Definition at line 36 of file SndlhcDigi.py.

◆ fn

SndlhcDigi.SndlhcDigi.fn

Definition at line 18 of file SndlhcDigi.py.

◆ header

SndlhcDigi.SndlhcDigi.header

Definition at line 22 of file SndlhcDigi.py.

◆ iEvent

SndlhcDigi.SndlhcDigi.iEvent

Definition at line 14 of file SndlhcDigi.py.

◆ makeClusterScifi

SndlhcDigi.SndlhcDigi.makeClusterScifi

Definition at line 40 of file SndlhcDigi.py.

◆ mufiDet

SndlhcDigi.SndlhcDigi.mufiDet

Definition at line 32 of file SndlhcDigi.py.

◆ scifiDet

SndlhcDigi.SndlhcDigi.scifiDet

Definition at line 31 of file SndlhcDigi.py.

◆ siPMFibres

SndlhcDigi.SndlhcDigi.siPMFibres

Definition at line 37 of file SndlhcDigi.py.

◆ sTree

SndlhcDigi.SndlhcDigi.sTree

Definition at line 19 of file SndlhcDigi.py.

The documentation for this class was generated from the following file:

python/SndlhcDigi.py

Public Attributes
	iEvent

	fn

	sTree

	header

	eventHeader

	digiScifi

	digiScifiBranch

	digiScifi2MCPoints

	digiScifi2MCPointsBranch

	scifiDet

	mufiDet

	fibresSiPM

	siPMFibres

	makeClusterScifi

	clusScifi

	clusScifiBranch

	digiMuFilter

	digiMuFilterBranch

	digiMuFilter2MCPoints

	digiMuFilter2MCPointsBranch

Public Member Functions

Public Attributes