@antv/gpt-vis 0.6.0 → 1.0.0-beta.1

package/src/vis/pie/index.ts

@@ -0,0 +1,163 @@
+import { Chart } from '@antv/g2';
+import type { VisualizationOptions } from '../../types';
+import { getBackgroundColor, getThemeColors, getThemeObject } from '../../util/theme';
+
+/**
+ * PieDataItem is the type for each data item in the pie chart.
+ */
+type PieDataItem = {
+  category: string;
+  value: number;
+};
+
+/**
+ * PieConfig defines the configuration for rendering the pie chart.
+ */
+export interface PieConfig {
+  type?: 'pie';
+  data: PieDataItem[];
+  innerRadius?: number;
+  theme?: 'default' | 'academy' | 'dark';
+  title?: string;
+  style?: {
+    backgroundColor?: string;
+    palette?: string[];
+  };
+}
+
+/**
+ * PieInstance represents a pie chart instance with render and destroy methods.
+ */
+export interface PieInstance {
+  render: (config: PieConfig) => void;
+  destroy: () => void;
+}
+
+/**
+ * Pie chart component using G2 5.0.
+ *
+ * @example
+ * ```ts
+ * const pie = Pie({
+ *   container: '#container',
+ *   width: 600,
+ *   height: 400,
+ * });
+ *
+ * pie.render({
+ *   type: 'pie',
+ *   data: [
+ *     { category: '分类一', value: 27 },
+ *     { category: '分类二', value: 25 },
+ *   ],
+ *   innerRadius: 0.6,
+ *   theme: 'academy'
+ * });
+ *
+ * pie.destroy();
+ * ```
+ */
+export const Pie = (options: VisualizationOptions): PieInstance => {
+  const { container, width, height, theme: chartTheme = 'default' } = options;
+  let chart: Chart | null = null;
+
+  // Precision multiplier for rounding percentages to 4 decimal places
+  const PERCENTAGE_PRECISION_MULTIPLIER = 10000;
+
+  /**
+   * Render the pie chart with the given configuration.
+   */
+  const render = (config: PieConfig): void => {
+    const { data = [], innerRadius = 0, theme = chartTheme, title, style = {} } = config;
+
+    // Clean up previous chart if exists
+    if (chart) {
+      chart.destroy();
+    }
+
+    // Get colors from style.palette or theme defaults
+    const colors = style.palette || getThemeColors(theme);
+    const backgroundColor = style.backgroundColor || getBackgroundColor(theme);
+
+    // Calculate sum for percentage labels
+    const sumValue = data.reduce((sum, item) => sum + item.value, 0);
+
+    // Create chart
+    chart = new Chart({
+      container,
+      width,
+      height,
+      autoFit: true,
+    });
+
+    // Configure chart options
+    // Note: Using 'any' type due to G2's complex type system with transformations
+    // This is consistent with how G2 5.0 is used elsewhere in the codebase (e.g., Radar component)
+    const chartOptions: any = {
+      type: 'interval',
+      data,
+      title: title ?? '',
+      encode: {
+        y: 'value',
+        color: 'category',
+      },
+      transform: [{ type: 'stackY' }],
+      coordinate: { type: 'theta', innerRadius: Math.max(0, Math.min(1, innerRadius)) },
+      scale: {
+        color: { range: colors },
+      },
+      legend: {
+        color: { position: 'top' },
+      },
+      labels: [
+        {
+          text: (d: any) => {
+            const percentage =
+              Math.round((d.value / sumValue) * 100 * PERCENTAGE_PRECISION_MULTIPLIER) /
+              PERCENTAGE_PRECISION_MULTIPLIER;
+            return `${d.category}: ${percentage}%`;
+          },
+          position: 'outside',
+          transform: [{ type: 'overlapHide' }],
+        },
+      ],
+      tooltip: {
+        items: [
+          (d: any) => ({
+            name: d.category,
+            value: d.value,
+          }),
+        ],
+      },
+      interaction: {
+        elementSelect: { single: true },
+      },
+      style: {
+        fillOpacity: 0.8,
+      },
+      viewStyle: {
+        viewFill: backgroundColor,
+      },
+      theme: getThemeObject(theme),
+      animate: { enter: { type: 'waveIn' } },
+    };
+
+    chart.options(chartOptions);
+    chart.render();
+  };
+
+  /**
+   * Destroy the chart instance and clean up resources.
+   */
+  const destroy = (): void => {
+    if (chart) {
+      chart.destroy();
+      chart = null;
+    }
+  };
+
+  return {
+    render,
+    destroy,
+  };
+};

