o
    aC                     @   s   d Z g dZddlmZ ddlmZ dd Zdd	 ZdddZdddZ	dd Z
dd Zdd Zd ddZG dd deZdd ZdddZedkridd
lZdd
lZeejdkr_ee  ee j d
S d
S )!z%Variation fonts interpolation models.)	nonNoneallNoneallEqual
allEqualTosubListnormalizeValuenormalizeLocationsupportScalarVariationModel    )noRound   )VariationModelErrorc                 C   s   dd | D S )Nc                 S   s   g | ]}|d ur|qS N .0lr   r   9/usr/lib/python3/dist-packages/fontTools/varLib/models.py
<listcomp>       znonNone.<locals>.<listcomp>r   lstr   r   r   r      s   r   c                 C   s   t dd | D S )Nc                 s   s    | ]}|d u V  qd S r   r   r   r   r   r   	<genexpr>       zallNone.<locals>.<genexpr>allr   r   r   r   r      s   r   Nc                    s>   d u rt fdd|D S  t  fdd|D S )Nc                 3   s    | ]} |kV  qd S r   r   r   item)refr   r   r      r   zallEqualTo.<locals>.<genexpr>c                 3   s    | ]	} |kV  qd S r   r   r   )mappedmapperr   r   r           r   )r   r   r    r   )r   r    r   r   r      s   r   c                 C   s@   | sdS t | }zt|}W n
 ty   Y dS w t|||dS )NT)r    )iternextStopIterationr   )r   r    itfirstr   r   r   r   #   s   r   c                 C   s(   t | t |ks
J dd t|| D S )Nc                 S   s   g | ]\}}|r|qS r   r   )r   r   tr   r   r   r   0   r   zsubList.<locals>.<listcomp>lenzip)truthr   r   r   r   r   .   s   r   c                 C   s   |\}}}||  kr|ks n t d|dd|dd|dtt| ||} | |kr0d} | S | |k r>| | ||  } | S | | ||  } | S )zNormalizes value based on a min/default/max triple.
    >>> normalizeValue(400, (100, 400, 900))
    0.0
    >>> normalizeValue(100, (100, 400, 900))
    -1.0
    >>> normalizeValue(650, (100, 400, 900))
    0.5
    z8Invalid axis values, must be minimum, default, maximum: z3.3fz,         )
ValueErrormaxmin)vtriplelowerdefaultupperr   r   r   r   3   s(   
	r   c                 C   s8   i }|  D ]\}}| ||d }t||||< q|S )a  Normalizes location based on axis min/default/max values from axes.
    >>> axes = {"wght": (100, 400, 900)}
    >>> normalizeLocation({"wght": 400}, axes)
    {'wght': 0.0}
    >>> normalizeLocation({"wght": 100}, axes)
    {'wght': -1.0}
    >>> normalizeLocation({"wght": 900}, axes)
    {'wght': 1.0}
    >>> normalizeLocation({"wght": 650}, axes)
    {'wght': 0.5}
    >>> normalizeLocation({"wght": 1000}, axes)
    {'wght': 1.0}
    >>> normalizeLocation({"wght": 0}, axes)
    {'wght': -1.0}
    >>> axes = {"wght": (0, 0, 1000)}
    >>> normalizeLocation({"wght": 0}, axes)
    {'wght': 0.0}
    >>> normalizeLocation({"wght": -1}, axes)
    {'wght': 0.0}
    >>> normalizeLocation({"wght": 1000}, axes)
    {'wght': 1.0}
    >>> normalizeLocation({"wght": 500}, axes)
    {'wght': 0.5}
    >>> normalizeLocation({"wght": 1001}, axes)
    {'wght': 1.0}
    >>> axes = {"wght": (0, 1000, 1000)}
    >>> normalizeLocation({"wght": 0}, axes)
    {'wght': -1.0}
    >>> normalizeLocation({"wght": -1}, axes)
    {'wght': -1.0}
    >>> normalizeLocation({"wght": 500}, axes)
    {'wght': -0.5}
    >>> normalizeLocation({"wght": 1000}, axes)
    {'wght': 0.0}
    >>> normalizeLocation({"wght": 1001}, axes)
    {'wght': 0.0}
    r   )itemsgetr   )locationaxesouttagr1   r0   r   r   r   r   L   s
   &r   Tc           	      C   s   d}|  D ]\\}\}}}|r-|dkrq||ks||krq|dk r&|dkr&q| |d}n
