diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 14 |
1 files changed, 9 insertions, 5 deletions
@@ -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)) |