diff options
Diffstat (limited to 'src/materials.h')
| -rw-r--r-- | src/materials.h | 45 |
1 files changed, 43 insertions, 2 deletions
diff --git a/src/materials.h b/src/materials.h index 05aa121..47b7d22 100644 --- a/src/materials.h +++ b/src/materials.h @@ -1,3 +1,20 @@ +#ifndef __MATERIALS_H__ +#define __MATERIALS_H__ + +__device__ float min_wavelength; +__device__ float max_wavelength; +__device__ float wavelength_step; +__device__ unsigned int wavelength_size; + +enum +{ + INIT = -1, + MAX_DEPTH_REACHED, + NO_HIT, + BULK_ABSORB, + SURFACE_ABSORB +}; + struct Material { float *refractive_index; @@ -12,12 +29,34 @@ struct Surface float *reflection_specular; }; -__device__ struct Material materials[100]; -__device__ struct Surface surfaces[100]; +__device__ Material materials[100]; +__device__ Surface surfaces[100]; + +__device__ float interp_property(const float &x, const float *fp) +{ + if (x < min_wavelength) + return fp[0]; + + if (x > max_wavelength) + return fp[wavelength_size-1]; + + unsigned int jl = (x-min_wavelength)/wavelength_step; + float xl = min_wavelength + jl*wavelength_step; + + return xl + (x-xl)*(fp[jl+1]-fp[jl])/wavelength_step; +} extern "C" { +__global__ void set_wavelength_range(float _min_wavelength, float _max_wavelength, float _wavelength_step, unsigned int _wavelength_size) +{ + min_wavelength = _min_wavelength; + max_wavelength = _max_wavelength; + wavelength_step = _wavelength_step; + wavelength_size = _wavelength_size; +} + __global__ void set_material(int material_index, float *refractive_index, float *absorption_length, float *scattering_length) { materials[material_index].refractive_index = refractive_index; @@ -33,3 +72,5 @@ __global__ void set_surface(int surface_index, float *absorption, float *reflect } } // extern "c" + +#endif |