diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 16 |
1 files changed, 5 insertions, 11 deletions
@@ -12,6 +12,7 @@ from geometry import * from materials import * from camera import * from vector import * +import detectors import pygame @@ -40,15 +41,7 @@ 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/lionsolid.stl'), vacuum, vacuum) -#geometry = Geometry() -#geometry.add_solid(solid) -#geometry.build(bits=3) - -import detectors - -geometry = detectors.build_lbne() -#geometry = detectors.load_lbne('./detectors/lbne_8bit.pkl') +geometry = detectors.build_lbne(bits=6) mesh = geometry.mesh mesh = mesh.reshape(mesh.shape[0]*3,3) @@ -90,11 +83,12 @@ states = np.empty(width*height, dtype=np.int32) states_gpu = cuda.to_device(states) block_size = 64 -for i in range(10): +for i in range(100000): 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)) - translate(np.int32(origin.size), origin_gpu, gpuarray.vec.make_float3(-.5*np.sin(i*np.pi/100), -.5*np.cos(i*np.pi/100), 0), block=(block_size,1,1), grid=(width*height//block_size+1,1)) + #translate(np.int32(origin.size), origin_gpu, gpuarray.vec.make_float3(-np.sin(i*np.pi/100)/2, -np.cos(i*np.pi/100)/2, 0), block=(block_size,1,1), grid=(width*height//block_size+1,1)) t0 = time.time() intersect_mesh(np.int32(origin.size), origin_gpu, direction_gpu, pixels_gpu, first_leaf, states_gpu, block=(block_size,1,1), grid=(width*height//block_size+1,1), texrefs=[mesh_tex, lower_bound_tex, upper_bound_tex, child_map_tex, child_len_tex]) |