package/src/vis/radar/README.md

@@ -0,0 +1,184 @@
+# Radar
+
+A radar chart component for visualizing multi-dimensional data, built with G2 5.0.
+
+## Usage
+
+```ts
+import { Radar } from '@antv/gpt-vis/ai';
+
+const radar = Radar({
+  container: '#container',
+  width: 600,
+  height: 400,
+});
+
+radar.render({
+  data: [
+    { name: '沟通能力', value: 2 },
+    { name: '协作能力', value: 3 },
+    { name: '领导能力', value: 2 },
+    { name: '学习能力', value: 5 },
+    { name: '创新能力', value: 6 },
+    { name: '技术能力', value: 9 },
+  ],
+});
+
+radar.destroy();
+```
+
+## Configuration
+
+### Constructor Options (RadarOptions)
+
+| Property  | Type                  | Default | Description                   |
+| --------- | --------------------- | ------- | ----------------------------- |
+| container | string \| HTMLElement | -       | Container element or selector |
+| width     | number                | 640     | Chart width in pixels         |
+| height    | number                | 480     | Chart height in pixels        |
+
+### Render Config (RadarConfig)
+
+| Property | Type   | Default   | Description               |
+| -------- | ------ | --------- | ------------------------- |
+| type     | string | 'radar'   | Chart type                |
+| data     | Array  | -         | Chart data array          |
+| title    | string | -         | Chart title               |
+| theme    | string | 'default' | Color theme               |
+| style    | object | -         | Chart style configuration |
+
+### Data Structure
+
+```ts
+type RadarDataItem = {
+  name: string; // Category name (axis label)
+  value: number; // Value for this category
+  group?: string; // Optional: group for multiple series
+};
+```
+
+### Style Options
+
+```ts
+style?: {
+  backgroundColor?: string; // Background color
+  palette?: string[]; // Color palette for groups
+  lineWidth?: number; // Line width (default: 2)
+}
+```
+
+## Examples
+
+### Basic Radar Chart
+
+```ts
+radar.render({
+  data: [
+    { name: '沟通能力', value: 2 },
+    { name: '协作能力', value: 3 },
+    { name: '领导能力', value: 2 },
+    { name: '学习能力', value: 5 },
+    { name: '创新能力', value: 6 },
+    { name: '技术能力', value: 9 },
+  ],
+});
+```
+
+### With Title
+
+```ts
+radar.render({
+  data: [
+    { name: 'Vitamin C', value: 7 },
+    { name: 'Fiber', value: 6 },
+    { name: 'Sugar', value: 5 },
+    { name: 'Protein', value: 4 },
+    { name: 'Iron', value: 3 },
+    { name: 'Calcium', value: 2 },
+  ],
+  title: 'Nutrition Analysis',
+});
+```
+
+### With Multiple Groups
+
+```ts
+radar.render({
+  data: [
+    { name: '语文', value: 95, group: '一班' },
+    { name: '数学', value: 96, group: '一班' },
+    { name: '外语', value: 85, group: '一班' },
+    { name: '物理', value: 63, group: '一班' },
+    { name: '化学', value: 91, group: '一班' },
+    { name: '语文', value: 75, group: '二班' },
+    { name: '数学', value: 93, group: '二班' },
+    { name: '外语', value: 66, group: '二班' },
+    { name: '物理', value: 85, group: '二班' },
+    { name: '化学', value: 88, group: '二班' },
+  ],
+});
+```
+
+### With Academy Theme
+
+```ts
+radar.render({
+  data: [
+    { name: 'Attack', value: 85, group: 'Hero A' },
+    { name: 'Defense', value: 70, group: 'Hero A' },
+    { name: 'Speed', value: 90, group: 'Hero A' },
+    { name: 'HP', value: 75, group: 'Hero A' },
+    { name: 'MP', value: 60, group: 'Hero A' },
+    { name: 'Attack', value: 65, group: 'Hero B' },
+    { name: 'Defense', value: 90, group: 'Hero B' },
+    { name: 'Speed', value: 70, group: 'Hero B' },
+    { name: 'HP', value: 85, group: 'Hero B' },
+    { name: 'MP', value: 80, group: 'Hero B' },
+  ],
+  theme: 'academy',
+});
+```
+
+### With Custom Styles
+
+```ts
+radar.render({
+  data: [
+    { name: 'Category A', value: 8, group: 'Product X' },
+    { name: 'Category B', value: 7, group: 'Product X' },
+    { name: 'Category C', value: 6, group: 'Product X' },
+    { name: 'Category D', value: 5, group: 'Product X' },
+    { name: 'Category E', value: 9, group: 'Product X' },
+    { name: 'Category A', value: 6, group: 'Product Y' },
+    { name: 'Category B', value: 9, group: 'Product Y' },
+    { name: 'Category C', value: 8, group: 'Product Y' },
+    { name: 'Category D', value: 7, group: 'Product Y' },
+    { name: 'Category E', value: 5, group: 'Product Y' },
+  ],
+  style: {
+    palette: ['#FF6B6B', '#4ECDC4'],
+    backgroundColor: '#F8F9FA',
+    lineWidth: 3,
+  },
+});
+```
+
+### Dark Theme
+
+```ts
+radar.render({
+  data: [
+    { name: 'Skill 1', value: 90 },
+    { name: 'Skill 2', value: 75 },
+    { name: 'Skill 3', value: 85 },
+    { name: 'Skill 4', value: 70 },
+    { name: 'Skill 5', value: 95 },
+  ],
+  theme: 'dark',
+});
+```
+
+## Methods
+
+- `render(config: RadarConfig): void` - Render or update the chart
+- `destroy(): void` - Destroy the chart instance and clean up resources

