eventDisplay.DrawTracks Class Reference

Inheritance diagram for eventDisplay.DrawTracks:

Collaboration diagram for eventDisplay.DrawTracks:

Public Member Functions
None	InitTask (self)
None	FinishEvent (self)
None	ExecuteTask (self, str option="")
None	DrawParticle (self, n)
None	DrawMCTrack (self, int n)
None	DrawMCTracks (self, str option="")
None	DrawFittedTracks (self, str option="")

Public Attributes
	comp
	trackColors
	z_end
	z_start
	z_mag
	niter
	dz
	parallelToZ
	evscene
	Targetz

Detailed Description

Definition at line 274 of file eventDisplay.py.

Member Function Documentation

DrawFittedTracks()

None eventDisplay.DrawTracks.DrawFittedTracks	(		self,
		str	option = ""
	)

Definition at line 485 of file eventDisplay.py.

    def DrawFittedTracks(self, option: str = "") -> None:
        n, ntot = -1, 0
        for fT in sTree.FitTracks:
            n += 1
            fst = fT.getFitStatus()
            if not fst.isFitConverged():
                continue
            if fst.getNdf() < 20:
                continue
            DTrack = ROOT.TEveLine()
            DTrack.SetPickable(ROOT.kTRUE)
            DTrack.SetTitle(fT.__repr__())
            fstate = fT.getFittedState(0)
            fPos = fstate.getPos()
            fMom = fstate.getMom()
            pos = fPos
            mom = fMom
            pid = fstate.getPDG()
            zs = self.z_start
            for i in range(self.niter):
                rc, newpos, _newmom = TrackExtrapolateTool.extrapolateToPlane(fT, zs)
                if rc:
                    DTrack.SetNextPoint(newpos.X(), newpos.Y(), newpos.Z())
                else:
                    print("error with extrapolation: z=", zs)
                    # use linear extrapolation
                    px, py, pz = mom.X(), mom.Y(), mom.Z()
                    lam = (zs - pos.Z()) / pz
                    DTrack.SetNextPoint(pos.X() + lam * px, pos.Y() + lam * py, zs)
                zs += self.dz
            DTrack.SetName("FitTrack_" + str(n))
            c = ROOT.kWhite
            if abs(pid) in self.trackColors:
                c = self.trackColors[abs(pid)]
            DTrack.SetMainColor(c)
            DTrack.SetLineWidth(3)
            self.comp.AddElement(DTrack)
            ntot += 1
        print("draw ", ntot, " fitted tracks")
        n = -1
        for aP in sTree.Particles:
            n += 1
            # check fitted tracks
            tracksOK = True
            if aP.GetMother(1) == 99:  # DOCA is set
                if aP.T() > 3 * u.cm:
                    continue
            for k in range(aP.GetNDaughters()):
                if k > 1:
                    break  # we don't have more than 2tracks/vertex yet, no idea why ROOT sometimes comes up with 4!
                fT = sTree.FitTracks[aP.GetDaughter(k)]
                fst = fT.getFitStatus()
                if not fst.isFitConverged():
                    tracksOK = False
                if fst.getNdf() < 20:
                    tracksOK = False
            if not tracksOK:
                continue
            DTrack = ROOT.TEveLine()
            DTrack.SetPickable(ROOT.kTRUE)
            DTrack.SetMainColor(ROOT.kCyan)
            DTrack.SetLineWidth(4)
            DTrack.SetName("Particle_" + str(n))
            DTrack.SetTitle(aP.__repr__())
            DTrack.SetNextPoint(aP.Vx(), aP.Vy(), aP.Vz())
            lam = (self.Targetz - aP.Vz()) / aP.Pz()
            DTrack.SetNextPoint(aP.Vx() + lam * aP.Px(), aP.Vy() + lam * aP.Py(), self.Targetz)
            self.comp.AddElement(DTrack)
 
 
#

DrawMCTrack()

None eventDisplay.DrawTracks.DrawMCTrack	(		self,
		int	n
	)

Definition at line 356 of file eventDisplay.py.

    def DrawMCTrack(self, n: int) -> None:
        self.comp.OpenCompound()
        fT = sTree.MCTrack[n]
        DTrack = ROOT.TEveLine()
        DTrack.SetPickable(ROOT.kTRUE)
        DTrack.SetTitle(fT.__repr__())
        p = pdg.GetParticle(fT.GetPdgCode())
        if p:
            pName = p.GetName()
        else:
            pName = str(fT.GetPdgCode())
        DTrack.SetName("MCTrck_" + str(n) + "_" + pName)
        fPos = ROOT.TVector3()
        fMom = ROOT.TVector3()
        fT.GetStartVertex(fPos)
        fT.GetMomentum(fMom)
        # check for end vertex
        evVx = False
        for da in sTree.MCTrack:
            if da.GetMotherId() == n:
                evVx = True
                break
        DTrack.SetNextPoint(fPos.X(), fPos.Y(), fPos.Z())
        if evVx and abs(da.GetStartZ() - fPos.Z()) > 1 * u.cm:
            DTrack.SetNextPoint(da.GetStartX(), da.GetStartY(), da.GetStartZ())
        else:
            zEx = 10 * u.m
            if evVx:
                zEx = -10 * u.m
            lam = (zEx + fPos.Z()) / fMom.Z()
            DTrack.SetNextPoint(fPos.X() + lam * fMom.X(), fPos.Y() + lam * fMom.Y(), zEx + fPos.Z())
        c = ROOT.kYellow
        DTrack.SetMainColor(c)
        DTrack.SetLineWidth(3)
        self.comp.AddElement(DTrack)
        self.comp.CloseCompound()
        gEve.ElementChanged(self.evscene, True, True)
 

