File: //opt/alt/python37/lib64/python3.7/site-packages/guppy/etc/__pycache__/RE_Rect.cpython-37.pyc
B
�]�) � @ s� d Z ddlmZmZmZ G dd� de�ZG dd� de�ZG dd� dee�Zd d
� Z d d � Z
G d
d� d�Z G dd� de e�Z G dd� de e�Z
dd� Zg dgfdd�Zdgfdd�Zdd� Zd d � ZdS ) a� Support functions for RE simplification.
This module is intended for use by the RE module.
It is in a separate module to keep RE itself cleaner
since the algorithm, 'rectangle selection', is a quite
separate part that depends on some tricky heuristics.
The primary entry function is
chooserects(lines, gauges)
It chooses 'the best' rectangles from lines to base simplification on.
A weight on atoms is given by gauges.
pr() gives some example usages of chooserects.
� )� immbitset� mutbitset� immbitrangec @ s: e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zd d � Zd
S )�Rect)�width�lines�gainmemo�lnos� all_lines� common_partc C s* || _ || _|r t|�dkr t�i | _d S )N� )r r �len�AssertionErrorr )�selfr r � r �D/opt/alt/python37/lib64/python3.7/site-packages/guppy/etc/RE_Rect.py�__init__ s z
Rect.__init__c s. || _ t� fdd�| jD ��| _| �� | _d S )Nc s g | ]}� t |� �qS r )�id)�.0�line)�lnobyidr r �
<listcomp> s z Rect.init2.<locals>.<listcomp>)r
r r r �get_common_partr )r r r
r )r r �init2 s z
Rect.init2c sR � j }|| @ }||kr2|� _ � fdd�|D �� _t|�dkrNt� jd �� _d S )Nc s g | ]}� j | �qS r )r
)r �lno)r r r r + s z$Rect.reducelines.<locals>.<listcomp>r r )r r r
r )r r �olnosr )r r � reducelines# s
zRect.reducelinesr c C s. g }x$| j | @ D ]}|�| j| � qW |S )N)r �appendr
)r � pickednosr �ir r r � get_lines/ s zRect.get_linesc C s d| j | jt| j�| jf S )Nz0<
dir = %d
width = %d
lnos = %s
lines = %s
>)�dirr �listr r )r r r r �__repr__5 s z
Rect.__repr__N)r ) �__name__�
__module__� __qualname__� __slots__r r r r r# r r r r r s
r c @ s& e Zd ZdZdZdd� Zddd�ZdS ) �LeftRectr r c C s | j d d | j� S )Nr )r r )r r r r r > s zLeftRect.get_common_partc C s0 g }x&| � |�D ]}|�|| jd � � qW |S )N)r r r )r r �ucr r r r �
get_uncommonsA s zLeftRect.get_uncommonsN)r )r$ r% r& r' r! r r* r r r r r( : s r( c @ s& e Zd ZdZdZdd� Zd dd�ZdS )
� RightRectr ���c C s& | j }|dkrg S | jd |d � S )Nr )r r )r �lor r r r L s zRightRect.get_common_partr c C sB g }| j }|dkrd }x$| �|�D ]}|�|d |� � q$W |S )Nr )r r r )r r r) �hir r r r r* R s zRightRect.get_uncommonsN)r )r$ r% r& r' r! r r* r r r r r+ H s r+ c C s2 | d krt |�S d}x|D ]}|| |�7 }qW |S )Nr )r
)�gauge�lstZgain�xr r r � sum_gauge\ s
r2 c C s8 x2|D ]*}t || �}t ||�}t||�}|r|S qW dS )Nr )r2 Zcmp)ZxsZys�gaugesr/ ZgxZgy�cr r r �
cmp_gaugedg s
r5 c @ s e Zd Zdd� ZdS )� InducedRectc C s& || _ |j| _|j| _|| _|| _d S )N)�sr r
r r )r r7 r r r r r r r s
zInducedRect.__init__N)r$ r% r& r r r r r r6 q s r6 c @ s e Zd ZdS )�InducedRightRectN)r$ r% r& r r r r r8 z s r8 c @ s e Zd ZdS )�InducedLeftRectN)r$ r% r& r r r r r9 ~ s r9 c
C s t | �dkrtd| �gS td| �td| �g}g }x�|�r|}g }x�|D ]�}|j}xnd}i }||jA }x6|jD ],} |t | �k r�|�| | g ��| � qld}qlW |s�t |�dkr�P |d |_}qRW |�|� |d7 }x:t|� � �D ]*\}
} t | �dkr�|�
|| �} |�| � q�W qFW q2W |S )Nr r ) r
r( r+ r r! r �
setdefaultr r" �items� __class__)
r ZnewrectsZ donerectsZoldrects�rr Zis_done�d�posr �k�vZnew_rr r r �brect� s6
rB Nc s� ����fdd�}� ���fdd�}� ���fdd�}dd� }� d d krV� d
d � � t d d � t��D ��}�}t|��x�D ]} | �|�� q~W �j|d
d� tt���}
t� �g } g �x6�|
kr�|� } �| jO �| � | � || � || � q�W | S )Nc s� | � � }t|�dk rd S g }xƈD ]�}|j| jkr4q"g }|� ��}|j� @ }t|�t|�ksbt�x(t||�D ]\}}||krn|�|� qnW t|�t|�kr"t|�} � fdd�|D �}|jdkr�t}
nt }
|
||| �} |�| � q"W |r��
|� d S )N� c s g | ] }� | �qS r r )r r )r r r r � s z*choose.<locals>.induce.<locals>.<listcomp>r, ) r* r
r! r r �zipr r r8 r9 �extend) r= Z uncommonsZirsr7 ZpssZuncsr Zuncr Zpslnosr4 Zir)r �newsr �rectsr r �induce� s0
zchoose.<locals>.inducec s� | j }g }x܈D ]�}|| kr q|j| jkr,q|j }||@ s<q|� M }|sLq|j}|sXqx��D ]n}||krlq^|j|jkrzq^|j � @ }||@ r�q^||@ }|s�q^||kr�q^|j} | s�q^t|| � �}
|
dkr�q^P q^W |�|� ��|� qW ��|� d S )Nr )r r! r r5 r �removerE ) r= �rlnosZtonewsr7 �slnosZscom�tZtlnosr Ztcomr4 )r3 rF r rG r r �overlap� sN
zchoose.<locals>.overlapc
s� �x��r�d } �d= �nLd } �x0t ��D �]"}|j� }|sJ��|� q*t|�d }|j}| d k �r.|slP |srq*|�r.| j}|j}d}x�� D ]�} y || }
W n( tk
r� t| | j�}
|
|| < Y nX |
| 9 }
y || } W n* tk
�r t| |j�} | || < Y nX | |9 } |
| }|r�P q�W |dk�r.q*|} |}
|�s>P |} |}|| }q*W | d k �rf��| � | d k r| � �� | jr| S qW d S )Nr r )
r" r rI r
r r �KeyErrorr2 r r )r= r7 rK Zsn�swZrwnZrmemoZsmemor4 r/ ZgrZrnZgsrJ �rw)r3 rF r rG r r �picknext� sh
zchoose.<locals>.picknextc S s | j | j| jd fS )Nr )r r! r )r1 r r r �cmpinit$ s zchoose.<locals>.cmpinitr r c S s g | ]\}}t |�|f�qS r )r )r r r r r r r * s zchoose.<locals>.<listcomp>T)�key�reverse)
�dict� enumerater" r �sortr r
r r r ) rG r r3 rH rM rQ rR r Zorectsr= ZallnosZ pickedrectsr )r3 r rF r rG r �choose� s. .7
rX c C s t | �}t|| |�}|S )N)rB rX )r r3 rG Zchoosenr r r � chooserectsD s rY c C s� t ddg�} t ddddg�} t| � t dddd g�} t| � t d
d d d
g�} t| � t ddddg�} td| � dddg} tt | �� tt ddg�� tt ddg�� tt ddddg�� tt ddddg�� tt dddd g�� tt d d d d d!g�� d S )"N�abcZade�abe�ace�xbyZab1Zab2Zac3Zac4Z1bfZ2bfZ3cfZ4cfZabdZbcxZbdyZTW�eZaxc�a� ZabcdZabce�fr> ZauvwZbuvw�bZaxuvwZbxuvwZaxyZbxy�cy)rY �print)r1 r r r �prJ sB
re c C s& x t d�D ]} tddddg�}q
W d S )N�d rZ r[ r\ r] )�rangerY )r r1 r r r �tmany� s
rh )�__doc__Z
guppy.setsr r r �objectr r( r+ r2 r5 r6 r8 r9 rB rX rY re rh r r r r �<module> s %
#F