o e+!@sPddlmZddlmZddlZ     d ddZddZd d Zd d ZdS)) spec_to_html)import_vl_convertNc Ksddlm}m} |dkrtd| r||}d}|dvr)t|||fd|i|S|dkr>t|f||||d |} d | iS|dkrS|d urJtd d |d|iS|dkr[d|iStd)atConvert a vega-lite specification to a mimebundle The mimebundle type is controlled by the ``format`` argument, which can be one of the following ['html', 'json', 'png', 'svg', 'pdf', 'vega', 'vega-lite'] Parameters ---------- spec : dict a dictionary representing a vega-lite plot spec format : string {'html', 'json', 'png', 'svg', 'pdf', 'vega', 'vega-lite'} the file format to be saved. mode : string {'vega-lite'} The rendering mode. vega_version : string The version of vega.js to use vegaembed_version : string The version of vegaembed.js to use vegalite_version : string The version of vegalite.js to use. Only required if mode=='vega-lite' engine: string {'vl-convert', 'altair_saver'} the conversion engine to use for 'png', 'svg', 'pdf', and 'vega' formats **kwargs : Additional arguments will be passed to the generating function Returns ------- output : dict a mime-bundle representing the image Note ---- The png, svg, pdf, and vega outputs require the altair_saver package r)compile_with_vegafusionusing_vegafusionz vega-litezmode must be 'vega-lite'vega)pngsvgpdfrenginehtml)mode vega_versionvegaembed_versionvegalite_versionz text/htmlNzMust specify vegalite_versionz!application/vnd.vegalite.v{}+jsonjsonzapplication/jsonzPformat must be one of ['html', 'json', 'png', 'svg', 'pdf', 'vega', 'vega-lite'])Zaltair.utils.displayrr ValueError_spec_to_mimebundle_with_enginerformat) specrr rrrr kwargsrrr rGD:\Projects\ConvertPro\env\Lib\site-packages\altair/utils/mimebundle.pyspec_to_mimebundlesF,rcKs||dd}t||}|dkrt}ddlm}d|ddd}|dkr=|dkr2|} d | iS|j||d } d | iS|d krY|dkrN||} d | iS|j ||d } d | iS|d kr| dd} d} | d| } |dkrx|j || | d}n |j ||| | d}| | }t |\}}d|id||||difStdj|d|dkrddl}|j||fd|i|Stdj|d)aHelper for Vega-Lite to mimebundle conversions that require an engine Parameters ---------- spec : dict a dictionary representing a vega-lite plot spec format : string {'png', 'svg', 'pdf', 'vega'} the format of the mimebundle to be returned mode : string {'vega-lite', 'vega'} The rendering mode. engine: string {'vl-convert', 'altair_saver'} the conversion engine to use **kwargs : Additional arguments will be passed to the conversion function r N vlconvert)SCHEMA_VERSION_.r) vl_versionzapplication/vnd.vega.v5+jsonr z image/svg+xmlrZ scale_factorrHppi)scaler!)rr"r!z image/png)widthheightzUnexpected format {fmt!r}fmt altairsaverrr z$Unexpected normalized_engine {eng!r})Zeng)pop_validate_normalize_enginerZvegaliterjoinsplitZvegalite_to_vegaZ vega_to_svgZvegalite_to_svggetZ vega_to_pngZvegalite_to_png_pngxyrr altair_saverrender)rrr rr normalized_enginevlcrrvgr r"Z default_ppir!rfactorwhr.rrrrUs\        rcCs$zt}Wn tyd}Ynwzddl}Wn ty"d}Ynw|dur)dn |dddd}|dkrO|durAtd|dkrMtd j|d |S|d kr]|dur[td |S|dur|durm|dkrmd}|S|durud }|S|dkrtd j|d tdj|d tdj|dd)zHelper to validate and normalize the user-provided engine engine : {None, 'vl-convert', 'altair_saver'} the user-provided engine string format : string {'png', 'svg', 'pdf', 'vega'} the format of the mimebundle to be returned Nr-rrzIThe 'vl-convert' conversion engine requires the vl-convert-python packager zmThe 'vl-convert' conversion engine does not support the {fmt!r} format. Use the 'altair_saver' engine insteadr%r'zFThe 'altair_saver' conversion engine requires the altair_saver packagezqSaving charts in {fmt!r} format requires the altair_saver package: see http://github.com/altair-viz/altair_saver/zSaving charts in {fmt!r} format requires the vl-convert-python or altair_saver package: see http://github.com/altair-viz/altair_saver/z?Invalid conversion engine {engine!r}. Expected one of {valid!r})z vl-convertr.)r Zvalid)r ImportErrorr.lowerreplacerr)r rr1r.r0rrrr)sj    "r)cCs&|d}td||d|dS)zOread the (width, height) from a PNG header Taken from IPython.display sIHDRz>ii )indexstructunpack)dataZihdrrrrr-s r-)NNNNN) r rZ _importersrr>rrr)r-rrrrs  OO B