a ij6) @sdZddlmZddlmZmZmZddlm Z ddl m Z ddlm Z mZddlZddlZddlmZddlZdd lmZd ZGd d d e jejZeeej_d dZejD]>\ZZeeZeej e<eejeksJej!deksJqz [[[Wn&e"y&Z#z WYdZ#[#n dZ#[#00ddZ$e ejddZe ej%ddZ%dZ&ddZ'ddZ(e ej)ddZ)e ej*ddZ*e ej+dd Z+e ej,d!d"Z,dS)#aImplementation of GLU Nurbs structure and callback methods Same basic pattern as seen with the gluTess* functions, just need to add some bookkeeping to the structure class so that we can keep the Python function references alive during the calling process. )GLU)platform converterswrapper) glustruct)lazy)arrayserrorN)PLATFORM) _configflags)GLUnurbsgluNewNurbsRenderergluNurbsCallbackgluNurbsCallbackDatagluNurbsCallbackDataEXT gluNurbsCurvegluNurbsSurface gluPwlCurvec@seZdZdZeejZiZe j ede j e j ede j e e je jede e je jede e je e je jede e je jede e je e je jede e je jede e je e je jede e je jede e je e je jede jede e je jede j i Ze j de j de jde jde jde jde jde jde jde jde jde jde jdi ZddZdd d Zd d Z ddZ!dS)r aGLU Nurbs structure with oor and callback storage support IMPORTANT NOTE: the texture coordinate callback receives a raw ctypes data-pointer, as without knowing what type of evaluation is being done (1D or 2D) we cannot safely determine the size of the array to convert it. This is a limitation of the C implementation. To convert to regular data-pointer, just call yourNurb.ptrAsArray( ptr, size, arrays.GLfloatArray ) with the size of data you expect. N_justOOR_vec3_vec4_texcsfdd}|S)z#Just do OOR on the last argument...cs$|dd|df}|S)N)originalObject)argsfunctionselfQ/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/OpenGL/GLU/glunurbs.pygetOORhsz!GLUnurbs._justOOR..getOORr)rrr rrrrfszGLUnurbs._justOORcsfdd}|S)zBConvert first arg to size-element array, do OOR on arg2 if presentcsD|dtj}t|dkr8|d}||S|SdS)Nr)Z ptrAsArrayr GLfloatArraylenr)rvecoorrrsizerrr%ns   zGLUnurbs._vec3..vecr)rrr(r%rr'rrlszGLUnurbs._vec3cCs ||dS)zSize-4 vector version...)r)rrrrrrvszGLUnurbs._vec4csfdd}|S)zTexture coordinate callback NOTE: there is no way for *us* to tell what size the array is, you will get back a raw data-point, not an array, as you do for all other callback types!!! cs8t|dkr(|d}|d|S|dSdS)Nr"r)r$r)rr&rrrr&s zGLUnurbs._tex..oorr)rrr&rrrrysz GLUnurbs._tex)r!)"__name__ __module__ __qualname____doc__r ZfunctionTypeForrZ FUNCTION_TYPECALLBACK_FUNCTION_REGISTRARS_simpleZGLU_NURBS_BEGINGLenumZGLU_NURBS_BEGIN_DATActypesPOINTERZGLvoidZGLU_NURBS_VERTEXZGLfloatZGLU_NURBS_VERTEX_DATAZGLU_NURBS_NORMALZGLU_NURBS_NORMAL_DATAZGLU_NURBS_COLORZGLU_NURBS_COLOR_DATAZGLU_NURBS_TEXTURE_COORDZGLU_NURBS_TEXTURE_COORD_DATAZ GLU_NURBS_ENDZGLU_NURBS_END_DATAZGLU_NURBS_ERRORCALLBACK_TYPESZWRAPPER_METHODSrrrrrrrrr sz      + r cCs6ttj}tttj|g|_|jd|ks2J|S)z4Get gluNurbsCallback function with set last arg-typer) rZcopyBaseFunctionr/rr1r2r r0argtypes)funcTyperesultrrr_callbackWithTypes r7rcCs |||S)z,Dispatch to the nurb's addCallback operation)Z addCallback)nurbwhichZ CallBackFuncrrrrsrcCs|}|d}|S)zAReturn a new nurbs renderer for the system (dereferences pointer)rr) baseFunctionZnewSetnewrrrr sr cCs||||Sz6Note the Python object for use as userData by the nurbZ noteObjectr:r8ZuserDatarrrrs rcCsb|dkrtd||fnB|tkr:td|t|fn$|d|kr^td||d|fdS)zCheck that order is valid...r"z%s should be 1 or more, is %sz%s should be %s or less, is %sz|d}|ddD] }||krtd|||fqdS)z'Check that knots are in ascending orderrr"Nz"%s has decreasing knot %s after %s)r$r rA)knotsrEZknotnextrrr checkKnotss rIcCs||||Sr<r=r>rrrrs rc Cstj|}tj|}tj|}ztj|\}}Wn.tyf}ztdWYd}~n d}~00||} tj rt | |dt |d||||||| |S)zaPythonic version of gluNurbsCurve Calculates knotCount, stride, and order automatically z"Need a 2-dimensional control arrayNzorder of NURBS curvezknots of NURBS curve rr#asArray arraySize dimensions ValueErrorr rAr ZERROR_CHECKINGrFrI) r:r8rGcontroltyperDlengthsteperrrCrrrrs      rcCs*tj|}tj|}tj|}tj|}tj|}ztj|\}} } Wn.ty} ztdWYd} ~ n d} ~ 00||} || } | | }| }tj rt | |dt | |dt |dt |d|| || || ks td||||||||| | | |||||||||| | | }|S)zcPythonic version of gluNurbsSurface Calculates knotCount, stride, and order automatically z"Need a 3-dimensional control arrayNzsOrder of NURBS surfaceztOrder of NURBS surfacezsKnots of NURBS surfaceztKnots of NURBS surfacezInvalid NURB structurerJ)r:r8ZsKnotsZtKnotsrOrPZ sKnotCountZ tKnotCountrQwidthrRrSZsOrderZtOrderZsStrideZtStrider6rrrrsB            rcCs`tj|}|tjkrd}n|tjkr,d}n td|tj|}t||}||||||S)zgluPwlCurve -- piece-wise linear curve within GLU context data -- the data-array type -- determines number of elements/data-point r@r!zUnrecognised type constant: %s) rr#rKr/ZGLU_MAP1_TRIM_2ZGLU_MAP_TRIM_3rNrLint)r:r8datarPZdivisorr(rrrrs      r)-r-Z OpenGL.rawrr/ZOpenGLrrrZ OpenGL.GLUrZOpenGL.lazywrapperrZ_lazyrr r1weakrefZOpenGL.platformr r __all__Z GLUStructr r2r restyper7r3itemscr5cbr.r4 NameErrorrSrrrBrFrIrrrrrrrrsN      k         )