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)
