File: //opt/alt/python37/lib64/python3.7/site-packages/guppy/etc/__pycache__/RE_Rect.cpython-37.opt-1.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 || _ || _i | _d S )N)r r r )�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 �lenr )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 )�gaugeZlstZgain�xr
r
r � sum_gauge\ s
r0 c C s8 x2|D ]*}t || �}t ||�}t||�}|r|S qW dS )Nr )r0 Zcmp)ZxsZys�gaugesr. ZgxZgy�cr
r
r �
cmp_gaugedg s
r3 c @ s e Zd Zdd� ZdS )� InducedRectc C s& || _ |j| _|j| _|| _|| _d S )N)�sr r
r r )r r5 r r r
r
r r r s
zInducedRect.__init__N)r# r$ r% r r
r
r
r r4 q s r4 c @ s e Zd ZdS )�InducedRightRectN)r# r$ r% r
r
r
r r6 z s r6 c @ s e Zd ZdS )�InducedLeftRectN)r# r$ r% r
r
r
r r7 ~ s r7 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
r@ 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� @ }x(t||�D ]\}}||krZ|�|� qZW 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 �zipr r r6 r7 �extend) r; Z uncommonsZirsr5 ZpssZuncsr Zuncr Zpslnosr2 Zir)r �newsr �rectsr
r �induce� s.
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 r3 r �removerC ) r; �rlnosZtonewsr5 �slnosZscom�tZtlnosr Ztcomr2 )r1 rD r rE 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 rG r r r �KeyErrorr0 r r )r; r5 rI Zsn�swZrwnZrmemoZsmemor2 r. ZgrZrnZgsrH Zrw)r1 rD r rE r
r �picknext� sh
zchoose.<locals>.picknextc S s | j | j| jd fS )Nr )r r r )r/ 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 ) rE r r1 rF rK rN rO r Zorectsr; ZallnosZ pickedrectsr
)r1 r rD r rE r �choose� s. .7
rU c C s t | �}t|| |�}|S )N)r@ rU )r r1 rE Zchoosenr
r
r � chooserectsD s rV 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�bZaxuvwZbxuvwZaxyZbxyZcy)rV �print)r/ r
r
r �prJ sB
ra c C s& x t d�D ]} tddddg�}q
W d S )N�d rW rX rY rZ )�rangerV )r r/ r
r
r �tmany� s
rd )�__doc__Z
guppy.setsr r r �objectr r' r* r0 r3 r4 r6 r7 r@ rU rV ra rd r
r
r
r �<module> s %
#F