B d@sdZddlZddlZddlZddlmZddl m Z m Z ddddZ dd iZ xe D]\ZZee e<qVWd d ZGd d d eZddZddZddZddZddZeeedZeedZdddZddZdS)zQ This Module contains generic utility functions for creating plots within PyCBC. N) HTMLParser)escapeunescapez"z'z@)"'@rcCs t|tS)z Sanitize text for html parsing )r escape_table)textr c/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/results/metadata.py html_escapesr c@s$eZdZddZddZddZdS) MetaParsercCsi|_t|dS)N)metadatar__init__)selfr r r rszMetaParser.__init__cCsdS)Nr )rdatar r r handle_dataszMetaParser.handle_datacCsVi}x|D]\}}|||<q W|dkrRd|krR|ddkrRt|dt|j|d<dS)Ndivclassz pycbc-metavaluekey)runescape_tabler)rtagattrsattrrrr r r handle_starttag!s  zMetaParser.handle_starttagN)__name__ __module__ __qualname__rrrr r r r r sr c Cszt|tr|}nddlm}||f|}t|d}x8|D],\}}t|t}dt||f} || qN) isinstancestrZmpld3ropenitemsrrwrite) figfilenamefig_kwdskwdsr rfrrliner r r save_html_with_metadata)s     r,cCsTt}t|d}d|kr(td|||t|j}|tj ||S)z Get metadata from html file rz pycbc-metazLOADING HTML FILE %s) r r#readprintfeed ConfigParserr add_sectionospathbasename)r'parserrcpr r r load_html_metadata9s   r8c Csjddlm}m}|j|f|||}|}x$|D]}|t|t||q6W|j|d|ddS)z5 Save a matplotlib figure to a png with metadata r)ImagePngImagePluginZpng)ZpnginfoN) PILr9r:savefigr#ZPngInfoadd_textr"save) r&r'r(r)r9r:Zimmetarr r r save_png_with_metadataEs  r@c Csddlm}||h}|j|fddi||}xB|D]:}t|dkr`t|||d<q:t|||t|<q:WWdQRXdS)z:Save a matplotlib figure to a PDF file with metadata. r)PdfPagesformatZpdftitleZTitleN)Zmatplotlib.backends.backend_pdfrAr<Zinfodictr"lower)r&r'r(r)rAZpdfprrr r r save_pdf_with_metadataSs   rEcCs8ddlm}||j}t|}|tj||S)Nr)r9) r;r9r#infor1r2r3r4r5)r'r9rr7r r r load_png_metadatacs    rG)z.pngz.htmlz.pdf)z.pngz.htmlcKsh|dkr i}y2tj|d}tjj|d<t|||||Wn$tk rbtd||fYnXdS)aA Save plot to file with metadata included. Kewords translate to metadata that is stored directly in the plot file. Limited format types available. Parameters ---------- fig: matplotlib figure The matplotlib figure to save to the file filename: str Name of file to store the plot. NversionzJCannot save file %s with metadata, extension %s not supported at this time) r3r4splitextpycbcrIZgit_verbose_msg_metadata_saverKeyError TypeError)r&r'r(r) extensionr r r save_fig_with_metadatars  rPcCsFytj|d}t||Stk r@td||fYnXdS)z Load the plot related metadata saved in a file Parameters ---------- filename: str Name of file load metadata from. Returns ------- cp: ConfigParser A configparser object containing the metadata rHzJCannot read metadata from file %s, extension %s not supported at this timeN)r3r4rJ_metadata_loaderrMrN)r'rOr r r load_metadata_from_files  rR)N)__doc__os.pathr3Z pycbc.versionrK configparserr1 html.parserrZxml.sax.saxutilsrrrrr$kvr r r,r8r@rErGrLrQrPrRr r r r s0