@antv/g2plot/src/adaptor/geometries/point.ts

An interactive and responsive charting library

import { Params } from '../../core/adaptor';
import { deepAssign } from '../../utils';
import { getTooltipMapping } from '../../utils/tooltip';
import { geometry, GeometryOptions, MappingOptions } from './base';

export interface PointGeometryOptions extends GeometryOptions {
  /** x 轴字段 */
  readonly xField?: string;
  /** y 轴字段 */
  readonly yField?: string;
  /** 分组字段 */
  readonly seriesField?: string;
  /** size 映射字段 */
  readonly sizeField?: string;
  /** shape 的映射字段 */
  readonly shapeField?: string;
  /** point 图形映射规则 */
  readonly point?: MappingOptions;
}

/**
 * point 辅助点的配置处理
 * @param params
 */
export function point<O extends PointGeometryOptions>(params: Params<O>): Params<O> {
  const { options } = params;
  const { point, xField, yField, seriesField, sizeField, shapeField, tooltip, useDeferredLabel } = options;

  const { fields, formatter } = getTooltipMapping(tooltip, [xField, yField, seriesField, sizeField, shapeField]);

  return point
    ? geometry(
        deepAssign({}, params, {
          options: {
            type: 'point',
            colorField: seriesField,
            shapeField: shapeField,
            tooltipFields: fields,
            mapping: {
              tooltip: formatter,
              ...point,
            },
            args: { useDeferredLabel },
          },
        })
      )
    : params;
}