o eF@sddlmZmZmZmZmZmZmZddlm 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*e  d(dee e fd ee&d eee(d ee#fd d Z+e  d(dee e e efd ee&d eee(d ee#fdd Z+d(dd Z+ d)dee e e e e efde&d eee(d ee(fddZ,de-de'd ee-fddZ.de-de'd ee(fddZ/de-de(de'd ee'fddZ0d*de-de'd e)fdd Z1d!ee(e'fd"e)d ee(e'ffd#d$Z2 d*de-d%ee(d"e)de'd ee(ee(e'fff d&d'Z3dS)+)ListOptionalTupleDictIterableoverloadUnion)Chart FacetChart LayerChart HConcatChart VConcatChart ConcatChartTopLevelUnitSpecFacetedUnitSpecUnitSpecUnitSpecWithFrameNonNormalizedSpecTopLevelLayerSpec LayerSpecTopLevelConcatSpecConcatSpecGenericSpecTopLevelHConcatSpecHConcatSpecGenericSpecTopLevelVConcatSpecVConcatSpecGenericSpecTopLevelFacetSpec FacetSpecdata_transformers)get_inline_tablesimport_vegafusion) DataFrameLike) Undefined.Nchart row_limitexcludereturncCdSNr#r$r%r)r)^/root/Voice_Assist/myprojectenv/lib/python3.10/site-packages/altair/utils/_transformed_data.pytransformed_data6r,cCr'r(r)r*r)r)r+r,?r-c Cst}t|tr|jtkr|}|jdd}t|d|d}t d|j dddid }t |}Wd n1s;wYt |}t |||}g} |D]} | |vr\| || qNtd |jj|| ||d \} } t|ttfr{| swd S| dS| 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_specr1 facet_mappingZdataset_mappingZ dataset_namesZ chart_namedatasetswarningsr)r)r+r,Hs8       icCs|durt|nt}t|ttst|ttr0|j|vr.|jdtfvr*t|_|jgSgSt|ttr;|j }n*t|tt rF|j }nt|tt rQ|j }nt|ttr\|j}n tdt|g}|D]}t||t||dD]}||qvqi|S)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: )rAr%)setr2_chart_class_mappingr r namer"Z _get_namer layerr Zhconcatr Zvconcatrconcatr;typer5lenr:)r#rAr%Z subchartsr<ZsubchartrDr)r)r+r5s8   r5r=scopecCsd|}|D]+}d}d}|dgD]}|ddkr%||kr!|}n|d7}q|dur-dS|}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'}]} rNmarksrGgroup)get)r=rIrKZ scope_value group_indexZ child_groupr3r)r)r+get_group_mark_for_scopes"rOcCs`t||pi}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)) [] datarDfromfacetN)rOrMr:)r=rIrKr?ZdatasetZ facet_datasetr)r)r+get_datasets_for_scopes7 rS data_name usage_scopecCsDttt|dD]}|d|}t||}||vr|Sq dS)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' rLN)reversedrangerHrS)r=rTrUrArIr?r)r)r+'get_definition_scope_for_data_referenceTs:  rXr)rKc Csi}d}t||p i}|dgD]R}|dddkrc||f}|didd}|durT|dd}|d d} |durT| durTt|| |} | durT| | 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', ())} rrJrGNrKrQrRrDrPrIrL)rOrMrXr8update) rKrIr>rN mark_groupr3Z group_scoperRZ facet_nameZ facet_dataZdefinition_scopeZ child_mappingr)r)r+r8s.%     r8scoped_datasetr>cCs||vr ||}||vs|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))r\r>r)r)r+get_from_facet_mappingsr]vl_chart_namescCs8i}d}t||p i}|dgD]}|D](}|dd|dkr=|diddd d} | |f} t| |||<nq|dd} |d dd krkt|||||fd } | D] \} }|| |q[|d 7}q|D]+}| |r| dr|did d} t|| |}|durt| |f|||<nqmq|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 rrJrD_cellrQrRNrPrGrKrYrL_marks) rOrMr]r9items setdefault startswithendswithrX)rKr^r>rIr?rNr[r3Z vl_chart_namerTZscoped_data_namerDZgroup_data_nameskvZ scoped_datar)r)r+r9sD  r9)NN)rN)r))4typingrrrrrrrZaltairr r r r r rrrrrrrrrrrrrrrrrZaltair.utils._vegafusion_datarr Zaltair.utils.corer!Zaltair.utils.schemapir"intZScopestrZ FacetMappingrCr,r5dictrOrSrXr8r]r9r)r)r)r+s$`             W  ?5E BB   !