a  d I@shddlmZmZmZmZmZmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlm Z m!Z!ddl"m#Z#ddl$m%Z%ee&dfZ'eee(e'fee(e'ffZ)e e eeeeefe e eefeeeefe e eefe e eefe e eefiZ*ed"ee e fee&eee(ee#dd d Z+ed#ee e e efee&eee(ee#dd d Z+d$d d Z+d%ee e e e e efe&eee(ee(d ddZ,e-e'ee-dddZ.e-e'ee(dddZ/e-e(e'ee'dddZ0d&e-e'e)dddZ1ee(e'fe)ee(e'fdddZ2d'e-ee(e)e'ee(ee(e'ffdd d!Z3dS)()ListOptionalTupleDictIterableoverloadUnion)Chart FacetChart LayerChart HConcatChart VConcatChart ConcatChartTopLevelUnitSpecFacetedUnitSpecUnitSpecUnitSpecWithFrameNonNormalizedSpecTopLevelLayerSpec LayerSpecTopLevelConcatSpecConcatSpecGenericSpecTopLevelHConcatSpecHConcatSpecGenericSpecTopLevelVConcatSpecVConcatSpecGenericSpecTopLevelFacetSpec FacetSpecdata_transformers)get_inline_tablesimport_vegafusion)_DataFrameLike) Undefined.N)chart row_limitexcludereturncCsdSNr#r$r%r(r(g/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/altair/utils/_transformed_data.pytransformed_data6sr+cCsdSr'r(r)r(r(r*r+?sc Cst}t|tr"|jtkr"|}|jdd}t|d|d}t d*|j dddid }t |}Wd n1sv0Yt |}t |||}g} |D]$} | |vr| || qtd q|jj|| ||d \} } t|ttfr| sd S| dSn| Sd S) aEvaluate a Chart's transforms Evaluate the data transforms associated with a Chart and return the transformed data as one or more DataFrames Parameters ---------- chart : Chart, FacetChart, LayerChart, HConcatChart, VConcatChart, or ConcatChart Altair chart to evaluate transforms on row_limit : int (optional) Maximum number of rows to return for each DataFrame. None (default) for unlimited exclude : iterable of str Set of the names of charts to exclude Returns ------- DataFrame or list of DataFrames or None If input chart is a Chart or Facet Chart, returns a DataFrame of the transformed data. Otherwise, returns a list of DataFrames of the transformed data T)deepr)r%Z vegafusionZvegaZ pre_transformF)formatcontextNzFailed to locate all datasets)r$inline_datasets)r isinstancer markr"Z mark_pointcopy name_viewsrenableto_dictrget_facet_mappingget_datasets_for_view_namesappend ValueErrorZruntimeZpre_transform_datasetsr ) r#r$r%Zvf chart_names vega_specr/ facet_mappingZdataset_mappingZ dataset_namesZ chart_namedatasetswarningsr(r(r*r+Hs6    &    )r#ir%r&cCs|durt|nt}t|tts2t|ttrb|j|vr\|jdtfvrTt|_|jgSgSnt|ttrx|j }nTt|tt r|j }n>t|tt r|j }n(t|ttr|j}ntdt|g}|D]*}t||t||dD]}||qq|SdS)aGName unnamed chart views Name unnamed charts views so that we can look them up later in the compiled Vega spec. Note: This function mutates the input chart by applying names to unnamed views. Parameters ---------- chart : Chart, FacetChart, LayerChart, HConcatChart, VConcatChart, or ConcatChart Altair chart to apply names to i : int (default 0) Starting chart index exclude : iterable of str Names of charts to exclude Returns ------- list of str List of the names of the charts and subcharts Nztransformed_data accepts an instance of Chart, FacetChart, LayerChart, HConcatChart, VConcatChart, or ConcatChart Received value of type: )r?r%)setr0_chart_class_mappingr r namer"Z _get_namer Zlayerr Zhconcatr Zvconcatrconcatr9typer3lenr8)r#r?r%Z subchartsr:ZsubchartrBr(r(r*r3s6  r3)r;scoper&cCsd|}|D]V}d}d}|dgD]*}|ddkr ||krB|}qL|d7}q |durZdS|}q|S)aGet the group mark at a particular scope Parameters ---------- vega_spec : dict Top-level Vega specification dictionary scope : tuple of int Scope tuple. If empty, the original Vega specification is returned. Otherwise, the nested group mark at the scope specified is returned. Returns ------- dict or None Top-level Vega spec (if scope is empty) or group mark (if scope is non-empty) or None (if group mark at scope does not exist) Examples -------- >>> spec = { ... "marks": [ ... { ... "type": "group", ... "marks": [{"type": "symbol"}] ... }, ... { ... "type": "group", ... "marks": [{"type": "rect"}]} ... ] ... } >>> get_group_mark_for_scope(spec, (1,)) {'type': 'group', 'marks': [{'type': 'rect'}]} rNmarksrDgroup)get)r;rFrHZ scope_value group_indexZ child_groupr1r(r(r*get_group_mark_for_scopes" rLcCs`t||p i}g}|dgD]}||dq|dididd}|r\|||S)aGet the names of the datasets that are defined at a given scope Parameters ---------- vega_spec : dict Top-leve Vega specification scope : tuple of int Scope tuple. If empty, the names of top-level datasets are returned Otherwise, the names of the datasets defined in the nested group mark at the specified scope are returned. Returns ------- list of str List of the names of the datasets defined at the specified scope Examples -------- >>> spec = { ... "data": [ ... {"name": "data1"} ... ], ... "marks": [ ... { ... "type": "group", ... "data": [ ... {"name": "data2"} ... ], ... "marks": [{"type": "symbol"}] ... }, ... { ... "type": "group", ... "data": [ ... {"name": "data3"}, ... {"name": "data4"}, ... ], ... "marks": [{"type": "rect"}] ... } ... ] ... } >>> get_datasets_for_scope(spec, ()) ['data1'] >>> get_datasets_for_scope(spec, (0,)) ['data2'] >>> get_datasets_for_scope(spec, (1,)) ['data3', 'data4'] Returns empty when no group mark exists at scope >>> get_datasets_for_scope(spec, (1, 3)) [] datarBfromfacetN)rLrJr8)r;rFrHr=ZdatasetZ facet_datasetr(r(r*get_datasets_for_scopes7 rP)r; data_name usage_scoper&cCsDttt|dD]*}|d|}t||}||vr|SqdS)aGReturn the scope that a dataset is defined at, for a given usage scope Parameters ---------- vega_spec: dict Top-level Vega specification data_name: str The name of a dataset reference usage_scope: tuple of int The scope that the dataset is referenced in Returns ------- tuple of int The scope where the referenced dataset is defined, or None if no such dataset is found Examples -------- >>> spec = { ... "data": [ ... {"name": "data1"} ... ], ... "marks": [ ... { ... "type": "group", ... "data": [ ... {"name": "data2"} ... ], ... "marks": [{ ... "type": "symbol", ... "encode": { ... "update": { ... "x": {"field": "x", "data": "data1"}, ... "y": {"field": "y", "data": "data2"}, ... } ... } ... }] ... } ... ] ... } data1 is referenced at scope [0] and defined at scope [] >>> get_definition_scope_for_data_reference(spec, "data1", (0,)) () data2 is referenced at scope [0] and defined at scope [0] >>> get_definition_scope_for_data_reference(spec, "data2", (0,)) (0,) If data2 is not visible at scope [] (the top level), because it's defined in scope [0] >>> repr(get_definition_scope_for_data_reference(spec, "data2", ())) 'None' rIN)reversedrangerErP)r;rQrRr?rFr=r(r(r*'get_definition_scope_for_data_referenceTs :   rUr()rHrFr&c Csi}d}t||pi}|dgD]}|dddkr"||f}|didd}|dur|dd}|d d} |dur| durt|| |} | dur| | f|||f<t||d } || |d 7}q"|S) aCreate mapping from facet definitions to source datasets Parameters ---------- group : dict Top-level Vega spec or nested group mark scope : tuple of int Scope of the group dictionary within a top-level Vega spec Returns ------- dict Dictionary from (facet_name, facet_scope) to (dataset_name, dataset_scope) Examples -------- >>> spec = { ... "data": [ ... {"name": "data1"} ... ], ... "marks": [ ... { ... "type": "group", ... "from": { ... "facet": { ... "name": "facet1", ... "data": "data1", ... "groupby": ["colA"] ... } ... } ... } ... ] ... } >>> get_facet_mapping(spec) {('facet1', (0,)): ('data1', ())} rrGrDNrHrNrOrBrMrFrI)rLrJrUr6update) rHrFr<rK mark_groupr1Z group_scoperOZ facet_nameZ facet_dataZdefinition_scopeZ child_mappingr(r(r*r6s,%       r6)scoped_datasetr<r&cCs||vr||}q|S)aApply facet mapping to a scoped dataset Parameters ---------- scoped_dataset : (str, tuple of int) A dataset name and scope tuple facet_mapping : dict from (str, tuple of int) to (str, tuple of int) The facet mapping produced by get_facet_mapping Returns ------- (str, tuple of int) Dataset name and scope tuple that has been mapped as many times as possible Examples -------- Facet mapping as produced by get_facet_mapping >>> facet_mapping = {("facet1", (0,)): ("data1", ()), ("facet2", (0, 1)): ("facet1", (0,))} >>> get_from_facet_mapping(("facet2", (0, 1)), facet_mapping) ('data1', ()) r()rYr<r(r(r*get_from_facet_mappings rZ)rHvl_chart_namesr<rFr&cCs:i}d}t||pi}|dgD]}|D]P}|dd|dkr,|diddd d} | |f} t| |||<q~q,|dd} |d dd krt|||||fd } | D]\} }|| |q|d 7}q"|D]V}| |r| dr|did d} t|| |}|durt| |f|||<q"qq"|S)aBGet the Vega datasets that correspond to the provided Altair view names Parameters ---------- group : dict Top-level Vega spec or nested group mark vl_chart_names : list of str List of the Vega-Lite facet_mapping : dict from (str, tuple of int) to (str, tuple of int) The facet mapping produced by get_facet_mapping scope : tuple of int Scope of the group dictionary within a top-level Vega spec Returns ------- dict from str to (str, tuple of int) Dict from Altair view names to scoped datasets rrGrBZ_cellrNrONrMrDrHrVrI_marks) rLrJrZr7items setdefault startswithendswithrU)rHr[r<rFr=rKrXr1Z vl_chart_namerQZscoped_data_namerBZgroup_data_nameskvZ scoped_datar(r(r*r7s@  r7)NN)NN)NN)rN)r()r()4typingrrrrrrrZaltairr r r r r rrrrrrrrrrrrrrrrrZaltair.utils._vegafusion_datarr Zaltair.utils.corer!Zaltair.utils.schemapir"intZScopestrZ FacetMappingrAr+r3dictrLrPrUr6rZr7r(r(r(r*s$`            W  ?5F BC  !