dumpEvent.py

Go to the documentation of this file.

# SPDX-License-Identifier: LGPL-3.0-or-later
# SPDX-FileCopyrightText: Copyright CERN for the benefit of the SHiP Collaboration
 
# example for dumping an MC event
 
import ROOT
import ShipGeoConfig
import shipunit as u
 
ship_geo = ShipGeoConfig.Config().loadpy("$FAIRSHIP/geometry/geometry_config.py")
PDG = ROOT.TDatabasePDG.Instance()
 
 
def printMCTrack(n: int, MCTrack) -> None:
    mcp = MCTrack[n]
    print(
        " %6i %7i %6.3F %6.3F %7.3F %7.3F %7.3F %7.3F %6i "
        % (
            n,
            mcp.GetPdgCode(),
            mcp.GetPx() / u.GeV,
            mcp.GetPy() / u.GeV,
            mcp.GetPz() / u.GeV,
            mcp.GetStartX() / u.m,
            mcp.GetStartY() / u.m,
            mcp.GetStartZ() / u.m,
            mcp.GetMotherId(),
        )
    )
 
 
def dump(i, pcut) -> None:
    tree = ROOT.gROOT.FindObjectAny("cbmsim")
    tree.GetEntry(i)
    print("   #         pid   px    py      pz     vx      vy       vz      mid")
    n = -1
    for mcp in tree.MCTrack:
        n += 1
        if mcp.GetP() / u.GeV < pcut:
            continue
        printMCTrack(n, tree.MCTrack)
 
 
def dumpStraw(i) -> None:
    tree = ROOT.gROOT.FindObjectAny("cbmsim")
    tree.GetEntry(i)
    for aStraw in tree.strawtubesPoint:
        trID = aStraw.GetTrackID()
        if not trID < 0:
            printMCTrack(trID, tree.MCTrack)