o
    i-                     @   s  d dl mZ d dlmZmZmZ d dlmZmZ d dl	m
Z
mZ d dlmZ d dlmZmZ d dlmZmZmZmZ d dlmZ d d	lZd d	lZed
ZG dd dZdd Zdd ZG dd deZG dd deZ G dd deZ!dd Z"dd Z#dd Z$dd Z%dd  Z&d!d" Z'z
d d#l(m)Z) e%Z*W n e+y   z
d d$l,m-Z- e&Z*W n e+y   e'Z*Y nw Y nw d%d& Z.d'd( Z/d)d* Z0d+Z1d,d- Z2d.d/ Z3e4 d0d1d2Z5d6d4d5Z6d	S )7    )LerpGlyphSet)AbstractPenBasePenDecomposingPen)AbstractPointPenSegmentToPointPen)RecordingPenDecomposingRecordingPen)	Transform)defaultdictdeque)sqrtcopysignatan2pi)EnumNzfontTools.varLib.interpolatablec                   @   sh   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZededededede	de
dedededediZdS )InterpolatableProblemZnothingmissingZ	open_pathZ
path_countZ
node_countZnode_incompatibilityZcontour_orderZwrong_start_pointZkinkZunderweightZ
overweight                        	   
      N)__name__
__module____qualname__ZNOTHINGMISSINGZ	OPEN_PATHZ
PATH_COUNTZ
NODE_COUNTZNODE_INCOMPATIBILITYZCONTOUR_ORDERZWRONG_START_POINTZKINKZUNDERWEIGHTZ
OVERWEIGHTseverity r$   r$   u/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/fontTools/varLib/interpolatableHelpers.pyr      s0    r   c                 C   s   t t|  dd ddS )zGSort problems by severity, then by glyph name, then by problem message.c                 S   s   t dd | d D  S )Nc                 s   s*    | ]}t j|d   |dd V  qdS )typeZ	tolerancer   N)r   r#   get).0pr$   r$   r%   	<genexpr>2   s
    
z2sort_problems.<locals>.<lambda>.<locals>.<genexpr>r   )min)_r$   r$   r%   <lambda>1   s
    zsort_problems.<locals>.<lambda>T)keyreverse)dictsorteditems)problemsr$   r$   r%   sort_problems,   s   r4   c                 C   s   | | d | d|   S )z{Rotate list by k items forward.  Ie. item at position 0 will be
    at position k in returned list.  Negative k is allowed.Nr$   )lkr$   r$   r%   rot_list<   s   r7   c                   @   sN   e Zd ZdddZdd Zdd Zdd	 Zd
d Zdd Zdd Z	dd Z
dS )PerContourPenNc                 C   s(   t | | || _|| _d | _g | _d S N)r   __init__Z	_glyphset_Pen_penvalue)selfZPenZglyphsetr$   r$   r%   r:   C   s
   
zPerContourPen.__init__c                 C   s   |    | j| d S r9   )_newItemr<   ZmoveTo)r>   p0r$   r$   r%   _moveToJ   s   zPerContourPen._moveToc                 C   s   | j | d S r9   )r<   ZlineTo)r>   p1r$   r$   r%   _lineToN   s   zPerContourPen._lineToc                 C   s   | j || d S r9   )r<   ZqCurveTo)r>   rB   p2r$   r$   r%   _qCurveToOneQ   s   zPerContourPen._qCurveToOnec                 C   s   | j ||| d S r9   )r<   ZcurveTo)r>   rB   rD   Zp3r$   r$   r%   _curveToOneT   s   zPerContourPen._curveToOnec                 C      | j   d | _ d S r9   )r<   Z	closePathr>   r$   r$   r%   
_closePathW      

zPerContourPen._closePathc                 C   rG   r9   )r<   endPathrH   r$   r$   r%   _endPath[   rJ   zPerContourPen._endPathc                 C   s   |    | _}| j| d S r9   )r;   r<   r=   append)r>   penr$   r$   r%   r?   _   s   zPerContourPen._newItemr9   )r   r    r!   r:   rA   rC   rE   rF   rI   rL   r?   r$   r$   r$   r%   r8   B   s    
r8   c                   @   s   e Zd Zdd ZdS )PerContourOrComponentPenc                 C   s   |    | jd || d S )N)r?   r=   addComponent)r>   Z	glyphNameZtransformationr$   r$   r%   rQ   e   s   z%PerContourOrComponentPen.addComponentN)r   r    r!   rQ   r$   r$   r$   r%   rO   d   s    rO   c                   @   s2   e Zd Zdd ZdddZdddZdd	d
ZdS )SimpleRecordingPointPenc                 C   s
   g | _ d S r9   )r=   rH   r$   r$   r%   r:   k   s   
