a ij @sdZddlZddlZddlmZddlmZmZmZddlmZddl m Z m Z ddl Z e dZddlmZdZejrz0dd lmZdd lmZdd lmZmZmZWn.eyZzed WYdZ[n dZ[00edur"d dZGdddejZGdddejZGdddejZnedZejs>d#ddZn d$ddZej sVeZ!nerbeZ!nddZ!ejs|ddZ"nddZ"Gddde#Z$d%dd Z%d!d"Z&dS)&zHelper functions for wrapping array-using operations These are functions intended to be used in wrapping GL functions that deal with OpenGL array data-types. N) _configflags) contextdataerror converters) arraydatatype)bytesunicodezOpenGL.arrays.arrayhelpers)acceleratesupport)AsArrayTypedSizeChecked)returnPyArgumentIndex) AsArrayOfType AsArrayTypedAsArrayTypedSizez>Unable to load arrayhelpers accelerator from OpenGL_acceleratecCs|dS)zwReturn the converted object as result of function Note: this is a hack that always returns pyArgs[0]! r)result baseOperationpyArgscArgsrrX/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/OpenGL/arrays/arrayhelpers.py returnPointer s rc@s*eZdZdZdZdZd ddZdd Zd S) r zGiven arrayName and typeName coerce arrayName to array of type typeName TODO: It should be possible to drop this if ERROR_ON_COPY, as array inputs always have to be the final objects in that case.  arrayNametypeName) arrayIndexr pyArgIndex) typeIndexrrpointertypecCs||_||_dSNr)selfrrrrr__init__:szAsArrayOfType.__init__cCs||j}tj|}||Sz/Get the arg as an array of the appropriate type)rrZGL_CONSTANT_TO_ARRAY_TYPEasArray)r argwrappedOperationargsr arrayTyperrr__call__>s  zAsArrayOfType.__call__N)rr__name__ __module__ __qualname____doc__ZargNamesZ indexLookupsr!r(rrrrr ,s  r c@s*eZdZdZdZdZd ddZdd ZdS) r zGiven arrayName and arrayType, convert arrayName to array of type TODO: It should be possible to drop this if ERROR_ON_COPY, as array inputs always have to be the final objects in that case. rr'rrNcCs||_||_dSrr.r rr'rrrr!OszAsArrayTyped.__init__cCs |j|Sr")r'r#)r r$r%r&rrrr(SszAsArrayTyped.__call__)rNr)rrrrr Ds  r c@s*eZdZdZdZdZd ddZdd ZdS) rz:Given arrayName and arrayType, determine size of arrayNamer.r/rNcCs||_||_dSrr.r0rrrr!]szAsArrayTypedSize.__init__cCs|j||jSr")r'Z arraySizer)r rindexr%rrrr(aszAsArrayTypedSize.__call__)rNr)rrrrrWs  rcCs t|jS)z9Create PyConverter to get first argument as array of type)rCallFuncPyConverterr#typsizerrr asArrayTypejsr6cCsdS)zNo converter requiredNrr3rrrr6pscs*jtj|fdd}|S)zCreate PyConverter function to get array as type and check size Produces a raw function, not a PyConverter instance cs>|}||}||}|kr:td|f||S)Nz)Expected %r byte array, got %r byte array)Z getHandlerr#ZarrayByteCount ValueError)incomingfunctionr&handlerrZbyteSizeZdataTypeZ expectedBytesr4rr asArraySizes   z$asArrayTypeSize..asArraySize)Z typeConstantctypesZsizeofZbaseType)r4r5r<rr;rasArrayTypeSize|sr>cCsddlm}t|jS)z=Create PyConverter returning incoming as an array of any typer ArrayDatatype) OpenGL.arraysr@rr2r#r?rrr asVoidArrays rBcCsdS)z9If there's no copying allowed, we can use default passingNrrrrrrBsc@s(eZdZdZddZddZddZdS) storePointerTypeafStore named pointer value in context indexed by constant pointerName -- named pointer argument constant -- constant used to index in the context storage Note: OpenGL.STORE_POINTERS can be set with ERROR_ON_COPY to ignore this storage operation. Stores the pyArgs (i.e. result of pyConverters) for the named pointer argument... cCs||_||_dSr) pointerNameconstant)r rDrErrrr!szstorePointerType.__init__cCs||j|_dSr)rrD pointerIndex)r wrapperrrrfinaliseszstorePointerType.finalisecCst|j||jdSr)rZsetValuerErF)r rrrrrrrr(szstorePointerType.__call__N)r*r+r,r-r!rHr(rrrrrCs rCcCsddlm}||||S)aDecorate function with vector-handling code for a single argument if OpenGL.ERROR_ON_COPY is False, then we return the named argument, converting to the passed array type, optionally checking that the array matches size. if OpenGL.ERROR_ON_COPY is True, then we will dramatically simplify this function, only wrapping if size is True, i.e. only wrapping if we intend to do a size check on the array. r)rG)OpenGLrGZsetInputArraySize)rr5rZargNamerGrrrsetInputArraySizeTypes rJcs|jfdd}|S)Ncs |d}|durS|SdS)z+Return the array size of the first argumentrNr)rr1rarraydefaultunitSizerrarraySizeOfFirstsz.arraySizeOfFirstType..arraySizeOfFirst)rN)r4rMrOrrLrarraySizeOfFirstTypesrP)N)N)r)'r-rIr=rrrrrArZ OpenGL._bytesrrlogging getLogger_logr r ZACCELERATE_AVAILABLEZOpenGL_accelerate.arraydatatypeZOpenGL_accelerate.wrapperr r r r ImportErrorerrwarningrZ PyConverterZ CConverterZ ERROR_ON_COPYr6ZARRAY_SIZE_CHECKINGr>rBobjectrCrJrPrrrrsJ