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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) Z.d*Z/d+d, Z0d-d. Z1e2 d/d0d1Z3dS )2    )LerpGlyphSet)AbstractPenBasePenDecomposingPen)AbstractPointPenSegmentToPointPen)RecordingPenDecomposingRecordingPen)	Transform)defaultdictdeque)sqrtcopysignatan2piNz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 )InterpolatableProblemnothingmissing	open_path
path_count
node_countnode_incompatibilitycontour_orderwrong_start_pointkinkunderweight
overweight                        	   
      N)__name__
__module____qualname__NOTHINGMISSING	OPEN_PATH
PATH_COUNT
NODE_COUNTNODE_INCOMPATIBILITYCONTOUR_ORDERWRONG_START_POINTKINKUNDERWEIGHT
OVERWEIGHTseverity r7   r7   d/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/fontTools/varLib/interpolatableHelpers.pyr      s0    r   c                 C   s   t t|  dd ddS )z6Sort problem groups by their most severe problem type.c                 S   s   t dd | d D  S )Nc                 s   s*    | ]}t j|d   |dd V  qdS )type	tolerancer   N)r   r6   get).0pr7   r7   r8   	<genexpr>1   s
    
z2sort_problems.<locals>.<lambda>.<locals>.<genexpr>r   )min)_r7   r7   r8   <lambda>0   s
    zsort_problems.<locals>.<lambda>T)keyreverse)dictsorteditems)problemsr7   r7   r8   sort_problems+   s   rH   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.Nr7   )lkr7   r7   r8   rot_list;   s   rK   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___Pen_penvalue)selfPenglyphsetr7   r7   r8   rN   B   s   
zPerContourPen.__init__c                 C   s   |    | j| d S rM   )_newItemrP   moveTo)rR   p0r7   r7   r8   _moveToH   s   zPerContourPen._moveToc                 C   s   | j | d S rM   )rP   lineTo)rR   p1r7   r7   r8   _lineToL   s   zPerContourPen._lineToc                 C   s   | j || d S rM   )rP   qCurveTo)rR   rZ   p2r7   r7   r8   _qCurveToOneO   s   zPerContourPen._qCurveToOnec                 C   s   | j ||| d S rM   )rP   curveTo)rR   rZ   r]   p3r7   r7   r8   _curveToOneR   s   zPerContourPen._curveToOnec                 C      | j   d | _ d S rM   )rP   	closePathrR   r7   r7   r8   
_closePathU      

zPerContourPen._closePathc                 C   rb   rM   )rP   endPathrd   r7   r7   r8   _endPathY   rf   zPerContourPen._endPathc                 C   s   |    | _}| j| d S rM   )rO   rP   rQ   append)rR   penr7   r7   r8   rU   ]   s   zPerContourPen._newItemrM   )r(   r)   r*   rN   rX   r[   r^   ra   re   rh   rU   r7   r7   r7   r8   rL   A   s    
rL   c                   @   s   e Zd Zdd ZdS )PerContourOrComponentPenc                 C   s   |    | jd || d S )N)rU   rQ   addComponent)rR   	glyphNametransformationr7   r7   r8   rm   c   s   z%PerContourOrComponentPen.addComponentN)r(   r)   r*   rm   r7   r7   r7   r8   rk   b   s    rk   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 rM   )rQ   rd   r7   r7   r8   rN   i   s   
z SimpleRecordingPointPen.__init__Nc                 K      d S rM   r7   )rR   
identifierkwargsr7   r7   r8   	beginPathl      z!SimpleRecordingPointPen.beginPathreturnc                 C   rq   rM   r7   rd   r7   r7   r8   rg   o   ru   zSimpleRecordingPointPen.endPathc                 C   s    | j ||d u r
dndf d S )NFT)rQ   ri   )rR   ptsegmentTyper7   r7   r8   addPointr   s    z SimpleRecordingPointPen.addPointrM   )rv   N)r(   r)   r*   rN   rt   rg   ry   r7   r7   r7   r8   rp   h   s
    

