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()
