From 438e4c497d5a9f8e9fd75ea63fb05eac860c2708 Mon Sep 17 00:00:00 2001 From: Tavian Barnes Date: Wed, 15 Jun 2011 23:27:19 -0600 Subject: Make python module more "pythonic". Use lower_case instead of mixedCase, and add docstrings. --- libdimension-python/dimension.pyx | 684 ++++++++++++++++++++++++++-------- libdimension-python/tests/canvas.py | 14 +- libdimension-python/tests/color.py | 2 +- libdimension-python/tests/demo.py | 42 +-- libdimension-python/tests/geometry.py | 10 +- 5 files changed, 562 insertions(+), 190 deletions(-) (limited to 'libdimension-python') diff --git a/libdimension-python/dimension.pyx b/libdimension-python/dimension.pyx index dd9ff17..fa57691 100644 --- a/libdimension-python/dimension.pyx +++ b/libdimension-python/dimension.pyx @@ -17,6 +17,10 @@ # along with this program. If not, see . # ######################################################################### +""" +Dimension: a high-performance photo-realistic 3D renderer. +""" + import os ########### @@ -24,40 +28,51 @@ import os ########### # Make warnings fatal -def dieOnWarnings(alwaysDie): - dmnsn_die_on_warnings(alwaysDie) +def die_on_warnings(always_die): + """Whether to treat Dimension warnings as errors.""" + dmnsn_die_on_warnings(always_die) ########## # Timers # ########## cdef class Timer: + """A timer for Dimension tasks.""" cdef dmnsn_timer *_timer def __init__(self): + """ + Create a Timer. + + Timing starts as soon as the object is created. + """ self._timer = dmnsn_new_timer() def __dealloc__(self): dmnsn_delete_timer(self._timer) def complete(self): + """Stop the Timer.""" dmnsn_complete_timer(self._timer) property real: + """Real (wall clock) time.""" def __get__(self): return self._timer.real property user: + """User (CPU) time.""" def __get__(self): return self._timer.user property system: + """System time.""" def __get__(self): return self._timer.system def __str__(self): - return '%.2fs (user: %.2fs; system: %.2fs)' % \ + return "%.2fs (user: %.2fs; system: %.2fs)" % \ (self._timer.real, self._timer.user, self._timer.system) -cdef _rawTimer(dmnsn_timer *timer): +cdef _Timer(dmnsn_timer *timer): cdef Timer self = Timer.__new__(Timer) self._timer = timer DMNSN_INCREF(self._timer) @@ -68,99 +83,121 @@ cdef _rawTimer(dmnsn_timer *timer): ############ cdef class Vector: + """A vector (or point or pseudovector) in 3D space.""" cdef dmnsn_vector _v def __init__(self, *args, **kwargs): + """ + Create a Vector. + + Keyword arguments: + x -- The x coordinate + y -- The y coordinate + z -- The z coordinate + + Alternatively, you can pass another Vector, the value 0, or a tuple or other + sequence (x, y, z). + """ if len(args) == 1: - if (isinstance(args[0], Vector)): + if isinstance(args[0], Vector): self._v = (args[0])._v - elif (isinstance(args[0], tuple)): - self._realInit(*args[0]) - elif (args[0] == 0): + elif hasattr(args[0], '__iter__'): # Faster than try: ... except: + self._real_init(*args[0]) + elif args[0] == 0: self._v = dmnsn_zero else: - raise TypeError, 'expected a tuple or 0' + raise TypeError, "expected a sequence or 0" else: - self._realInit(*args, **kwargs) + self._real_init(*args, **kwargs) - def _realInit(self, double x, double y, double z): + def _real_init(self, double x, double y, double z): self._v = dmnsn_new_vector(x, y, z) property x: + """The x coordinate.""" def __get__(self): return self._v.x property y: + """The y coordinate.""" def __get__(self): return self._v.y property z: + """The z coordinate.""" def __get__(self): return self._v.z def __pos__(self): return self def __neg__(self): - return _rawVector(dmnsn_vector_negate(self._v)) + return _Vector(dmnsn_vector_negate(self._v)) def __nonzero__(self): return dmnsn_vector_norm(self._v) >= dmnsn_epsilon def __add__(lhs, rhs): - return _rawVector(dmnsn_vector_add(Vector(lhs)._v, Vector(rhs)._v)) + return _Vector(dmnsn_vector_add(Vector(lhs)._v, Vector(rhs)._v)) def __sub__(lhs, rhs): - return _rawVector(dmnsn_vector_sub(Vector(lhs)._v, Vector(rhs)._v)) + return _Vector(dmnsn_vector_sub(Vector(lhs)._v, Vector(rhs)._v)) def __mul__(lhs, rhs): - if (isinstance(lhs, Vector)): - return _rawVector(dmnsn_vector_mul(rhs, (lhs)._v)) + if isinstance(lhs, Vector): + return _Vector(dmnsn_vector_mul(rhs, (lhs)._v)) else: - return _rawVector(dmnsn_vector_mul(lhs, (rhs)._v)) + return _Vector(dmnsn_vector_mul(lhs, (rhs)._v)) def __truediv__(Vector lhs not None, double rhs): - return _rawVector(dmnsn_vector_div(lhs._v, rhs)) + return _Vector(dmnsn_vector_div(lhs._v, rhs)) def __richcmp__(lhs, rhs, int op): equal = (Vector(lhs) - Vector(rhs)).norm() < dmnsn_epsilon - if (op == 2): # == + if op == 2: # == return equal - elif (op == 3): # != + elif op == 3: # != return not equal else: return NotImplemented def norm(self): + """Return the magnitude of the vector.""" return dmnsn_vector_norm(self._v) def normalized(self): - return _rawVector(dmnsn_vector_normalized(self._v)) + """Return the direction of the vector.""" + return _Vector(dmnsn_vector_normalized(self._v)) def __repr__(self): - return 'dimension.Vector(%r, %r, %r)' % (self.x, self.y, self.z) + return "dimension.Vector(%r, %r, %r)" % (self.x, self.y, self.z) def __str__(self): - return '<%s, %s, %s>' % (self.x, self.y, self.z) + return "<%s, %s, %s>" % (self.x, self.y, self.z) -cdef _rawVector(dmnsn_vector v): +cdef _Vector(dmnsn_vector v): cdef Vector self = Vector.__new__(Vector) self._v = v return self def cross(Vector lhs not None, Vector rhs not None): - return _rawVector(dmnsn_vector_cross(lhs._v, rhs._v)) + """Vector cross product.""" + return _Vector(dmnsn_vector_cross(lhs._v, rhs._v)) def dot(Vector lhs not None, Vector rhs not None): + """Vector dot product.""" return dmnsn_vector_dot(lhs._v, rhs._v) def proj(Vector u not None, Vector d not None): - return _rawVector(dmnsn_vector_proj(u._v, d._v)) + """Vector projection (of u onto d).""" + return _Vector(dmnsn_vector_proj(u._v, d._v)) -X = _rawVector(dmnsn_x) -Y = _rawVector(dmnsn_y) -Z = _rawVector(dmnsn_z) +X = _Vector(dmnsn_x) +Y = _Vector(dmnsn_y) +Z = _Vector(dmnsn_z) cdef class Matrix: + """An affine transformation matrix.""" cdef dmnsn_matrix _m def __init__(self, double a1, double a2, double a3, double a4, double b1, double b2, double b3, double b4, double c1, double c2, double c3, double c4): - self._m = dmnsn_new_matrix(a1, a2, a3, a4, - b1, b2, b3, b4, - c1, c2, c3, c4) + """Create a Matrix.""" + self._m = dmnsn_new_matrix(a1, a2, a3, a4, + b1, b2, b3, b4, + c1, c2, c3, c4) def __nonzero__(self): cdef double sum = 0.0 @@ -170,10 +207,10 @@ cdef class Matrix: return sqrt(sum) >= dmnsn_epsilon def __mul__(Matrix lhs not None, rhs): - if (isinstance(rhs, Matrix)): - return _rawMatrix(dmnsn_matrix_mul(lhs._m, (rhs)._m)) + if isinstance(rhs, Matrix): + return _Matrix(dmnsn_matrix_mul(lhs._m, (rhs)._m)) else: - return _rawVector(dmnsn_transform_vector(lhs._m, (rhs)._v)) + return _Vector(dmnsn_transform_vector(lhs._m, (rhs)._v)) def __richcmp__(Matrix lhs not None, Matrix rhs not None, int op): cdef double sum = 0.0 @@ -183,100 +220,155 @@ cdef class Matrix: sum += diff*diff equal = sqrt(sum) < dmnsn_epsilon - if (op == 2): # == + if op == 2: # == return equal - elif (op == 3): # != + elif op == 3: # != return not equal else: return NotImplemented cpdef Matrix inverse(self): - return _rawMatrix(dmnsn_matrix_inverse(self._m)); + """Return the inverse of a matrix.""" + return _Matrix(dmnsn_matrix_inverse(self._m)); def __repr__(self): return \ - 'dimension.Matrix(%r, %r, %r, %r, %r, %r, %r, %r, %r, %r, %r, %r)' % \ + "dimension.Matrix(%r, %r, %r, %r, %r, %r, %r, %r, %r, %r, %r, %r)" % \ (self._m.n[0][0], self._m.n[0][1], self._m.n[0][2], self._m.n[0][3], self._m.n[1][0], self._m.n[1][1], self._m.n[1][2], self._m.n[1][3], self._m.n[2][0], self._m.n[2][1], self._m.n[2][2], self._m.n[2][3]) def __str__(self): return \ - '\n[%s\t%s\t%s\t%s]' \ - '\n[%s\t%s\t%s\t%s]' \ - '\n[%s\t%s\t%s\t%s]' \ - '\n[%s\t%s\t%s\t%s]' %\ + "\n[%s\t%s\t%s\t%s]" \ + "\n[%s\t%s\t%s\t%s]" \ + "\n[%s\t%s\t%s\t%s]" \ + "\n[%s\t%s\t%s\t%s]" %\ (self._m.n[0][0], self._m.n[0][1], self._m.n[0][2], self._m.n[0][3], self._m.n[1][0], self._m.n[1][1], self._m.n[1][2], self._m.n[1][3], self._m.n[2][0], self._m.n[2][1], self._m.n[2][2], self._m.n[2][3], 0.0, 0.0, 0.0, 1.0) -cdef Matrix _rawMatrix(dmnsn_matrix m): +cdef Matrix _Matrix(dmnsn_matrix m): cdef Matrix self = Matrix.__new__(Matrix) self._m = m return self def scale(*args, **kwargs): - return _rawMatrix(dmnsn_scale_matrix(Vector(*args, **kwargs)._v)) + """ + Return a scale transformation. + + Accepts the same arguments that Vector(...) does. The transformation scales + by a factor of x in the x direction, y in the y direction, and z in the z + direction. In particular, this means that scale(2*X) is probably a mistake, + as the y and z coordinates will disappear. + + Alternatively, a single argument may be passed, which specifies the scaling + factor in every component. + """ + cdef Vector s + try: + s = Vector(*args, **kwargs) + except: + s = args[0]*(X + Y + Z) + return _Matrix(dmnsn_scale_matrix(s._v)) def translate(*args, **kwargs): - return _rawMatrix(dmnsn_translation_matrix(Vector(*args, **kwargs)._v)) + """ + Return a translation. + + Accepts the same arguments that Vector(...) does. + """ + return _Matrix(dmnsn_translation_matrix(Vector(*args, **kwargs)._v)) def rotate(*args, **kwargs): + """ + Return a rotation. + + Accepts the same arguments that Vector(...) does. theta.norm() is the left- + handed angle of rotation, and theta.normalized() is the axis of rotation. + theta is specified in degrees. + """ cdef Vector rad = dmnsn_radians(1.0)*Vector(*args, **kwargs) - return _rawMatrix(dmnsn_rotation_matrix(rad._v)) -def _rawRotate(*args, **kwargs): - return _rawMatrix(dmnsn_rotation_matrix(Vector(*args, **kwargs)._v)) + return _Matrix(dmnsn_rotation_matrix(rad._v)) +def _Rotate(*args, **kwargs): + return _Matrix(dmnsn_rotation_matrix(Vector(*args, **kwargs)._v)) ########## # Colors # ########## cdef class Color: + """ + An sRGB color. + + Note that 0.5*White == Color(0.5, 0.5, 0.5), which is not technically a half- + intensity white, due to sRGB gamma. Dimension handles the gamma correctly + when rendering, though. + """ cdef dmnsn_color _c cdef dmnsn_color _sRGB def __init__(self, *args, **kwargs): + """ + Create a Color. + + Keyword arguments: + red -- The red component + green -- The green component + blue -- The blue component + trans -- The transparency of the color, 0.0 meaning opaque (default 0.0) + filter -- How filtered the transparency is (default 0.0) + + Alternatively, you can pass another Color, a gray intensity like 0.5, or a + tuple or other sequence (red, green, blue[, trans[, filter]]). + """ if len(args) == 1: - if (isinstance(args[0], Color)): + if isinstance(args[0], Color): self._sRGB = (args[0])._sRGB - elif (isinstance(args[0], tuple)): - self._realInit(*args[0]) + elif hasattr(args[0], '__iter__'): + self._real_init(*args[0]) else: self._sRGB = dmnsn_color_mul(args[0], dmnsn_white) else: - self._realInit(*args, **kwargs) + self._real_init(*args, **kwargs) self._c = dmnsn_color_from_sRGB(self._sRGB) - def _realInit(self, double red, double green, double blue, - double trans = 0.0, double filter = 0.0): + def _real_init(self, double red, double green, double blue, + double trans = 0.0, double filter = 0.0): self._sRGB = dmnsn_new_color5(red, green, blue, trans, filter) property red: + """The red component.""" def __get__(self): return self._sRGB.R property green: + """The green component.""" def __get__(self): return self._sRGB.G property blue: + """The blue component.""" def __get__(self): return self._sRGB.B property trans: + """The transparency of the color.""" def __get__(self): return self._sRGB.trans property filter: + """How filtered the transparency is.""" def __get__(self): return self._sRGB.filter def __nonzero__(self): + """Return whether a color is not black.""" return not dmnsn_color_is_black(self._c) def __add__(lhs, rhs): - return _rawsRGBColor(dmnsn_color_add(Color(lhs)._sRGB, Color(rhs)._sRGB)) + return _sRGBColor(dmnsn_color_add(Color(lhs)._sRGB, Color(rhs)._sRGB)) def __mul__(lhs, rhs): - if (isinstance(lhs, Color)): - return _rawsRGBColor(dmnsn_color_mul(rhs, (lhs)._sRGB)) + if isinstance(lhs, Color): + return _sRGBColor(dmnsn_color_mul(rhs, (lhs)._sRGB)) else: - return _rawsRGBColor(dmnsn_color_mul(lhs, (rhs)._sRGB)) + return _sRGBColor(dmnsn_color_mul(lhs, (rhs)._sRGB)) def __richcmp__(lhs, rhs, int op): cdef clhs = Color(lhs) @@ -290,81 +382,95 @@ cdef class Color: cdef double sum = rdiff*rdiff + gdiff*gdiff + bdiff*bdiff \ + tdiff*tdiff + fdiff*fdiff equal = sqrt(sum) < dmnsn_epsilon - if (op == 2): # == + if op == 2: # == return equal - elif (op == 3): # != + elif op == 3: # != return not equal else: return NotImplemented def __repr__(self): - return 'dimension.Color(%r, %r, %r, %r, %r)' % \ + return "dimension.Color(%r, %r, %r, %r, %r)" % \ (self.red, self.green, self.blue, self.trans, self.filter) def __str__(self): - if (self.trans >= dmnsn_epsilon): - return '' % \ + if self.trans >= dmnsn_epsilon: + return "" % \ (self.red, self.green, self.blue, self.trans, self.filter) else: - return '' % \ + return "" % \ (self.red, self.green, self.blue) -cdef _rawsRGBColor(dmnsn_color sRGB): +cdef _sRGBColor(dmnsn_color sRGB): cdef Color self = Color.__new__(Color) self._sRGB = sRGB self._c = dmnsn_color_from_sRGB(sRGB) return self -cdef _rawColor(dmnsn_color c): +cdef _Color(dmnsn_color c): cdef Color self = Color.__new__(Color) self._c = c self._sRGB = dmnsn_color_to_sRGB(c) return self -Black = _rawColor(dmnsn_black) -White = _rawColor(dmnsn_white) -Clear = _rawColor(dmnsn_clear) -Red = _rawColor(dmnsn_red) -Green = _rawColor(dmnsn_green) -Blue = _rawColor(dmnsn_blue) -Magenta = _rawColor(dmnsn_magenta) -Orange = _rawColor(dmnsn_orange) -Yellow = _rawColor(dmnsn_yellow) -Cyan = _rawColor(dmnsn_cyan) +Black = _Color(dmnsn_black) +White = _Color(dmnsn_white) +Clear = _Color(dmnsn_clear) +Red = _Color(dmnsn_red) +Green = _Color(dmnsn_green) +Blue = _Color(dmnsn_blue) +Magenta = _Color(dmnsn_magenta) +Orange = _Color(dmnsn_orange) +Yellow = _Color(dmnsn_yellow) +Cyan = _Color(dmnsn_cyan) ############ # Canvases # ############ cdef class Canvas: + """A rendering target.""" cdef dmnsn_canvas *_canvas - def __cinit__(self, size_t width, size_t height): + def __init__(self, size_t width, size_t height): + """ + Create a Canvas. + + Keyword arguments: + width -- the width of the canvas + height -- the height of the canvas + """ self._canvas = dmnsn_new_canvas(width, height) def __dealloc__(self): dmnsn_delete_canvas(self._canvas) property width: + """The width of the canvas.""" def __get__(self): return self._canvas.width property height: + """The height of the canvas.""" def __get__(self): return self._canvas.height - def optimizePNG(self): + def optimize_PNG(self): + """Optimize a canvas for PNG output.""" if dmnsn_png_optimize_canvas(self._canvas) != 0: raise OSError(errno, os.strerror(errno)) - def optimizeGL(self): + def optimize_GL(self): + """Optimize a canvas for OpenGL output.""" if dmnsn_gl_optimize_canvas(self._canvas) != 0: raise OSError(errno, os.strerror(errno)) def clear(self, c): + """Clear a canvas with a solid color.""" dmnsn_clear_canvas(self._canvas, Color(c)._c) - def writePNG(self, str path not None): - bpath = path.encode('UTF-8') + def write_PNG(self, path): + """Export the canvas as a PNG file.""" + bpath = path.encode("UTF-8") cdef char *cpath = bpath cdef FILE *file = fopen(cpath, "wb") if file == NULL: @@ -373,7 +479,8 @@ cdef class Canvas: if dmnsn_png_write_canvas(self._canvas, file) != 0: raise OSError(errno, os.strerror(errno)) - def drawGL(self): + def draw_GL(self): + """Export the canvas to the current OpenGL context.""" if dmnsn_gl_write_canvas(self._canvas) != 0: raise OSError(errno, os.strerror(errno)) @@ -382,6 +489,7 @@ cdef class Canvas: ############ cdef class Pattern: + """A function which maps points in 3D space to scalar values.""" cdef dmnsn_pattern *_pattern def __cinit__(self): @@ -391,19 +499,28 @@ cdef class Pattern: dmnsn_delete_pattern(self._pattern) def transform(self, Matrix trans not None): + """Transform a pattern.""" if self._pattern == NULL: - raise TypeError('attempt to transform base Pattern') + raise TypeError("attempt to transform base Pattern") self._pattern.trans = dmnsn_matrix_mul(trans._m, self._pattern.trans) return self cdef class Checker(Pattern): + """A checkerboard pattern.""" def __init__(self): self._pattern = dmnsn_new_checker_pattern() Pattern.__init__(self) cdef class Gradient(Pattern): + """A gradient pattern.""" def __init__(self, orientation): + """ + Create a gradient pattern. + + Keyword arguments: + orientation -- The direction of the linear gradient. + """ self._pattern = dmnsn_new_gradient_pattern(Vector(orientation)._v) Pattern.__init__(self) @@ -412,14 +529,21 @@ cdef class Gradient(Pattern): ############ cdef class Pigment: + """Object surface coloring.""" cdef dmnsn_pigment *_pigment def __cinit__(self): self._pigment = NULL def __init__(self, arg = None): + """ + Create a Pigment. + + With an arguement, create a solid pigment of that color. Otherwise, create + a base Pigment. + """ if arg is not None: - if (isinstance(arg, Pigment)): + if isinstance(arg, Pigment): self._pigment = (arg)._pigment DMNSN_INCREF(self._pigment) else: @@ -429,22 +553,34 @@ cdef class Pigment: dmnsn_delete_pigment(self._pigment) def transform(self, Matrix trans not None): + """Transform a pigment.""" if self._pigment == NULL: - raise TypeError('attempt to transform base Pigment') + raise TypeError("attempt to transform base Pigment") self._pigment.trans = dmnsn_matrix_mul(trans._m, self._pigment.trans) return self -cdef _rawPigment(dmnsn_pigment *pigment): +cdef _Pigment(dmnsn_pigment *pigment): cdef Pigment self = Pigment.__new__(Pigment) self._pigment = pigment DMNSN_INCREF(self._pigment) return self cdef class ColorMap(Pigment): + """A color map""" def __init__(self, Pattern pattern not None, map, bool sRGB not None = True): + """ + Create a ColorMap. + + Keyword arguments: + pattern -- the pattern to use for the mapping + map -- a dictionary of the form { val1: color1, val2: color2, ... }, + or a list of the form [color1, color2, ...] + sRGB -- whether the gradients should be in sRGB or linear space + (default True) + """ cdef dmnsn_map *color_map = dmnsn_new_color_map() - if hasattr(map, 'items'): + if hasattr(map, "items"): for i, color in map.items(): dmnsn_add_map_entry(color_map, i, &Color(color)._c) else: @@ -463,21 +599,32 @@ cdef class ColorMap(Pigment): Pigment.__init__(self) cdef class PigmentMap(Pigment): + """A pigment map.""" def __init__(self, Pattern pattern not None, map, bool sRGB not None = True): + """ + Create a PigmentMap. + + Keyword arguments: + pattern -- the pattern to use for the mapping + map -- a dictionary of the form { val1: color1, val2: pigment2, ... }, + or a list of the form [color1, pigment2, ...] + sRGB -- whether the gradients should be in sRGB or linear space + (default True) + """ cdef dmnsn_map *pigment_map = dmnsn_new_pigment_map() - cdef dmnsn_pigment *realPigment - if hasattr(map, 'items'): + cdef dmnsn_pigment *real_pigment + if hasattr(map, "items"): for i, pigment in map.items(): pigment = Pigment(pigment) - realPigment = (pigment)._pigment - DMNSN_INCREF(realPigment) - dmnsn_add_map_entry(pigment_map, i, &realPigment) + real_pigment = (pigment)._pigment + DMNSN_INCREF(real_pigment) + dmnsn_add_map_entry(pigment_map, i, &real_pigment) else: for i, pigment in enumerate(map): pigment = Pigment(pigment) - realPigment = (pigment)._pigment - DMNSN_INCREF(realPigment) - dmnsn_add_map_entry(pigment_map, i/len(map), &realPigment) + real_pigment = (pigment)._pigment + DMNSN_INCREF(real_pigment) + dmnsn_add_map_entry(pigment_map, i/len(map), &real_pigment) cdef dmnsn_pigment_map_flags flags if sRGB: @@ -495,6 +642,7 @@ cdef class PigmentMap(Pigment): ############ cdef class Finish: + """Object surface qualities.""" cdef dmnsn_finish _finish def __cinit__(self): @@ -504,33 +652,66 @@ cdef class Finish: dmnsn_delete_finish(self._finish) def __add__(Finish lhs not None, Finish rhs not None): + """ + Combine two finishes. + + In lhs + rhs, the attributes of rhs override those of lhs if any conflict; + thus, Ambient(0.1) + Ambient(0.2) is the same as Ambient(0.2) + """ cdef Finish ret = Finish() dmnsn_finish_cascade(&lhs._finish, &ret._finish) dmnsn_finish_cascade(&rhs._finish, &ret._finish) # rhs gets priority return ret -cdef _rawFinish(dmnsn_finish finish): +cdef _Finish(dmnsn_finish finish): cdef Finish self = Finish.__new__(Finish) self._finish = finish dmnsn_finish_incref(&self._finish) return self cdef class Ambient(Finish): + """Ambient light reflected.""" def __init__(self, color): + """ + Create an Ambient finish. + + Keyword arguments: + color -- the color and intensity of the ambient light + """ self._finish.ambient = dmnsn_new_basic_ambient(Color(color)._c) cdef class Diffuse(Finish): + """Lambertian diffuse reflection.""" def __init__(self, double diffuse): + """ + Create a Diffuse finish. + + Keyword arguments: + diffuse -- the intensity of the diffuse reflection + """ cdef dmnsn_color gray = dmnsn_color_mul(diffuse, dmnsn_white) diffuse = dmnsn_color_intensity(dmnsn_color_from_sRGB(gray)) self._finish.diffuse = dmnsn_new_lambertian(diffuse) cdef class Phong(Finish): + """Phong specular highlight.""" def __init__(self, double strength, double size = 40.0): + """ + Create a Phong highlight. + """ self._finish.specular = dmnsn_new_phong(strength, size) cdef class Reflection(Finish): + """Reflective finish.""" def __init__(self, min, max = None, double falloff = 1.0): + """ + Create a Reflection. + + Keyword arguments: + min -- color and intensity of reflection at indirect angles + max -- color and intensity of reflection at direct angles (default: min) + falloff -- exponent for reflection falloff (default: 1.0) + """ if max is None: max = min self._finish.reflection = dmnsn_new_basic_reflection(Color(min)._c, @@ -542,12 +723,19 @@ cdef class Reflection(Finish): ############ cdef class Texture: + """Object surface properties.""" cdef dmnsn_texture *_texture def __init__(self, pigment = None, Finish finish = None): + """ + Create a Texture. + + Keyword arguments: + pigment -- the Pigment for the texture, or a color (default: None) + finish -- the Finish for the texture (default: None) + """ self._texture = dmnsn_new_texture() - cdef Pigment realPigment if pigment is not None: self.pigment = Pigment(pigment) @@ -558,22 +746,24 @@ cdef class Texture: dmnsn_delete_texture(self._texture) property pigment: + """The texture's pigment.""" def __get__(self): - return _rawPigment(self._texture.pigment) + return _Pigment(self._texture.pigment) def __set__(self, Pigment pigment not None): dmnsn_delete_pigment(self._texture.pigment) self._texture.pigment = pigment._pigment DMNSN_INCREF(self._texture.pigment) property finish: + """The texture's finish.""" def __get__(self): - return _rawFinish(self._texture.finish) + return _Finish(self._texture.finish) def __set__(self, Finish finish not None): dmnsn_delete_finish(self._texture.finish) self._texture.finish = finish._finish dmnsn_finish_incref(&self._texture.finish) -cdef _rawTexture(dmnsn_texture *texture): +cdef _Texture(dmnsn_texture *texture): cdef Texture self = Texture.__new__(Texture) self._texture = texture DMNSN_INCREF(self._texture) @@ -584,16 +774,30 @@ cdef _rawTexture(dmnsn_texture *texture): ############# cdef class Interior: + """Object interior properties.""" cdef dmnsn_interior *_interior def __init__(self, double ior = 1.0): + """ + Create an Interior. + + Keyword arguments: + ior -- index of reflection + """ self._interior = dmnsn_new_interior() self._interior.ior = ior def __dealloc__(self): dmnsn_delete_interior(self._interior) -cdef _rawInterior(dmnsn_interior *interior): + property ior: + """Index of reflection.""" + def __get__(self): + return self._interior.ior + def __set__(self, double ior): + self._interior.ior = ior + +cdef _Interior(dmnsn_interior *interior): cdef Interior self = Interior.__new__(Interior) self._interior = interior DMNSN_INCREF(self._interior) @@ -604,12 +808,23 @@ cdef _rawInterior(dmnsn_interior *interior): ########### cdef class Object: + """Physical objects.""" cdef dmnsn_object *_object def __cinit__(self): self._object = NULL def __init__(self, Texture texture = None, Interior interior = None): + """ + Initialize an Object. + + Keyword arguments: + texture -- the object's Texture + interior -- the object's Interior + """ + if self._object == NULL: + raise TypeError("attempt to initialize base Object") + if texture is not None: self._object.texture = texture._texture DMNSN_INCREF(self._object.texture) @@ -621,37 +836,66 @@ cdef class Object: dmnsn_delete_object(self._object) def transform(self, Matrix trans not None): - if self._object == NULL: - raise TypeError('attempt to transform base Object') - + """Transform an object.""" self._object.trans = dmnsn_matrix_mul(trans._m, self._object.trans) return self # Transform an object without affecting the texture - cdef _intrinsicTransform(self, Matrix trans): + cdef _intrinsic_transform(self, Matrix trans): self._object.trans = dmnsn_matrix_mul(self._object.trans, trans._m) if self._object.texture != NULL: self._object.texture.trans = dmnsn_matrix_mul(self._object.texture.trans, trans.inverse()._m) cdef class Plane(Object): + """A plane.""" def __init__(self, normal, double distance, *args, **kwargs): + """ + Create a Plane. + + Keyword arguments: + normal -- a vector perpendicular to the plane + distance -- the distance from the origin to the plane, in the direction of + normal + + Additionally, Plane() accepts any arguments that Object() accepts. + """ self._object = dmnsn_new_plane(Vector(normal)._v) Object.__init__(self, *args, **kwargs) - self._intrinsicTransform(translate(distance*Vector(normal))) + self._intrinsic_transform(translate(distance*Vector(normal))) cdef class Sphere(Object): + """A sphere.""" def __init__(self, center, double radius, *args, **kwargs): + """ + Create a Sphere. + + Keyword arguments: + center -- the center of the sphere + radius -- the radius of the sphere + + Additionally, Sphere() accepts any arguments that Object() accepts. + """ self._object = dmnsn_new_sphere() Object.__init__(self, *args, **kwargs) cdef Matrix trans = translate(Vector(center)) trans *= scale(radius, radius, radius) - self._intrinsicTransform(trans) + self._intrinsic_transform(trans) cdef class Box(Object): + """An axis-aligned rectangular prism.""" def __init__(self, min, max, *args, **kwargs): + """ + Create a Box. + + Keyword arguments: + min -- the coordinate-wise minimal extent of the box + max -- the coordinate-wise maximal extent of the box + + Additionally, Box() accepts any arguments that Object() accepts. + """ self._object = dmnsn_new_cube() Object.__init__(self, *args, **kwargs) @@ -659,12 +903,25 @@ cdef class Box(Object): max = Vector(max) cdef Matrix trans = translate((max + min)/2) trans *= scale((max - min)/2) - self._intrinsicTransform(trans) + self._intrinsic_transform(trans) cdef class Cone(Object): - def __init__(self, bottom, double bottomRadius, top, double topRadius, + """A cone or cone slice.""" + def __init__(self, bottom, double bottom_radius, top, double top_radius = 0.0, bool open not None = False, *args, **kwargs): - self._object = dmnsn_new_cone(bottomRadius, topRadius, open) + """ + Create a Cone. + + Keyword arguments: + bottom -- the location of the bottom of the cone + bottom_radius -- the radius at the bottom of the cone + top -- the location of the top of the cone + top_radius -- the radius at the top of the cone/cone slice (default 0.0) + open -- whether to draw the cone cap(s) + + Additionally, Cone() accepts any arguments that Object() accepts. + """ + self._object = dmnsn_new_cone(bottom_radius, top_radius, open) Object.__init__(self, *args, **kwargs) # Lift the cone to start at the origin, then scale, rotate, and translate @@ -674,27 +931,59 @@ cdef class Cone(Object): cdef Matrix trans = translate(Y) trans = scale(1.0, dir.norm()/2, 1.0)*trans - trans = _rawMatrix(dmnsn_alignment_matrix(dmnsn_y, dir._v, dmnsn_x, dmnsn_z))*trans + trans = _Matrix(dmnsn_alignment_matrix(dmnsn_y, dir._v, dmnsn_x, dmnsn_z))*trans trans = translate(bottom)*trans - self._intrinsicTransform(trans) + self._intrinsic_transform(trans) cdef class Cylinder(Cone): + """A cylinder.""" def __init__(self, bottom, top, double radius, bool open not None = False): + """ + Create a Cylinder. + + Keyword arguments: + bottom -- the location of the bottom of the cylinder + top -- the location of the top of the cylinder + radius -- the radius of the cylinder + open -- whether to draw the cylinder caps + + Additionally, Cylinder() accepts any arguments that Object() accepts. + """ Cone.__init__(self, - bottom = bottom, bottomRadius = radius, - top = top, topRadius = radius, + bottom = bottom, bottom_radius = radius, + top = top, top_radius = radius, open = open) cdef class Torus(Object): - def __init__(self, double majorRadius, double minorRadius, *args, **kwargs): - self._object = dmnsn_new_torus(majorRadius, minorRadius) + """A torus.""" + def __init__(self, double major_radius, double minor_radius, *args, **kwargs): + """ + Create a Torus. + + Keyword arguments: + major_radius -- the distance from the center of the torus to the center of + a circular cross-section of the torus + minor_radius -- the radius of the circular cross-sections of the torus + + Additionally, Torus() accepts any arguments that Object() accepts. + """ + self._object = dmnsn_new_torus(major_radius, minor_radius) Object.__init__(self, *args, **kwargs) cdef class Union(Object): + """A CSG union.""" def __init__(self, objects, *args, **kwargs): + """ + Create a Union. + + Keyword arguments: + objects -- a list of objects to include in the union + + Additionally, Union() accepts any arguments that Object() accepts. + """ if len(objects) < 2: - raise TypeError('expected a list of two or more Objects') + raise TypeError("expected a list of two or more Objects") cdef dmnsn_array *array = dmnsn_new_array(sizeof(dmnsn_object *)) cdef dmnsn_object *o @@ -712,9 +1001,18 @@ cdef class Union(Object): Object.__init__(self, *args, **kwargs) cdef class Intersection(Object): + """A CSG intersection.""" def __init__(self, objects, *args, **kwargs): + """ + Create an Intersection. + + Keyword arguments: + objects -- a list of objects to include in the intersection + + Additionally, Intersection() accepts any arguments that Object() accepts. + """ if len(objects) < 2: - raise TypeError('expected a list of two or more Objects') + raise TypeError("expected a list of two or more Objects") cdef dmnsn_object *o @@ -730,9 +1028,18 @@ cdef class Intersection(Object): Object.__init__(self, *args, **kwargs) cdef class Difference(Object): + """A CSG difference.""" def __init__(self, objects, *args, **kwargs): + """ + Create a Difference. + + Keyword arguments: + objects -- a list of objects to include in the difference + + Additionally, Difference() accepts any arguments that Object() accepts. + """ if len(objects) < 2: - raise TypeError('expected a list of two or more Objects') + raise TypeError("expected a list of two or more Objects") cdef dmnsn_object *o @@ -748,9 +1055,18 @@ cdef class Difference(Object): Object.__init__(self, *args, **kwargs) cdef class Merge(Object): + """A CSG merge.""" def __init__(self, objects, *args, **kwargs): + """ + Create a Merge. + + Keyword arguments: + objects -- a list of objects to include in the merge + + Additionally, Merge() accepts any arguments that Object() accepts. + """ if len(objects) < 2: - raise TypeError('expected a list of two or more Objects') + raise TypeError("expected a list of two or more Objects") cdef dmnsn_object *o @@ -770,13 +1086,22 @@ cdef class Merge(Object): ########## cdef class Light: + """A light.""" cdef dmnsn_light *_light def __dealloc__(self): dmnsn_delete_light(self._light) cdef class PointLight(Light): + """A point light.""" def __init__(self, location, color): + """ + Create a PointLight. + + Keyword arguments: + location -- the origin of the light rays + color -- the color and intensity of the light + """ # Take the sRGB component because "color = 0.5*White" should really mean # a half-intensity white light self._light = dmnsn_new_point_light(Vector(location)._v, Color(color)._sRGB) @@ -787,6 +1112,7 @@ cdef class PointLight(Light): ########### cdef class Camera: + """A camera.""" cdef dmnsn_camera *_camera def __cinit__(self): @@ -796,33 +1122,44 @@ cdef class Camera: dmnsn_delete_camera(self._camera) def transform(self, Matrix trans not None): + """Transform a camera.""" if self._camera == NULL: - raise TypeError('attempt to transform base Camera') + raise TypeError("attempt to transform base Camera") self._camera.trans = dmnsn_matrix_mul(trans._m, self._camera.trans) return self cdef class PerspectiveCamera(Camera): - def __init__(self, location = -Z, lookAt = 0, sky = Y, - angle = dmnsn_degrees(atan(0.5))): + """A regular perspective camera.""" + def __init__(self, location = -Z, look_at = 0, sky = Y, + angle = dmnsn_degrees(atan(1.0))): + """ + Create a PerspectiveCamera. + + Keyword arguments: + location -- the location of the camera (default: -Z) + look_at -- where to aim the camera (default: 0) + sky -- the direction of the top of the camera (default: Y) + angle -- the field of view angle (from bottom to top) (default: 45) + """ self._camera = dmnsn_new_perspective_camera() Camera.__init__(self) # Apply the field of view angle - self.transform(scale(2*tan(dmnsn_radians(angle))*(X + Y) + Z)) + self.transform(scale(tan(dmnsn_radians(angle))*(X + Y) + Z)) - cdef Vector dir = Vector(lookAt) - Vector(location) + cdef Vector dir = Vector(look_at) - Vector(location) cdef Vector vsky = Vector(sky) # Line up the top of the viewport with the sky vector - cdef Matrix alignSky = _rawMatrix(dmnsn_alignment_matrix(dmnsn_y, vsky._v, - dmnsn_z, dmnsn_x)) - cdef Vector forward = alignSky*Z - cdef Vector right = alignSky*X + cdef Matrix align_sky = _Matrix(dmnsn_alignment_matrix(dmnsn_y, vsky._v, + dmnsn_z, dmnsn_x)) + cdef Vector forward = align_sky*Z + cdef Vector right = align_sky*X - # Line up the look at point with lookAt - self.transform(_rawMatrix(dmnsn_alignment_matrix(forward._v, dir._v, - vsky._v, right._v))) + # Line up the look at point with look_at + self.transform(_Matrix(dmnsn_alignment_matrix(forward._v, dir._v, + vsky._v, right._v))) # Move the camera into position self.transform(translate(Vector(location))) @@ -832,22 +1169,31 @@ cdef class PerspectiveCamera(Camera): ############### cdef class SkySphere: - cdef dmnsn_sky_sphere *_skySphere + """A scene background.""" + cdef dmnsn_sky_sphere *_sky_sphere def __init__(self, pigments): - self._skySphere = dmnsn_new_sky_sphere() + """ + Create a SkySphere. + + Keyword arguments: + pigments -- the list of pigments that make up the background, in back-to- + front order + """ + self._sky_sphere = dmnsn_new_sky_sphere() - cdef Pigment realPigment + cdef Pigment real_pigment for pigment in pigments: - realPigment = Pigment(pigment) - DMNSN_INCREF(realPigment._pigment) - dmnsn_array_push(self._skySphere.pigments, &realPigment._pigment) + real_pigment = Pigment(pigment) + DMNSN_INCREF(real_pigment._pigment) + dmnsn_array_push(self._sky_sphere.pigments, &real_pigment._pigment) def __dealloc__(self): - dmnsn_delete_sky_sphere(self._skySphere) + dmnsn_delete_sky_sphere(self._sky_sphere) def transform(self, Matrix trans not None): - self._skySphere.trans = dmnsn_matrix_mul(trans._m, self._skySphere.trans) + """Transform a sky sphere.""" + self._sky_sphere.trans = dmnsn_matrix_mul(trans._m, self._sky_sphere.trans) return self ########## @@ -855,10 +1201,20 @@ cdef class SkySphere: ########## cdef class Scene: + """An entire scene.""" cdef dmnsn_scene *_scene def __init__(self, Canvas canvas not None, objects, lights, Camera camera not None): + """ + Create a Scene. + + Keyword arguments: + canvas -- the rendering Canvas + objects -- the list of objects in the scene + lights -- the list of lights in the scene + camera -- the camera for the scene + """ self._scene = dmnsn_new_scene() self._scene.canvas = canvas._canvas @@ -884,59 +1240,75 @@ cdef class Scene: self._scene.camera = camera._camera DMNSN_INCREF(self._scene.camera) - property defaultTexture: + property default_texture: + """The default Texture for objects.""" def __get__(self): - return _rawTexture(self._scene.default_texture) + return _Texture(self._scene.default_texture) def __set__(self, Texture texture not None): dmnsn_delete_texture(self._scene.default_texture) self._scene.default_texture = texture._texture DMNSN_INCREF(self._scene.default_texture) - property defaultInterior: + property default_interior: + """The default Interior for objects.""" def __get__(self): - return _rawInterior(self._scene.default_interior) + return _Interior(self._scene.default_interior) def __set__(self, Interior interior not None): dmnsn_delete_interior(self._scene.default_interior) self._scene.default_interior = interior._interior DMNSN_INCREF(self._scene.default_interior) property background: + """The solid background color of the scene (default: Black).""" def __get__(self): - return _rawColor(self._scene.background) + return _Color(self._scene.background) def __set__(self, color): self._scene.background = Color(color)._c - property skySphere: - def __set__(self, SkySphere skySphere not None): + property sky_sphere: + """The background sky pattern of the scene.""" + def __set__(self, SkySphere sky_sphere not None): dmnsn_delete_sky_sphere(self._scene.sky_sphere) - self._scene.sky_sphere = skySphere._skySphere + self._scene.sky_sphere = sky_sphere._sky_sphere DMNSN_INCREF(self._scene.sky_sphere) - property adcBailout: + property adc_bailout: + """The adaptive depth control bailout (default: 1/255).""" def __get__(self): return self._scene.adc_bailout def __set__(self, double bailout): self._scene.adc_bailout = bailout - property recursionLimit: + property recursion_limit: + """The rendering recursion limit (default: 5).""" def __get__(self): return self._scene.reclimit def __set__(self, level): self._scene.reclimit = level - property nThreads: + property nthreads: + """The number of threads to use for the render.""" def __get__(self): return self._scene.nthreads def __set__(self, n): self._scene.nthreads = n - property boundingTimer: + property bounding_timer: + """The Timer for building the bounding hierarchy.""" def __get__(self): - return _rawTimer(self._scene.bounding_timer) - property renderTimer: + if self._scene.bounding_timer == NULL: + raise RuntimeError('scene has not been rendered yet') + + return _Timer(self._scene.bounding_timer) + property render_timer: + """The Timer for the actual render.""" def __get__(self): - return _rawTimer(self._scene.render_timer) + if self._scene.render_timer == NULL: + raise RuntimeError('scene has not been rendered yet') + + return _Timer(self._scene.render_timer) def raytrace(self): + """Render the scene.""" # Ensure the default texture is complete cdef Texture default = Texture(Black) dmnsn_texture_cascade(default._texture, &self._scene.default_texture) diff --git a/libdimension-python/tests/canvas.py b/libdimension-python/tests/canvas.py index e69d2d9..97f322d 100755 --- a/libdimension-python/tests/canvas.py +++ b/libdimension-python/tests/canvas.py @@ -23,25 +23,25 @@ from dimension import * import errno # Treat warnings as errors for tests -dieOnWarnings(True) +die_on_warnings(True) canvas = Canvas(768, 480) assert canvas.width == 768, canvas.width assert canvas.height == 480, canvas.height -havePNG = True +have_PNG = True try: - canvas.optimizePNG() + canvas.optimize_PNG() except OSError as e: if e.errno == errno.ENOSYS: havePNG = False else: raise -haveGL = True +have_GL = True try: - canvas.optimizeGL() + canvas.optimize_GL() except OSError as e: if e.errno == errno.ENOSYS: haveGL = False @@ -50,8 +50,8 @@ except OSError as e: canvas.clear(Blue) -if havePNG: - canvas.writePNG('png.png') +if have_PNG: + canvas.write_PNG('png.png') #if haveGL: # canvas.drawGL() diff --git a/libdimension-python/tests/color.py b/libdimension-python/tests/color.py index 2d8545d..bdfb65f 100755 --- a/libdimension-python/tests/color.py +++ b/libdimension-python/tests/color.py @@ -22,7 +22,7 @@ from dimension import * # Treat warnings as errors for tests -dieOnWarnings(True) +die_on_warnings(True) c = Color(0, 0.5, 1, trans = 0.5, filter = 0.35) assert repr(c) == 'dimension.Color(0.0, 0.5, 1.0, 0.5, 0.35)', repr(c) diff --git a/libdimension-python/tests/demo.py b/libdimension-python/tests/demo.py index c024812..d73410f 100755 --- a/libdimension-python/tests/demo.py +++ b/libdimension-python/tests/demo.py @@ -22,23 +22,23 @@ from dimension import * # Treat warnings as errors for tests -dieOnWarnings(True) +die_on_warnings(True) # Canvas canvas = Canvas(width = 768, height = 480) -havePNG = True +have_PNG = True try: - canvas.optimizePNG() + canvas.optimize_PNG() except OSError as e: if e.errno == errno.ENOSYS: - havePNG = False + have_PNG = False else: raise # Camera camera = PerspectiveCamera(location = (0, 0.25, -4), - lookAt = 0) + look_at = 0) camera.transform(rotate(53*Y)) # Lights @@ -48,7 +48,7 @@ lights = [ # Objects -hollowCube = Difference( +hollow_cube = Difference( [ Box( (-1, -1, -1), (1, 1, 1), @@ -77,8 +77,8 @@ arrow = Union( [ Cylinder(bottom = -1.25*Y, top = 1.25*Y, radius = 0.1), Cone( - bottom = 1.25*Y, bottomRadius = 0.1, - top = 1.5*Y, topRadius = 0, + bottom = 1.25*Y, bottom_radius = 0.1, + top = 1.5*Y, top_radius = 0, open = True ), ], @@ -103,12 +103,12 @@ arrow.transform(rotate(-45*X)) torii = Union( [ - Torus(majorRadius = 0.15, minorRadius = 0.05) + Torus(major_radius = 0.15, minor_radius = 0.05) .transform(translate(-Y)), - Torus(majorRadius = 0.15, minorRadius = 0.05), + Torus(major_radius = 0.15, minor_radius = 0.05), - Torus(majorRadius = 0.15, minorRadius = 0.05) + Torus(major_radius = 0.15, minor_radius = 0.05) .transform(translate(Y)), ], texture = Texture( @@ -126,21 +126,21 @@ ground = Plane( Checker(), [ White, - ColorMap(Checker(), [Black, White]).transform(scale(1/3, 1/3, 1/3)) + ColorMap(Checker(), [Black, White]).transform(scale(1/3)) ], ), ), ) objects = [ - hollowCube, + hollow_cube, arrow, torii, ground, ] # Sky sphere -skySphere = SkySphere( +sky_sphere = SkySphere( [ ColorMap( pattern = Gradient(Y), @@ -157,12 +157,12 @@ scene = Scene(canvas = canvas, objects = objects, lights = lights, camera = camera) -scene.defaultTexture = Texture(finish = Ambient(0.1) + Diffuse(0.6)) -scene.background = Clear -scene.skySphere = skySphere -scene.adcBailout = 1/255 -scene.recursionLimit = 5 +scene.default_texture = Texture(finish = Ambient(0.1) + Diffuse(0.6)) +scene.background = Clear +scene.sky_sphere = sky_sphere +scene.adc_bailout = 1/255 +scene.recursion_limit = 5 scene.raytrace() -if havePNG: - canvas.writePNG('demo.png') +if have_PNG: + canvas.write_PNG('demo.png') diff --git a/libdimension-python/tests/geometry.py b/libdimension-python/tests/geometry.py index c1d4f48..e18d17a 100755 --- a/libdimension-python/tests/geometry.py +++ b/libdimension-python/tests/geometry.py @@ -22,7 +22,7 @@ from dimension import * # Treat warnings as errors for tests -dieOnWarnings(True) +die_on_warnings(True) assert 0 == Vector(0, 0, 0), Vector(0) assert X == Vector(1, 0, 0), X @@ -66,10 +66,10 @@ assert str(m) == '\n' \ '[9.0\t10.0\t11.0\t12.0]\n' \ '[0.0\t0.0\t0.0\t1.0]', str(m) -s = scale(Vector(1, 2, 3)) -assert s == Matrix(1, 0, 0, 0, - 0, 2, 0, 0, - 0, 0, 3, 0), s +s = scale(0.5) +assert s == Matrix(0.5, 0, 0, 0, + 0, 0.5, 0, 0, + 0, 0, 0.5, 0), s t = translate(1, 2, 3) assert t == Matrix(1, 0, 0, 1, -- cgit v1.2.3