rp   c                 C   s0   d}t | |D ]\}}|| }||| 7 }q|S Nr   )zipv0v1sx0x1dr7   r7   r8   vdiff_hypot2v   s
   r   c                 C   s@   d}t | |D ]\}}|| }||j|j |j|j  7 }q|S rz   )r{   realimagr|   r7   r7   r8   vdiff_hypot2_complex~   s
   r   c                    s   t  fddt|D S )Nc                 3   s     | ]\}} | | V  qd S rM   r7   )r<   ijGr7   r8   r>      s    z matching_cost.<locals>.<genexpr>)sum	enumerate)r   matchingr7   r   r8   matching_cost   s   r   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 rM   )int)r<   er7   r7   r8   r>      s    z<min_cost_perfect_bipartite_matching_scipy.<locals>.<genexpr>)lenlinear_sum_assignmentlistrangeallr   )r   nrowscolsr7   r7   r8   )min_cost_perfect_bipartite_matching_scipy   s
   r   c                 C   s>   t | }d g| }t | D ]\}}|||< q|t| |fS rM   )r   Munkrescomputer   )r   r   r   rowcolr7   r7   r8   +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')r   	Exception	itertoolspermutationsr   r   nextr   )r   r   r   best	best_costr=   costr7   r7   r8   .min_cost_perfect_bipartite_matching_bruteforce   s   

r   )r   )r   c                 C   s<   t t| j}t|| j| j| j| jd | jd | j| fS )Nr   )	r   absarear   meanXmeanYstddevXstddevYcorrelation)statssizer7   r7   r8   contour_vector_from_stats   s   
r   c                    sJ   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 r7   )r   )r<   r~   r}   r7   r8   
<listcomp>   s    z3matching_for_vectors.<locals>.<listcomp>.<listcomp>r7   r<   )m1r   r8   r      s    z(matching_for_vectors.<locals>.<listcomp>c                 3   s    | ]	} | | V  qd S rM   r7   )r<   r   )costsr7   r8   r>          z'matching_for_vectors.<locals>.<genexpr>)r   #min_cost_perfect_bipartite_matchingr   r   )m0r   r   r   r   identity_costr7   )r   r   r8   matching_for_vectors   s   
r   c                 C   s&   d}t | D ]
\}}|d> |B }q|S )Nr   r   )reversed)pointsbitsrw   br7   r7   r8   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 r7   )complex)r<   rw   r@   r7   r7   r8   r      s    z)points_complex_vector.<locals>.<listcomp>r    r   r   r   )
r   $_NUM_ITEMS_PER_POINTS_COMPLEX_VECTORextendr   ri   r   r   r   r   r   )
r   vectorr   r   rW   rZ   d0r]   d1crossr7   r7   r8   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 )Nrl   r   r   )r   r   r   r   ri   rK   )r   isomorphismsrC   reference_bitsr   r   r   multmaskr   r   r7   r7   r8   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   Nr7   r<   rJ   vr   r7   r8   r>   /  s   " 4find_parents_and_order.<locals>.<listcomp>.<genexpr>)r   rF   )r<   r   rI   r   r7   r8   r   ,  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   )r   )r<   r@   )	locationsr7   r8   r   :  s    c                    s"   g | ] t  fd dD qS )c                 3   s    | ]	}  |d V  qdS r   )r;   )r<   rJ   rI   r7   r8   r>   ?  r   r   )tupler   )axesr   r8   r   ?  s   " r   c                       i | ]\}}| v r||qS r7   r7   r   r   r7   r8   
<dictcomp>A      z*find_parents_and_order.<locals>.<dictcomp>c                    r   r7   r7   r   r   r7   r8   r   D  r   T)	overwritez.Not all masters are reachable; report an issuezParents: %sz	Order: %s)r   r   r   r   loginfowarningscipy.sparse.csgraphr   setupdatekeysrE   r   combinationsr{   r   nonzeror   addr   popleftri   ImportError)	glyphsetsr   r   parentsorderbasesr   graphrI   vectorsr   r   i_discrete_locationj_discrete_locationtreer   r   r   r   visitedqueuer7   )r   r   r   r8   find_parents_and_order'  st   







r   )4fontTools.ttLib.ttGlyphSetr   fontTools.pens.basePenr   r   r   fontTools.pens.pointPenr   r   fontTools.pens.recordingPenr   r	   fontTools.misc.transformr
   collectionsr   r   mathr   r   r   r   r   logging	getLoggerr   r   rH   rK   rL   rk   rp   r   r   r   r   r   r   scipy.optimizer   r   r   munkresr   r   r   r   r   r   r   r   r   r7   r7   r7   r8   <module>   sV    
!

$