def __init__(self, *args, **kwargs): super(PFSim, self).__init__(*args, **kwargs) self.detector = self.cfg_ana.detector self.simulator = Simulator(self.detector, self.mainLogger) self.simname = '_'.join( [self.instance_label, self.cfg_ana.sim_particles]) self.recname = '_'.join( [self.instance_label, self.cfg_ana.rec_particles]) self.is_display = self.cfg_ana.display if self.is_display: self.init_display()
# display.register(gtraj,1) def draw(self, projection): for traj in self: traj.draw(projection) if __name__ == '__main__': import math from heppy_fcc.fastsim.detectors.CMS import CMS from heppy_fcc.fastsim.simulator import Simulator from heppy_fcc.fastsim.vectors import Point from heppy_fcc.fastsim.toyevents import particles from heppy_fcc.display.core import Display from heppy_fcc.display.geometry import GDetector cms = CMS() simulator = Simulator(cms) particles = list( particles(5, 211, math.pi / 5., 4 * math.pi / 5., 10., 10., Point(0.5, 0.5, 0))) simulator.simulate(particles) display = Display() gcms = GDetector(cms) display.register(gcms, 0) gtrajectories = GTrajectories(particles) display.register(gtrajectories, 1) display.draw()