interactive_python.py

#!/usr/bin/env python3
 

 

 
from basf2 import Module, Path, process, B2INFO
import interactive
from ROOT import Belle2
 
 
class MinModule(Module):
    """
    Example module to drop into ipython and create some objects to look at.
    If you just want to start IPython and create PyStoreArray etc.
    interactively in your own steering file, the 'Interactive' module
    might be of interest.
    """
 
    def event(self):
        """
        reimplement Module::event()
        """
        evtmetadata = Belle2.PyStoreObj("EventMetaData")  # noqa
        particles = Belle2.PyStoreArray("MCParticles")  # noqa
        B2INFO(80 * '=')
        B2INFO("Dropping into interactive python shell. Try:"
               "\n  print evtmetadata.obj().getEvent()"
               "\n  particles[0].Dump()"
               "\n  help(particles[0])"
               "\nTo continue non-interactively (until next event), press Ctrl+D."
               "\nPress Ctrl+C followed by Ctrl+D to exit basf2.")
        B2INFO(80 * '=')
        interactive.embed()
 
 
main = Path()
main.add_module('EventInfoSetter', evtNumList=[2])
main.add_module('ParticleGun')
main.add_module(MinModule())
process(main)