&Kc@sdZddkZddkZddkZddkZddkZddkZddkZddkZdZ dZ dZ dZ ddZdZd Zd Zd Zd Zd ZdZdddddddddgddgddZddeeddddZdddZdddd Zeeeed!Zd"Zd#Zd$Zd%Z d&Z!dS('s\ deco.py Routines and Matplotlib Settings for Various Useful Plot Formats Lev, 8 Apr 2009 iNcCstiddtdS(sTurn TeX rendering onttexttusetexN(tpylabtrctTrue(((sc:\py\lib\deco.pyttexon#scCstiddtdS(sTurn TeX rendering offRRN(RRtFalse(((sc:\py\lib\deco.pyttexoff$scCs tidS(s/Check if TeX rendering is on. Return True/Falses text.usetex(RtrcParams(((sc:\py\lib\deco.pytistex%scCs:x3ttiiD]}d|ti|fGHqWdS(s/Print matplotlib settings in alphabetical orders%s: %sN(tsortedRRtkeys(tk((sc:\py\lib\deco.pytdumprc'sc KsP|djoti}ntidddti|dddd|dS(s Add a date stamp to the top right corner of the plot. Use today if 'tm' is not specified. Extra keywords are passed to 'pylab.figtext' g\(\?sPlotted tvattopthatrightN(tNonettimeRtfigtexttflibtdt2str(ttmtvargs((sc:\py\lib\deco.pyt datestamp,s c Kstii}toDtiid}d||i|i|iddf}nti|id|dddd |d S( s1 Add a comment to the bottom of the page is\parbox{%.2fin}{%s}s s\\g{Gz?RtbottomRtleftN( Rtgcft subplotparsR tget_size_inchesRRtreplaceR(RRtparst width_inches((sc:\py\lib\deco.pytfooter4s  .c Kst|}t|djotdntt|\}}}tii}|i|i }|i |i } | ||i |d} |i | } |||i |d} |i | } t|d|\}}|i |d| || }|i || | }|| }to:tiid}d|| |iddf}nti|||dd d d |d S( s* Add a text in place of a subplot is#Subplot index must be 3 digits longiis\parbox{%.2fin}{%s}s s\\RRRRN(tstrtlent ValueErrortmaptintRRRRRRRthspacetwspacetdivmodR RRR(tsubplotRRtstrowstcolstnumR ttotWidtht totHeighttfigHtsepHtfigWtsepWtrowNumtcolNumt figBottomtfigLefttfigTopR!((sc:\py\lib\deco.pytsubtext>s&     $cKs$tiihdd<dd<dd<dd<dd<d d <d gd <d gd <d gd<dd<dd<dd<dd<dd<dd<dd<dd<d|jotii|d|d=nd|jo|d=d"|d sfont.sans-serifs font.serifsfont.monospaceislegend.numpointsg?sfigure.subplot.leftgQ?sfigure.subplot.bottomgףp= ?sfigure.subplot.topgffffff?sfigure.subplot.rightg333333?sfigure.subplot.wspaceg?slines.markeredgewidthg?slines.linewidthsaxes.linewidthR?R@g%@RAi (g%@i(i i(RRRBRCR(RRD((sc:\py\lib\deco.pyt landscapeA4s:                       cKs3ttiihdd<dd<dd<dd<dd<d d <d gd <d gd <d gd<dd<td<dd<dd<dd<dd<dd<dd<dd<d|jotii|d|d=nd |ijod#|d sfont.sans-serifs font.serifsfont.monospaceislegend.numpointss text.usetexg333333?sfigure.subplot.leftg?sfigure.subplot.bottomgffffff?sfigure.subplot.topsfigure.subplot.rightg333333?slines.markeredgewidthg?slines.linewidthg?saxes.linewidthR?RAigffffff?(igffffff?(RRRRBRR RCtset_size_inches(RRD((sc:\py\lib\deco.pytpubplots8                     cKs<ttiihdd<dd<dd<dd<dd<d d <d gd <d gd <d gd<dd<td<dd<dd<dd<dd<dd<dd<dd<dd<d |jotii|d |d =nd!|ijod$|d!sfont.sans-serifs font.serifsfont.monospaceislegend.numpointss text.usetexg?sfigure.subplot.leftg333333?sfigure.subplot.bottomgffffff?sfigure.subplot.topsfigure.subplot.righti,s savefig.dpig?slines.markeredgewidthslines.linewidthg?saxes.linewidthR?RAii(ii(RRRRBRR RCRH(RRD((sc:\py\lib\deco.pyt overheadplots:                      cCs=|idd}x$dD]}|i|d|}qW|S(sL Escape TeX special characters &, $, %, #, _, {, }, \ in the string s\s\textbackslash s&$%#_{}(R(R#tch((sc:\py\lib\deco.pyt escapetex s i,iiiii i icCs=|i}x=ddgD]/} |i| od|t| }PqqW|idd}d} |dj o| |d7} n| dt|idd id d 7} |id jo| d |i7} n| d|i7} |dj ox|D]} x|D]} |id| d| } | i |i ||i |} | i |i |i } t| djoqnt i| i}t i| i}| d| | |idddd|iddddf7} qWqWn|dj o|i |i ||i |}|i |i |i }t|d joZ| dt i|iiddddt i|iiddddf7} qn|dj o|i |i ||i |}|i |i |i }t|d jo3| dt i|iidddd7} q/n| d7} | S(sY Convert trace info to TeX. If 'plm'/'mct'/'paro' is specified, temperature/pressure during taking this log is returned. 'maxextratime' is the maximum extention of time interval around the trace aquisition time in a search for thermometry/pressure points. 'xmits', 'gains' are PLM4 XMIT pulses and GAINs to be considered s/data/archive/phpc329/exps/mnt/phpc329/exps...t*t#s\begin{flushleft}s\\sFilename: \texttt{%s}\\s\\\-t/s/\-is %d averages, s captured: %stxmittgainis'\\PLM4 xmit %d gain %d: $%s$ [mK], $%s$tvarnametTtfmts%.3ftM_0s%.1fs0\\MCTNS $%s$ [mK,Greywall] / $%s$ [mK,PLTS high]s%.2fs\\Paroscientific: $%s$ [mbar]tPs\end{flushleft}N(tpatht startswithR$RRRLtNavttimespantattperiodt starttimet finaltimetaroundtavertaveragetM0RSttotextT_grwltT_pltsHRV(tinfotplmtmcttparot maxextratimetxmitstgainstcaptionRWtlocttexRPRQtpRbRS((sc:\py\lib\deco.pyt traceinfo2texsR    ,  J  $4  7 t.s%d/%m/%y %H:%M:%SiZc Ks|dj oti|nti||||||| } tii||} |oOtiii | tii t t t ti id|ddn|o3tiii | tii t t t nti| S(s A version of pylab.plot_date() Use a matplotlib.dates.DateFormatter with datefmt format. Call autofmt_xdate(rotation=45, ha='center') for a current figure. trotationRtcenterN(RRtaxest plot_datet matplotlibtdatest DateFormattertgcatxaxistset_major_formattertautoscale_viewRRRt autofmt_xdatetyaxistdraw_if_interactive( txtyRTttztxdatetydatetdatefmtRuRstkwargsRptfmtr((sc:\py\lib\deco.pytplotdateMs !  Rs%+gcCst|}d}|djoHxEtd|dD],}|||||d||d7}q3Wn|djo|||d|7}n|djo|||d7}n|djo d}n|dd jo|d}n|S( s Convert a list of polynomial coefficients 'p' into a TeX-style string using 'argname' for the argument. 'p' is assumed to be in power-decending order, as returned by 'numpy.polyfit' routine. When overriding coefficient format 'format', please remember to use '+' in the format abbreviation that puts '+' signes in front of positive coefficients. No extra '+' are added between terms in the polynomial. '+' is automatically deleted in front of the first argument, '0' is returned for a zeroth-order polynomail (empty array 'p') The string returned is not surrounded by math-mode delimiters (e.g. '$'). tiis^%diiit0t+(R$trange(RptargnametformattNR,ti((sc:\py\lib\deco.pytpoly2strfs  .   s%.3fcCsftitit|}||jp ||jo"|d|:}||d|Sn ||SdS(s? Convert a number to scientific representation in TeX i s\times 10^{%d}N(tnumpytfloortlog10tabs(tnRTt min_ordert max_ordertpower10((sc:\py\lib\deco.pyt float2texs c Csg}t|tio |g}nxh|D]`}t|tiio|i|q-t|tiio||i7}q-tdq-Wt |djodSnt i g}|D]}||i q~}t i g} |D]}| |i q~ } |ot|t|fnt|t|f}|ot| t| fnt| t| f} x>|D]6}|o|i|n|o|i| qqWt i|| gS(ss Equalise limits of all given axes and all given figures to show all the data. 'axes_and_figures' - an array of axes and figures to modify, can be a figure, then all subplots are adjusted 'x','y' - determine whenever to adjust x/y range 'x_inverse', 'y_inverse' - determine whenever the range should be from small to large or the other way round. sQ'axes_and_figures' contains an element that is neither a set of axes nor a figureiN(t isinstanceRtFigureRwRutAxestappendRCR%R$Rt concatenatetget_xlimtget_ylimtmaxtmintset_xlimtset_ylimtvstack( taxes_and_figuresRRt x_inverset y_inverseRutat_[1]txlimt_[2]tylim((sc:\py\lib\deco.pyteqlimss.  0088cCs3tiiitiid|tidS(sJ Set a LinearLocator with 'n' ticks as the current x-axis locator tnumticksN(RRzR{tset_major_locatorRwttickert LinearLocatortdraw(R((sc:\py\lib\deco.pyt setnumxtickss%cCs3tiiitiid|tidS(sJ Set a LinearLocator with 'n' ticks as the current y-axis locator RN(RRzRRRwRRR(R((sc:\py\lib\deco.pyt setnumytickss%cOs?dtids0<$      # - . ( ) 37 '