z SimpleRecordingPointPen.__init__Nc                 K      d S r9   r$   )r>   
identifierkwargsr$   r$   r%   	beginPathn      z!SimpleRecordingPointPen.beginPathreturnc                 C   rS   r9   r$   rH   r$   r$   r%   rK   q   rW   zSimpleRecordingPointPen.endPathc                 C   s    | j ||d u r
dndf d S )NFT)r=   rM   )r>   ptZsegmentTyper$   r$   r%   addPointt   s    z SimpleRecordingPointPen.addPointr9   )rX   N)r   r    r!   r:   rV   rK   rZ   r$   r$   r$   r%   rR   j   s
    

rR   c                 C   s0   d}t | |D ]\}}|| }||| 7 }q|S Nr   )zipv0v1sZx0x1dr$   r$   r%   vdiff_hypot2x   s
   rc   c                 C   s@   d}t | |D ]\}}|| }||j|j |j|j  7 }q|S r[   )r\   realimagr]   r$   r$   r%   vdiff_hypot2_complex   s
   rf   c                    s   t  fddt|D S )Nc                 3   s     | ]\}} | | V  qd S r9   r$   )r(   ijGr$   r%   r*      s    z matching_cost.<locals>.<genexpr>)sum	enumerate)rj   matchingr$   ri   r%   matching_cost   s   rn   c                 C   sP   t | }t| \}}|tt|k sJ tdd |D }t|t| |fS )Nc                 s   s    | ]}t |V  qd S r9   )int)r(   er$   r$   r%   r*      s    z<min_cost_perfect_bipartite_matching_scipy.<locals>.<genexpr>)lenlinear_sum_assignmentlistrangeallrn   )rj   nrowscolsr$   r$   r%   )min_cost_perfect_bipartite_matching_scipy   s
   ry   c                 C   s>   t | }d g| }t | D ]\}}|||< q|t| |fS r9   )rq   MunkresZcomputern   )rj   rv   rx   rowcolr$   r$   r%   +min_cost_perfect_bipartite_matching_munkres   s
   

r}   c                 C   sn   t | }|dkrtdtt|}tt|}t| |}|D ]}t| |}||k r2t||}}q ||fS )Nr   z4Install Python module 'munkres' or 'scipy >= 0.17.0')rq   	Exception	itertoolspermutationsrt   rs   nextrn   )rj   rv   r   bestZ	best_costr)   Zcostr$   r$   r%   .min_cost_perfect_bipartite_matching_bruteforce   s   

r   )rr   )rz   c                 C   s<   t t| j}t|| j| j| j| jd | jd | j| fS )Nr   )	r   absZarear   meanXmeanYZstddevXZstddevYZcorrelation)statssizer$   r$   r%   contour_vector_from_stats   s   
r   c                    sV   t | }tt|}fdd| D  t \}}t fddt|D }|||fS )Nc                    s   g | ]  fd dD qS )c                    s   g | ]}t  |qS r$   )rc   )r(   r_   r^   r$   r%   
<listcomp>   s    z3matching_for_vectors.<locals>.<listcomp>.<listcomp>r$   r(   )m1r   r%   r      s    z(matching_for_vectors.<locals>.<listcomp>c                 3   s    | ]	} | | V  qd S r9   r$   )r(   rg   )costsr$   r%   r*          z'matching_for_vectors.<locals>.<genexpr>)rq   rs   rt   #min_cost_perfect_bipartite_matchingrk   )Zm0r   rv   Zidentity_matchingrm   rn   Zidentity_costr$   )r   r   r%   matching_for_vectors   s   
r   c                 C   s&   d}t | D ]
\}}|d> |B }q|S )Nr   r   )reversed)pointsbitsrY   br$   r$   r%   points_characteristic_bits   s   r   r   c           
      C   s  g }| s|S dd | D } t | }tdksJ | | d td   t | tk r9| | d td   t | tk s(t|D ]I}| | }|| | |d  }|| }||d  | |d  }|| }|||  |j|j |j|j  }	ttt	|	|	}	||	d  q=|S )Nc                 S   s   g | ]\}}t | qS r$   )complex)r(   rY   r,   r$   r$   r%   r      s    z)points_complex_vector.<locals>.<listcomp>r   r   r   r   )
