added stl mesh viewer

1 files changed, 2 insertions, 1 deletions

diff --git a/src/kernel.cu b/src/kernel.cu
index d8f2300..c2b3fb2 100644
--- a/src/kernel.cu
+++ b/src/kernel.cu
@@ -163,7 +163,7 @@ __global__ void rotate(int max_idx, float3 *pt, float phi, float3 axis)
 	pt[idx] = rotate(pt[idx], phi, axis);
 }
 
-__global__ void intersect_mesh(int max_idx, float3 *origin_arr, float3 *direction_arr, int *pixel_arr, int first_leaf, int *state_arr)
+__global__ void intersect_mesh(int max_idx, float3 *origin_arr, float3 *direction_arr, int first_leaf, int *state_arr, int *pixel_arr)
 {
 	int idx = blockIdx.x*blockDim.x + threadIdx.x;
 
@@ -172,6 +172,7 @@ __global__ void intersect_mesh(int max_idx, float3 *origin_arr, float3 *directio
 
 	float3 origin = origin_arr[idx];
 	float3 direction = direction_arr[idx];
+	direction /= norm(direction);
 
 	int *pixel = pixel_arr+idx;
 	int *state = state_arr+idx;