mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-11-03 18:00:55 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Move argset to code.util

Browse Source
This commit is contained in:
Steven Robertson
2012-01-21 00:03:28 -05:00
parent acbde65b9f
commit a803216551
2 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cuburn/code/util.py
View File

@ -11,6 +11,11 @@ def crep(s):
s = s.encode('utf-8') s = s.encode('utf-8')
return '"%s"' % s.encode("string_escape") return '"%s"' % s.encode("string_escape")
def argset(obj, **kwargs):
for k, v in kwargs.items():
setattr(obj, k, v)
return obj
class Template(tempita.Template): class Template(tempita.Template):
default_namespace = tempita.Template.default_namespace.copy() default_namespace = tempita.Template.default_namespace.copy()
Template.default_namespace.update({'np': np, 'crep': crep}) Template.default_namespace.update({'np': np, 'crep': crep})

17
cuburn/render.py
View File

@ -25,11 +25,6 @@ Dimensions = namedtuple('Dimensions', 'w h aw ah astride')
def _sync_stream(dst, src): def _sync_stream(dst, src):
dst.wait_for_event(cuda.Event(cuda.event_flags.DISABLE_TIMING).record(src)) dst.wait_for_event(cuda.Event(cuda.event_flags.DISABLE_TIMING).record(src))
def argset(obj, **kwargs):
for k, v in kwargs.items():
setattr(obj, k, v)
return obj
class Renderer(object): class Renderer(object):
""" """
Control structure for rendering a series of frames. Control structure for rendering a series of frames.
@ -197,7 +192,7 @@ class Renderer(object):
d_atom = cuda.mem_alloc(8 * nbins) d_atom = cuda.mem_alloc(8 * nbins)
flush_fun = self.mod.get_function("flush_atom") flush_fun = self.mod.get_function("flush_atom")
obuf_copy = argset(cuda.Memcpy2D(), obuf_copy = util.argset(cuda.Memcpy2D(),
src_y=self.gutter, src_x_in_bytes=16*self.gutter, src_y=self.gutter, src_x_in_bytes=16*self.gutter,
src_pitch=16*astride, dst_pitch=16*width, src_pitch=16*astride, dst_pitch=16*width,
width_in_bytes=16*width, height=height) width_in_bytes=16*width, height=height)
@ -260,8 +255,9 @@ class Renderer(object):
if self.acc_mode in ('deferred', 'atomic'): if self.acc_mode in ('deferred', 'atomic'):
palette_fun = self.mod.get_function("interp_palette_flat") palette_fun = self.mod.get_function("interp_palette_flat")
dsc = argset(cuda.ArrayDescriptor3D(), height=self.palette_height, dsc = util.argset(cuda.ArrayDescriptor3D(),
width=256, depth=0, format=cuda.array_format.SIGNED_INT32, height=self.palette_height, width=256, depth=0,
format=cuda.array_format.SIGNED_INT32,
num_channels=2, flags=cuda.array3d_flags.SURFACE_LDST) num_channels=2, flags=cuda.array3d_flags.SURFACE_LDST)
palarray = cuda.Array(dsc) palarray = cuda.Array(dsc)
@ -269,8 +265,9 @@ class Renderer(object):
tref.set_array(palarray, 0) tref.set_array(palarray, 0)
else: else:
palette_fun = self.mod.get_function("interp_palette") palette_fun = self.mod.get_function("interp_palette")
dsc = argset(cuda.ArrayDescriptor(), height=self.palette_height, dsc = util.argset(cuda.ArrayDescriptor(),
width=256, format=cuda.array_format.UNSIGNED_INT8, height=self.palette_height, width=256,
format=cuda.array_format.UNSIGNED_INT8,
num_channels=4) num_channels=4)
d_palmem = cuda.mem_alloc(256 * self.palette_height * 4) d_palmem = cuda.mem_alloc(256 * self.palette_height * 4)