ó ÄCõRc@sdZddlmZddlmZmZddlZdd d„ƒYZdd d„ƒYZ d e fd „ƒYZ e ƒZ dS( sQ The classes here provide support for using custom classes with matplotlib, eg those that do not expose the array interface but know how to converter themselves to arrays. It also supoprts classes with units and units conversion. Use cases include converters for custom objects, eg 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 a ConversionInterface, and the register with the Registry converter dictionary. 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 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 a converter units.registry[datetime.date] = DateConverter() iÿÿÿÿ(tprint_function(titerablet is_numlikeNtAxisInfocBs)eZdZddddddd„ZRS(sYinformation to support default axis labeling and tick labeling, and default limitscCs:||_||_||_||_||_||_dS(sW majloc and minloc: TickLocators for the major and minor ticks majfmt and minfmt: TickFormatters for the major and minor ticks label: the default axis label default_limits: the default min, max of the axis if no data is present If any of the above are None, the axis will simply use the default N(tmajloctminloctmajfmttminfmttlabeltdefault_limits(tselfRRRRRR ((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyt__init__5s     N(t__name__t __module__t__doc__tNoneR (((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyR2s tConversionInterfacecBsJeZdZed„ƒZed„ƒZed„ƒZed„ƒZRS(s‚ The minimal interface for a converter to take custom instances (or sequences) and convert them to values mpl can use cCsdS(sCreturn an units.AxisInfo instance for axis with the specified unitsN(R(tunittaxis((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pytaxisinfoLscCsdS(s8return the default unit for x or None for the given axisN(R(txR((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyt default_unitsQscCs|S(sÙ convert obj using unit for the specified axis. If obj is a sequence, return the converted sequence. The ouput must be a sequence of scalars that can be used by the numpy array layer ((tobjRR((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pytconvertVscCs5t|ƒr'x"|D] }t|ƒSWn t|ƒSdS(s® 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. Derived conversion interfaces may opt to pass plain-ol unitless numbers through the conversion interface and this is a helper function for them. N(RR(Rtthisx((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyR_s  (R R Rt staticmethodRRRR(((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyRGs  tRegistrycBs eZdZd„Zd„ZRS(s2 register types with conversion interface cCstj|ƒi|_dS(N(tdictR t_cached(R ((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyR us cCs2t|ƒsdSd}t|ddƒ}|dk rF|j|ƒ}nt|tjƒrÑ|jrÑ|jƒ}y9tj |j ƒs¥|j |tj |j ƒƒ}|SWqÑt k rÍ|j |dƒ}|SXn|dkr.t|ƒr.xB|D]7}|rð|t|ddƒkrð|j |ƒ}|SqðWn|S(s3get the converter interface instance for x, or Nonet __class__iN(tlenRtgetattrtgett isinstancetnptndarraytsizetraveltalltmaskt get_convertertargmintAttributeErrorR(R Rt convertertclassxtxravelR((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyR(ys,      (R R RR R((((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyRqs ((( Rt __future__Rtmatplotlib.cbookRRtnumpyR"RRRRtregistry(((sJ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/units.pyt,s *3