File: //usr/lib/python3.6/site-packages/glances/plugins/__pycache__/glances_gpu.cpython-36.opt-1.pyc
3
Eo�c[, � @ s� d Z ddlmZmZ ddlmZ ddlmZ yddlj Z
W n6 e k
rp Z zdZ
ejdje �� W Y ddZ [ X nX dZ
d d
d d �d
dd d �gZG dd� de�Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )z(GPU plugin (limited to NVIDIA chipsets).� )� nativestr�
to_fahrenheit)�logger)�
GlancesPluginNTz6Missing Python Lib ({}), Nvidia GPU plugin is disabledF�procz
GPU processor�%)�name� descriptionZy_unit�memzMemory consumptionc sp e Zd ZdZd� fdd� Zdd� Zdd� Zejej d d
� ��Z
� fd d �Z dd
d�Z dd� Z
� fdd�Z� ZS )�PluginzqGlances GPU plugin (limited to NVIDIA chipsets).
stats is a list of dictionaries with one entry per GPU
Nc s( t t| �j||g d� | j� d| _dS )zInit the plugin.)�args�configZstats_init_valueTN)�superr �__init__� init_nvidiaZ
display_curse)�selfr r
)� __class__� �!/usr/lib/python3.6/glances_gpu.pyr ( s zPlugin.__init__c C sP t r
d| _ytj� t� | _d| _W n$ tk
rH tjd� d| _Y nX | jS )zInit the NVIDIA API.FTz pynvml could not be initialized.) �import_error_tag�
nvml_ready�pynvmlZnvmlInit�get_device_handles�device_handles� Exceptionr �debug)r r r r r 2 s
zPlugin.init_nvidiac C s dS )zReturn the key of the list.�gpu_idr )r r r r �get_keyA s zPlugin.get_keyc C s> | j � }| js| jS | jdkr(| j� }n
| jdkr2|| _| jS )zUpdate the GPU stats.ZlocalZsnmp)Zget_init_valuer �statsZ input_method�get_device_stats)r r r r r �updateE s !
z
Plugin.updatec s� t t| �j� x�| jD ]�}i i i d�| j|| j� < d|krh| j|d dd�}|| j|| j� d d< d|kr�| j|d dd�}|| j|| j� d d< d|kr| j|d dd�}|| j|| j� d d< qW dS )zUpdate stats views.)r r
� temperaturer )�header�
decorationr
r! T)r r � update_viewsr Zviewsr Z get_alert)r �iZalert)r r r r$ z s zPlugin.update_viewsc s� g }� j s � j g ks � j� r"|S t� fdd�� j D ��}� j d }d}t� j �dkrj|djt� j ��7 }|r�|djd|d �7 }n|d
jd�7 }|d d � }|j� j|d
�� t� j �dks�|j�r
|j� j� � y"t dd� � j D ��t� j � }W n t
k
�r djd�} Y n X dj|�} t� j �dk�r>djd�}n
djd�}|j� j|�� |j� j| � j |� j � ddd��� |j� j� � y"t dd� � j D ��t� j � }
W n t
k
�r� djd�} Y n X dj|
�} t� j �dk�r�djd�}n
djd�}|j� j|�� |j� j| � j |� j � ddd��� |j� j� � y"t d d� � j D ��t� j � } W n t
k
�r� djd�}
Y n&X d }|j
�r�t| �} d }d j| |�}
t� j �dk�r�djd �}n
djd!�}|j� j|�� |j� j|
� j |� j � d"dd��� n�x�� j D ]�}|j� j� � d
j|d# �}ydj|d �}W n$ tt
fk
�rh djd�}Y nX ydj|d �}W n$ tt
fk
�r� djd�}Y nX d$j|||�}|j� j|�� �qW |S )%z2Return the dict to display in the curse interface.c 3 s$ | ] }|d � j d d kV qdS )r r N)r )�.0�s)r r r � <genexpr>� s z#Plugin.msg_curse.<locals>.<genexpr>r � � z{} z{} {}ZGPUr z{}N� ZTITLEc s s | ]}|d k r|d V qd S )Nr r )r& r' r r r r( � s z{:>4}zN/Az {:>3.0f}%z{:13}z
proc mean:zproc:r r# )�item�keyZoptionc s s | ]}|d k r|d V qd S )Nr
r )r& r' r r r r( � s z mem mean:zmem:r
c s s | ]}|d k r|d V qd S )Nr! r )r& r' r r r r( � s �C�Fz
{:>3.0f}{}z
temp mean:z temperature:r! r z{}: {} mem: {})r Z is_disabled�all�len�format�appendZcurse_add_lineZmeangpuZcurse_new_line�sum� TypeErrorZ get_viewsr Z
fahrenheitr �
ValueError)r r Z max_width�retZ same_nameZ gpu_statsr" �msgZ mean_procZ
mean_proc_msgZmean_memZ mean_mem_msgZmean_temperatureZmean_temperature_msgZunitZid_msgZproc_msgZmem_msgr )r r � msg_curse� s�
"
"
"
zPlugin.msg_cursec C st g }xjt | j�D ]\\}}t� }| j� |d< ||d< t|�|d< t|�|d< t|�|d< t|�|d< |j|� qW |S )zGet GPU stats.r- r r r
r r! ) � enumerater �dictr �get_device_name�get_mem�get_proc�get_temperaturer3 )r r �index�
device_handleZ device_statsr r r r s zPlugin.get_device_statsc sX | j rFy tj� W n2 tk
rD } ztjdj|�� W Y dd}~X nX tt| �j � dS )z/Overwrite the exit method to close the GPU API.z(pynvml failed to shutdown correctly ({})N)
r r Z nvmlShutdownr r r r2 r r �exit)r �e)r r r rB s "z Plugin.exit)NN)NN)�__name__�
__module__� __qualname__�__doc__r r r r Z_check_decoratorZ_log_result_decoratorr r$ r9 r rB �
__classcell__r r )r r r " s
4
pr c C s dd� t tj� �D �S )zQGet a list of NVML device handles, one per device.
Can throw NVMLError.
c S s g | ]}t j|��qS r )r ZnvmlDeviceGetHandleByIndex)r& r% r r r �
<listcomp>* s z&get_device_handles.<locals>.<listcomp>)�ranger ZnvmlDeviceGetCountr r r r r % s r c C s* yt tj| ��S tjk
r$ dS X dS )zGet GPU device name.ZNVIDIAN)r r ZnvmlDeviceGetNameZ NVMlError)rA r r r r< - s r< c C s6 yt j| �}|jd |j S t jk
r0 dS X dS )z-Get GPU device memory consumption in percent.g Y@N)r ZnvmlDeviceGetMemoryInfoZusedZtotal� NVMLError)rA Z memory_infor r r r= 5 s
r= c C s( y t j| �jS t jk
r" dS X dS )z*Get GPU device CPU consumption in percent.N)r Z nvmlDeviceGetUtilizationRatesZgpurK )rA r r r r> > s r> c C s* yt j| t j�S t jk
r$ dS X dS )z*Get GPU device CPU consumption in percent.N)r ZnvmlDeviceGetTemperatureZNVML_TEMPERATURE_GPUrK )rA r r r r? F s r? )rG Zglances.compatr r Zglances.loggerr Z glances.plugins.glances_pluginr Zpy3nvml.py3nvmlZpy3nvmlr r rC r Zwarningr2 Zitems_history_listr r r< r= r> r? r r r r �<module>
s& "