Edit File: 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