|| v s3J | | }||kr<q||ksD||krId} |S ||k rX||| ||  9 }q||| ||  9 }q|S )a;  Returns the scalar multiplier at location, for a master
    with support.  If ot is True, then a peak value of zero
    for support of an axis means "axis does not participate".  That
    is how OpenType Variation Font technology works.
    >>> supportScalar({}, {})
    1.0
    >>> supportScalar({'wght':.2}, {})
    1.0
    >>> supportScalar({'wght':.2}, {'wght':(0,2,3)})
    0.1
    >>> supportScalar({'wght':2.5}, {'wght':(0,2,4)})
    0.75
    >>> supportScalar({'wght':2.5, 'wdth':0}, {'wght':(0,2,4), 'wdth':(-1,0,+1)})
    0.75
    >>> supportScalar({'wght':2.5, 'wdth':.5}, {'wght':(0,2,4), 'wdth':(-1,0,+1)}, ot=False)
    0.375
    >>> supportScalar({'wght':2.5, 'wdth':0}, {'wght':(0,2,4), 'wdth':(-1,0,+1)})
    0.75
    >>> supportScalar({'wght':2.5, 'wdth':.5}, {'wght':(0,2,4), 'wdth':(-1,0,+1)})
    0.75
    g      ?r,   )r5   r6   )	r7   supportZotscalaraxisr2   peakr4   r0   r   r   r   r   y   s,   r   c                   @   s   e Zd ZdZd ddZdd Zeg fddZd	d
 Zdd Z	dd Z
dd ZedddZedddZdd Zedd Zdd ZedddZedddZdS )!r	   a"  
  Locations must be in normalized space.  Ie. base master
  is at origin (0)::

      >>> from pprint import pprint
      >>> locations = [       {'wght':100},       {'wght':-100},       {'wght':-180},       {'wdth':+.3},       {'wght':+120,'wdth':.3},       {'wght':+120,'wdth':.2},       {},       {'wght':+180,'wdth':.3},       {'wght':+180},       ]
      >>> model = VariationModel(locations, axisOrder=['wght'])
      >>> pprint(model.locations)
      [{},
       {'wght': -100},
       {'wght': -180},
       {'wght': 100},
       {'wght': 180},
       {'wdth': 0.3},
       {'wdth': 0.3, 'wght': 180},
       {'wdth': 0.3, 'wght': 120},
       {'wdth': 0.2, 'wght': 120}]
      >>> pprint(model.deltaWeights)
      [{},
       {0: 1.0},
       {0: 1.0},
       {0: 1.0},
       {0: 1.0},
       {0: 1.0},
       {0: 1.0, 4: 1.0, 5: 1.0},
       {0: 1.0, 3: 0.75, 4: 0.25, 5: 1.0, 6: 0.6666666666666666},
       {0: 1.0,
        3: 0.75,
        4: 0.25,
        5: 0.6666666666666667,
        6: 0.4444444444444445,
        7: 0.6666666666666667}]
	Nc                    s   t tdd  D t  krtd _|d ur|ng _dd  D  j jd}t |d_fdd D _ fd	djD _	
  i _d S )
Nc                 s   s     | ]}t t| V  qd S r   )tuplesortedr5   r   r   r   r   r      s    z*VariationModel.__init__.<locals>.<genexpr>zLocations must be unique.c                 S      g | ]}d d |  D qS )c                 S      i | ]\}}|d kr||qS r,   r   r   kr0   r   r   r   
<dictcomp>       z6VariationModel.__init__.<locals>.<listcomp>.<dictcomp>r5   r   locr   r   r   r      rG   z+VariationModel.__init__.<locals>.<listcomp>	axisOrder)keyc                       g | ]} j |qS r   	locationsindexr   selfr   r   r      r   c                       g | ]}  |qS r   rQ   r   rP   r   r   r          )r)   setr   origLocationsrL   getMasterLocationsSortKeyFuncr@   rP   mappingreverseMapping_computeMasterSupports
_subModels)rS   rP   rL   ZkeyFuncr   )rP   rS   r   __init__   s   
zVariationModel.__init__c                 C   sb   d |vr| |fS t dd |D }| j|}|d u r*tt|| j| j}|| j|< |t||fS )Nc                 s   s    | ]}|d uV  qd S r   r   r   r0   r   r   r   r      r   z-VariationModel.getSubModel.<locals>.<genexpr>)r?   r^   r6   r	   r   rY   rL   )rS   r5   rM   ZsubModelr   r   r   getSubModel   s   
zVariationModel.getSubModelc                 C   s   i | vrt di }| D ]2}t|dkrqtt|}|| }||vr(dh||< ||| vs7J d|||f || | qdd }|||}|S )NzBase master not found.r   r,   z&Value "%s" in axisPoints["%s"] -->  %sc                    s   dd  fdd}|S )Nc                 S   s   | dk rdS | dkrdS dS )Nr
   r   r   r0   r   r   r   sign  s   zJVariationModel.getMasterLocationsSortKeyFunc.<locals>.getKey.<locals>.signc              	      s   t  }fdd  D } fddD }|fddt  D  |t | tfdd|D t|t fdd|D t fdd|D fS )	Nc                    s(   g | ]\}}| v r| | v r|qS r   r   )r   r=   value)
