Aluode/PerceptionLabPortable
0
1o
2 �Yi�% � @ s� d Z ddlZddlZddlZddlZddlmZ ddlZ ddl3m Z m Z m
Z
e �dd� ddlZzddlZe �dd� dZW n
eyK dZd ZY nw zddlZdZW n
eya dZd ZY nw zdd4lmZ W n eyu dZY nw ejd d �Zejd
d dddd� ejddg d�dd� ejdde dd� ejdd d d gd d d � ejd!dd"d#� d$d� ejd%d dd&d'� ejd(d dd)d*d� ejd+d,d-d#� d.d� ejd/e d0�e d1d� e� ej d2d� �Z e�!� Z"edur�e �#� Z$e$�%d� e j#�&e$� e �'d3�Z(e
�)� Z*e*�+d4� e�,� a-t-�.e*� e*�/� edu �r-t-j0�1d � t-j0�2t-j0�3� d5 � ne� t-_4t-j5�6t-j4� e*j7�8t-j4j9� t-j:�;e j<�oHe� t-j:�1e� t-j=�;e�oZe �>d6�� t-j=�1e� t-j?�@e jA� t-jB�@e jCd � t-jD�@e jCd2 � t-jE�Fe jG� t-jH�;e jId k� t-jJjKd7d2d8� t-jLjKdd2d8� t-jLt-jJt-jMt-jNt-jOt-jPgaQe j<�r�e�r�eaRneaRe jSdu �r�t-jT�;d � t-jU�;d � nOt-jT�;d� eVe jS�d9k�r�t-jU�;d � t-jL�@e jSd � t-jJ�@e jSd2 � n(eVe jS�d:k�rt-jU�;d� eWtQe jS�D ]5\ZXZYeX�@eY� �qneZd;e jSf ��t-j[�;e j\� t-j]�;e j^� e �_� a`t-ja�bt`� t`�c� e �d� Zet`�fee� dagd<d=� Zht-jijj�8eh� eh� t-j]jk�8eh� d>d?� Zlel� t-jUjk�8el� i amd@dA� ZndBdC� ZodDdE� ZpdFdG� Zqen� t-jEjr�8en� t-jHjk�8en� t-jBjs�8en� t-jDjs�8en� t-j?js�8en� t-jBjt�8eo� t-jDjt�8eo� t-j?jt�8eo� t-j:jk�8ep� t-j=jk�8eq� daudHdI� Zve �w� Zxexjy�8ev� ex�zd� e� Z{e{j|�8dJd#� � e}dKk�r�e �~� dS dS )Lz�6Tests the speed of image updates for an ImageItem and RawImageWidget.7The speed will generally depend on the type of data being shown, whether8it is being scaled and/or converted by lookup table, and whether OpenGL9is used by the view widget10� N)� FrameCounter)�QtCore�QtGui� QtWidgetsZimageAxisOrderz row-majorZuseCupyTF)�RawImageGLWidgetz'Benchmark for testing video performance)� descriptionz--cuda�11store_truez Use CUDA to process on the GPU�cuda)�default�action�help�destz--dtype�uint8)r �uint16�floatz%Image dtype (uint8, uint16, or float))r12 �choicesr z--frames� z.Number of image frames to generate (default=3))r13 �typer z --image-mode�mono�rgbz Image data mode (mono or rgb)�14image_mode)r15 r r r
z--levelsc C � t dd� | �d�D ��S )Nc S � g | ]}t |��qS � )r ��.0�xr r ��E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\pyqtgraph/examples/VideoSpeedTest.py�16<listcomp>/ � � <lambda>.<locals>.<listcomp>�,��tuple�split��sr r r �<lambda>/ � r' zhmin,max levels to scale monochromatic image dynamic range, or rmin,rmax,gmin,gmax,bmin,bmax to scale rgbz--lutzUse color lookup table)r17 r r z --lut-alphaz Use alpha color lookup table� lut_alphaz--sizeZ512x512c C r )Nc S r r )�intr r r r r 2 r r r r" r% r r r r' 2 r( z&WxH image dimensions default='512x512'z --iterations�infz*Number of iterations to run before exiting� zVideo Speed Test Examplez!pyqtgraph example: VideoSpeedTestz (OpenGL not available)�useNumba�� )�value�step� � z>levels argument must be 2 or 6 comma-separated values (got %r)c C sT t j�� } | dkr d}nd}t jj|t j�� d�atr&t t18kr(t19� t�ad S d S d S )Nr � � )�alpha) �ui�20dtypeCombo� currentText�gradientZgetLookupTable�21alphaCheck� isChecked�LUT� _has_cupy�xp�cpZasarray)�dtype�nr r r � updateLUT~ s 22 �rB c C sJ t j�� rtdd � D ]} | �d� q d S tdd � D ]} | �d� qd S )Nr1 TF)r6 �rgbLevelsCheckr; �23levelSpins�24setEnabledr% r r r � updateScale� s 25 � �rF c
C s� t �� �� tj�� } tj�� }tj�� }tj�� tj �26� | ||f}|t vr�|d dkr5t j
}d}d}d}n)|d dkrEt j}d}d}d }n|d d27krUt j}d }d }d }n td
|d � ���||f}tj �28� rl|d } n|} t j| f| |d�at�d| �}29t|30jd �D ] } t jj|||d�} |d d31kr�t j| d|| d� | |32| < q�|td d �dd�df< |td d �ddd�f< |td d �ddd�f< |tia t | at� t� W d � d S 1 s�w Y d S )Nr r � �@ r. r r4 i i�� r g �?g�������?zunable to handle dtype: )r )r@ )�����)�loc�scale�size)�out�33 �2 �0 � � �/ � �
)�pgZ34BusyCursorr6 �35framesSpinr/ � widthSpin�36heightSpinr7 r8 �rgbCheckr; �cacher> r r Zfloat32�37ValueError�empty�dataZreshape�range�shape�random�normalZcliprB �38updateSize)
�frames�width�heightZcacheKey�dtrJ rK ZmxZ39chan_shapeZ frame_shape�view�idxZsubdatar r r �mkData� sR 40414243 4445 46"�rj c C s� t j�� } t j�� }t j�� }t�tt j� � ��}t j47� � r dnd}t j �
d| | | | |j d � t�t�dd||�� d S )Nr r, z%d MBg ��.Ar )r6 rW r/ rX rY r> r@ �strr7 r8 rZ r; Z sizeLabel�setText�itemsize�vbZsetRanger ZQRectF)rd re rf r@ r r r r rc � s 484950&rc c C s8 i a tj�� rtr tan tatj�d� ntat � d S )NF)51r[ r6 � cudaCheckr; r= r? r> �np�52setCheckedrj r r r r �noticeCudaCheck� s 5354rr c C s t �dto tj�� � d S )Nr- )rV �setConfigOption�55_has_numbar6 �56numbaCheckr; r r r r �noticeNumbaCheck� s rv c C s\ t t�tjkrt�� t�� d S tj �57� rt } nd } tj �58� }tj
�59� rStj�60� rHtj�� tj�� gtj�� tj�� gtj�� tj�� gg}n
tj�� tj�� g}nd }tj�61� rqtjjtttjd | |d� tj � d� n3tj �62� r�tj jtttjd | |d� tj � d� nt jtttjd d|| |d� tj � d� td7 at!�"� d S )Nr )�lut�levelsr, r1 F)Z63autoLevelsrx rw ZautoDownsample)#�next�iterations_counter�argsZ64iterations�timer�stop�app�quitr6 �lutCheckr; r< ZdownsampleCheck�65scaleCheckrC �minSpin1r/ �maxSpin1�minSpin2�maxSpin2�minSpin3�maxSpin3ZrawRadioZrawImgZsetImager^ �ptrr` �stackZsetCurrentIndex�66rawGLRadio�rawGLImg�img�framecnt�update)ZuseLutZ67downsampleZuseScaler r r r� � s4 68697071�72 73 " r� c C s t j�| d�d��S )Nz.1fz fps)r6 ZfpsLabelrl )Zfpsr r r r' ' r �__main__)�__doc__�argparse� itertools�sys�numpyrp �utilsr Z pyqtgraphrV Z pyqtgraph.Qtr r r rs ZVideoTemplate_genericZ ui_templateZcupyr? r= � ImportErrorZnumbart Z pyqtgraph.widgets.RawImageWidgetr �ArgumentParser�parser� add_argumentr* r �74parse_args�argvr{ �countrz ZQSurfaceFormatZsfmtZsetSwapIntervalZsetDefaultFormatZmkQAppr~ Z QMainWindow�winZsetWindowTitleZ
Ui_MainWindowr6 ZsetupUi�showr� rE rl �textr� r� Z addWidgetZ destroyed�connect�cleanupro rq r ru ZgetConfigOptionrW ZsetValuerd rX rL rY r7 ZsetCurrentTextr@ rZ r r� ZsetOptsr� r� r� r� r� rD r> rx r� rC �len�zip�spin�valr\ r� rw r: r) ZViewBoxrn Z graphicsViewZsetCentralItemZsetAspectLockedZ ImageItemr� ZaddItemr< rB r9 ZsigGradientChangedZtoggledrF r[ rj rc rr rv ZcurrentIndexChangedZeditingFinishedZ valueChangedr� r� ZQTimerr| �timeout�startr� Z sigFpsUpdate�__name__�execr r r r �<module> s � � � �75 76777879 � 8081 2 ( 8283 