ó m[c@`s«dZddlmZmZmZmZddlZddlmZm Z m Z ddl Z de fd„ƒYZde fd„ƒYZd efd „ƒYZeƒZdS( us The classes here provide support for using custom classes with Matplotlib, e.g., those that do not expose the array interface but know how to convert themselves to arrays. It also supports classes with units and units conversion. Use cases include converters for custom objects, e.g., a list of datetime objects, as well as for objects that are unit aware. We don't assume any particular units implementation; rather a units implementation must provide the register with the Registry converter dictionary and a `ConversionInterface`. For example, here is a complete implementation which supports plotting with native datetime objects:: import matplotlib.units as units import matplotlib.dates as dates import matplotlib.ticker as ticker import datetime class DateConverter(units.ConversionInterface): @staticmethod def convert(value, unit, axis): 'Convert a datetime value to a scalar or array' return dates.date2num(value) @staticmethod def axisinfo(unit, axis): 'Return major and minor tick locators and formatters' if unit!='date': return None majloc = dates.AutoDateLocator() majfmt = dates.AutoDateFormatter(majloc) return AxisInfo(majloc=majloc, majfmt=majfmt, label='date') @staticmethod def default_units(x, axis): 'Return the default unit for x or None' return 'date' # Finally we register our object type with the Matplotlib units registry. units.registry[datetime.date] = DateConverter() i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(titerablet is_numliketsafe_first_elementtAxisInfocB`s)eZdZddddddd„ZRS(u¶ Information to support default axis labeling, tick labeling, and default limits. An instance of this class must be returned by :meth:`ConversionInterface.axisinfo`. cC`s:||_||_||_||_||_||_dS(u: Parameters ---------- majloc, minloc : Locator, optional Tick locators for the major and minor ticks. majfmt, minfmt : Formatter, optional Tick formatters for the major and minor ticks. label : str, optional The default axis label. default_limits : optional The default min and max limits of the axis if no data has been plotted. Notes ----- If any of the above are ``None``, the axis will simply use the default value. N(tmajloctminloctmajfmttminfmttlabeltdefault_limits(tselfRR R R R R ((s/lib/python2.7/site-packages/matplotlib/units.pyt__init__;s      N(t__name__t __module__t__doc__tNoneR(((s/lib/python2.7/site-packages/matplotlib/units.pyR5s tConversionInterfacecB`sJeZdZed„ƒZed„ƒZed„ƒZed„ƒZRS(u‹ The minimal interface for a converter to take custom data types (or sequences) and convert them to values Matplotlib can use. cC`sdS(ue Return an `~units.AxisInfo` instance for the axis with the specified units. N(R(tunittaxis((s/lib/python2.7/site-packages/matplotlib/units.pytaxisinfo]scC`sdS(uQ Return the default unit for *x* or ``None`` for the given axis. N(R(txR((s/lib/python2.7/site-packages/matplotlib/units.pyt default_unitsescC`s|S(ué Convert *obj* using *unit* for the specified *axis*. If *obj* is a sequence, return the converted sequence. The output must be a sequence of scalars that can be used by the numpy array layer. ((tobjRR((s/lib/python2.7/site-packages/matplotlib/units.pytconvertlscC`s5t|ƒr'x"|D] }t|ƒSWn t|ƒSdS(u The Matplotlib datalim, autoscaling, locators etc work with scalars which are the units converted to floats given the current unit. The converter may be passed these floats, or arrays of them, even when units are set. N(RR(Rtthisx((s/lib/python2.7/site-packages/matplotlib/units.pyRvs  (RRRt staticmethodRRRR(((s/lib/python2.7/site-packages/matplotlib/units.pyRXs  tRegistrycB`s eZdZd„Zd„ZRS(u> A register that maps types to conversion interfaces. cC`stj|ƒi|_dS(N(tdictRt_cached(R((s/lib/python2.7/site-packages/matplotlib/units.pyR‰s cC`s”t|ƒsdSd}t|ddƒ}|dk rF|j|ƒ}n|dkrmt|dƒrm|j}nt|tjƒr&|j r&|j ƒ}y9tj |j ƒsÌ|j |tj|j ƒƒ}|SWq&tk r"|d}t|tjƒ s |j|jkr|j |ƒ}n|SXn|dkryt|ƒ}Wnttfk r[qX|r|t|ddƒkr|j |ƒ}|Sn|S(uŽ Get the converter for data that has the same type as *x*. If no converters are registered for *x*, returns ``None``. u __class__uvaluesiN(tlenRtgetattrtgetthasattrtvaluest isinstancetnptndarraytsizetraveltalltmaskt get_convertertargmintAttributeErrortshapeRt TypeErrort StopIteration(RRt convertertclassxtxravelt next_itemR((s/lib/python2.7/site-packages/matplotlib/units.pyR-s<       (RRRRR-(((s/lib/python2.7/site-packages/matplotlib/units.pyR…s (Rt __future__RRRRtsixtmatplotlib.cbookRRRtnumpyR'tobjectRRRRtregistry(((s/lib/python2.7/site-packages/matplotlib/units.pyt+s"  #-C