def __init__(self, color_mgr=RGBManager()):
# def __init__(self, color_mgr=SampledManager()):
self._color_mgr = color_mgr
register_prototypes(color_mgr.zero())
self.sampler = RegularSampler(width=512,
height=512,
pixelsize=1.0,
nthreads=1)
self.camera = Camera(eye=Vector3(5.0, 5.0, 5.0),
lookat=Vector3(0.0, 0.0, 0.0),
distance=200)
self.camera.load('pinhole')
self.shapes = ShapeManager()
self.intersector = LinearIsect(self.shapes)
self.materials = MaterialManager()
self.lights = LightManager()
self.filter = SampleFilter()
self.filter.load('box')
self.integrator = Integrator()
self.integrator.load('test', self._color_mgr)
self.tone_mapping = Tmo()
self.tone_mapping.load('exp')
self._ready = False
self._create_hdr_buffer()
def __init__(self, color_mgr=RGBManager()):
# def __init__(self, color_mgr=SampledManager()):
self._color_mgr = color_mgr
register_prototypes(color_mgr.zero())
self.sampler = RegularSampler(width=512, height=512,
pixelsize=1.0, nthreads=1)
self.camera = Camera(eye=Vector3(5.0, 5.0, 5.0),
lookat=Vector3(0.0, 0.0, 0.0), distance=200)
self.camera.load('pinhole')
self.shapes = ShapeManager()
self.intersector = LinearIsect(self.shapes)
self.materials = MaterialManager()
self.lights = LightManager()
self.filter = SampleFilter()
self.filter.load('box')
self.integrator = Integrator()
self.integrator.load('test', self._color_mgr)
self.tone_mapping = Tmo()
self.tone_mapping.load('exp')
self._ready = False
self._create_hdr_buffer()
from sdl import ImagePRGBA
from imldr import load_image
from hdr import Tmo
tm = Tmo()
tm.load('exp')
in_img = load_image('E:/hdr_images/Desk_oBA2.hdr')
w, h = in_img.size()
out_img = ImagePRGBA(w, h)
tm.tmo(in_img, out_img)
class Renderer:
def __init__(self, color_mgr=RGBManager()):
# def __init__(self, color_mgr=SampledManager()):
self._color_mgr = color_mgr
register_prototypes(color_mgr.zero())
self.sampler = RegularSampler(width=512,
height=512,
pixelsize=1.0,
nthreads=1)
self.camera = Camera(eye=Vector3(5.0, 5.0, 5.0),
lookat=Vector3(0.0, 0.0, 0.0),
distance=200)
self.camera.load('pinhole')
self.shapes = ShapeManager()
self.intersector = LinearIsect(self.shapes)
self.materials = MaterialManager()
self.lights = LightManager()
self.filter = SampleFilter()
self.filter.load('box')
self.integrator = Integrator()
self.integrator.load('test', self._color_mgr)
self.tone_mapping = Tmo()
self.tone_mapping.load('exp')
self._ready = False
self._create_hdr_buffer()
@property
def color_mgr(self):
return self._color_mgr
def _create_defaults(self):
pass
def _create_hdr_buffer(self):
width, height = self.sampler.get_resolution()
self._hdr_buffer = ImagePRGBA(width, height)
self._hdr_buffer.clear()
self._hdr_output = ImagePRGBA(width, height)
def prepare(self):
self._create_hdr_buffer()
self.sync_area_lights()
runtimes = [Runtime(code=4) for i in range(self.sampler.nthreads)]
shaders_funcs = shaders_functions()
for s in shaders_funcs:
s.compile()
s.prepare(runtimes)
self.sampler.create_shader()
self.sampler.compile()
self.sampler.prepare(runtimes)
self.sampler.reset()
self.camera.compile()
self.camera.prepare(runtimes)
self.filter.compile()
self.filter.prepare(runtimes)
self.intersector.prepare_accel()
self.intersector.compile()
self.intersector.prepare(runtimes)
self.lights.compile_shaders(self._color_mgr)
self.lights.prepare_shaders(runtimes)
for shape in self.lights.shapes_to_update():
self.shapes.update(shape)
lum_shader = luminance_shader(self._color_mgr)
lum_shader.compile(color_mgr=self._color_mgr)
lum_shader.prepare(runtimes)
spec_to_rgb_shader = spectrum_to_rgb_shader(self._color_mgr)
spec_to_rgb_shader.compile(color_mgr=self._color_mgr)
spec_to_rgb_shader.prepare(runtimes)
rgb_to_spec_shader = rgb_to_spectrum_shader(self._color_mgr)
rgb_to_spec_shader.compile(color_mgr=self._color_mgr)
rgb_to_spec_shader.prepare(runtimes)
mat_shader_funcs = shaders_funcs + [lum_shader, rgb_to_spec_shader]
self.materials.compile_shaders(self._color_mgr, mat_shader_funcs)
self.materials.prepare_shaders(runtimes)
shaders = [
self.sampler.shader, self.camera.shader, self.intersector.shader,
self.lights.rad_shader, self.lights.nlights_shader,
self.lights.env_shader, self.lights.emission_shader,
self.materials.ref_shader, spec_to_rgb_shader,
self.intersector.visible_shader, lum_shader,
self.materials.sampling_shader, self.materials.pdf_shader,
self.filter.shader
]
self.integrator.compile(shaders)
self.integrator.prepare(runtimes)
self._ready = True
def reset(self):
if self._hdr_buffer is not None:
self._hdr_buffer.clear()
self.sampler.reset()
def render(self):
"""Execute rendering of one sample for each pixel. If we have
multiple samples per pixel we must call this function multiple
times. Function return True if all samples are processed otherwise
False.
"""
if not self._ready:
self.prepare()
if not self.sampler.has_more_samples():
return True
self.integrator.execute(self._hdr_buffer)
# print (self.integrator.shader._asm_code)
# self.integrator.shader._mc.print_machine_code()
self.sampler.increment_pass()
if not self.sampler.has_more_samples():
return True
return False
def tmo(self):
if self._hdr_buffer is not None:
self.tone_mapping.tmo(self._hdr_buffer, self._hdr_output)
def save(self, filename):
f = open(filename, mode='w')
directory = os.path.dirname(os.path.abspath(f.name))
mesh_dir = os.path.join(directory, 'meshes')
if not os.path.isdir(mesh_dir):
os.mkdir(mesh_dir)
txt = self.sampler.output()
f.write(txt)
f.write('\n')
txt = self.camera.output()
f.write(txt)
f.write('\n')
txt = self.lights.output()
f.write(txt)
f.write('\n')
txt = self.materials.output()
f.write(txt)
f.write('\n')
for shape in self.shapes:
text = 'Shape\n'
fname = self.shapes.name(shape) + '.data'
relative_name = os.path.join('meshes', fname)
text += shape.output(directory, relative_name)
text += 'name = %s\n' % self.shapes.name(shape)
text += 'material = %s\n' % self.materials.name(shape.mat_idx)
text += 'End\n\n'
f.write(text)
f.close()
def load(self, filename):
import_scene(filename, self)
self._ready = False
def sync_area_lights(self):
for shape in self.shapes:
material = self.materials.material(index=shape.mat_idx)
area_light = self.lights.arealight(shape)
if material.is_emissive() and area_light is None:
light = AreaLight(shape=shape, material=material)
light.load('general', self._color_mgr)
name = 'arealight_%i' % id(light)
self.lights.add(name, light)
elif not material.is_emissive() and area_light is not None:
area_light.shape.light_id = -1
self.lights.remove(light=area_light)
self.shapes.update(area_light.shape)
def __init__(self):
self.tmapping = Tmo()
self.tmapping.load('exp')
class ITmo:
def __init__(self):
self.tmapping = Tmo()
self.tmapping.load('exp')
def load(self, objects, args):
shader_name = args
self.tmapping.load(shader_name)
return ""
def tmo(self, objects, args):
in_id, out_id = args.split(',')
in_img = objects[in_id]
out_img = objects[out_id]
self.tmapping.tmo(in_img, out_img)
return ""
def get_prop(self, objects, args):
key = args
if key in self.tmapping.shader._args_map:
arg = self.tmapping.shader._args_map[key]
if isinstance(arg, FloatArg):
return str(self.tmapping.shader.get_value(key))
raise ValueError("Wrong key %s" % key)
def set_prop(self, objects, args):
key, val = args.split(',')
if key in self.tmapping.shader._args_map:
arg = self.tmapping.shader._args_map[key]
if isinstance(arg, FloatArg):
self.tmapping.shader.set_value(arg.name, float(val))
else:
raise ValueError("Wrong key %s" % key)
return ""
def get_public_props(self, objects, args):
public_props = ''
for a in self.tmapping.shader._args:
if isinstance(a, FloatArg):
public_props += ',' + a.name + ',' + 'float'
if public_props == '':
return ''
return public_props[1:]
def save_image(self, objects, args):
fname, id_img = args.split(',')
image = objects[id_img]
if isinstance(image, ImagePRGBA):
width, height = image.size()
new_img = ImageRGBA(width, height)
blt_prgba_to_rgba(image, new_img)
save_image(fname, new_img)
elif isinstance(image, ImageRGBA):
save_image(fname, image)
elif isinstance(image, ImageBGRA):
new_img = image.to_rgba()
save_image(fname, new_img)
return ""
def shader_code(self, objects, args):
return self.tmapping.shader._code
def assembly_code(self, objects, args):
return self.tmapping.shader._asm_code
def __init__(self):
self.tmapping = Tmo()
self.tmapping.load('exp')
class ITmo:
def __init__(self):
self.tmapping = Tmo()
self.tmapping.load('exp')
def load(self, objects, args):
shader_name = args
self.tmapping.load(shader_name)
return ""
def tmo(self, objects, args):
in_id, out_id = args.split(',')
in_img = objects[in_id]
out_img = objects[out_id]
self.tmapping.tmo(in_img, out_img)
return ""
def get_prop(self, objects, args):
key = args
if key in self.tmapping.shader._args_map:
arg = self.tmapping.shader._args_map[key]
if isinstance(arg, FloatArg):
return str(self.tmapping.shader.get_value(key))
raise ValueError("Wrong key %s" % key)
def set_prop(self, objects, args):
key, val = args.split(',')
if key in self.tmapping.shader._args_map:
arg = self.tmapping.shader._args_map[key]
if isinstance(arg, FloatArg):
self.tmapping.shader.set_value(arg.name, float(val))
else:
raise ValueError("Wrong key %s" % key)
return ""
def get_public_props(self, objects, args):
public_props = ''
for a in self.tmapping.shader._args:
if isinstance(a, FloatArg):
public_props += ',' + a.name + ',' + 'float'
if public_props == '':
return ''
return public_props[1:]
def save_image(self, objects, args):
fname, id_img = args.split(',')
image = objects[id_img]
if isinstance(image, ImagePRGBA):
width, height = image.size()
new_img = ImageRGBA(width, height)
blt_prgba_to_rgba(image, new_img)
save_image(fname, new_img)
elif isinstance(image, ImageRGBA):
save_image(fname, image)
elif isinstance(image, ImageBGRA):
new_img = image.to_rgba()
save_image(fname, new_img)
return ""
def shader_code(self, objects, args):
return self.tmapping.shader._code
def assembly_code(self, objects, args):
return self.tmapping.shader._asm_code
class Renderer:
def __init__(self, color_mgr=RGBManager()):
# def __init__(self, color_mgr=SampledManager()):
self._color_mgr = color_mgr
register_prototypes(color_mgr.zero())
self.sampler = RegularSampler(width=512, height=512,
pixelsize=1.0, nthreads=1)
self.camera = Camera(eye=Vector3(5.0, 5.0, 5.0),
lookat=Vector3(0.0, 0.0, 0.0), distance=200)
self.camera.load('pinhole')
self.shapes = ShapeManager()
self.intersector = LinearIsect(self.shapes)
self.materials = MaterialManager()
self.lights = LightManager()
self.filter = SampleFilter()
self.filter.load('box')
self.integrator = Integrator()
self.integrator.load('test', self._color_mgr)
self.tone_mapping = Tmo()
self.tone_mapping.load('exp')
self._ready = False
self._create_hdr_buffer()
@property
def color_mgr(self):
return self._color_mgr
def _create_defaults(self):
pass
def _create_hdr_buffer(self):
width, height = self.sampler.get_resolution()
self._hdr_buffer = ImagePRGBA(width, height)
self._hdr_buffer.clear()
self._hdr_output = ImagePRGBA(width, height)
def prepare(self):
self._create_hdr_buffer()
self.sync_area_lights()
runtimes = [Runtime(code=4) for i in range(self.sampler.nthreads)]
shaders_funcs = shaders_functions()
for s in shaders_funcs:
s.compile()
s.prepare(runtimes)
self.sampler.create_shader()
self.sampler.compile()
self.sampler.prepare(runtimes)
self.sampler.reset()
self.camera.compile()
self.camera.prepare(runtimes)
self.filter.compile()
self.filter.prepare(runtimes)
self.intersector.prepare_accel()
self.intersector.compile()
self.intersector.prepare(runtimes)
self.lights.compile_shaders(self._color_mgr)
self.lights.prepare_shaders(runtimes)
for shape in self.lights.shapes_to_update():
self.shapes.update(shape)
lum_shader = luminance_shader(self._color_mgr)
lum_shader.compile(color_mgr=self._color_mgr)
lum_shader.prepare(runtimes)
spec_to_rgb_shader = spectrum_to_rgb_shader(self._color_mgr)
spec_to_rgb_shader.compile(color_mgr=self._color_mgr)
spec_to_rgb_shader.prepare(runtimes)
rgb_to_spec_shader = rgb_to_spectrum_shader(self._color_mgr)
rgb_to_spec_shader.compile(color_mgr=self._color_mgr)
rgb_to_spec_shader.prepare(runtimes)
mat_shader_funcs = shaders_funcs + [lum_shader, rgb_to_spec_shader]
self.materials.compile_shaders(self._color_mgr, mat_shader_funcs)
self.materials.prepare_shaders(runtimes)
shaders = [self.sampler.shader, self.camera.shader,
self.intersector.shader, self.lights.rad_shader,
self.lights.nlights_shader, self.lights.env_shader,
self.lights.emission_shader, self.materials.ref_shader,
spec_to_rgb_shader, self.intersector.visible_shader,
lum_shader, self.materials.sampling_shader,
self.materials.pdf_shader, self.filter.shader]
self.integrator.compile(shaders)
self.integrator.prepare(runtimes)
self._ready = True
def reset(self):
if self._hdr_buffer is not None:
self._hdr_buffer.clear()
self.sampler.reset()
def render(self):
"""Execute rendering of one sample for each pixel. If we have
multiple samples per pixel we must call this function multiple
times. Function return True if all samples are processed otherwise
False.
"""
if not self._ready:
self.prepare()
if not self.sampler.has_more_samples():
return True
self.integrator.execute(self._hdr_buffer)
# print (self.integrator.shader._asm_code)
# self.integrator.shader._mc.print_machine_code()
self.sampler.increment_pass()
if not self.sampler.has_more_samples():
return True
return False
def tmo(self):
if self._hdr_buffer is not None:
self.tone_mapping.tmo(self._hdr_buffer, self._hdr_output)
def save(self, filename):
f = open(filename, mode='w')
directory = os.path.dirname(os.path.abspath(f.name))
mesh_dir = os.path.join(directory, 'meshes')
if not os.path.isdir(mesh_dir):
os.mkdir(mesh_dir)
txt = self.sampler.output()
f.write(txt)
f.write('\n')
txt = self.camera.output()
f.write(txt)
f.write('\n')
txt = self.lights.output()
f.write(txt)
f.write('\n')
txt = self.materials.output()
f.write(txt)
f.write('\n')
for shape in self.shapes:
text = 'Shape\n'
fname = self.shapes.name(shape) + '.data'
relative_name = os.path.join('meshes', fname)
text += shape.output(directory, relative_name)
text += 'name = %s\n' % self.shapes.name(shape)
text += 'material = %s\n' % self.materials.name(shape.mat_idx)
text += 'End\n\n'
f.write(text)
f.close()
def load(self, filename):
import_scene(filename, self)
self._ready = False
def sync_area_lights(self):
for shape in self.shapes:
material = self.materials.material(index=shape.mat_idx)
area_light = self.lights.arealight(shape)
if material.is_emissive() and area_light is None:
light = AreaLight(shape=shape, material=material)
light.load('general', self._color_mgr)
name = 'arealight_%i' % id(light)
self.lights.add(name, light)
elif not material.is_emissive() and area_light is not None:
area_light.shape.light_id = -1
self.lights.remove(light=area_light)
self.shapes.update(area_light.shape)