U
K��^� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ e� d�Z
dZ dZ dZ
d\ZZZG dd � d ejj�ZG d
d � d e�ZG d d
� d
e�Zdadd� Zddd�ZdS )� N)�COLORS�
urwid_8_nameszsubiquitycore.screeni3K ipK iqK )�� �T � c s( e Zd Z� fdd�Z� fdd�Z� ZS )�SubiquityScreenc s< t � �� t�| tjj� | j�� }t� |�r8t
� |� d S �N) �super�start�urwid� emit_signal�display_common�INPUT_DESCRIPTORS_CHANGED�_term_input_file�fileno�os�isatty�ttyZsetraw)�self�fd�� __class__� �,/usr/share/subiquity/subiquitycore/screen.pyr
8 s
zSubiquityScreen.startc s t � �� t�| tjj� d S r )r �stopr r r
r �r r r r r @ s
zSubiquityScreen.stop)�__name__�
__module__� __qualname__r
r �
__classcell__r r r r r $ s r c s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )� LinuxScreenc s || _ t� jf |� d S r )�_colorsr �__init__�r �colors�kwargsr r r r" G s zLinuxScreen.__init__c s� t d�| _t�tj�� t| j� | j�� }t d�D ]0}t d�D ]"}| j
| d | ||d | < q>q2t�| j �� t |� t
� �� d S )N�0 � � � )� bytearray�curpal�fcntl�ioctl�sys�stdoutr �GIO_CMAP�copy�ranger! r �PIO_CMAPr r
)r Znewpal�i�jr r r r
K s
"zLinuxScreen.startc s$ t �| j�� t| j� t� �� d S r )r, r- r r r3 r+ r r r r r r r U s zLinuxScreen.stop)r r r r" r
r r r r r r r E s
r c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�TwentyFourBitScreenc s* � fdd�t t�D �| _t� jf |� d S )Nc s i | ]\}}|� | d �qS )r) r )�.0r4 �n�r$ r r �
<dictcomp>] s z0TwentyFourBitScreen.__init__.<locals>.<dictcomp>)� enumerater �_urwid_name_to_rgbr r" r# r r9 r r" \ s
�z TwentyFourBitScreen.__init__c C s8 |dkr dS |dkrdS |dkr$dS dj | j| � S dS ) a0 Return the "SGR" parameter for selecting color.
See https://en.wikipedia.org/wiki/ANSI_escape_code#SGR for an
explanation. We use the basic codes for black/white/default for
maximum compatibility; they are the only colors used when the
mono palette is selected.
Zwhite�7Zblack�0�default�9z 8;2;{};{};{}N)�formatr< )r Zcolorr r r �_cca s zTwentyFourBitScreen._ccc C s d� | �|j�| �|j��S )Nz [0;3{};4{}m)rA rB Z
foregroundZ
background)r �ar r r �_attrspec_to_escapeu s
�z'TwentyFourBitScreen._attrspec_to_escape)r r r r" rB rD r r r r r r6 Z s r6 c
C st t d krpzt�tj�� td�} W n4 tk
rT } zt� d|� W Y �dS d }~X Y nX t� d| | dk� | dka t S )N� zKDGKBTYPE failed %rFz&KDGKBTYPE returned %r, is_linux_tty %s� )
�
_is_linux_ttyr, r- r. r/ r � KDGKBTYPE�IOError�log�debug)�r�er r r � is_linux_tty~ s rN Fc C sR |dkrt j}|dkr t j}t� r0tt||d�S | r@t||d�S tt||d�S dS )rE N)�input�output)r. �stdinr/ rN r r r r6 )�asciiZinputfZoutputfr r r � make_screen� s rS )FNN)r, Zloggingr r. r r Zsubiquitycore.paletter r Z getLoggerrJ rH r0 r3 ZUO_RZUO_GZUO_BZ raw_displayZScreenr r r6 rG rN rS r r r r �<module> s"
!!
|