From 084dfd08b714faefaea77cb7dc04d2e93dc04b1d Mon Sep 17 00:00:00 2001 From: Stan Seibert Date: Fri, 16 Sep 2011 14:27:46 -0400 Subject: File reorganization to move toward standard python package layout --- event.py | 76 ---------------------------------------------------------------- 1 file changed, 76 deletions(-) delete mode 100644 event.py (limited to 'event.py') diff --git a/event.py b/event.py deleted file mode 100644 index a2b99cc..0000000 --- a/event.py +++ /dev/null @@ -1,76 +0,0 @@ -import numpy as np - -class Vertex(object): - def __init__(self, particle_name, pos, dir, ke, t0=0.0, pol=None): - self.particle_name = particle_name - self.pos = pos - self.dir = dir - self.pol = pol - self.ke = ke - self.t0 = t0 - -class Photons(object): - def __init__(self, pos, dir, pol, wavelengths, t=None, last_hit_triangles=None, flags=None): - self.pos = pos - self.dir = dir - self.pol = pol - self.wavelengths = wavelengths - - if t is None: - self.t = np.zeros(len(pos), dtype=np.float32) - else: - self.t = t - - if last_hit_triangles is None: - self.last_hit_triangles = np.empty(len(pos), dtype=np.int32) - self.last_hit_triangles.fill(-1) - else: - self.last_hit_triangles = last_hit_triangles - - if flags is None: - self.flags = np.zeros(len(pos), dtype=np.uint32) - else: - self.flags = flags - - def __add__(self, other): - pos = np.concatenate((self.pos, other.pos)) - dir = np.concatenate((self.dir, other.dir)) - pol = np.concatenate((self.pol, other.pol)) - wavelengths = np.concatenate((self.wavelengths, other.wavelengths)) - t = np.concatenate((self.t, other.t)) - last_hit_triangles = np.concatenate((self.last_hit_triangles, other.last_hit_triangles)) - flags = np.concatenate((self.flags, other.flags)) - return Photons(pos, dir, pol, wavelengths, t, last_hit_triangles, flags) - - def __len__(self): - return len(self.pos) - -class Channels(object): - def __init__(self, hit, t, q, flags=None): - self.hit = hit - self.t = t - self.q = q - self.flags = flags - - def hit_channels(self): - return self.hit.nonzero(), self.t[self.hit], self.q[self.hit] - -class Event(object): - def __init__(self, id=0, primary_vertex=None, vertices=None, photons_beg=None, photons_end=None, channels=None): - self.id = id - - self.nphotons = None - - self.primary_vertex = primary_vertex - - if vertices is not None: - if np.iterable(vertices): - self.vertices = vertices - else: - self.vertices = [vertices] - else: - self.vertices = [] - - self.photons_beg = photons_beg - self.photons_end = photons_end - self.channels = channels -- cgit