home *** CD-ROM | disk | FTP | other *** search
/ Hackers Magazine 57 / CdHackersMagazineNr57.iso / Software / Multimedia / k3d-setup-0.7.11.0.exe / lib / site-packages / OpenGL / GLU / tess.py < prev   
Encoding:
Python Source  |  2008-12-07  |  7.7 KB  |  217 lines
  1. """Wrapper/Implementation of the GLU tessellator objects for PyOpenGL"""
  2. from OpenGL.raw import GLU as simple
  3. from OpenGL.platform import GLU,createBaseFunction
  4. from OpenGL.GLU import glustruct
  5. from OpenGL import arrays, constants
  6. from OpenGL.platform import PLATFORM
  7. from OpenGL.lazywrapper import lazy
  8. import ctypes
  9.  
  10. class GLUtesselator( glustruct.GLUStruct, simple.GLUtesselator):
  11.     """Implementation class for GLUTessellator structures in OpenGL-ctypes"""
  12.     FUNCTION_TYPE = PLATFORM.functionTypeFor(PLATFORM.GLU)
  13.     CALLBACK_TYPES = {
  14.         # mapping from "which" GLU enumeration to a ctypes function type
  15.         simple.GLU_TESS_BEGIN: FUNCTION_TYPE( None, simple.GLenum ),
  16.         simple.GLU_TESS_BEGIN_DATA: FUNCTION_TYPE( 
  17.             None, simple.GLenum, ctypes.c_void_p 
  18.         ),
  19.         simple.GLU_TESS_EDGE_FLAG: FUNCTION_TYPE( None, simple.GLboolean),
  20.         simple.GLU_TESS_EDGE_FLAG_DATA: FUNCTION_TYPE( 
  21.             None, simple.GLboolean, ctypes.c_void_p 
  22.         ),
  23.         simple.GLU_TESS_VERTEX: FUNCTION_TYPE( None, ctypes.c_void_p ),
  24.         simple.GLU_TESS_VERTEX_DATA: FUNCTION_TYPE( 
  25.             None, ctypes.c_void_p, ctypes.c_void_p
  26.         ),
  27.         simple.GLU_TESS_END: FUNCTION_TYPE( None ),
  28.         simple.GLU_TESS_END_DATA: FUNCTION_TYPE( None, ctypes.c_void_p), 
  29.         simple.GLU_TESS_COMBINE: FUNCTION_TYPE( 
  30.             None, 
  31.             ctypes.POINTER(simple.GLdouble), 
  32.             ctypes.POINTER(ctypes.c_void_p), 
  33.             ctypes.POINTER(simple.GLfloat),
  34.             ctypes.POINTER(ctypes.c_void_p)
  35.         ),
  36.         simple.GLU_TESS_COMBINE_DATA: FUNCTION_TYPE( 
  37.             None, 
  38.             ctypes.POINTER(simple.GLdouble), 
  39.             ctypes.POINTER(ctypes.c_void_p), 
  40.             ctypes.POINTER(simple.GLfloat),
  41.             ctypes.POINTER(ctypes.c_void_p),
  42.             ctypes.c_void_p,
  43.         ),
  44.         simple.GLU_TESS_ERROR: FUNCTION_TYPE( None, simple.GLenum), 
  45.         simple.GLU_TESS_ERROR_DATA: FUNCTION_TYPE( 
  46.             None, simple.GLenum, ctypes.c_void_p
  47.         ), 
  48.         simple.GLU_ERROR : FUNCTION_TYPE( None, simple.GLenum )
  49.     }
  50.     WRAPPER_METHODS = {
  51.         simple.GLU_TESS_BEGIN_DATA: 'dataWrapper',
  52.         simple.GLU_TESS_EDGE_FLAG_DATA: 'dataWrapper',
  53.         simple.GLU_TESS_VERTEX: 'vertexWrapper',
  54.         simple.GLU_TESS_VERTEX_DATA: 'vertexWrapper',
  55.         simple.GLU_TESS_END_DATA: 'dataWrapper', 
  56.         simple.GLU_TESS_COMBINE: 'combineWrapper',
  57.         simple.GLU_TESS_COMBINE_DATA: 'combineWrapper',
  58.         simple.GLU_TESS_ERROR_DATA: 'dataWrapper', 
  59.     }
  60.     def gluTessVertex( self, location, data=None ):
  61.         """Add a vertex to this tessellator, storing data for later lookup"""
  62.         vertexCache = getattr( self, 'vertexCache', None )
  63.         if vertexCache is None:
  64.             self.vertexCache = []
  65.             vertexCache = self.vertexCache
  66.         location = arrays.GLdoubleArray.asArray( location, constants.GL_DOUBLE )
  67.         if arrays.GLdoubleArray.arraySize( location ) != 3:
  68.             raise ValueError( """Require 3 doubles for array location, got: %s"""%(location,))
  69.         oorValue = self.noteObject(data)
  70.         vp = ctypes.c_void_p( oorValue )
  71.         self.vertexCache.append( location )
  72.         return gluTessVertexBase( self, location, vp )
  73.     def gluTessBeginPolygon( self, data ):
  74.         """Note the object pointer to return it as a Python object"""
  75.         return simple.gluTessBeginPolygon( 
  76.             self, ctypes.c_void_p(self.noteObject( data ))
  77.         )
  78.     def combineWrapper( self, function ):
  79.         """Wrap a Python function with ctypes-compatible wrapper for combine callback
  80.         
  81.         For a Python combine callback, the signature looks like this:
  82.             def combine(
  83.                 GLdouble coords[3], 
  84.                 void *vertex_data[4], 
  85.                 GLfloat weight[4]
  86.             ):
  87.                 return data
  88.         While the C signature looks like this:
  89.             void combine( 
  90.                 GLdouble coords[3], 
  91.                 void *vertex_data[4], 
  92.                 GLfloat weight[4], 
  93.                 void **outData 
  94.             )
  95.         """
  96.         if (function is not None) and (not callable( function )):
  97.             raise TypeError( """Require a callable callback, got:  %s"""%(function,))
  98.         def wrap( coords, vertex_data, weight, outData, *args ):
  99.             """The run-time wrapper around the function"""
  100.             coords = self.ptrAsArray( coords, 3, arrays.GLdoubleArray )
  101.             weight = self.ptrAsArray( weight, 4, arrays.GLfloatArray )
  102.             # find the original python objects for vertex data
  103.             vertex_data = [ self.originalObject( vertex_data[i] ) for i in range(4) ]
  104.             args = tuple( [ self.originalObject( x ) for x in args ] )
  105.             try:
  106.                 result = function( coords, vertex_data, weight, *args )
  107.             except Exception, err:
  108.                 raise err.__class__(
  109.                     """Failure during combine callback %r with args( %s,%s,%s,*%s):\n%s"""%(
  110.                         function, coords, vertex_data, weight, args, str(err),
  111.                     )
  112.                 )
  113.             outP = ctypes.c_void_p(self.noteObject(result))
  114.             outData.value = ctypes.addressof( outP )
  115.             return None
  116.         return wrap
  117.     def dataWrapper( self, function ):
  118.         """Wrap a function which only has the one data-pointer as last arg"""
  119.         if (function is not None) and (not callable( function )):
  120.             raise TypeError( """Require a callable callback, got:  %s"""%(function,))
  121.         def wrap( *args ):
  122.             """Just return the original object for polygon_data"""
  123.             args = args[:-1] + ( self.originalObject(args[-1]), )
  124.             try:
  125.                 return function( *args )
  126.             except Exception, err:
  127.                 err.args += (function,args)
  128.                 raise
  129.         return wrap
  130.     def dataWrapper2( self, function ):
  131.         """Wrap a function which has two data-pointers as last args"""
  132.         if (function is not None) and (not callable( function )):
  133.             raise TypeError( """Require a callable callback, got:  %s"""%(function,))
  134.         def wrap( *args ):
  135.             """Just return the original object for polygon_data"""
  136.             args = args[:-2] + ( self.originalObject(args[-2]), self.originalObject(args[-1]), )
  137.             try:
  138.                 return function( *args )
  139.             except Exception, err:
  140.                 err.args += (function,args)
  141.                 raise
  142.         return wrap
  143.     def vertexWrapper( self, function ):
  144.         """Converts a vertex-pointer into an OOR vertex for processing"""
  145.         if (callable is not None) and (not callable( function )):
  146.             raise TypeError( """Require a callable callback, got:  %s"""%(function,))
  147.         def wrap( vertex, data=None ):
  148.             """Just return the original object for polygon_data"""
  149.             vertex = self.originalObject(vertex)
  150.             try:
  151.                 if data is not None:
  152.                     data = self.originalObject(data)
  153.                     return function( vertex, data )
  154.                 else:
  155.                     return function( vertex )
  156.             except Exception, err:
  157.                 err.args += (function,(vertex,data))
  158.                 raise
  159.         return wrap
  160.  
  161. GLUtesselator.CALLBACK_FUNCTION_REGISTRARS = dict([
  162.     (c,createBaseFunction( 
  163.         'gluTessCallback', dll=GLU, resultType=None, 
  164.         argTypes=[ctypes.POINTER(GLUtesselator), simple.GLenum,funcType],
  165.         doc='gluTessCallback( POINTER(GLUtesselator)(tess), GLenum(which), _GLUfuncptr(CallBackFunc) ) -> None', 
  166.         argNames=('tess', 'which', 'CallBackFunc'),
  167.     ))
  168.     for (c,funcType) in GLUtesselator.CALLBACK_TYPES.items()
  169. ])
  170. del c, funcType
  171.  
  172. def gluTessCallback( tess, which, function ):
  173.     """Set a given gluTessellator callback for the given tessellator"""
  174.     return tess.addCallback( which, function )
  175. def gluTessBeginPolygon( tess, data ):
  176.     """Start definition of polygon in the tessellator"""
  177.     return tess.gluTessBeginPolygon( data )
  178. def gluTessVertex( tess, location, data=None ):
  179.     """Add a vertex to the tessellator's current polygon"""
  180.     return tess.gluTessVertex( location, data )
  181.  
  182. # /usr/include/GL/glu.h 293
  183. @lazy( 
  184.     createBaseFunction( 
  185.         'gluNewTess', dll=GLU, resultType=ctypes.POINTER(GLUtesselator), 
  186.         doc='gluNewTess(  ) -> POINTER(GLUtesselator)', 
  187.     ) 
  188. )
  189. def gluNewTess( baseFunction ):
  190.     """Get a new tessellator object (just unpacks the pointer for you)"""
  191.     return baseFunction()[0]
  192.  
  193. @lazy( simple.gluGetTessProperty )
  194. def gluGetTessProperty( baseFunction, tess, which, data=None ):
  195.     """Retrieve single double for a tessellator property"""
  196.     if data is None:
  197.         data = simple.GLdouble( 0.0 )
  198.         baseFunction( tess, which, data )
  199.         return data.value 
  200.     else:
  201.         return baseFunction( tess, which, data )
  202.  
  203. gluTessVertexBase = arrays.setInputArraySizeType(
  204.     simple.gluTessVertex,
  205.     3,
  206.     arrays.GLdoubleArray,
  207.     'location',
  208. )
  209.  
  210. __all__ = (
  211.     'gluNewTess',
  212.     'gluGetTessProperty',
  213.     'gluTessBeginPolygon',
  214.     'gluTessCallback',
  215.     'gluTessVertex',
  216. )
  217.