o OZhE8@s dZddlmZmZmZmZddlZddlm Z m Z m ZmZddlZdddDZeehdZgd Zd d Zd d ZddZddZddZddZddZddZddZddZ ddZ!d d!Z"d"d#Z#d$d%Z$d&d'Z%d(d)Zd*d+Z&d,d-Z'd.d/Z(dS)0apygame module for accessing surface pixel data using array interfaces Functions to convert between NumPy arrays and Surface objects. This module will only be functional when pygame can use the external NumPy package. If NumPy can't be imported, surfarray becomes a MissingModule object. Every pixel is stored as a single integer value to represent the red, green, and blue colors. The 8bit images use a value that looks into a colormap. Pixels with higher depth use a bit packing process to place three or four values into a single number. The arrays are indexed by the X axis first, followed by the Y axis. Arrays that treat the pixels as a single integer are referred to as 2D arrays. This module can also separate the red, green, and blue color values into separate indices. These types of arrays are referred to as 3D arrays, and the last index is 0 for red, 1 for green, and 2 for blue. )array_to_surfacesurface_to_array map_array make_surfaceN)arrayemptyuint32ndarraycCs g|] }tt|rtt|qS)hasattrnumpygetattr).0 type_namer r ?/var/www/auris/lib/python3.10/site-packages/pygame/surfarray.py ;s rzfloat32 float64 float96> )array2darray3d array_alpha array_bluearray_colorkey array_green array_redr blit_array get_arraytypeget_arraytypesrrpixels2dpixels3d pixels_alpha pixels_blue pixels_green pixels_redr use_arraytypecCs.t|tr|jtvr|dt}t||S)apygame.surfarray.blit_array(Surface, array): return None Blit directly from a array values. Directly copy values from an array into a Surface. This is faster than converting the array into a Surface and blitting. The array must be the same dimensions as the Surface and will completely replace all pixel values. Only integer, ascii character and record arrays are accepted. This function will temporarily lock the Surface as the new values are copied. r) isinstance numpy_ndarraydtype numpy_floatsroundastype numpy_uint32r)surfacerr r rras  rcCs,t|tr|jtvr|dt}t|S)zpygame.surfarray.make_surface (array): return Surface Copy an array to a new surface. Create a new Surface that best resembles the data and format on the array. The array can be 2D or 3D with any sized integer values. r)r&r'r(r)r*r+r,pix_make_surface)rr r rrssrcCsn|}ztjtjtjtjf|d}Wnty%td|ddw|}t||}t |||S)apygame.surfarray.array2d(Surface): return array copy pixels into a 2d array Copy the pixels from a Surface into a 2D array. The bit depth of the surface will control the size of the integer values, and will work for any type of pixel format. This function will temporarily lock the Surface as pixels are copied (see the Surface.lock - lock the Surface memory for pixel access method). zunsupported bit depth rz for 2D array) Z get_bytesizer uint8Zuint16int32 IndexError ValueErrorget_sizerr)r-Zbppr(sizerr r rrs     rc CsP|tvr tdz t|dddWSttfy'td|dw)a.pygame.surfarray.pixels2d(Surface): return array reference pixels into a 2d array Create a new 2D array that directly references the pixel values in a Surface. Any changes to the array will affect the pixels in the Surface. This is a fast operation since no data is copied. Pixels from a 24-bit Surface cannot be referenced, but all other Surface bit depths can. The Surface this references will remain locked for the lifetime of the array (see the Surface.lock - lock the Surface memory for pixel access method). z,unsupported bit depth for 2D reference array2Fcopyz bit depth z# unsupported for 2D reference array)Z get_bitsize_pixel2d_bitdepthsr3 numpy_arrayget_view TypeErrorr-r r rrs rcCs.|\}}t||dftj}t|||S)apygame.surfarray.array3d(Surface): return array copy pixels into a 3d array Copy the pixels from a Surface into a 3D array. The bit depth of the surface will control the size of the integer values, and will work for any type of pixel format. This function will temporarily lock the Surface as pixels are copied (see the Surface.lock - lock the Surface memory for pixel access method). r4r rr0r)r-widthheightrr r rrs  rcCst|dddS)aCpygame.surfarray.pixels3d(Surface): return array reference pixels into a 3d array Create a new 3D array that directly references the pixel values in a Surface. Any changes to the array will affect the pixels in the Surface. This is a fast operation since no data is copied. This will only work on Surfaces that have 24-bit or 32-bit formats. Lower pixel formats cannot be referenced. The Surface this references will remain locked for the lifetime of the array (see the Surface.lock - lock the Surface memory for pixel access method). 3Fr7)r:r;r=r r rr sr cC&|}t|tj}t||d|S)apygame.surfarray.array_alpha(Surface): return array copy pixel alphas into a 2d array Copy the pixel alpha values (degree of transparency) from a Surface into a 2D array. This will work for any type of Surface format. Surfaces without a pixel alpha will return an array with all opaque values. This function will temporarily lock the Surface as pixels are copied (see the Surface.lock - lock the Surface memory for pixel access method). Ar?r-r5rr r rrs rcCtj|dddS)apygame.surfarray.pixels_alpha(Surface): return array reference pixel alphas into a 2d array Create a new 2D array that directly references the alpha values (degree of transparency) in a Surface. Any changes to the array will affect the pixels in the Surface. This is a fast operation since no data is copied. This can only work on 32-bit Surfaces with a per-pixel alpha value. The Surface this array references will remain locked for the lifetime of the array. rDFr7r rr;r=r r rr!sr!cCrF)apygame.surfarray.pixels_red(Surface): return array Reference pixel red into a 2d array. Create a new 2D array that directly references the red values in a Surface. Any changes to the array will affect the pixels in the Surface. This is a fast operation since no data is copied. This can only work on 24-bit or 32-bit Surfaces. The Surface this array references will remain locked for the lifetime of the array. RFr7rGr=r r rr$r$cCrC)ahpygame.surfarray.array_red(Surface): return array copy pixel red into a 2d array Copy the pixel red values from a Surface into a 2D array. This will work for any type of Surface format. This function will temporarily lock the Surface as pixels are copied (see the Surface.lock - lock the Surface memory for pixel access method). rHr?rEr r rr  rcCrF)apygame.surfarray.pixels_green(Surface): return array Reference pixel green into a 2d array. Create a new 2D array that directly references the green values in a Surface. Any changes to the array will affect the pixels in the Surface. This is a fast operation since no data is copied. This can only work on 24-bit or 32-bit Surfaces. The Surface this array references will remain locked for the lifetime of the array. GFr7rGr=r r rr#!rIr#cCrC)anpygame.surfarray.array_green(Surface): return array copy pixel green into a 2d array Copy the pixel green values from a Surface into a 2D array. This will work for any type of Surface format. This function will temporarily lock the Surface as pixels are copied (see the Surface.lock - lock the Surface memory for pixel access method). rKr?rEr r rr2rJrcCrF)apygame.surfarray.pixels_blue(Surface): return array Reference pixel blue into a 2d array. Create a new 2D array that directly references the blue values in a Surface. Any changes to the array will affect the pixels in the Surface. This is a fast operation since no data is copied. This can only work on 24-bit or 32-bit Surfaces. The Surface this array references will remain locked for the lifetime of the array. BFr7rGr=r r rr"DrIr"cCrC)akpygame.surfarray.array_blue(Surface): return array copy pixel blue into a 2d array Copy the pixel blue values from a Surface into a 2D array. This will work for any type of Surface format. This function will temporarily lock the Surface as pixels are copied (see the Surface.lock - lock the Surface memory for pixel access method). rLr?rEr r rrUrJrcCrC)apygame.surfarray.array_colorkey(Surface): return array copy the colorkey values into a 2d array Create a new array with the colorkey transparency value from each pixel. If the pixel matches the colorkey it will be fully transparent; otherwise it will be fully opaque. This will work on any type of Surface format. If the image has no colorkey a solid opaque array will be returned. This function will temporarily lock the Surface as pixels are copied. Cr?rEr r rrgs rcCsP|jdkr td|j}|ddkrtdt|ddtj}t||||S)apygame.surfarray.map_array(Surface, array3d): return array2d map a 3d array into a 2d array Convert a 3D array into a 2D array. This will use the given Surface format to control the conversion. Note: arrays do not need to be 3D, as long as the minor axis has three elements giving the component colours, any array shape can be used (for example, a single colour can be mapped, or an array of colours). The array shape is limited to eleven dimensions maximum, including the three element minor axis. rz$array must have at least 1 dimensionr>z.array must be a 3d array of 3-value color dataN)ndimr3shape numpy_emptyr r1 pix_map_array)r-rrPtargetr r rr|s   rcCs*ttd|}|dkrtddS)zrpygame.surfarray.use_arraytype(arraytype): return None DEPRECATED - only numpy arrays are now supported. donly numpy arrays are now supported, this function will be removed in a future version of the moduler zinvalid array typeN)warningswarnDeprecationWarninglowerr3)Z arraytyper r rr%sr%cCttddS)zhpygame.surfarray.get_arraytype(): return str DEPRECATED - only numpy arrays are now supported. rTr rUrVrWr r r rr rcCrY)zkpygame.surfarray.get_arraytypes(): return tuple DEPRECATED - only numpy arrays are now supported. rT)r rZr r r rrr[r))__doc__Zpygame.pixelcopyrrrrRrr.r rr:rrQrr,r r'rUsplitr)appendfloatr9__all__rrrrr rr!r$rr#rr"rrr%rrr r r rs<