12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
import {AggregateOp} from 'vega';import {isString, toSet} from 'vega-util';import {contains, Flag, keys} from './util';
const AGGREGATE_OP_INDEX: Flag<AggregateOp> = {  argmax: 1,  argmin: 1,  average: 1,  count: 1,  distinct: 1,  product: 1,  max: 1,  mean: 1,  median: 1,  min: 1,  missing: 1,  q1: 1,  q3: 1,  ci0: 1,  ci1: 1,  stderr: 1,  stdev: 1,  stdevp: 1,  sum: 1,  valid: 1,  values: 1,  variance: 1,  variancep: 1};
export const MULTIDOMAIN_SORT_OP_INDEX = {  count: 1,  min: 1,  max: 1};
export interface ArgminDef {  argmin: string;}
export interface ArgmaxDef {  argmax: string;}
export type NonArgAggregateOp = Exclude<AggregateOp, 'argmin' | 'argmax'>;
export type Aggregate = NonArgAggregateOp | ArgmaxDef | ArgminDef;
export function isArgminDef(a: Aggregate | string): a is ArgminDef {  return !!a && !!a['argmin'];}
export function isArgmaxDef(a: Aggregate | string): a is ArgmaxDef {  return !!a && !!a['argmax'];}
export const AGGREGATE_OPS = keys(AGGREGATE_OP_INDEX);
export function isAggregateOp(a: string | ArgminDef | ArgmaxDef): a is AggregateOp {  return isString(a) && !!AGGREGATE_OP_INDEX[a];}
export const COUNTING_OPS: NonArgAggregateOp[] = ['count', 'valid', 'missing', 'distinct'];
export function isCountingAggregateOp(aggregate?: string | Aggregate): boolean {  return isString(aggregate) && contains(COUNTING_OPS, aggregate);}
export function isMinMaxOp(aggregate?: Aggregate | string): boolean {  return isString(aggregate) && contains(['min', 'max'], aggregate);}
/** Additive-based aggregation operations. These can be applied to stack. */export const SUM_OPS: NonArgAggregateOp[] = ['count', 'sum', 'distinct', 'valid', 'missing'];
/** * Aggregation operators that always produce values within the range [domainMin, domainMax]. */export const SHARED_DOMAIN_OPS: AggregateOp[] = ['mean', 'average', 'median', 'q1', 'q3', 'min', 'max'];
export const SHARED_DOMAIN_OP_INDEX = toSet(SHARED_DOMAIN_OPS);