def test_sampled_to_rgb(self): runtimes = [Runtime()] color_mgr = SampledManager() s_to_rgb = spectrum_to_rgb_shader(color_mgr) s_to_rgb.compile(color_mgr=color_mgr) s_to_rgb.prepare(runtimes) code = """ value = spectrum_to_rgb(color) """ spec = RGBSpectrum(0.3, 0.5, 0.4) spec = color_mgr.rgb_to_sampled(spec) col = SampledArg('color', spec) val = Vec3Arg('value', Vector3(0.0, 0.0, 0.0)) shader = Shader(code=code, args=[col, val]) shader.compile(shaders=[s_to_rgb], color_mgr=color_mgr) shader.prepare(runtimes) shader.execute() value = shader.get_value('value') vv = color_mgr.sampled_to_rgb(spec) self.assertAlmostEqual(value.x, vv.r) self.assertAlmostEqual(value.y, vv.g) self.assertAlmostEqual(value.z, vv.b, places=6)
def test_sampled_point_light(self): sam_mgr = SampledManager() register_sampled_shadepoint(sam_mgr) runtimes = [Runtime()] lgt = GeneralLight() lgt.load('point', sam_mgr, spectral=True) lgt.set_value('intensity', RGBSpectrum(0.3, 0.3, 0.3)) lgt.set_value('position', Vector3(2.0, 2.0, 2.0)) lgt.compile() lgt.prepare(runtimes) ptrs = lgt.shader.get_ptrs() ptr_func = PointerArg('ptr_func', ptrs[0]) spec = SampledArg('spec', sam_mgr.zero()) wi = Vec3Arg('wi', Vector3(0.0, 0.0, 0.0)) pos = Vec3Arg('position', Vector3(0.0, 0.0, 0.0)) code = """ hp = HitPoint() hp.hit = (4.0, 5, 6) sp = ShadePoint() __light_radiance(hp, sp, ptr_func) spec = sp.light_intensity wi = sp.wi position = sp.light_position """ shader = Shader(code=code, args=[ptr_func, wi, spec, pos]) shader.compile() shader.prepare(runtimes) shader.execute() wi = Vector3(2.0, 2.0, 2.0) - Vector3(4.0, 5.0, 6.0) wi.normalize() wi_s = shader.get_value('wi') self.assertAlmostEqual(wi.x, wi_s.x) self.assertAlmostEqual(wi.y, wi_s.y) self.assertAlmostEqual(wi.z, wi_s.z) p = Vector3(2.0, 2.0, 2.0) p_s = shader.get_value('position') self.assertAlmostEqual(p.x, p_s.x) self.assertAlmostEqual(p.y, p_s.y) self.assertAlmostEqual(p.z, p_s.z) s = sam_mgr.rgb_to_sampled(RGBSpectrum(0.3, 0.3, 0.3), illum=True) s_s = shader.get_value('spec') for i in range(len(s.samples)): self.assertAlmostEqual(s.samples[i], s_s.samples[i])
def test_sampled_point_light(self): sam_mgr = SampledManager() register_sampled_shadepoint(sam_mgr) runtimes = [Runtime()] lgt = GeneralLight() lgt.load('point', sam_mgr, spectral=True) lgt.set_value('intensity', RGBSpectrum(0.3, 0.3, 0.3)) lgt.set_value('position', Vector3(2.0, 2.0, 2.0)) lgt.compile() lgt.prepare(runtimes) ptrs = lgt.shader.get_ptrs() ptr_func = PointerArg('ptr_func', ptrs[0]) spec = SampledArg('spec', sam_mgr.zero()) wi = Vec3Arg('wi', Vector3(0.0, 0.0, 0.0)) pos = Vec3Arg('position', Vector3(0.0, 0.0, 0.0)) code = """ hp = HitPoint() hp.hit = (4.0, 5, 6) sp = ShadePoint() __light_radiance(hp, sp, ptr_func) spec = sp.light_intensity wi = sp.wi position = sp.light_position """ shader = Shader(code=code, args=[ptr_func, wi, spec, pos]) shader.compile() shader.prepare(runtimes) shader.execute() wi = Vector3(2.0, 2.0, 2.0) - Vector3(4.0, 5.0, 6.0) wi.normalize() wi_s = shader.get_value('wi') self.assertAlmostEqual(wi.x, wi_s.x) self.assertAlmostEqual(wi.y, wi_s.y) self.assertAlmostEqual(wi.z, wi_s.z) p = Vector3(2.0, 2.0, 2.0) p_s = shader.get_value('position') self.assertAlmostEqual(p.x, p_s.x) self.assertAlmostEqual(p.y, p_s.y) self.assertAlmostEqual(p.z, p_s.z) s = sam_mgr.rgb_to_sampled(RGBSpectrum(0.3, 0.3, 0.3), illum=True) s_s = shader.get_value('spec') for i in range(len(s.samples)): self.assertAlmostEqual(s.samples[i], s_s.samples[i])
def test_sampled_luminance(self): runtimes = [Runtime()] color_mgr = SampledManager() lum_shader = luminance_shader(color_mgr) lum_shader.compile(color_mgr=color_mgr) lum_shader.prepare(runtimes) code = """ value = luminance(color) """ spec = RGBSpectrum(0.3, 0.5, 0.7) spec = color_mgr.rgb_to_sampled(spec) col = SampledArg('color', spec) val = FloatArg('value', 0.0) shader = Shader(code=code, args=[col, val]) shader.compile(shaders=[lum_shader], color_mgr=color_mgr) shader.prepare(runtimes) shader.execute() value = shader.get_value('value') vv = color_mgr.luminance(spec) self.assertAlmostEqual(value, vv)
def test_rgb_to_sampled_spectrum(self): runtimes = [Runtime()] color_mgr = SampledManager() rgb_to_spec = rgb_to_spectrum_shader(color_mgr) rgb_to_spec.compile(color_mgr=color_mgr) rgb_to_spec.prepare(runtimes) code = """ spec = rgb_to_spectrum(color) """ val = Vector3(0.3, 0.5, 0.7) col = Vec3Arg('color', val) spec = SampledArg('spec', color_mgr.zero()) shader = Shader(code=code, args=[col, spec]) shader.compile(shaders=[rgb_to_spec], color_mgr=color_mgr) shader.prepare(runtimes) shader.execute() value = shader.get_value('spec') vv = color_mgr.rgb_to_sampled(RGBSpectrum(val.x, val.y, val.z)) for i in range(len(value.samples)): self.assertAlmostEqual(value.samples[i], vv.samples[i], places=6)
def test_rgb_to_sampled_spectrum(self): runtimes = [Runtime()] color_mgr = SampledManager() rgb_to_spec = rgb_to_spectrum_shader(color_mgr) rgb_to_spec.compile(color_mgr=color_mgr) rgb_to_spec.prepare(runtimes) code = """ spec = rgb_to_spectrum(color) """ val = Vector3(0.3, 0.5, 0.7) col = Vec3Arg('color', val) spec = SampledArg('spec', color_mgr.zero()) shader = Shader(code=code, args=[col, spec]) shader.compile(shaders=[rgb_to_spec], color_mgr=color_mgr) shader.prepare(runtimes) shader.execute() value = shader.get_value('spec') vv = color_mgr.rgb_to_sampled(RGBSpectrum(val.x, val.y, val.z)) for i in range(len(value.samples)): self.assertAlmostEqual(value.samples[i], vv.samples[i], places=6)