3
�<�e� � @ s� d dl mZmZ d dlmZ d dlmZ d dlmZm Z m
Z
d dl m Z d dl
Z
d dlZd dlZej� ZdZdZd ZG d
d � d e �ZG d d
� d
e�ZG dd� de�ZG dd� dejj�ZdS )� )�exports�logs)�commands)�PPD_CONFIG_FILE)� PPDConfig�PPD_PERFORMANCE�PPD_POWER_SAVER)�StrEnumN�tunedz-/sys/devices/system/cpu/intel_pstate/no_turboz4/sys/bus/platform/devices/thinkpad_acpi/dytc_lapmodec @ s e Zd ZdZdZdZdS )�PerformanceDegraded� z lap-detectedzhigh-operating-temperatureN)�__name__�
__module__� __qualname__�NONE� LAP_DETECTED�HIGH_OPERATING_TEMPERATURE� r r � /usr/lib/python3.6/controller.pyr s r c @ s e Zd Zdd� Zdd� ZdS )� ProfileHoldc C s || _ || _|| _|| _d S )N)�profile�reason�app_id�watch)�selfr r r r r r r �__init__ s zProfileHold.__init__c C s | j | j| jd�S )N)�ProfileZReasonZ
ApplicationId)r r r )r r r r �as_dict s zProfileHold.as_dictN)r
r r r r r r r r r s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�ProfileHoldManagerc C s i | _ d| _|| _d S )Nr )�_holds�_cookie_counter� _controller)r Z
controllerr r r r ' s zProfileHoldManager.__init__c s � ��fdd�}|S )Nc s( | dkr$t jd� �f � �j�� d S )Nr z1Application '%s' disappeared, releasing hold '%s')�log�info�remove)�name)r �cookier r r �callback- s z.ProfileHoldManager._callback.<locals>.callbackr )r r& r r' r )r r& r r � _callback, s z ProfileHoldManager._callbackc C s t dd� | jj� D ��r tS tS )Nc s s | ]}|j tkV qd S )N)r r )�.0�holdr r r � <genexpr>5 s z=ProfileHoldManager._effective_hold_profile.<locals>.<genexpr>)�anyr �valuesr r )r r r r �_effective_hold_profile4 s z*ProfileHoldManager._effective_hold_profilec C s\ || j krd S | j j|�}|jj� tjd|� tjd| j� � tj d||j
|j f � d S )N�ProfileReleased�ActiveProfileHoldsz5Releasing hold '%s': profile '%s' by application '%s') r �popr Zcancelr Z send_signal�property_changed�
as_dbus_arrayr" r# r r )r r& r* r r r �_cancel9 s
zProfileHoldManager._cancelc C s t jdd� | jj� D �dd�S )Nc S s g | ] }|j � �qS r )r )r) r* r r r �
<listcomp>C s z4ProfileHoldManager.as_dbus_array.<locals>.<listcomp>za{sv})� signature)�dbus�Arrayr r- )r r r r r3 B s z ProfileHoldManager.as_dbus_arrayc C st | j }| j d7 _ | jjj|| j||��}tjd|||f � t||||�| j|< t j
d| j � � | jj |� |S )N� z2Adding hold '%s': profile '%s' by application '%s'r0 )
r r! �busZwatch_name_ownerr( r" r# r r r r2 r3 �switch_profile)r r r r �callerr& r r r r �addE s zProfileHoldManager.addc C s
|| j kS )N)r )r r& r r r �hasO s zProfileHoldManager.hasc C s: | j |� t| j�dkr"| j� }n| jj}| jj|� d S )Nr )r4 �lenr r. r! � base_profiler; )r r& Z new_profiler r r r$ R s
zProfileHoldManager.removec C s x| j D ]}| j|� qW d S )N)r r4 )r r& r r r �clearZ s zProfileHoldManager.clearN) r
r r r r( r. r4 r3 r= r> r$ rA r r r r r & s
r c s
e Zd Z� fdd�Zdd� Zdd� Zedd� �Zed d
� �Zd d � Z d
d� Z
dd� Z dd� Z e
jdd�dd� �Ze
jdd�dd� �Ze
jd�dd� �Ze
jd �d d � �Ze
jd �d d � �Ze
jd!�d"d#� �Ze
jd$�d%d&� �Ze
jd'�d(d)� �Ze
jd*�d+d,� �Z� ZS )-�
Controllerc sJ t t| �j� || _|| _t| �| _tj| _ t
� | _ t j
� | _| j� d S )N)�superrB r �_bus�_tuned_interfacer �_profile_holdsr r �_performance_degradedr �_cmd� threadingZEvent�
_terminate� load_config)r r: Ztuned_interface)� __class__r r r ` s
zController.__init__c C s� t j}tjjt�r,t| jjt��dkr,t j }tjjt
�rRt| jjt
��dkrRt j }|| j kr|t
jd| � || _ tjd|� d S )Nr9 zPerformance degraded: %sr )r r �os�path�exists�
NO_TURBO_PATH�intrH Z read_filer �
LAP_MODE_PATHr rG r" r# r r2 )r Zperformance_degradedr r r �_check_performance_degradedj s
z&Controller._check_performance_degradedc C s2 t j� x | jj| jd�s$| j� q
W t j� d S )Nr9 )r �startrH �waitrJ rS �stop)r r r r �runw s zController.runc C s | j S )N)rD )r r r r r: } s zController.busc C s | j S )N)�
_base_profile)r r r r r@ � s zController.base_profilec C s | j j� d S )N)rJ �set)r r r r � terminate� s zController.terminatec C s& t t�| _| jj| _| j| jj� d S )N)r r �_configZdefault_profilerX r; )r r r r rK � s
zController.load_configc C sF | j � |krd S | jj| }tjd| � | jj|� tjd|� d S )NzSwitching to profile '%s'�
ActiveProfile) �active_profiler[ � ppd_to_tunedr" r# rE r; r r2 )r r �
tuned_profiler r r r; � s zController.switch_profilec C s | j j� }| jjj|d�S )N�unknown)rE r] r[ Z tuned_to_ppd�get)r r_ r r r r] � s
zController.active_profileZsss�uc C s6 |t kr$|tkr$tjjdt tf ��| jj||||�S )Nz'Only '%s' and '%s' profiles may be held)r r r7 �
exceptions�
DBusExceptionrF r= )r r r r r< r r r � HoldProfile� s zController.HoldProfiler c C s, | j j|�s tjjd| ��| j j|� d S )Nz No active hold for cookie '%s')rF r> r7 rc rd r$ )r r& r< r r r �ReleaseProfile� s zController.ReleaseProfilec C s d S )Nr )r r& r r r r/ � s zController.ProfileReleasedr\ c C s: || j jkr tjjd| ��|| _| jj� | j|� d S )NzInvalid profile '%s') r[ r^ r7 rc rd rX rF rA r; )r r r r r �set_active_profile� s
z Controller.set_active_profilec C s | j � S )N)r] )r r r r �get_active_profile� s z Controller.get_active_profileZProfilesc C s t jdd� | jjj� D �dd�S )Nc S s g | ]}|t d ��qS ))r ZDriver)�DRIVER)r) r r r r r5 � s z+Controller.get_profiles.<locals>.<listcomp>za{sv})r6 )r7 r8 r[ r^ �keys)r r r r � get_profiles� s zController.get_profilesZActionsc C s t jg dd�S )N�s)r6 )r7 r8 )r r r r � get_actions� s zController.get_actionsr c C s | j S )N)rG )r r r r �get_performance_degraded� s z#Controller.get_performance_degradedr0 c C s
| j j� S )N)rF r3 )r r r r �get_active_profile_holds� s z#Controller.get_active_profile_holds) r
r r r rS rW �propertyr: r@ rZ rK r; r] r Zexportre rf �signalr/ Zproperty_setterrg Zproperty_getterrh rk rm rn ro �
__classcell__r r )rL r rB _ s$
rB ) r
r r Ztuned.utils.commandsr Z tuned.constsr Ztuned.ppd.configr r r �enumr rI r7 rM ra r" ri rP rR r �objectr r Z
interfacesZExportableInterfacerB r r r r �<module> s 9