rq   $_NUM_ITEMS_PER_POINTS_COMPLEX_VECTORextendrt   rM   rd   re   r   r   r   )
r   vectorrv   rg   r@   rB   Zd0rD   Zd1crossr$   r$   r%   points_complex_vector   s.   
r   c                 C   s   t | }t| }|r| d d d } t | }n|}t| }t|| dks&J t|| }d|> d }t|D ](}	|||	 > |@ ||	? B }
|
|kr^|t||	 | |rY|d |	 n|	|f q6d S )NrP   r   r   )r   rq   r   rt   rM   r7   )r   Zisomorphismsr/   Zreference_bitsrv   r   r   Zmultmaskrg   r   r$   r$   r%   add_isomorphisms  s$   
$r   discrete_axesc                   sV  d gt tt| d  }t tt| }r'fddtD }|r/tdt|| ntd zddlm} fddttD }t	  D ]	} 
|  qMt   fd	dD }	tttd
D ]2\}
}fddt |	|
 D }fddt |	| D }||krqnt|	|
 |	| ||
 |< qn||dd}| \}}tt	}t||D ]\}}|| | || | qd gt }g }t	 }t|}|r| }
||
 ||
 t||
 D ]}||vr|
||< || q|st|t|ksJ dW n
 ty   Y nw td| td| ||fS )Nr   c                    s.   g | ]\}}t  fd d| D r|qS )c                 3   s$    | ]\}}| vr|d kV  qdS r   Nr$   r(   r6   vr   r$   r%   r*   3  s   " 4find_parents_and_order.<locals>.<listcomp>.<genexpr>)ru   r2   )r(   rg   r5   r   r$   r%   r   0  s    z*find_parents_and_order.<locals>.<listcomp>zFound %s base masters: %szNo base master location foundr   )minimum_spanning_treec                    s   g | ]	}d gt   qS )r   )rq   )r(   r,   )	locationsr$   r%   r   >  s    c                    s"   g | ] t  fd dD qS )c                 3   s    | ]	}  |d V  qdS r   )r'   )r(   r6   r5   r$   r%   r*   C  r   r   )tupler   )axesr   r%   r   C  s   " r   c                       i | ]\}}| v r||qS r$   r$   r   r   r$   r%   
<dictcomp>E      z*find_parents_and_order.<locals>.<dictcomp>c                    r   r$   r$   r   r   r$   r%   r   H  r   T)	overwritez.Not all masters are reachable; report an issuezParents: %sz	Order: %s)rs   rt   rq   rl   logginginfowarningZscipy.sparse.csgraphr   setupdatekeysr1   r   combinationsr\   rc   Znonzeror   addr   popleftrM   ImportErrorlog)Z	glyphsetsr   r   parentsorderbasesr   graphr5   Zvectorsrg   rh   Zi_discrete_locationZj_discrete_locationtreerw   rx   r{   r|   visitedqueuer$   )r   r   r   r%   find_parents_and_order+  st   







r   Fc           
      C   s   | j }| j}| j}|| d d ||  d }|| d | }|| d | }|dkr2t|| |n	||k r:td nd}t }	|dk rEd}|rg|	| j | j }	|		| }	|	
dt| dt| }	|	S |	
t|t|}	|		|}	|	| j| j}	|	S )Ng      ?r   r   r   )Z	varianceXZ
covarianceZ	varianceYr   r   r
   	translater   r   rotatescaler   )
r   Zinversear   cdeltaZlambda1Zlambda2thetaZtransr$   r$   r%   transform_from_statso  s&   *	
r   )F)7ZfontTools.ttLib.ttGlyphSetr   ZfontTools.pens.basePenr   r   r   ZfontTools.pens.pointPenr   r   ZfontTools.pens.recordingPenr   r	   ZfontTools.misc.transformr
   collectionsr   r   mathr   r   r   r   enumr   r   r   	getLoggerr   r   r4   r7   r8   rO   rR   rc   rf   rn   ry   r}   r   Zscipy.optimizerr   r   r   Zmunkresrz   r   r   r   r   r   r   r   r   r   r$   r$   r$   r%   <module>   sZ    
"

$D