@quantlife/qlchart 0.0.1

package/src/plugins/drawing/DrawingToolRegistry.ts

@@ -0,0 +1,684 @@
+import type { DrawingToolType, DrawingToolConfig } from '../../types/index.js';
+import { DrawingToolCategory } from '../../types/index.js';
+
+/**
+ * 绘图工具注册表
+ * 从 lightweight-charts-drawing 的 68 种工具中筛选出我们需要的 21 种
+ */
+export const DRAWING_TOOLS: DrawingToolConfig[] = [
+  // ── P0 必须工具 (priority: 0) ──
+  {
+    type: 'trend-line',
+    label: '趋势线',
+    icon: '📈',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 2,
+    priority: 0,
+  },
+  {
+    type: 'horizontal-line',
+    label: '水平线',
+    icon: '➖',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 1,
+    priority: 0,
+  },
+  {
+    type: 'vertical-line',
+    label: '垂直线',
+    icon: '↕️',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 1,
+    priority: 0,
+  },
+  {
+    type: 'ray',
+    label: '射线',
+    icon: '↗️',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 2,
+    priority: 0,
+  },
+  {
+    type: 'rectangle',
+    label: '矩形',
+    icon: '⬜',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 2,
+    priority: 0,
+  },
+  {
+    type: 'fib-retracement',
+    label: '斐波那契回撤',
+    icon: '📐',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 2,
+    priority: 0,
+  },
+  {
+    type: 'path',
+    label: '路径',
+    icon: '✏️',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: -1, // 动态锚点
+    priority: 0,
+  },
+
+  // ── P1 重要工具 (priority: 1) ──
+  {
+    type: 'parallel-channel',
+    label: '平行通道',
+    icon: '📊',
+    category: DrawingToolCategory.Channels,
+    anchorCount: 3,
+    priority: 1,
+  },
+  {
+    type: 'circle',
+    label: '圆形',
+    icon: '⭕',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'ellipse',
+    label: '椭圆',
+    icon: '🔵',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'triangle',
+    label: '三角形',
+    icon: '🔺',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 3,
+    priority: 1,
+  },
+  {
+    type: 'fib-extension',
+    label: '斐波那契扩展',
+    icon: '📏',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 3,
+    priority: 1,
+  },
+  {
+    type: 'arrow',
+    label: '箭头',
+    icon: '➡️',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'brush',
+    label: '画笔',
+    icon: '🖌️',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: -1, // 动态锚点
+    priority: 1,
+  },
+
+  // ── P2 扩展工具 (priority: 2) ──
+  {
+    type: 'cross-line',
+    label: '十字线',
+    icon: '✚',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'callout',
+    label: '标注',
+    icon: '💬',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'long-position',
+    label: '多头仓位',
+    icon: '🟢',
+    category: DrawingToolCategory.Positions,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'short-position',
+    label: '空头仓位',
+    icon: '🔴',
+    category: DrawingToolCategory.Positions,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'fib-fan',
+    label: '斐波那契扇形',
+    icon: '🌀',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 2,
+    priority: 2,
+  },
+  // ── 新增工具 ──
+  {
+    type: 'date-price-range',
+    label: '测量器',
+    icon: '📐',
+    category: DrawingToolCategory.Measure,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'fib-circles',
+    label: '斐波那契圆形',
+    icon: '🎯',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'fib-time-zone',
+    label: '斐波那契时区',
+    icon: '⏱️',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'rotated-rectangle',
+    label: '旋转矩形',
+    icon: '🔷',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 3,
+    priority: 1,
+  },
+  // ── 形态工具组 ──
+  {
+    type: 'pattern-abcd',
+    label: 'ABCD形态',
+    icon: '🔤',
+    category: DrawingToolCategory.Patterns,
+    anchorCount: 4,
+    priority: 1,
+  },
+  {
+    type: 'pattern-xabcd',
+    label: 'XABCD形态',
+    icon: '🔠',
+    category: DrawingToolCategory.Patterns,
+    anchorCount: 5,
+    priority: 1,
+  },
+  {
+    type: 'pattern-head-shoulders',
+    label: '头肩顶/底',
+    icon: '👤',
+    category: DrawingToolCategory.Patterns,
+    anchorCount: 5,
+    priority: 1,
+  },
+  {
+    type: 'pattern-triangle',
+    label: '三角形',
+    icon: '📐',
+    category: DrawingToolCategory.Patterns,
+    anchorCount: 4,
+    priority: 1,
+  },
+  {
+    type: 'pattern-cypher',
+    label: '赛福形态',
+    icon: '🌀',
+    category: DrawingToolCategory.Patterns,
+    anchorCount: 5,
+    priority: 1,
+  },
+
+  // ═══════════════════════════════════════════
+  // ★ 新增工具（约40种）
+  // ═══════════════════════════════════════════
+
+  // ── 线条扩展 ──
+  {
+    type: 'info-line',
+    label: '信息线',
+    icon: 'ℹ️',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'extended-line',
+    label: '延长线',
+    icon: '↔️',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'trend-angle',
+    label: '趋势角',
+    icon: '📐',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'horizontal-ray',
+    label: '水平射线',
+    icon: '➡️',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 1,
+    priority: 1,
+  },
+
+  // ── 通道扩展 ──
+  {
+    type: 'regression-trend',
+    label: '回归趋势',
+    icon: '📈',
+    category: DrawingToolCategory.Channels,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'flat-top-bottom',
+    label: '平底/平顶',
+    icon: '〰️',
+    category: DrawingToolCategory.Channels,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'disjoint-channel',
+    label: '分离通道',
+    icon: '↕️',
+    category: DrawingToolCategory.Channels,
+    anchorCount: 4,
+    priority: 2,
+  },
+
+  // ── 螺叉 ──
+  {
+    type: 'andrews-pitchfork',
+    label: '安德鲁斯螺叉',
+    icon: '🔱',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'schiff-pitchfork',
+    label: '席夫螺叉',
+    icon: '🔱',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'modified-schiff-pitchfork',
+    label: '修正席夫螺叉',
+    icon: '🔱',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'inside-pitchfork',
+    label: '内部螺叉',
+    icon: '🔱',
+    category: DrawingToolCategory.Lines,
+    anchorCount: 3,
+    priority: 2,
+  },
+
+  // ── 斐波那契扩展 ──
+  {
+    type: 'fib-channel',
+    label: '斐波那契通道',
+    icon: '📊',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'fib-time-extension',
+    label: '斐波那契时间扩展',
+    icon: '⏰',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'fib-spiral',
+    label: '斐波那契螺旋',
+    icon: '🌀',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'fib-arcs',
+    label: '斐波那契弧线',
+    icon: '🌗',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'fib-wedge',
+    label: '斐波那契楔形',
+    icon: '🔻',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'pitchfan',
+    label: '螺叉扇形',
+    icon: '🌀',
+    category: DrawingToolCategory.Fibonacci,
+    anchorCount: 3,
+    priority: 2,
+  },
+
+  // ── 江恩 ──
+  {
+    type: 'gann-box',
+    label: '江恩方块',
+    icon: '⬛',
+    category: DrawingToolCategory.Measure,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'gann-fan',
+    label: '江恩扇形',
+    icon: '✳️',
+    category: DrawingToolCategory.Measure,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'gann-square-fixed',
+    label: '江恩方形(固定)',
+    icon: '🔲',
+    category: DrawingToolCategory.Measure,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'gann-square',
+    label: '江恩方形',
+    icon: '🔲',
+    category: DrawingToolCategory.Measure,
+    anchorCount: 2,
+    priority: 2,
+  },
+
+  // ── 预测扩展 ──
+  {
+    type: 'forecast',
+    label: '预测',
+    icon: '🔮',
+    category: DrawingToolCategory.Positions,
+    anchorCount: 2,
+    priority: 2,
+  },
+  {
+    type: 'bars-pattern',
+    label: 'K线形态',
+    icon: '📊',
+    category: DrawingToolCategory.Patterns,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'projection',
+    label: '投影',
+    icon: '📐',
+    category: DrawingToolCategory.Positions,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'price-range',
+    label: '价格范围',
+    icon: '📏',
+    category: DrawingToolCategory.Measure,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'date-range',
+    label: '日期范围',
+    icon: '📅',
+    category: DrawingToolCategory.Measure,
+    anchorCount: 2,
+    priority: 1,
+  },
+
+  // ── 形状扩展 ──
+  {
+    type: 'polyline',
+    label: '多段线',
+    icon: '〰️',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: -1,
+    priority: 1,
+  },
+  {
+    type: 'curve',
+    label: '曲线',
+    icon: '〰️',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 4,
+    priority: 2,
+  },
+  {
+    type: 'double-curve',
+    label: '双曲线',
+    icon: '≈',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'arc',
+    label: '弧形',
+    icon: '⌒',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: 3,
+    priority: 2,
+  },
+  {
+    type: 'polygon',
+    label: '多边形',
+    icon: '⬡',
+    category: DrawingToolCategory.Shapes,
+    anchorCount: -1,
+    priority: 2,
+  },
+
+  // ── 标注 ──
+  {
+    type: 'text',
+    label: '文字',
+    icon: '📝',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 0,
+  },
+  {
+    type: 'text-annotation',
+    label: '文字标注',
+    icon: '📝',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'anchored-text',
+    label: '锚定文字',
+    icon: '📌',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 2,
+    priority: 1,
+  },
+  {
+    type: 'note',
+    label: '便签',
+    icon: '🗒️',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 1,
+  },
+  {
+    type: 'price-note',
+    label: '价格标签',
+    icon: '🏷️',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 1,
+  },
+  {
+    type: 'price-label',
+    label: '价格标注',
+    icon: '💲',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 1,
+  },
+  {
+    type: 'flag-mark',
+    label: '旗标',
+    icon: '🚩',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'pin',
+    label: '图钉',
+    icon: '📍',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'comment',
+    label: '评论',
+    icon: '💬',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'signpost',
+    label: '路标',
+    icon: '🪧',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'table',
+    label: '表格',
+    icon: '📋',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'highlighter',
+    label: '高亮笔',
+    icon: '🖍️',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: -1,
+    priority: 1,
+  },
+  {
+    type: 'arrow-marker',
+    label: '箭头标记',
+    icon: '📌',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 1,
+  },
+  {
+    type: 'arrow-mark-up',
+    label: '上涨标记',
+    icon: '🔼',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 2,
+  },
+  {
+    type: 'arrow-mark-down',
+    label: '下跌标记',
+    icon: '🔽',
+    category: DrawingToolCategory.Annotations,
+    anchorCount: 1,
+    priority: 2,
+  },
+];
+
+/** 获取按分类分组的工具 */
+export function getToolsByCategory(): Map<DrawingToolCategory, DrawingToolConfig[]> {
+  const map = new Map<DrawingToolCategory, DrawingToolConfig[]>();
+  for (const tool of DRAWING_TOOLS) {
+    const list = map.get(tool.category) ?? [];
+    list.push(tool);
+    map.set(tool.category, list);
+  }
+  return map;
+}
+
+/** 获取按优先级分组的工具 */
+export function getToolsByPriority(priority: number): DrawingToolConfig[] {
+  return DRAWING_TOOLS.filter((t) => t.priority === priority);
+}
+
+/** 获取工具配置 */
+export function getToolConfig(type: DrawingToolType): DrawingToolConfig | undefined {
+  return DRAWING_TOOLS.find((t) => t.type === type);
+}
+
+/** 分类中文标签 */
+export const CATEGORY_LABELS: Record<DrawingToolCategory, string> = {
+  [DrawingToolCategory.Lines]: '线条',
+  [DrawingToolCategory.Channels]: '通道',
+  [DrawingToolCategory.Fibonacci]: '斐波那契',
+  [DrawingToolCategory.Shapes]: '形状',
+  [DrawingToolCategory.Annotations]: '标注',
+  [DrawingToolCategory.Positions]: '仓位',
+  [DrawingToolCategory.Measure]: '测量',
+  [DrawingToolCategory.Patterns]: '形态',
+  [DrawingToolCategory.VolumeProfile]: '成交量',
+};
+
+/**
+ * 工具类型映射：将我们的 DrawingToolType 映射到库的 tool type string
+ * 注意：fib-fan 在库中是 fib-speed-fan
+ */
+export function mapToolTypeToLib(type: DrawingToolType): string | null {
+  if (type === 'cursor') return null;
+  if (type === 'fib-fan') return 'fib-speed-fan';
+  // 形态工具使用 path 作为底层渲染
+  if (type.startsWith('pattern-')) return 'path';
+  // 新增工具直接透传
+  return type;
+}
+
+/**
+ * 从库的 tool type 映射回我们的 DrawingToolType
+ */
+export function mapLibTypeToOurs(libType: string): DrawingToolType | null {
+  if (libType === 'fib-speed-fan') return 'fib-fan';
+  const found = DRAWING_TOOLS.find((t) => t.type === libType);
+  return found ? found.type : null;
+}