o h2@sFUddlmZddlZddlmZmZmZmZmZddl m Z m Z ddl m Z er?ddlZejdkr9ddlmZnddlmZed Zd ed <e d9d:ddZe d9d;d!dZe d9dd?d(dZ  d@dAd0d1ZdBd3d4Zd5d6ZdCd7d8ZdS)D) annotationsN) TYPE_CHECKINGAnyLiteralcastoverload)import_vl_convertvl_version_for_vl_convert) spec_to_html) ) TypeAlias)htmljsonpngsvgpdfvega vega-literMimeBundleFormat.specdict[str, Any]formatLiteral['json', 'vega-lite']modeLiteral['vega-lite'] | None vega_version str | Nonevegaembed_versionvegalite_version embed_optionsdict[str, Any] | NoneengineLiteral['vl-convert'] | Nonereturndict[str, dict[str, Any]]c KdSN rrrrrr r!r#kwargsr)r)K/var/www/vscode/kcb/lib/python3.10/site-packages/altair/utils/mimebundle.pyspec_to_mimebundle r-Literal['html']dict[str, str]c Kr'r(r)r*r)r)r,r-"r.Literal['pdf', 'svg', 'vega']c Kr'r(r)r*r)r)r,r-.r.Literal['png']%tuple[dict[str, Any], dict[str, Any]]c Kr'r(r)r*r)r)r,r-:r.6dict[str, Any] | tuple[dict[str, Any], dict[str, Any]]cKs ddlm} ddlm} m} |dkrd} t| |} | r#| |}d} |dur/| jdi}n|}t|}|d vrTt |t t d || f||d d|d dd |S|dkrjt |f| ||||d|}d|iS|dkr|durxd} t| d|dd|iS|dkrd|iSd} t| )aE Convert 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' embed_options : dict (optional) The vegaEmbed options dictionary. Defaults to the embed options set with alt.renderers.set_embed_options(). (See https://github.com/vega/vega-embed for details) engine: string {'vl-convert'} 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 vl-convert package r) renderers)compile_with_vegafusionusing_vegafusionrzmode must be 'vega-lite'rNr!>rrrr)rrrr formatLocaletimeFormatLocale)r# format_localetime_format_localer)rrrr r!z text/htmlzMust specify vegalite_versionzapplication/vnd.vegalite.vz+jsonrzapplication/jsonzPformat must be one of ['html', 'json', 'png', 'svg', 'pdf', 'vega', 'vega-lite']) altairr5altair.utils.displayr6r7 ValueErroroptionsgetpreprocess_embed_options_spec_to_mimebundle_with_enginerrr )rrrrrr r!r#r+r5r6r7msg internal_modefinal_embed_optionsrr)r)r,r-Fs` 2     dict$Literal['png', 'svg', 'pdf', 'vega']Literal['vega-lite', 'vega']r:str | dict | Noner;rcKs|dd}t||}|dkrt}t} |dkr.|dkr#|} d| iS|j|| d} d| iS|dkrO|dkrB|j|||d} d | iS|j|| ||d } d | iS|d kr|d d } d} |d| }|dkrp|j|| |||d}n |j || | |||d}|| }t |\}}d|id||||difS|dkr|d d } |dkr|j || ||d}d|iS|j || | ||d}d|iSd|}t |d|}t |)a Helper 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'} the conversion engine to use format_locale : str or dict d3-format locale name or dictionary. Defaults to "en-US" for United States English. See https://github.com/d3/d3-format/tree/main/locale for available names and example definitions. time_format_locale : str or dict d3-time-format locale name or dictionary. Defaults to "en-US" for United States English. See https://github.com/d3/d3-time-format/tree/main/locale for available names and example definitions. **kwargs : Additional arguments will be passed to the conversion function r#N vlconvertr) vl_versionzapplication/vnd.vega.v5+jsonr)r:r;)rKr:r;z image/svg+xmlr scale_factorrHppi)scalerNr:r;)rKrOrNr:r;z image/png)widthheightr)rOr:r;)rKrOr:r;zapplication/pdfzUnexpected format zUnexpected normalized_engine )pop_validate_normalize_enginer r vegalite_to_vega vega_to_svgvegalite_to_svgr@ vega_to_pngvegalite_to_png_pngxy vega_to_pdfvegalite_to_pdfr>)rrrr:r;r+r#normalized_enginevlcrKvgrrO default_ppirNrfactorwhrrCr)r)r,rBs "        rBstrcCszt}Wn tyd}Ynw|durdn |dddd}|dkr3|dur1d}t||S|durI|dur?d}|Sd|d}t|d |d }t|) z Helper to validate and normalize the user-provided engine. engine : {None, 'vl-convert'} the user-provided engine string format : string {'png', 'svg', 'pdf', 'vega'} the format of the mimebundle to be returned N-_rJzIThe 'vl-convert' conversion engine requires the vl-convert-python packagezSaving charts in z format requires the vl-convert-python package: see https://altair-viz.github.io/user_guide/saving_charts.html#png-svg-and-pdf-formatzInvalid conversion engine z. Expected vl-convert)r ImportErrorlowerreplacer>)r#rr]r\rCr)r)r,rS&s*  "   rScCs&|d}td||d|dS)zU read the (width, height) from a PNG header. Taken from IPython.display sIHDRz>ii )indexstructunpack)dataihdrr)r)r,rYQs rYcCsd|pi}|dd}t|trt}|||d<|dd}t|tr0t}|||d<|S)a Preprocess embed options to a form compatible with Vega Embed. Parameters ---------- embed_options : dict The embed options dictionary to preprocess. Returns ------- embed_opts : dict The preprocessed embed options dictionary. r8Nr9)copyr@ isinstancercr get_format_localeget_time_format_locale)r!r:r]r;r)r)r,rA\s     rA)......)rrrrrrrrrrr rr!r"r#r$r%r&)rrrr/rrrrrrr rr!r"r#r$r%r0)rrrr1rrrrrrr rr!r"r#r$r%r)rrrr2rrrrrrr rr!r"r#r$r%r3)NNNNNN)rrrrrrrrrrr rr!r"r#r$r%r4)NN) rrFrrGrrHr:rIr;rIr%r)r#r$rrGr%rc)r!rFr%rF) __future__rrmtypingrrrrr _importersr r rr sys version_infortyping_extensionsr__annotations__r-rBrSrYrAr)r)r)r,st        n v+