Python Task示例

示例#1

文件： ShipAna.py 项目： mikeharrison12/FairShip

ROOT.genfit.MaterialEffects.getInstance().init(geoMat)
volDict = {}
i = 0
for x in ROOT.gGeoManager.GetListOfVolumes():
    volDict[i] = x.GetName()
    i += 1

bfield = ROOT.genfit.BellField(ShipGeo.Bfield.max, ShipGeo.Bfield.z, 2,
                               ShipGeo.Yheight / 2.)
fM = ROOT.genfit.FieldManager.getInstance()
fM.init(bfield)

# prepare veto decisions
import shipVeto

veto = shipVeto.Task()
vetoDets = {}

# fiducial cuts
vetoStation = ROOT.gGeoManager.GetTopVolume().GetNode('Veto_5')
vetoStation_zDown = vetoStation.GetMatrix().GetTranslation(
)[2] + vetoStation.GetVolume().GetShape().GetDZ()
T1Station = ROOT.gGeoManager.GetTopVolume().GetNode('Tr1_1')
T1Station_zUp = T1Station.GetMatrix().GetTranslation(
)[2] - T1Station.GetVolume().GetShape().GetDZ()

h = {}
ut.bookHist(h, 'delPOverP', 'delP / P', 400, 0., 200., 100, -0.5, 0.5)
ut.bookHist(h, 'pullPOverPx', 'delPx / sigma', 400, 0., 200., 100, -3., 3.)
ut.bookHist(h, 'pullPOverPy', 'delPy / sigma', 400, 0., 200., 100, -3., 3.)
ut.bookHist(h, 'pullPOverPz', 'delPz / sigma', 400, 0., 200., 100, -3., 3.)

示例#2

ROOT.genfit.MaterialEffects.getInstance().init(geoMat)
volDict = {}
i = 0
for x in ROOT.gGeoManager.GetListOfVolumes():
    volDict[i] = x.GetName()
    i += 1

bfield = ROOT.genfit.BellField(ShipGeo.Bfield.max, ShipGeo.Bfield.z, 2,
                               ShipGeo.Yheight / 2.)
fM = ROOT.genfit.FieldManager.getInstance()
fM.init(bfield)

# prepare veto decisions
import shipVeto

veto = shipVeto.Task(sTree)
vetoDets = {}

# fiducial cuts
vetoStation = ROOT.gGeoManager.GetTopVolume().GetNode('Veto_5')
vetoStation_zDown = vetoStation.GetMatrix().GetTranslation(
)[2] + vetoStation.GetVolume().GetShape().GetDZ()
T1Station = ROOT.gGeoManager.GetTopVolume().GetNode('Tr1_1')
T1Station_zUp = T1Station.GetMatrix().GetTranslation(
)[2] - T1Station.GetVolume().GetShape().GetDZ()

#DECLARING HISTOGRAMS
h = {}
ut.bookHist(h, 'delPOverP', 'delP / P', 400, 0., 200., 100, -0.5, 0.5)
ut.bookHist(h, 'pullPOverPx', 'delPx / sigma', 400, 0., 200., 100, -3., 3.)
ut.bookHist(h, 'pullPOverPy', 'delPy / sigma', 400, 0., 200., 100, -3., 3.)