axisPointsr   r   r     s    
z]VariationModel.getMasterLocationsSortKeyFunc.<locals>.getKey.<locals>.key.<locals>.<listcomp>c                    s   g | ]}| v r|qS r   r   r   r=   rJ   r   r   r     r   c                    s   g | ]}| vr|qS r   r   rg   rK   r   r   r     r   c                 3   s&    | ]}| v r  |nd V  qdS )i   NrU   rg   rK   r   r   r     s
    
z\VariationModel.getMasterLocationsSortKeyFunc.<locals>.getKey.<locals>.key.<locals>.<genexpr>c                 3   s    | ]	} | V  qd S r   r   rg   )rJ   rd   r   r   r         
c                 3   s    | ]	}t  | V  qd S r   )absrg   rh   r   r   r      ri   )r)   r5   extendr@   keysr?   )rJ   ZrankZonPointAxesZorderedAxesrL   rf   rd   rh   r   rM   
  s*   
zIVariationModel.getMasterLocationsSortKeyFunc.<locals>.getKey.<locals>.keyr   )rf   rL   rM   r   rm   r   getKey  s   z<VariationModel.getMasterLocationsSortKeyFunc.<locals>.getKey)r   r)   r#   r"   add)rP   rL   rf   rJ   r=   re   rn   retr   r   r   rZ      s$   


!z,VariationModel.getMasterLocationsSortKeyFuncc                    sj   fdd|D }fdd|D _ dd j D  fdd D _ fddjD _i _|S )Nc                    s   g | ]} | qS r   r   r   idx)master_listr   r   r   -      z1VariationModel.reorderMasters.<locals>.<listcomp>c                    s   g | ]} j | qS r   )rY   rq   rR   r   r   r   .  rW   c                 S   rA   )c                 S   rB   rC   r   rD   r   r   r   rF   0  rG   z<VariationModel.reorderMasters.<locals>.<listcomp>.<dictcomp>rH   rI   r   r   r   r   /  s    c                    rN   r   rO   r   rR   r   r   r   2  r   c                    rT   r   rU   r   rV   r   r   r   3  rW   )rY   r[   rP   r\   r^   )rS   rs   r[   Znew_listr   )rP   rs   rS   r   reorderMasters*  s   zVariationModel.reorderMastersc                 C   s  g | _ |  }t|D ]\}}t| }|d | D ]}t| |s'qd}| D ]&\}\}}	}
||vsO|| d |	ksS||| d   k rN|
k sSn d} nq-|sWqi }d}| D ]L}|| d }||v smJ || \}}}
||
}}||k r|}|| ||  }n||k r|}|| |
|  }nq_||kri }|}||kr|||f||< q_| D ]\}}|||< qq| j | q|   d S )NTr   Frb   )	supports_locationsToRegions	enumeraterX   rl   issubsetr5   append_computeDeltaWeights)rS   regionsiregionZlocAxesZprev_regionZrelevantr=   r2   r>   r4   ZbestAxesZ	bestRatiovallocVZnewLowerZnewUpperZratior1   r   r   r   r]   7  s\   	

z%VariationModel._computeMasterSupportsc                 C   s   | j }i }i }|D ]!}| D ]\}}t||||||< t||||||< qq	g }|D ])}i }	| D ]\}
}|dkrId|||
 f|	|
< q7||
 |df|	|
