summaryrefslogtreecommitdiff
path: root/fileio/root.py
diff options
context:
space:
mode:
authorAnthony LaTorre <tlatorre9@gmail.com>2011-08-16 20:25:17 -0400
committerAnthony LaTorre <tlatorre9@gmail.com>2011-08-16 20:25:17 -0400
commit1476f921813e60cf3749a5d03b9ed5cbf1951db6 (patch)
tree605d0d68418e0df4d3674da388fe47d2078ed836 /fileio/root.py
parentea3a50431499633b3b367427d63db5507df7dd2e (diff)
parentcc478cf132d554642345ada31d3f801f004bf184 (diff)
downloadchroma-1476f921813e60cf3749a5d03b9ed5cbf1951db6.tar.gz
chroma-1476f921813e60cf3749a5d03b9ed5cbf1951db6.tar.bz2
chroma-1476f921813e60cf3749a5d03b9ed5cbf1951db6.zip
merge
Diffstat (limited to 'fileio/root.py')
-rw-r--r--fileio/root.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/fileio/root.py b/fileio/root.py
index 5bebda0..77c23ff 100644
--- a/fileio/root.py
+++ b/fileio/root.py
@@ -20,9 +20,10 @@ class RootWriter(object):
self.ev.event_id = pyev.event_id
self.ev.mc.particle = pyev.particle_name
- self.ev.mc.gen_pos.SetXYZ(*pyev.gen_position)
- self.ev.mc.gen_dir.SetXYZ(*pyev.gen_direction)
+ self.ev.mc.gen_position.SetXYZ(*pyev.gen_position)
+ self.ev.mc.gen_direction.SetXYZ(*pyev.gen_direction)
self.ev.mc.gen_total_energy = pyev.gen_total_energy
+ self.ev.mc.nphoton = pyev.nphoton
if pyev.photon_start is not None:
photons = pyev.photon_start
@@ -46,8 +47,8 @@ class RootWriter(object):
self.ev.mc.subtrack.resize(len(pyev.subtracks))
for i, subtrack in enumerate(pyev.subtracks):
self.ev.mc.subtrack[i].name = subtrack.particle_name
- self.ev.mc.subtrack[i].pos.SetXYZ(*subtrack.position)
- self.ev.mc.subtrack[i].dir.SetXYZ(*subtrack.direction)
+ self.ev.mc.subtrack[i].position.SetXYZ(*subtrack.position)
+ self.ev.mc.subtrack[i].direction.SetXYZ(*subtrack.direction)
self.ev.mc.subtrack[i].start_time = subtrack.start_time
self.ev.mc.subtrack[i].total_energy = subtrack.total_energy