`i5ddlZddlmZddlmZddlmZejddddd Zejd d d dd Z ddededefdZ ddedefdZ dS)N)_core)texture)runtimez%U texObj, raw float32 m, uint64 widthzT transformed_imagea# float3 pixel = make_float3( (float)(i / width), (float)(i % width), 1.0f ); float x = dot(pixel, make_float3(m[0], m[1], m[2])) + .5f; float y = dot(pixel, make_float3(m[3], m[4], m[5])) + .5f; transformed_image = tex2D(texObj, y, x); ,cupyx_texture_affine_transformation_2d_arrayz inline __host__ __device__ float dot(float3 a, float3 b) { return a.x * b.x + a.y * b.y + a.z * b.z; } )preamblez4U texObj, raw float32 m, uint64 height, uint64 widthzT transformed_volumea float4 voxel = make_float4( (float)(i / (width * height)), (float)((i % (width * height)) / width), (float)((i % (width * height)) % width), 1.0f ); float x = dot(voxel, make_float4(m[0], m[1], m[2], m[3])) + .5f; float y = dot(voxel, make_float4(m[4], m[5], m[6], m[7])) + .5f; float z = dot(voxel, make_float4(m[8], m[9], m[10], m[11])) + .5f; transformed_volume = tex3D(texObj, z, y, x); ,cupyx_texture_affine_transformation_3d_arrayz inline __host__ __device__ float dot(float4 a, float4 b) { return a.x * b.x + a.y * b.y + a.z * b.z + a.w * b.w; } address_mode filter_mode read_modectj|jtjr tj}nytj|jtjr tj}nHtj|jtjr tj }ntd|j|dkr tj }n&|dkr tj }ntd|d|dkr tj }n&|dkr tj}ntd|d|d kr tj}n&|d kr tj}ntd |d t#j|jd zddd|}t#j|g|jdddR}t#jtj|}t#j|f|jz|||f} t#j|| } ||| S)NzUnsupported data type nearestconstantzUnsupported address mode z (supported: constant, nearest)linearzUnsupported filter mode z (supported: nearest, linear) element_typenormalized_floatzUnsupported read mode z, (supported: element_type, normalized_float)r)cuArr) borderColors)cupy issubdtypedtypeunsignedintegerrcudaChannelFormatKindUnsignedintegercudaChannelFormatKindSignedfloatingcudaChannelFormatKindFloat ValueErrorcudaAddressModeClampcudaAddressModeBordercudaFilterModePointcudaFilterModeLinearcudaReadModeElementTypecudaReadModeNormalizedFloatrChannelFormatDescriptoritemsize CUDAarrayshapeResourceDescriptorcudaResourceTypeArrayTextureDescriptorndim TextureObject copy_from) datar r r border_colorfmt_kind texture_fmtarrayres_desctex_desctex_objs b/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/_texture.py_create_texture_objectr95sI  tz4#788@8 T\ 2 2@6 T] 3 3@5>$*>>???y  3  # #4  -  - - -.. .i1  2  ,{ , , ,-- -N""3 ( ( (7  :Y : : :;; ;1 1aH..K  k =DJttt,< = = =E)%U444H( 49$k9"%'''H#Hh77G OOD Nrr interpolationmodec |j}|dks|dkrtd|j}|tjkrtd|d|dvrtd|d|j|d z|d zfkrtd t |||d | } |dkrt} nt} ||j}|tj ||}nt|ttjfr:|tjkrtd|dtj ||}nDt|tj r|j|krtdntd| | |g|d d |R|S)a Apply an affine transformation. The method uses texture memory and supports only 2D and 3D float32 arrays without channel dimension. Args: data (cupy.ndarray): The input array or texture object. transformation_matrix (cupy.ndarray): Affine transformation matrix. Must be a homogeneous and have shape ``(ndim + 1, ndim + 1)``. output_shape (tuple of ints): Shape of output. If not specified, the input array shape is used. Default is None. output (cupy.ndarray or ~cupy.dtype): The array in which to place the output, or the dtype of the returned array. If not specified, creates the output array with shape of ``output_shape``. Default is None. interpolation (str): Specifies interpolation mode: ``'linear'`` or ``'nearest'``. Default is ``'linear'``. mode (str): Specifies addressing mode for points outside of the array: (`'constant'``, ``'nearest'``). Default is ``'constant'``. border_value: Specifies value to be used for coordinates outside of the array for ``'constant'`` mode. Default is 0. Returns: cupy.ndarray: The transformed input. .. seealso:: :func:`cupyx.scipy.ndimage.affine_transform` zdTexture memory affine transformation is defined only for 2D and 3D arrays without channel dimension.zRTexture memory affine transformation is available only for float32 data type (not ))rr zUnsupported interpolation z (supported: linear, nearest)z.Matrix must be have shape (ndim + 1, ndim + 1)r)r r r r1N)rzOutput shapes do not matchz/Output must be None, cupy.ndarray or cupy.dtype) r-rrrfloat32r)r9!_affine_transform_2d_array_kernel!_affine_transform_3d_array_kernelzeros isinstancetypendarray) r0transformation_matrix output_shapeoutputr;r< border_valuer-rtexture_objectkernels r8affine_transformationrOpsJ 9D qdQhh :;; ; JE  E0L<3CLVLLLL Mr:)r)NNrrr) rr cupy.cudarrElementwiseKernelrCrDstrr9rOr:r8rTs0 %