package/src/vis/radar/index.ts

@@ -0,0 +1,214 @@
+import { Chart } from '@antv/g2';
+import type { VisualizationOptions } from '../../types';
+import { getBackgroundColor, getThemeColors, getThemeObject } from '../../util/theme';
+
+/**
+ * RadarDataItem is the type for each data item in the radar chart.
+ */
+export type RadarDataItem = {
+  name: string;
+  value: number;
+  group?: string;
+};
+
+/**
+ * RadarConfig defines the configuration for rendering the radar chart.
+ */
+export interface RadarConfig {
+  type?: 'radar';
+  data: RadarDataItem[];
+  title?: string;
+  theme?: 'default' | 'academy' | 'dark';
+  style?: {
+    backgroundColor?: string;
+    palette?: string[];
+    lineWidth?: number;
+  };
+}
+
+/**
+ * RadarInstance represents a radar chart instance with render and destroy methods.
+ */
+export interface RadarInstance {
+  render: (config: RadarConfig) => void;
+  destroy: () => void;
+}
+
+/**
+ * Transform radar data to parallel coordinates format.
+ * Groups data by group field and converts to position-based encoding.
+ */
+function transformRadarToParallel(data: RadarDataItem[]) {
+  if (!data || data.length === 0) {
+    return [];
+  }
+
+  // Group data by group field
+  const groups: { [key: string]: RadarDataItem[] } = {};
+  data.forEach((item) => {
+    const groupKey = item.group || '';
+    if (!groups[groupKey]) {
+      groups[groupKey] = [];
+    }
+    groups[groupKey].push(item);
+  });
+
+  // Transform to parallel format
+  return Object.entries(groups).map(([group, values]) => {
+    const parallelValues: any = {};
+    values.forEach(({ name, value }) => {
+      parallelValues[name] = value;
+    });
+    return {
+      ...parallelValues,
+      group,
+    };
+  });
+}
+
+/**
+ * Radar chart component using G2 5.0.
+ *
+ * @example
+ * ```ts
+ * const radar = Radar({
+ *   container: '#container',
+ *   width: 600,
+ *   height: 400,
+ * });
+ *
+ * radar.render({
+ *   type: 'radar',
+ *   data: [
+ *     { name: '沟通能力', value: 2 },
+ *     { name: '协作能力', value: 3 },
+ *     { name: '领导能力', value: 2 },
+ *     { name: '学习能力', value: 5 },
+ *     { name: '创新能力', value: 6 },
+ *     { name: '技术能力', value: 9 },
+ *   ],
+ * });
+ *
+ * radar.destroy();
+ * ```
+ */
+export const Radar = (options: VisualizationOptions): RadarInstance => {
+  const { container, width, height, theme: chartTheme = 'default' } = options;
+  let chart: Chart | null = null;
+
+  /**
+   * Render the radar chart with the given configuration.
+   */
+  const render = (config: RadarConfig): void => {
+    const { data = [], theme = chartTheme, title, style = {} } = config;
+
+    // Clean up previous chart if exists
+    if (chart) {
+      chart.destroy();
+    }
+
+    const { lineWidth = 2 } = style;
+    const colors = style.palette || getThemeColors(theme);
+    const backgroundColor = style.backgroundColor || getBackgroundColor(theme);
+
+    // Transform data to parallel format
+    const parallelData = transformRadarToParallel(data);
+    const position = Object.keys(parallelData[0] || {}).filter((key) => key !== 'group');
+
+    // Create chart
+    chart = new Chart({
+      container,
+      width,
+      height,
+      autoFit: true,
+    });
+
+    // Configure chart options
+    // Note: Using 'any' type due to G2's complex type system with transformations
+    // This is consistent with how G2 5.0 is used elsewhere in the codebase
+    const chartOptions: any = {
+      type: 'line',
+      data: parallelData,
+      title: title ?? '',
+      coordinate: { type: 'radar' },
+      inset: 18,
+      encode: {
+        position,
+        color: 'group',
+      },
+      scale: {
+        // Configure scales for each position axis
+        ...Object.fromEntries(
+          Array.from({ length: position.length }, (_, i) => [
+            `position${i === 0 ? '' : i}`,
+            {
+              domainMin: 0,
+              nice: true,
+            },
+          ]),
+        ),
+        color: {
+          range: colors,
+        },
+      },
+      axis: Object.fromEntries(
+        Array.from({ length: position.length }, (_, i) => [
+          `position${i === 0 ? '' : i}`,
+          {
+            zIndex: 1,
+            titleFontSize: 10,
+            titleSpacing: 8,
+            label: true,
+            labelFill: theme === 'dark' ? '#fff' : '#000',
+            labelOpacity: 0.45,
+            labelFontSize: 10,
+            line: true,
+            lineFill: '#000',
+            lineStrokeOpacity: 0.25,
+            tickFilter: (_: string, idx: number) => {
+              return !(i !== 0 && idx === 0);
+            },
+            tickCount: 4,
+            gridStrokeOpacity: 0.45,
+            gridStroke: theme === 'dark' ? '#fff' : '#000',
+            gridLineWidth: 1,
+            gridLineDash: [4, 4],
+          },
+        ]),
+      ),
+      style: {
+        lineWidth,
+        lineCap: 'round',
+        lineJoin: 'round',
+      },
+      legend: parallelData.length > 1 ? { color: { position: 'top', itemMarker: 'point' } } : false,
+      interaction: {
+        tooltip: {
+          series: false,
+        },
+      },
+      viewStyle: {
+        viewFill: backgroundColor,
+      },
+      theme: getThemeObject(theme),
+    };
+
+    chart.options(chartOptions);
+    chart.render();
+  };
+
+  /**
+   * Destroy the chart instance and clean up resources.
+   */
+  const destroy = (): void => {
+    if (chart) {
+      chart.destroy();
+      chart = null;
+    }
+  };
+
+  return {
+    render,
+    destroy,
+  };
+};

