def load(self, name):
props = self._loader.load(name, 'props.txt')
args = []
if props is not None:
args = parse_args(props)
in_img = StructArg('input_image', ImagePRGBA(1, 1))
out_img = StructArg('output_image', ImagePRGBA(1, 1))
args.extend([in_img, out_img])
code = self._loader.load(name, 'code.py')
self._shader = Shader(code=code, args=args)
self._shader.compile()
self._runtime = Runtime()
self._shader.prepare([self._runtime])
def _parse_image(line, image_factory=None):
if image_factory is None:
img = ImagePRGBA(1, 1) # image of 1 pixel in size
else:
img = image_factory()
name = line
arg = StructArg(name, img)
return arg
def load(self, shader_name, color_mgr):
text = self._loader.load(shader_name, 'props.txt')
args = []
if text is not None:
args = parse_args(text)
code = self._loader.load(shader_name, 'code.py')
if code is None:
raise ValueError("Integrator %s code is missing!" % shader_name)
#array of shadepoints for debuging purpose
sp = ShadePoint.factory(color_mgr.zero())
arr = ObjArray(sp)
for i in range(10):
arr.append(sp)
arg = ArrayArg('path_array', arr)
args.append(arg)
hdr_buffer = StructArg('hdr_buffer', ImagePRGBA(1, 1))
args.append(hdr_buffer)
self.shader = Shader(code=code, args=args)
self._color_mgr = color_mgr
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)
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)