DrawMCTracks()

None eventDisplay.DrawTracks.DrawMCTracks	(		self,
		str	option = ""
	)

Definition at line 394 of file eventDisplay.py.

    def DrawMCTracks(self, option: str = "") -> None:
        n = -1
        ntot = 0
        fPos = ROOT.TVector3()
        fMom = ROOT.TVector3()
        for fT in sTree.MCTrack:
            n += 1
            DTrack = ROOT.TEveLine()
            DTrack.SetPickable(ROOT.kTRUE)
            DTrack.SetTitle(fT.__repr__())
            fT.GetStartVertex(fPos)
            hitlist = {}
            hitlist[fPos.Z()] = [fPos.X(), fPos.Y()]
            # look for HNL
            if abs(fT.GetPdgCode()) == options.HiddenParticleID:
                for da in sTree.MCTrack:
                    if da.GetMotherId() == n:
                        break
                # end vertex of HNL
                da.GetStartVertex(fPos)
                hitlist[fPos.Z()] = [fPos.X(), fPos.Y()]
            # loop over all sensitive volumes to find hits
            for P in [
                "vetoPoint",
                "strawtubesPoint",
                "ShipRpcPoint",
                "TargetPoint",
                "MTCDetPoint",
                "SiliconTargetPoint",
                "TimeDetPoint",
            ]:
                if not sTree.GetBranch(P):
                    continue
                c = eval("sTree." + P)
                for p in c:
                    if p.GetTrackID() == n:
                        if hasattr(p, "LastPoint"):
                            lp = p.LastPoint()
                            if lp.x() == lp.y() and lp.x() == lp.z() and lp.x() == 0:
                                # must be old data, don't expect hit at 0,0,0
                                hitlist[p.GetZ()] = [p.GetX(), p.GetY()]
                            else:
                                hitlist[lp.z()] = [lp.x(), lp.y()]
                                hitlist[2.0 * p.GetZ() - lp.z()] = [
                                    2.0 * p.GetX() - lp.x(),
                                    2.0 * p.GetY() - lp.y(),
                                ]
                        else:
                            hitlist[p.GetZ()] = [p.GetX(), p.GetY()]
            if len(hitlist) == 1:
                if fT.GetMotherId() < 0:
                    continue
                if abs(sTree.MCTrack[fT.GetMotherId()].GetPdgCode()) == options.HiddenParticleID:
                    # still would like to draw track stub
                    # check for end vertex
                    evVx = False
                    for da in sTree.MCTrack:
                        if da.GetMotherId() == n:
                            evVx = True
                            break
                    if evVx:
                        hitlist[da.GetStartZ()] = [da.GetStartX(), da.GetStartY()]
                    else:
                        zEx = 10 * u.m
                        fT.GetMomentum(fMom)
                        lam = (zEx + fPos.Z()) / fMom.Z()
                        hitlist[zEx + fPos.Z()] = [
                            fPos.X() + lam * fMom.X(),
                            fPos.Y() + lam * fMom.Y(),
                        ]
            # sort in z
            lz = list(hitlist.keys())
            if len(lz) > 1:
                lz.sort()
                for z in lz:
                    DTrack.SetNextPoint(hitlist[z][0], hitlist[z][1], z)
                p = pdg.GetParticle(fT.GetPdgCode())
                if p:
                    pName = p.GetName()
                else:
                    pName = str(fT.GetPdgCode())
                DTrack.SetName("MCTrack_" + str(n) + "_" + pName)
                c = ROOT.kYellow
                if abs(fT.GetPdgCode()) == options.HiddenParticleID:
                    c = ROOT.kMagenta
                DTrack.SetMainColor(c)
                DTrack.SetLineWidth(3)
                self.comp.AddElement(DTrack)
                ntot += 1
        print("draw ", ntot, " MC tracks")
 

DrawParticle()

None eventDisplay.DrawTracks.DrawParticle	(		self,
			n
	)

