added a detector folder to put detector geometries

1 files changed, 9 insertions, 5 deletions

diff --git a/test.py b/test.py
index a484032..157e5a8 100644
--- a/test.py
+++ b/test.py
@@ -39,10 +39,14 @@ origin, direction = make_vector(origin), make_vector(direction)
 origin_gpu = cuda.to_device(origin)
 direction_gpu = cuda.to_device(direction)
 
-solid = Solid(read_stl('models/tie_interceptor6.stl'), vacuum, vacuum)
-geometry = Geometry()
-geometry.add_solid(solid)
-geometry.build()
+#solid = Solid(read_stl('models/lionsolid.stl'), vacuum, vacuum)
+#geometry = Geometry()
+#geometry.add_solid(solid)
+#geometry.build(bits=3)
+
+import detectors
+
+geometry = detectors.build_lbne()
 
 mesh = geometry.mesh
 mesh = mesh.reshape(mesh.shape[0]*3,3)
@@ -84,7 +88,7 @@ states = np.empty(width*height, dtype=np.int32)
 states_gpu = cuda.to_device(states)
 
 block_size = 64
-for i in range(100):
+for i in range(10000):
     rotate(np.int32(origin.size), origin_gpu, np.float32(np.pi/100), gpuarray.vec.make_float3(0,0,1), block=(block_size,1,1), grid=(width*height//block_size+1,1))
     rotate(np.int32(direction.size), direction_gpu, np.float32(np.pi/100), gpuarray.vec.make_float3(0,0,1), block=(block_size,1,1), grid=(width*height//block_size+1,1))