Updated read_source script.

This commit is contained in:
Paul Romano 2012-07-07 10:04:08 -04:00
parent 4dd159eca2
commit bf708e24ed

View file

@ -15,16 +15,18 @@ class SourceFile(object):
for i in range(self.n_sites):
# Read position, angle, and energy
uid = struct.unpack('q', self.f.read(8))
(uid,) = struct.unpack('q', self.f.read(8))
(weight,) = struct.unpack('d', self.f.read(8))
xyz = struct.unpack('3d', self.f.read(24))
uvw = struct.unpack('3d', self.f.read(24))
E = struct.unpack('d', self.f.read(8))
(E,) = struct.unpack('d', self.f.read(8))
# Create source site and append to list
self.sites.append(SourceSite(xyz, uvw, E))
self.sites.append(SourceSite(weight, xyz, uvw, E))
class SourceSite(object):
def __init__(self, xyz, uvw, E):
def __init__(self, weight, xyz, uvw, E):
self.weight = weight
self.xyz = xyz
self.uvw = uvw
self.E = E