< q7||	 q/|S )Nr
   )rP   r5   r/   r6   r.   rz   )rS   rP   ZminVZmaxVr   rE   r0   r|   rJ   r~   r=   r   r   r   r   rw   o  s"   z"VariationModel._locationsToRegionsc                 C   s`   g | _ t| jD ]%\}}i }t| jd | D ]\}}t||}|r&|||< q| j | qd S r   )deltaWeightsrx   rP   rv   r   rz   )rS   r}   rJ   ZdeltaWeightjr;   r<   r   r   r   r{     s   
z#VariationModel._computeDeltaWeightsroundc          
      C   s   t |t | jksJ | j}g }t| jD ]-\}}|||  }| D ]\}}	|	dkr2||| 8 }q#||| |	 8 }q#||| q|S )Nr   )r)   r   r\   rx   r5   rz   )
rS   masterValuesr   r[   r9   r}   Zweightsdeltar   Zweightr   r   r   	getDeltas  s   zVariationModel.getDeltasc                C   s"   |  |\}}|j||d|jfS Nr   )ra   r   rv   )rS   r5   r   modelr   r   r   getDeltasAndSupports  s   z#VariationModel.getDeltasAndSupportsc                    s    fdd| j D S )Nc                    s   g | ]}t  |qS r   )r   )r   r;   rh   r   r   r     rW   z-VariationModel.getScalars.<locals>.<listcomp>)rv   )rS   rJ   r   rh   r   
getScalars  s   zVariationModel.getScalarsc                 C   sT   d }t | t |ksJ t| |D ]\}}|sq|| }|d u r#|}q||7 }q|S r   r(   )deltasscalarsr0   r   r<   Zcontributionr   r   r   interpolateFromDeltasAndScalars  s   
z.VariationModel.interpolateFromDeltasAndScalarsc                 C   s   |  |}| ||S r   )r   r   )rS   rJ   r   r   r   r   r   interpolateFromDeltas  s   
z$VariationModel.interpolateFromDeltasc                C   s   | j ||d}| ||S r   )r   r   )rS   rJ   r   r   r   r   r   r   interpolateFromMasters     z%VariationModel.interpolateFromMastersc                C   s   | j ||d}| ||S r   )r   r   )rS   r   r   r   r   r   r   r    interpolateFromMastersAndScalars  r   z/VariationModel.interpolateFromMastersAndScalarsr   )__name__
__module____qualname____doc__r_   ra   staticmethodrZ   ru   r]   rw   r{   r   r   r   r   r   r   r   r   r   r   r   r   r	      s$    
,
48
r	   c                    s   |  }|s S  |v r|  S t|} |k r  ||  | S t|} |kr0 ||  | S t fdd|D }t fdd|D }|| }|| }|||  |  ||   S )Nc                 3   s    | ]	}| k r|V  qd S r   r   r   rE   rc   r   r   r     r!   z%piecewiseLinearMap.<locals>.<genexpr>c                 3   s    | ]	}| kr|V  qd S r   r   r   rc   r   r   r     r!   )rl   r/   r.   )r0   r[   rl   rE   abZvaZvbr   rc   r   piecewiseLinearMap  s    r   c           
         s\  ddl m} ddl}|jdtjd}|jdddd	d
 |jdd}|jdddtd |jdddddd |	| } || j
d ddlm} | jr|ddlm} | }|| j dd |jD }td || |  td dd |jD }|| ndd ttd td!d" D   fd#d| jD }t|}	td$ ||	j td% ||	j dS )&z*Normalize locations on a given designspacer
   )configLoggerNzfonttools varLib.models)descriptionz
--loglevelZLEVELINFOz Logging level (defaults to INFO))metavarr3   helpT)requiredz-dz--designspaceZDESIGNSPACE)r   typez-lz--locationsZLOCATION+zFMaster locations as comma-separate coordinates. One must be all zeros.)r   nargsr   )level)pprint)DesignSpaceDocumentc                 S      g | ]}|j qS r   r7   r   sr   r   r   r         zmain.<locals>.<listcomp>zOriginal locations:zNormalized locations:c                 S   r   r   r   r   r   r   r   r     r   c                 S   s   g | ]}t |qS r   )chr)r   cr   r   r   r      rt   AZr   c              	      s*   g | ]}t t d d |dD qS )c                 s   s    | ]}t |V  qd S r   )floatr`   r   r   r   r     r   z"main.<locals>.<listcomp>.<genexpr>,)dictr*   splitr   r8   r   r   r     s    zSorted locations:z	Supports:)Z	fontToolsr   argparseArgumentParsermainr   add_argumentadd_mutually_exclusive_groupstr
parse_argsZloglevelr   ZdesignspaceZfontTools.designspaceLibr   readZsourcesprint	normalizerangeordrP   r	   rv   )
argsr   r   parsergroupr   r   docZlocsr   r   r   r   r     sX   

 

r   __main__r   )T)r   __all__ZfontTools.misc.roundToolsr   errorsr   r   r   r   r   r   r   r   r   objectr	   r   r   r   Zdoctestsysr)   argvexitZtestmodZfailedr   r   r   r   <module>   s.    


-0  
8