@wOGc@ sdZddklZddkZddkZddkZddki i Z ddk i Z ddkiZddkiZddkiZddkiZddkiZddkiZddkiZddkiZdfdYZdeiefdYZ dS(sQ These are classes to support contour plotting and labelling for the axes class i(tdivisionNtContourLabelercB sheZdZdZdZdZdZdZdZdZ dZ d Z d Z RS( s3Mixin to provide labelling capability to ContourSetcO s|idd}|idd}|idd|_|idd}t|djo"|i}tt|i}nt|djot|d}gg}}xHt|iD]7\} } | |jo|i| |i| qqWt|t|jo=dt |} | d 7} | t |i7} t | qan t d ||_ ||_ ti|_|djot|ii} not|ttt gjot d nCt|t jot|ii} n|ii||} | gt||_|djo(||_ti|i|i |_nXti|d t|i } tt|i |_tid | dti |_g|_!|i"|x!|i#D]}|i$i%|qWt&i'd|i#|_(|i(S(s clabel(CS, **kwargs) - add labels to line contours in CS, where CS is a ContourSet object returned by contour. clabel(CS, V, **kwargs) - only label contours listed in V keyword arguments: * fontsize = None: as described in http://matplotlib.sf.net/fonts.html * colors = None: - a tuple of matplotlib color args (string, float, rgb, etc), different labels will be plotted in different colors in the order specified - one string color, e.g. colors = 'r' or colors = 'red', all labels will be plotted in this color - if colors == None, the color of each label matches the color of the corresponding contour * inline = True: controls whether the underlying contour is removed (inline = True) or not (False) * fmt = '%1.3f': a format string for the label tfontsizetinlineitfmts%1.3ftcolorsisSpecified levels s don't match available levels s-Illegal arguments to clabel, see help(clabel)s$Font size must be an integer number.tNtcmaptnorms text.TextN()tgettNoneRtlentlevelstrangetlistt enumeratetappendtstrt ValueErrort TypeErrort label_levelst label_indicest font_managertFontPropertiestfptinttget_size_in_pointsttypetfloattset_sizetfslisttlabel_mappabletnpyttaketcvaluest label_cvaluesRtListedColormaptcmtScalarMappabletNoNormtcl_xytlabelstcltaxt add_artisttcbookt silent_listt label_list(tselftargstkwargsRRt_colorsR tindicestlevlabstitlevtmsgt font_sizeRtlabel((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pytclabels^               c C st|}|d|jodSntiti|dddf}titi|dddf}titi|dddf}titi|dddf}|}||d|jp||d|jodSndSdS(s-if contours are too short, don't plot a labeli iNig333333?(R R tamaxtarraytamin( R0t linecontourt labelwidthtlcsizetxmaxtxmintymaxtymintlw((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt print_label|s ((((*cC s|igjo{g}|iD]2}|ti||dd||ddq~}x3|D]#}|d|jodSq`dSq`WndSdS(s6if there's a label already nearby, find a better placeiiig333333?N(R(R tsqrt(R0txtyRFt_[1]tloctdisttd((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt too_closesB c C st|d}ti|}xk|D]c}||||||} } |i| | |oq&q&|ii| | f| | |fSq&W|d}||||||} } |ii| | f| | |fS(sB labels are ploted at a location with the smallest dispersion of the contour from a straight line unless there's another label nearby, in which case the second best place on the contour is picked up if there's no good place a label isplotted at the beginning of the contour ii(RR targsortROR(R( R0t distancestXXtYYtysizeRFthysizetadisttindRIRJ((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pytget_label_coordss  cC s<ti|ot||}nt|||}|S(s$get the width of the label in points(R-tis_string_likeR (R0R7RtfsizeRF((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pytget_label_widthscC sA|i||i||i|i|i|iidS(s0set the label properties - color, fontsize, textN(tset_textt set_colortset_fontpropertiesRt set_clip_boxR+tbbox(R0R:ttexttcolor((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pytset_label_propss  cC s$ti|o|Sn ||SdS(sget the text of the labelN(R-RY(R0R7R((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pytget_textscC s%t|}t|d}t|ti|tid}t|ti|tid}|ii} | i |} | |\} } ti | dddfi } ti | dddfi }ti | | |j| | |j@|| |j|| |j@@\}t|djoti |d|d dj\}|ddjoDt|djo1||d}||d||d!}g}q||d }||dd}n|| }||d}|djo.| || |}}| || |}}n+| || |}}| || |}}| i ||f\}}| i ||f\}}ti||ff}ti||ff}|djot|djoC|dd|jo.|dd|joti||f}nt|djoC|dd|jo.|dd|joti||f}qnt|djoC|dd|jo.|dd|joti||f}nt|djoC|dd|jo.|dd|joti||f}n||gS(s<break a contour in two contours at the location of the labeliiNiii(R RtabsR tsintpitcosR+t transDatat seq_xy_tupstasarraytcopytnonzerotinverse_xy_tupR=t concatenate(R0R?trotR@RWRAthlwtylabeltxlabelttranstslcRIRJtxxtyytindstlindstiitlc1tlc2tnew_x1tnew_y1tnew_x2tnew_y2tnew_x1dtnew_y1dtnew_x2dtnew_y2dtnew_xy1tnew_xy2((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pytbreak_linecontoursX $$ %%!%$$    ((((cC st|}|djotti||}nd}|djo |}n|}titi|dddf||f}titi|dddf||f}|dddfi|d}|dddfi|d} |dddfi|d} |dddfi|d} ||| | | || |} ti| | d| |di} ti i g}t |D]}|t | || |q~d}|i |||||\}}}ti| || | i}||dti}|djo|d}n|djod|}ng}|D]}|t|qp~}|i||f}||||fS( sfind a good place to plot a label (relatively flat part of the contour) and the angle of rotation for the text object iNiiiiiZi(R RR tceiltresizeRktreshapeRHtraveltaddtreduceR ReRXtarctan2Rgttupletindex(R0R?R@tnsizetxsizeRTRRRStyfirsttylasttxfirsttxlasttstLRKR6RMRIRJRWtangletrotationt_[2]tltlctdind((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt locate_labels4    11""""")H$   'c C s.|i}|i}|ii}|ii|i}|i}xt|i |i||i|D]\}}} } } |i |} |i ||| } g}xxt | i D]g\}}ti|d|djo0ti||dtiddff}n|i|}|i|| o|i|| \}}}}|i||f\}}ti||d|dddd}|i||}|i||| |ii||ii| |o>|i||| |}|d| i |<|i|dqqqW| i i|q^WdS(NiiiRthorizontalalignmenttcentertverticalalignment(RRR+RiRtto_rgbaR#RtzipRt collectionsR[Rt _segmentsR tallRotnewaxisRjRGRRnRatTextRdRcR*Rt cl_cvaluesRtextend(R0RR RRtR3RticonR7RbtcvalueRZtconRFt additionstsegNumR?RuRIRJRRWtdxtdyttt_texttnew((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyR)CsF         '   ( t__name__t __module__t__doc__R;RGRORXR[RcRdRRR)(((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyRs ]     N +t ContourSetcB sweZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d ZRS( s Create and store a set of contour lines or filled regions. User-callable method: clabel Useful attributes: ax - the axes object in which the contours are drawn collections - a silent_list of LineCollections or PolyCollections levels - contour levels layers - same as levels for line contours; half-way between levels for filled contours. See _process_colors method. c O s||_|idd(|_|idt|_|idd(|_|idd|_|idd|_|idd(|_ |id d(|_ |id d(}|id d(|_ |id d(}|id d|_ |idt |_|idd|_|idd(|_t|t ipt|itioJt |_|d(jot i}n|i dj otdqn t|_|i d(j o|i d)jptn|i d(j o!t|i djptn|d(j ot|t iptn|i d(j o|d(j otdn|i djotid|_ n|i|\}}}|i d(j o%t i|i dt|i}n|ioti d|_!nti d|_!g|_"g|_#h|d <} |d(j o|| d |}t(i=|}t(i>|}|ii?||f||fg|ii@||f|iiA||fd(S(*s5 Draw contour lines or filled regions, depending on whether keyword arg 'filled' is False (default) or True. The first argument of the initializer must be an axes object. The remaining arguments and keyword arguments are described in ContourSet.contour_doc. R tfilledt linewidthst linestylestsolidtalphag?torigintextentRRRRtneithert antialiasedtnchunkitlocators-extend kwarg does not work yet with log scaletlowertuppertimageis"Either colors or cmap must be Nones image.originRscollections.PolyCollectionscollections.LineCollections!linewidths is ignored by contourfiitpointst antialiasedst edgecolorsR t linestylegscontour.negative_linestylet _nolegend_N(slowersuppersimage(BR+R R R tFalseRRRRRRRRtTrueRRRt isinstancetLogNormttickert LogLocatortlogscaleRtAssertionErrorR tColormaptmpltrcParamst _contour_argsR$tlayersR-R.RR*RR%R&t__init__t_process_colorstmatgetmasktnomasktwarningstwarnt_cntrtCntrt_levelsRttracetPolyCollectiontadd_collectionRt_process_linewidthst tlinewidthst_process_linestylestLineCollectiont monochromet set_linestylet set_labeltchangedtminimumtmaximumtupdate_datalimtset_xlimtset_ylim(R0R+R1R2RRRIRJtztkwt_masktCtlowerstupperstlevelt level_uppertnlisttcolRt tlinestylestwidthtlstyletlstx0tx1ty0ty1((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyR{s    1+$%                      "cC sg}|i|id|iD]}|t|fq ~}||_x-t||iD]\}}|i|q[Wx<t|i|i D]%\}}|i|i i|qWt i i |dS(NR(RR"RRttcolorsRRR]R*RRR%R&R(R0RKtrgbaRRbt collectionR:tcv((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyRs8   cC s|idjo7|ioti|_qGti|d|_n|i}|i}|i}|i|||}||d}||djo|dc|7iigGz?N( RR RRRt MaxNLocatortzmaxtzmint set_boundsRt_autoR(R0RRRR R R7tzmargin((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt_autolevs(         c C s|idjotdn|i\}}|idjo|idjo&titi|ti|Sq|i\}}}}ti |||}ti |||} ti|| Sn|idjo"d|d|f\}}}}n|i\}}}}t |||} t |||} |ti|d| }|ti|d| } |idjo| ddd} nti|| S(s Return X, Y arrays such that contour(Z) will match imshow(Z) if origin is not None. The center of pixel Z[i,j] depends on origin: if origin is None, x = j, y = i; if origin is 'lower', x = j + 0.5, y = i + 0.5; if origin is 'upper', x = j + 0.5, y = Nrows - i - 0.5 If extent is not None, x and y will be scaled to match, as in imshow. If origin is None and extent is not None, then extent will give the minimum and maximum values of x and y. isInput must be a 2D array.ig?RNi( tndimRtshapeRR RR tmeshgridtarangetlinspaceR( R0RtNytNxRRRRRIRJRR((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt_initialize_x_ys( &"c C sUti|ddti}ti|ddti}ti|ddti}|idjotdn|i\}}|i|ijo$|i|ijo|||fSn|idjp|idjotdn|i\}|i\}||jp ||jotddnti||\}}|||fS( s  For functions like contour, check that the dimensions of the input arrays match; if x and y are 1D, convert them to 2D using meshgrid. Possible change: I think we should make and use an ArgumentError Exception class (here and elsewhere). itdtypeiisInput z must be a 2D array.s Inputs x and y must be 1D or 2D.s,Length of x must be number of columns in z, s'and length of y must be number of rows.(R Rktfloat64RRRRR( R0R1RIRJRRRtnxtny((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt _check_xyz8s" &   c G s|io d}nd}t|}|djo5ti|ddti}|i|\}}nD|djo |i|d \}}}ntd||fti ||_ ti ||_ |i oH|i djo8ti|dj|}tid |i|_ nt|_|idjo|d jp |djo|i|d }ns|d }yHt|tjo|i||}nti|iti}Wntd ||fnX|io#t|djotdn||_nt|i|_|idjo.|iidt|id|i d n|idjo+|iit |id |i d nti|i|_ti!|i|_"ti#|i|_$|idjo#d|id|id |_"n|idjo#d|id |id|_$n|i|_%|iod|id |id |_%|idjo#d|i"|id |i%dtvminR<tvmaxR( R0R1tfntNargsRRIRJR7t level_arg((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyRUsd        .+##  #'cC s|ii|_|idj owdt|i}}|idjo d}n|id jo|d}nt|||_|i ti n |i|_|i i p|i |i|in|id jot|i _n|i|idS( s Color argument processing for contouring. Note that we base the color mapping on the contour levels, not on the actual range of the Z values. This means we don't have to worry about bad values in Z, and we always have the full dynamic range available for the selected levels. The color is based on the midpoint of the layer, except for an extended end layers. iR R!iR"iN(sbothsmin(sbothsmax(sbothsmaxsmin(RRRR R RRR R"tset_normR'Rtscaledtset_climR&R'Rtclipt set_array(R0ti0ti1((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyRs   cC s|i}t|i}|djotidfg|}nti|o@t||jo-t|t t i |t|}n<ti| o*t |t t gjo|g|}ng}|D]}||fq~}|S(Nslines.linewidth(RR R R RRR-titerableRRR RRR(R0RtNlevRRKtw((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyRs  #-*$cC s|i}t|i}|djodg|}nrti|o|g|}nQti|o@t||jo-t|tt i |t|}n|S(NR( RR R R R-RYR2RRR R(R0RR3R((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyRs  #-cC s|iS(s1For compatibility with artists, return self.alpha(R(R0((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt get_alphascC s||_|idS(s.For compatibility with artists, set self.alphaN(RR(R0R((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyt set_alphas sM contour and contourf draw contour lines and filled contours, respectively. Except as noted, function signatures and return values are the same for both versions. contourf differs from the Matlab (TM) version in that it does not draw the polygon edges, because the contouring engine yields simply connected regions with branch cuts. To draw the edges, add line contours with calls to contour. Function signatures contour(Z) - make a contour plot of an array Z. The level values are chosen automatically. contour(X,Y,Z) - X,Y specify the (x,y) coordinates of the surface contour(Z,N) and contour(X,Y,Z,N) - contour N automatically-chosen levels. contour(Z,V) and contour(X,Y,Z,V) - draw len(V) contour lines, at the values specified in sequence V contourf(..., V) - fill the (len(V)-1) regions between the values in V contour(Z, **kwargs) - Use keyword args to control colors, linewidth, origin, cmap ... see below X, Y, and Z must be arrays with the same dimensions. Z may be a masked array, but filled contouring may not handle internal masked regions correctly. C = contour(...) returns a ContourSet object. Optional keyword args are shown with their defaults below (you must use kwargs for these): * colors = None; or one of the following: - a tuple of matplotlib color args (string, float, rgb, etc), different levels will be plotted in different colors in the order specified - one string color, e.g. colors = 'r' or colors = 'red', all levels will be plotted in this color - if colors == None, the colormap specified by cmap will be used * alpha=1.0 : the alpha blending value * cmap = None: a cm Colormap instance from matplotlib.cm. - if cmap == None and colors == None, a default Colormap is used. * norm = None: a matplotlib.colors.Normalize instance for scaling data values to colors. - if norm == None, and colors == None, the default linear scaling is used. * origin = None: 'upper'|'lower'|'image'|None. If 'image', the rc value for image.origin will be used. If None (default), the first value of Z will correspond to the lower left corner, location (0,0). This keyword is active only if contourf is called with one or two arguments, that is, without explicitly specifying X and Y. * extent = None: (x0,x1,y0,y1); also active only if X and Y are not specified. If origin is not None, then extent is interpreted as in imshow: it gives the outer pixel boundaries. In this case, the position of Z[0,0] is the center of the pixel, not a corner. If origin is None, then (x0,y0) is the position of Z[0,0], and (x1,y1) is the position of Z[-1,-1]. * locator = None: an instance of a ticker.Locator subclass; default is MaxNLocator. It is used to determine the contour levels if they are not given explicitly via the V argument. * extend = 'neither', 'both', 'min', 'max' Unless this is 'neither' (default), contour levels are automatically added to one or both ends of the range so that all data are included. These added ranges are then mapped to the special colormap values which default to the ends of the colormap range, but can be set via Colormap.set_under() and Colormap.set_over() methods. **************** contour only: * linewidths = None: or one of these: - a number - all levels will be plotted with this linewidth, e.g. linewidths = 0.6 - a tuple of numbers, e.g. linewidths = (0.4, 0.8, 1.2) different levels will be plotted with different linewidths in the order specified - if linewidths == None, the default width in lines.linewidth in matplotlibrc is used contourf only: * antialiased = True (default) or False * nchunk = 0 (default) for no subdivision of the domain; specify a positive integer to divide the domain into subdomains of roughly nchunk by nchunk points. This may never actually be advantageous, so this option may be removed. Chunking introduces artifacts at the chunk boundaries unless antialiased = False (RRRRRRRRRRRRR5R6t contour_doc(((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyRls  i " &  ?   t(!Rt __future__RRt matplotlibRtnumpyR tmatplotlib.numerix.npymatnumerixtnpymaRtmatplotlib._cntrRtmatplotlib.tickerRtmatplotlib.transformst transformst matplotlib.cmR%tmatplotlib.colorsRtmatplotlib.collectionsRtmatplotlib.font_managerRtmatplotlib.textRatmatplotlib.cbookR-RR&R(((s;/Users/cmoad/workspace/matplotlib/lib/matplotlib/contour.pyss"    Q