package/src/vis/sankey/README.md

@@ -0,0 +1,144 @@
+# Sankey
+
+A sankey chart component for visualizing flow, energy, fund transfer relationships between nodes, built with G2 5.0.
+
+## Usage
+
+```ts
+import { Sankey } from '@antv/gpt-vis/ai';
+
+const sankey = Sankey({
+  container: '#container',
+  width: 600,
+  height: 400,
+});
+
+sankey.render({
+  data: [
+    { source: '煤炭', target: '发电厂', value: 120 },
+    { source: '天然气', target: '发电厂', value: 80 },
+    { source: '发电厂', target: '工业', value: 100 },
+    { source: '发电厂', target: '居民', value: 60 },
+    { source: '发电厂', target: '商业', value: 40 },
+  ],
+});
+
+sankey.destroy();
+```
+
+## Configuration
+
+### Constructor Options (SankeyOptions)
+
+| Property  | Type                  | Default | Description                   |
+| --------- | --------------------- | ------- | ----------------------------- |
+| container | string \| HTMLElement | -       | Container element or selector |
+| width     | number                | 640     | Chart width in pixels         |
+| height    | number                | 480     | Chart height in pixels        |
+
+### Render Config (SankeyConfig)
+
+| Property  | Type   | Default   | Description                                                |
+| --------- | ------ | --------- | ---------------------------------------------------------- |
+| data      | Array  | -         | Chart data array                                           |
+| nodeAlign | string | 'center'  | Node alignment: 'left' \| 'center' \| 'right' \| 'justify' |
+| theme     | string | 'default' | Color theme: 'default' \| 'academy' \| 'dark'              |
+| title     | string | -         | Chart title                                                |
+| style     | object | -         | Chart style configuration                                  |
+
+### Data Structure
+
+```ts
+type SankeyDataItem = {
+  source: string; // Source node name
+  target: string; // Target node name
+  value: number; // Flow value
+};
+```
+
+### Style Options
+
+```ts
+style?: {
+  backgroundColor?: string;  // Background color
+  palette?: string[];        // Color palette
+}
+```
+
+## Examples
+
+### Basic Example
+
+```ts
+sankey.render({
+  data: [
+    { source: '煤炭', target: '发电厂', value: 120 },
+    { source: '天然气', target: '发电厂', value: 80 },
+    { source: '发电厂', target: '工业', value: 100 },
+  ],
+});
+```
+
+### With Title and Node Alignment
+
+```ts
+sankey.render({
+  data: [
+    { source: '煤炭', target: '发电厂', value: 120 },
+    { source: '天然气', target: '发电厂', value: 80 },
+    { source: '发电厂', target: '工业', value: 100 },
+  ],
+  title: '能源流动关系',
+  nodeAlign: 'justify',
+});
+```
+
+### With Academy Theme
+
+```ts
+sankey.render({
+  data: [
+    { source: '投资人', target: '创业公司', value: 200 },
+    { source: '创业公司', target: '市场营销', value: 80 },
+    { source: '创业公司', target: '研发', value: 120 },
+  ],
+  theme: 'academy',
+  title: '资金流转路径',
+});
+```
+
+### With Dark Theme
+
+```ts
+sankey.render({
+  data: [
+    { source: '投资人', target: '创业公司', value: 200 },
+    { source: '创业公司', target: '市场营销', value: 80 },
+    { source: '创业公司', target: '研发', value: 120 },
+  ],
+  theme: 'dark',
+  title: '资金流转路径',
+});
+```
+
+### With Custom Styles
+
+```ts
+sankey.render({
+  data: [
+    { source: '首页', target: '产品页', value: 300 },
+    { source: '产品页', target: '购物车', value: 150 },
+    { source: '购物车', target: '结算页', value: 100 },
+  ],
+  title: '用户行为路径',
+  style: {
+    palette: ['#5B8FF9', '#61DDAA', '#65789B', '#F6BD16', '#7262FD'],
+    backgroundColor: '#f0f2f5',
+  },
+});
+```
+
+## Methods
+
+- `render(config: SankeyConfig): void` - Render or update the chart
+- `destroy(): void` - Destroy the chart instance and clean up resources