ó ±éRc@svdZddlZddlZddlZddljZddlj Z d„Z ie _ dddddd„Z dS(sÆ Some io tools for excel -- requires xlwt Example usage: import matplotlib.mlab as mlab import mpl_toolkits.exceltools as exceltools r = mlab.csv2rec('somefile.csv', checkrows=0) formatd = dict( weight = mlab.FormatFloat(2), change = mlab.FormatPercent(2), cost = mlab.FormatThousands(2), ) exceltools.rec2excel(r, 'test.xls', formatd=formatd) mlab.rec2csv(r, 'test.csv', formatd=formatd) iÿÿÿÿNcCs-t|ƒ}tjj|ƒ}|dk r.|Stj|ƒ}tjƒ}t |t j ƒr“dj dg|j ƒ}d||f|_d|_n€t |t jƒrï|j dkrãdj dg|j ƒ}d||f|_qd|_n$t |t jƒr d|_nd}||_|tj|<|S( sn copy the format, perform any overrides, and attach an xlstyle instance copied format is returned tt0s0.%s%%;[RED]-0.%s%%gð?is#,##0.%s;[RED]-#,##0.%ss#,##;[RED]-#,##N(thashtxlformat_factorytcreated_formatstgettNonetcopytdeepcopytexceltXFStylet isinstancetmlabt FormatPercenttjoint precisiontnum_format_strtscalet FormatFloatt FormatInttxlstyle(tformattkeytfmt_Rtzeros((sQ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/exceltools.pyRs*        itNaNtInfcCszt}tj|ƒr?|}tjƒ} | jdƒ}t}n|dkrWtƒ}ng} tj ƒ} t| _ tj ƒ} | | _ xœt |jjƒD]ˆ\} }|j|}|j|ƒ}|dkrïtjj|jtjƒƒ}nt|ƒ}|j||| || ƒ| j|ƒqšW|d7}tjt|jjƒƒ}x|D] }x÷|D]ï} || }| | }|j|ƒ}tj|ƒr±|j||| |ƒq_tj|ƒrtj|ƒ}|dkrä|}n d|}|j||| |ƒq_|jdkr1|j||| |ƒq_|j||| ||jƒq_W|d7}qRW|rv| j |ƒn|S(sv save record array r to excel xlwt worksheet ws starting at rownum. if ws is string like, assume it is a filename and save to it start writing at rownum, colnum formatd is a dictionary mapping dtype name -> mlab.Format instances nanstr is the string that mpl will put into excel for np.nan value The next rownum after writing is returned t worksheetiis-%sN(!tFalsetcbooktis_string_likeR tWorkbookt add_sheettTrueRtdicttFonttboldR tfontt enumeratetdtypetnamesRR tdefaultformatdttypet FormatObjRtwritetappendtnptarangetlenttovalt safe_isnant safe_isinftsignRtsave(trtwstformatdtrownumtcolnumtnanstrtinfstrtautosavetfilenametwbtformatsR%tstylehdrtitnametdtRtindtrowtvaltsgnts((sQ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/exceltools.pyt rec2excelGsV          !        !(t__doc__RtnumpyR.txlwtR tmatplotlib.cbookRtmatplotlib.mlabR RRRRJ(((sQ/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/exceltools.pyts    &