Definition at line 342 of file eventDisplay.py.

    def DrawParticle(self, n) -> None:
        self.comp.OpenCompound()
        DTrack = ROOT.TEveLine()
        DTrack.SetPickable(ROOT.kTRUE)
        DTrack.SetMainColor(ROOT.kCyan)
        DTrack.SetLineWidth(4)
        aP = sTree.Particles[n]
        DTrack.SetTitle(aP.__repr__())
        DTrack.SetName("Prtcle_" + str(n))
        DTrack.SetNextPoint(aP.Vx(), aP.Vy(), aP.Vz())
        lam = (self.Targetz - aP.Vz()) / aP.Pz()
        DTrack.SetNextPoint(aP.Vx() + lam * aP.Px(), aP.Vy() + lam * aP.Py(), self.Targetz)
        self.comp.AddElement(DTrack)
 

ExecuteTask()

None eventDisplay.DrawTracks.ExecuteTask	(		self,
		str	option = ""
	)

Definition at line 329 of file eventDisplay.py.

    def ExecuteTask(self, option: str = "") -> None:
        self.comp.DestroyElements()
        self.comp.OpenCompound()
        if (sTree.FindBranch("FitTracks") or sTree.FindBranch("FitTracks_PR")) and len(sTree.FitTracks) > 0:
            self.DrawFittedTracks()
        if not sTree.FindBranch("GeoTracks") and len(sTree.MCTrack) > 0 and globals()["withMCTracks"]:
            if top.GetNode("Tunnel_1"):
                DrawSimpleMCTracks()  # for sndlhc, until more details are simulated
            else:
                self.DrawMCTracks()
        self.comp.CloseCompound()
        gEve.ElementChanged(self.evscene, True, True)
 

FinishEvent()

None eventDisplay.DrawTracks.FinishEvent

(

self

)

Definition at line 326 of file eventDisplay.py.

    def FinishEvent(self) -> None:
        pass
 

InitTask()

None eventDisplay.DrawTracks.InitTask

(

self

)

Definition at line 277 of file eventDisplay.py.

    def InitTask(self) -> None:
        # prepare container for fitted tracks
        self.comp = ROOT.TEveCompound("Tracks")
        gEve.AddElement(self.comp)
        self.trackColors = {
            13: ROOT.kGreen,
            211: ROOT.kRed,
            11: ROOT.kOrange,
            321: ROOT.kMagenta,
        }
        dv = top.GetNode("DecayVolume_1")
        self.z_end = 500.0
        if dv:
            ns = dv.GetNodes()
            try:
                T1Lid = ns.FindObject("T1Lid_1").GetMatrix()
                self.z_start = T1Lid.GetTranslation()[2]
            except AttributeError:
                self.z_start = 0
        else:
            self.z_start = 0
        muonDet = top.GetNode("MuonDetector_1")
        if muonDet:
            self.z_end = muonDet.GetMatrix().GetTranslation()[2] + muonDet.GetVolume().GetShape().GetDZ()
        elif hasattr(ShipGeo, "MuonStation3"):
            self.z_end = ShipGeo["MuonStation3"].z
        elif top.GetNode("VMuonBox_1"):
            xx = top.GetNode("VMuonBox_1")
            self.z_end = xx.GetMatrix().GetTranslation()[2] + xx.GetVolume().GetShape().GetDZ()
        magNode = top.GetNode("MCoil_1")
        if magNode:
            self.z_mag = magNode.GetMatrix().GetTranslation()[2]
        elif hasattr(ShipGeo, "Bfield"):
            self.z_mag = ShipGeo["Bfield"].z
        else:
            self.z_mag = 0
        self.niter = 100
        self.dz = (self.z_end - self.z_start) / float(self.niter)
        self.parallelToZ = ROOT.TVector3(0.0, 0.0, 1.0)
        sc = gEve.GetScenes()
        self.evscene = sc.FindChild("Event scene")
        targetNode = top.GetNode("TargetArea_1")
        if targetNode:
            self.Targetz = targetNode.GetMatrix().GetTranslation()[2]
        elif hasattr(ShipGeo, "target"):
            self.Targetz = ShipGeo["target"].z0
        else:
            self.Targetz = 0
 

Member Data Documentation

comp

eventDisplay.DrawTracks.comp

Definition at line 279 of file eventDisplay.py.

dz

eventDisplay.DrawTracks.dz

Definition at line 314 of file eventDisplay.py.

evscene

eventDisplay.DrawTracks.evscene

Definition at line 317 of file eventDisplay.py.

niter

eventDisplay.DrawTracks.niter

Definition at line 313 of file eventDisplay.py.

parallelToZ

eventDisplay.DrawTracks.parallelToZ

Definition at line 315 of file eventDisplay.py.

Targetz

eventDisplay.DrawTracks.Targetz

Definition at line 320 of file eventDisplay.py.

trackColors

eventDisplay.DrawTracks.trackColors

Definition at line 281 of file eventDisplay.py.

z_end

eventDisplay.DrawTracks.z_end

Definition at line 288 of file eventDisplay.py.

z_mag

eventDisplay.DrawTracks.z_mag

Definition at line 308 of file eventDisplay.py.

z_start

eventDisplay.DrawTracks.z_start

Definition at line 293 of file eventDisplay.py.

---

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

• macro/eventDisplay.py