from vertigo import RaveVertexFactory, EventFactory, LoopSettings, VertexMetaFactory_Instance
def printVertex ( vertex ):
print "pos=( %(x).4f, %(y).4f, %(z).4f ) chi2=%(chi2).1f ndf=%(n).1f" % \
{ "x": vertex.position().x(), "y": vertex.position().y(), "z": vertex.position().z(), \
"chi2": vertex.chiSquared(), "n": vertex.ndf() }
LoopSettings.Instance().setVerbosity(0)
ravefactory=VertexMetaFactory_Instance().get ("dummy" )
eventfactory=EventFactory ( "input3.txt" )
# methods = [ "linpt" ]
methods = [ "kalman", "avf-smoothing:0", "avf-smoothing:1", "avr-primcut:2.0-seccut:6.0-smoothing:0-weightthreshold:0.001", "avr-primcut:2.0-seccut:6.0-smoothing:0-weightthreshold:0.01", "tkf" ]
spotmethods= [ "avr", "tkf" ]
# methods = [ "linpt", "kalman", "avf-smoother:none", "avf-smoother:kalman", "mvf", "avr" ]
for event in eventfactory:
for method in methods:
print method+" no beamspot:"
vertices=ravefactory.create ( event.tracks(), method, False )
for vertex in vertices:
printVertex ( vertex )
for method in spotmethods:
print method+" beamspot:"
vertices=ravefactory.create ( event.tracks(), method, True )
for vertex in vertices:
printVertex ( vertex )
def printVertex ( vertex ):
print "pos=( %(x).4f, %(y).4f, %(z).4f ) chi2=%(chi2).3f ntrks=%(n)d" % \
{ "x": vertex.position().x(), "y": vertex.position().y(), "z": vertex.position().z(), \
"chi2": vertex.chiSquared(), "n": len (vertex.tracks()) }
eventfactory=EventFactory ( "input1.txt" )
# methods = [ "kalman", "avf", "mvf" ]
methods = [ "kalman", "avf-smoothing:0", "avf-smoothing:1", "avr", "tkf", "mvf" ]
methods = [ "avf" ]
vis=ObserverManager_Instance().get("VertexHistogrammer")
for event in eventfactory:
for method in methods:
print method+":"
vertices=ravefactory.create ( event.tracks(), method )
for vertex in vertices:
printVertex ( vertex )
event.add ( vertex )
simvertices=event.simVertices()
print simvertices.size()
event.associate ( simvertices[0], vertex )
#vis=ObserverManager_Instance().get("Visualiser")
vis.process ( event )
print "done."
ObserverManager_Instance().close()
##VertigoHarvest_close()
print "bs=",beamspot.isValid()
print "Has Beamspot?",ravefactory.hasBeamspot()
print "dzz",ravefactory.getBeamspot().covariance().dzz()
print "mz=",ravefactory.getMagneticField().inTesla ( RavePoint3D() ).z()
btagfactory=RaveFlavorTagFactory ( ravefactory )
set=LoopSettings.Instance()
set.setVerbosity(0)
eventfactory=EventFactory ( "bb.1.txt" )
converter=VertigoToCmsObjects()
for event in eventfactory:
primary=ravefactory.create ( event.tracks(), "avf", True )
jets=event.jets()
print len(jets),"jets. Primary vertex at ",primary[0].position()
for jet in jets:
# print "has flavor tagging: ", btagfactory.hasFlavorTagging()
# print jet.tracks().__class__
# print primary[0].__class__
# print jet.momentum().__class__
tag=btagfactory.tag ( jet.tracks(), primary[0], jet.momentum() )
print "FIXME: check track filter!"
print "Tagging jet: %(mc)s %(tag).4f jet axis: %(axis)s (pt %(pt).4f) %(sz)d tracks" % \
{ "mc": jet.getMCTag(), "tag": tag, "axis": jet.momentum(), "pt":jet.momentum().perp(), "sz" : len(jet.tracks() ) }
