diff options
| -rwxr-xr-x | camera.py | 13 | ||||
| -rw-r--r-- | images/doomhud.png | bin | 0 -> 17911 bytes | |||
| -rw-r--r-- | src/photon.h | 6 |
3 files changed, 19 insertions, 0 deletions
@@ -79,6 +79,17 @@ class Camera(Thread): pygame.display.set_caption('') self.clock = pygame.time.Clock() + self.doom_mode = False + try: + if self.width == 640: # SECRET DOOM MODE! + print 'shotgun activated!' + self.doom_hud = pygame.image.load('images/doomhud.png').convert_alpha() + rect = self.doom_hud.get_rect() + self.doom_rect = rect.move(0, self.height - rect.height) + self.doom_mode = True + except: + pass + lower_bound, upper_bound = self.geometry.mesh.get_bounds() self.scale = np.linalg.norm(upper_bound-lower_bound) @@ -212,6 +223,8 @@ class Camera(Thread): self.kernels.ray_trace(np.int32(self.pixels_gpu.size), self.origins_gpu, self.directions_gpu, self.pixels_gpu, block=(self.nblocks,1,1), grid=(self.pixels_gpu.size//self.nblocks+1,1)) pygame.surfarray.blit_array(self.screen, self.pixels_gpu.get().reshape(self.size)) + if self.doom_mode: + self.screen.blit(self.doom_hud, self.doom_rect) pygame.display.flip() if self.movie: diff --git a/images/doomhud.png b/images/doomhud.png Binary files differnew file mode 100644 index 0000000..3c11b78 --- /dev/null +++ b/images/doomhud.png diff --git a/src/photon.h b/src/photon.h index fb29ffe..fa3b85b 100644 --- a/src/photon.h +++ b/src/photon.h @@ -1,6 +1,7 @@ #ifndef __PHOTON_H__ #define __PHOTON_H__ +#include "stdio.h" #include "linalg.h" #include "materials.h" #include "rotate.h" @@ -102,6 +103,9 @@ __device__ void fill_state(State &s, Photon &p) s.absorption_length = interp_property(p.wavelength, material1.absorption_length); s.scattering_length = interp_property(p.wavelength, material1.scattering_length); + printf("wavelength = %f", p.wavelength); + printf("scattering length = %f\n", s.scattering_length); + } // fill_state __device__ void rayleigh_scatter(Photon &p, curandState &rng) @@ -152,6 +156,8 @@ __device__ int propagate_to_boundary(Photon &p, State &s, curandState &rng) { if (scattering_distance <= s.distance_to_boundary) { + printf("scattering distance = %f\n", scattering_distance); + p.time += scattering_distance/(SPEED_OF_LIGHT/s.refractive_index1); p.position += scattering_distance*p.direction; |
