B Afxa @sPdZddlZddlmZmZGdddejjZGdddejj Z d d Z dS) a pint.matplotlib ~~~~~~~~~~~~~~~ Functions and classes related to working with Matplotlib's support for plotting with units. :copyright: 2017 by Pint Authors, see AUTHORS for more details. :license: BSD, see LICENSE for more details. N)iterablesizedcs eZdZdZfddZZS) PintAxisInfoz:Support default axis and tick labeling and default limits.cstjd|ddS)z6Set the default label to the pretty-print of the unit.z{:P})ZlabelN)super__init__format)selfunits) __class__.lib/python3.7/site-packages/pint/matplotlib.pyrszPintAxisInfo.__init__)__name__ __module__ __qualname____doc__r __classcell__r r )r r rsrcsHeZdZdZfddZddZddZedd Zed d Z Z S) PintConverterzIImplement support for pint within matplotlib's unit conversion framework.cst||_dS)N)rr_reg)r registry)r r r rs zPintConverter.__init__cs0t|rfdd|DS|SdS)z4Convert :`Quantity` instances for matplotlib to use.csg|]}|qSr )_convert_value).0v)axisr unitr r #sz)PintConverter.convert..N)rr)r valuerrr )rr rr convert szPintConverter.convertcCs4t|dr||jS|j|||jSdS)zDHandle converting using attached unit or falling back to axis units.r N)hasattrtoZ magnituderQuantityZ get_units)r rrrr r r r's  zPintConverter._convert_valuecCst|S)z1Return axis information for this particular unit.)r)rrr r r axisinfo.szPintConverter.axisinfocCs,t|r t|r t|dddSt|ddS)zGGet the default unit to use for the given combination of unit and axis.rr N)rrgetattr)xrr r r default_units4szPintConverter.default_units) rrrrrrr staticmethodr!r$rr r )r r rs   rcCs@tjdkrtd|r*t|tjj|j<ntjj|jddS)aSet up matplotlib's unit support to handle units from a registry. Parameters ---------- registry : pint.UnitRegistry The registry that will be used. enable : bool Whether support should be enabled or disabled. Returns ------- z2.0z-Matplotlib >= 2.0 required to work with pint.N) matplotlib __version__ RuntimeErrorrr rr pop)renabler r r setup_matplotlib_handlers<s  r+) rZmatplotlib.unitsr&utilrrr ZAxisInforZConversionInterfacerr+r r r r  s #