@363045841yyt/klinechart-core 0.8.8 → 0.8.9

package/src/engine/drawing/plugin.ts

@@ -10,29 +10,76 @@ import type { PrimitiveRendererSet } from './index'
 import type { KLineData } from '../../types/price'
 import { getPhysicalKLineConfig } from '../utils/klineConfig'
 
+type SafeViewport = { scrollLeft: number; plotWidth: number; plotHeight: number }
+type ToScreenFn = (anchor: { index: number; price: number }) => { x: number; y: number }
+
+function resolveViewport(context: RenderContext, paneHeight: number): SafeViewport {
+  return context.viewport ?? {
+    scrollLeft: context.scrollLeft,
+    plotWidth: context.paneWidth,
+    plotHeight: paneHeight,
+  }
+}
+
+function createToScreen(
+  kWidth: number, kGap: number, dpr: number,
+  pane: RenderContext['pane'], viewport: SafeViewport
+): ToScreenFn {
+  const { startXPx, unitPx } = getPhysicalKLineConfig(kWidth, kGap, dpr)
+  return (anchor) => {
+    if (!Number.isFinite(anchor.index) || anchor.index < 0) {
+      return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
+    }
+    const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
+    return { x, y: pane.yAxis.priceToY(anchor.price) }
+  }
+}
+
+function renderPrimitives(
+  ctx: CanvasRenderingContext2D,
+  primitives: DrawingPrimitive[],
+  renderers: PrimitiveRendererSet,
+  viewportClip: { left: number; top: number; right: number; bottom: number },
+  dpr: number
+): void {
+  for (const primitive of primitives) {
+    if (primitive.kind === 'point') {
+      renderers.point(ctx, primitive, dpr)
+      continue
+    }
+    if (primitive.kind === 'line') {
+      renderers.line(ctx, primitive, viewportClip, dpr)
+      continue
+    }
+    if (primitive.kind === 'area') {
+      renderers.area(ctx, primitive, dpr)
+      continue
+    }
+    renderers.text(ctx, primitive, dpr)
+  }
+}
+
 /**
- * 为选中的绘图推送锚点轴标签
- * 提取为独立函数供主插件和 overlay 插件共用
+ * 查找当前选中且在指定 pane 内的绘图
  */
-function pushSelectedDrawingLabels(
+function findSelectedAnchorDrawing(
   store: DrawingStore,
   definitions: DrawingDefinitionRegistry,
-  context: RenderContext,
   pane: RenderContext['pane'],
   seriesData: KLineData[],
   range: RenderContext['range'],
+  context: RenderContext,
   kWidth: number,
   kGap: number,
   dpr: number,
-  viewport: NonNullable<RenderContext['viewport']>,
-  startXPx: number,
-  unitPx: number
-): void {
+  viewport: SafeViewport,
+  toScreen: ToScreenFn
+) {
   const selectedId = store.getSelectedId()
-  if (!selectedId) return
+  if (!selectedId) return null
 
   const selectedDrawing = store.getAll().find(d => d.id === selectedId)
-  if (!selectedDrawing || !selectedDrawing.visible || selectedDrawing.paneId !== pane.id) return
+  if (!selectedDrawing || !selectedDrawing.visible || selectedDrawing.paneId !== pane.id) return null
 
   const geometry = definitions.compute(selectedDrawing, {
     pane,
@@ -47,65 +94,88 @@ function pushSelectedDrawingLabels(
     dpr,
     paneWidth: context.paneWidth,
     viewport,
-    toScreen(anchor) {
-      if (!Number.isFinite(anchor.index) || anchor.index < 0) {
-        return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
-      }
-      const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
-      return { x, y: pane.yAxis.priceToY(anchor.price) }
-    },
+    toScreen,
   })
+  if (!geometry) return null
 
-  if (!geometry) return
+  const allAnchors = [...selectedDrawing.anchors, ...(geometry.computedAnchors ?? [])]
+  if (allAnchors.length === 0) return null
 
-  // 只在价格 pane 时添加标签
-  if (pane.role !== 'price') return
+  return { drawing: selectedDrawing, allAnchors }
+}
 
-  // 合并用户锚点和计算锚点
-  const allAnchors = [...selectedDrawing.anchors, ...(geometry.computedAnchors ?? [])]
-  if (allAnchors.length === 0) return
+/**
+ * 单次遍历锚点，同时计算价格范围和索引范围，推送到 context
+ */
+function pushAnchorBands(
+  allAnchors: ReadonlyArray<{ price: number; index: number }>,
+  context: RenderContext,
+  pane: RenderContext['pane'],
+  style: DrawingStyle | undefined,
+  startXPx: number,
+  unitPx: number,
+  dpr: number
+): void {
+  if (allAnchors.length < 2) return
 
-  // 计算锚点价格范围，用于Y轴价格范围带
-  if (allAnchors.length >= 2) {
-    let minP = Infinity, maxP = -Infinity
-    for (const a of allAnchors) {
-      if (!Number.isFinite(a.price)) continue
+  let minP = Infinity, maxP = -Infinity
+  let minIdx = Infinity, maxIdx = -Infinity
+
+  for (const a of allAnchors) {
+    if (Number.isFinite(a.price)) {
       if (a.price < minP) minP = a.price
       if (a.price > maxP) maxP = a.price
     }
-    if (Number.isFinite(minP) && Number.isFinite(maxP) && minP !== maxP) {
-      if (!context.yAxisRanges) context.yAxisRanges = []
-      context.yAxisRanges.push({
-        topY: pane.yAxis.priceToY(maxP),
-        bottomY: pane.yAxis.priceToY(minP),
-        color: selectedDrawing.style?.stroke ?? '#2962ff',
-        opacity: 0.15,
-      })
-    }
-  }
-
-  // 计算锚点X坐标范围，用于X轴时间范围带
-  if (allAnchors.length >= 2) {
-    let minIdx = Infinity, maxIdx = -Infinity
-    for (const a of allAnchors) {
-      if (!Number.isFinite(a.index) || a.index < 0) continue
+    if (Number.isFinite(a.index) && a.index >= 0) {
       if (a.index < minIdx) minIdx = a.index
       if (a.index > maxIdx) maxIdx = a.index
     }
-    if (Number.isFinite(minIdx) && Number.isFinite(maxIdx) && minIdx !== maxIdx) {
-      if (!context.xAxisRanges) context.xAxisRanges = []
-      const leftX = (startXPx + minIdx * unitPx + (unitPx - 1) / 2) / dpr
-      const rightX = (startXPx + maxIdx * unitPx + (unitPx - 1) / 2) / dpr
-      context.xAxisRanges.push({
-        leftX,
-        rightX,
-        color: selectedDrawing.style?.stroke ?? '#2962ff',
-        opacity: 0.15,
-      })
-    }
   }
 
-  // 辅助函数：根据index获取时间戳
+  const strokeColor = style?.stroke ?? '#2962ff'
+
+  if (Number.isFinite(minP) && Number.isFinite(maxP) && minP !== maxP) {
+    if (!context.yAxisRanges) context.yAxisRanges = []
+    context.yAxisRanges.push({
+      topY: pane.yAxis.priceToY(maxP),
+      bottomY: pane.yAxis.priceToY(minP),
+      color: strokeColor,
+      opacity: 0.15,
+    })
+  }
+
+  if (Number.isFinite(minIdx) && Number.isFinite(maxIdx) && minIdx !== maxIdx) {
+    if (!context.xAxisRanges) context.xAxisRanges = []
+    const leftX = (startXPx + minIdx * unitPx + (unitPx - 1) / 2) / dpr
+    const rightX = (startXPx + maxIdx * unitPx + (unitPx - 1) / 2) / dpr
+    context.xAxisRanges.push({
+      leftX,
+      rightX,
+      color: strokeColor,
+      opacity: 0.15,
+    })
+  }
+}
+
+/**
+ * 遍历锚点，推送 Y 轴和 X 轴标签
+ */
+function pushAnchorLabels(
+  allAnchors: ReadonlyArray<{ index: number; price: number; time?: number | string }>,
+  context: RenderContext,
+  pane: RenderContext['pane'],
+  viewport: SafeViewport,
+  kWidth: number,
+  style: DrawingStyle | undefined,
+  seriesData: KLineData[],
+  startXPx: number,
+  unitPx: number,
+  dpr: number
+): void {
+  if (pane.role !== 'price') return
+
+  const strokeColor = style?.stroke ?? '#2962ff'
+
   const getTimestampForIndex = (idx: number): number | null => {
     if (idx >= 0 && idx < seriesData.length) {
       return seriesData[idx]?.timestamp ?? null
@@ -121,7 +191,6 @@ function pushSelectedDrawingLabels(
     return null
   }
 
-  // 推送每个锚点的标签
   for (const anchor of allAnchors) {
     if (!Number.isFinite(anchor.index) || !Number.isFinite(anchor.price)) continue
 
@@ -133,7 +202,6 @@ function pushSelectedDrawingLabels(
       return { x, y: pane.yAxis.priceToY(anchor.price) }
     })()
 
-    // Y轴标签
     if (screenPoint.y >= 0 && screenPoint.y <= pane.height) {
       if (!context.yAxisLabels) context.yAxisLabels = []
       context.yAxisLabels.push({
@@ -141,14 +209,13 @@ function pushSelectedDrawingLabels(
         price: anchor.price,
         y: screenPoint.y,
         style: {
-          bgColor: selectedDrawing.style?.stroke ?? '#2962ff',
-          borderColor: selectedDrawing.style?.stroke ?? '#2962ff',
+          bgColor: strokeColor,
+          borderColor: strokeColor,
           textColor: '#ffffff',
         }
       })
     }
 
-    // X轴标签
     if (screenPoint.x >= -kWidth && screenPoint.x <= viewport.plotWidth + kWidth) {
       const timestamp = anchor.time
         ? (typeof anchor.time === 'string' ? new Date(anchor.time).getTime() : anchor.time)
@@ -161,7 +228,7 @@ function pushSelectedDrawingLabels(
         timestamp,
         x: screenPoint.x + viewport.scrollLeft,
         style: {
-          bgColor: selectedDrawing.style?.stroke ?? '#2962ff',
+          bgColor: strokeColor,
           textColor: '#ffffff',
         }
       })
@@ -169,6 +236,32 @@ function pushSelectedDrawingLabels(
   }
 }
 
+/**
+ * 为选中的绘图推送锚点轴标签
+ * 提取为独立函数供主插件和 overlay 插件共用
+ */
+function pushSelectedDrawingLabels(
+  store: DrawingStore,
+  definitions: DrawingDefinitionRegistry,
+  context: RenderContext,
+  pane: RenderContext['pane'],
+  seriesData: KLineData[],
+  range: RenderContext['range'],
+  kWidth: number,
+  kGap: number,
+  dpr: number,
+  viewport: SafeViewport,
+  startXPx: number,
+  unitPx: number
+): void {
+  const toScreen = createToScreen(kWidth, kGap, dpr, pane, viewport)
+  const found = findSelectedAnchorDrawing(store, definitions, pane, seriesData, range, context, kWidth, kGap, dpr, viewport, toScreen)
+  if (!found) return
+
+  pushAnchorBands(found.allAnchors, context, pane, found.drawing.style, startXPx, unitPx, dpr)
+  pushAnchorLabels(found.allAnchors, context, pane, viewport, kWidth, found.drawing.style, seriesData, startXPx, unitPx, dpr)
+}
+
 /**
  * 创建绘图渲染器插件（主层，负责绘制 shape）
  * 注意：此插件不再推送轴标签，标签由 createDrawingLabelOverlayPlugin 负责
@@ -193,11 +286,7 @@ export function createDrawingRendererPlugin(options: {
     priority: 55,
     draw(context: RenderContext) {
       const { ctx, pane, data, range, dpr, paneWidth, kLinePositions, kLineCenters, kBarRects, kWidth, kGap } = context
-      const viewport = context.viewport ?? {
-        scrollLeft: context.scrollLeft,
-        plotWidth: paneWidth,
-        plotHeight: pane.height,
-      }
+      const viewport = resolveViewport(context, pane.height)
       const seriesData = data as KLineData[]
       const visibleData = seriesData.slice(range.start, range.end)
       const drawings = store.getVisibleByPane(pane.id)
@@ -209,6 +298,7 @@ export function createDrawingRendererPlugin(options: {
         right: viewport.plotWidth,
         bottom: pane.height,
       }
+      const toScreen = createToScreen(kWidth, kGap, dpr, pane, viewport)
 
       ctx.save()
       ctx.beginPath()
@@ -229,14 +319,7 @@ export function createDrawingRendererPlugin(options: {
           dpr,
           paneWidth,
           viewport,
-          toScreen(anchor) {
-            const { startXPx, unitPx } = getPhysicalKLineConfig(kWidth, kGap, dpr)
-            if (!Number.isFinite(anchor.index) || anchor.index < 0) {
-              return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
-            }
-            const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
-            return { x, y: pane.yAxis.priceToY(anchor.price) }
-          },
+          toScreen,
         })
         if (!geometry) continue
 
@@ -245,22 +328,7 @@ export function createDrawingRendererPlugin(options: {
           ? geometry.primitives.map((p) => applySelectedStyle(p, drawing.style))
           : geometry.primitives
 
-        // 绘制形状（不再推送轴标签，标签由 overlay 插件负责）
-        for (const primitive of primitives) {
-          if (primitive.kind === 'point') {
-            renderers.point(ctx, primitive, dpr)
-            continue
-          }
-          if (primitive.kind === 'line') {
-            renderers.line(ctx, primitive, viewportClip, dpr)
-            continue
-          }
-          if (primitive.kind === 'area') {
-            renderers.area(ctx, primitive, dpr)
-            continue
-          }
-          renderers.text(ctx, primitive, dpr)
-        }
+        renderPrimitives(ctx, primitives, renderers, viewportClip, dpr)
       }
 
       ctx.restore()
@@ -297,15 +365,10 @@ export function createDrawingLabelOverlayPlugin(options: {
     priority: -24, // 比主绘图层稍晚，确保在其他 overlay 插件之后
     draw(context: RenderContext) {
       const { pane, data, range, dpr, paneWidth, kWidth, kGap } = context
-      const viewport = context.viewport ?? {
-        scrollLeft: context.scrollLeft,
-        plotWidth: paneWidth,
-        plotHeight: pane.height,
-      }
+      const viewport = resolveViewport(context, pane.height)
       const seriesData = data as KLineData[]
       const { startXPx, unitPx } = getPhysicalKLineConfig(kWidth, kGap, dpr)
 
-      // 推送选中绘图的锚点标签
       pushSelectedDrawingLabels(
         store,
         definitions,

package/src/engine/indicators/__tests__/calculators.test.ts

@@ -6,16 +6,24 @@ import {
   calcENEData,
   calcRSIData,
   DEFAULT_MA_PERIODS,
+} from '../calculators'
+import {
   DEFAULT_BOLL_PERIOD,
   DEFAULT_BOLL_MULTIPLIER,
+} from '../state/bollState'
+import {
   DEFAULT_EXPMA_FAST_PERIOD,
   DEFAULT_EXPMA_SLOW_PERIOD,
+} from '../state/expmaState'
+import {
   DEFAULT_ENE_PERIOD,
   DEFAULT_ENE_DEVIATION,
+} from '../state/eneState'
+import {
   DEFAULT_RSI_PERIOD1,
   DEFAULT_RSI_PERIOD2,
   DEFAULT_RSI_PERIOD3,
-} from '../calculators'
+} from '../state/rsiState'
 import type { KLineData } from '@/types/price'
 
 /**

package/src/engine/indicators/__tests__/indicatorRegistry.test.ts

@@ -1,7 +1,12 @@
-import { describe, expect, it, vi } from 'vitest'
+import { describe, expect, it, vi, beforeEach } from 'vitest'
 import { IndicatorRegistry } from '../indicatorRegistry'
+import { clearRegisteredIndicatorDefinitionsForTest } from '../indicatorDefinitionRegistry'
 import type { IndicatorMetadata } from '../indicatorMetadata'
 
+beforeEach(() => {
+  clearRegisteredIndicatorDefinitionsForTest()
+})
+
 function createMeta(overrides: Partial<IndicatorMetadata> = {}): IndicatorMetadata {
   return {
     name: 'rsi',
@@ -17,7 +22,7 @@ function createMeta(overrides: Partial<IndicatorMetadata> = {}): IndicatorMetada
 
 describe('IndicatorRegistry', () => {
   it('resolves indicators by name, displayName, aliases, and case-insensitive ids', () => {
-    const registry = new IndicatorRegistry()
+    const registry = new IndicatorRegistry(false)
     const meta = createMeta()
 
     registry.register(meta)
@@ -30,7 +35,7 @@ describe('IndicatorRegistry', () => {
   })
 
   it('throws for missing required metadata fields', () => {
-    const registry = new IndicatorRegistry()
+    const registry = new IndicatorRegistry(false)
 
     expect(() => registry.register(createMeta({ name: '' }))).toThrow('Indicator name is required')
     expect(() => registry.register(createMeta({ displayName: '' }))).toThrow("displayName is required")
@@ -38,7 +43,7 @@ describe('IndicatorRegistry', () => {
   })
 
   it('replaces old aliases when unregistering', () => {
-    const registry = new IndicatorRegistry()
+    const registry = new IndicatorRegistry(false)
     const meta = createMeta()
 
     registry.register(meta)
@@ -51,7 +56,7 @@ describe('IndicatorRegistry', () => {
   })
 
   it('returns required metadata or throws a descriptive error', () => {
-    const registry = new IndicatorRegistry()
+    const registry = new IndicatorRegistry(false)
     const meta = createMeta()
 
     registry.register(meta)

package/src/engine/indicators/__tests__/scheduler.test.ts

@@ -1,11 +1,11 @@
 import { describe, it, expect, vi, beforeEach, beforeAll } from 'vitest'
 import { IndicatorScheduler } from '../scheduler'
 import { getBuiltinIndicatorDefinitions, loadBuiltinIndicators } from '../registerBuiltins'
-import { MA_STATE_KEY, EMPTY_MA_STATE, type MARenderState } from '../maState'
-import { BOLL_STATE_KEY, EMPTY_BOLL_STATE, type BOLLRenderState } from '../bollState'
-import { EXPMA_STATE_KEY, EMPTY_EXPMA_STATE, type EXPMARenderState } from '../expmaState'
-import { ENE_STATE_KEY, EMPTY_ENE_STATE, type ENERenderState } from '../eneState'
-import { createRSIStateKey, EMPTY_RSI_STATE, type RSIRenderState } from '../rsiState'
+import { MA_STATE_KEY, EMPTY_MA_STATE, type MARenderState } from '../state/maState'
+import { BOLL_STATE_KEY, EMPTY_BOLL_STATE, type BOLLRenderState } from '../state/bollState'
+import { EXPMA_STATE_KEY, EMPTY_EXPMA_STATE, type EXPMARenderState } from '../state/expmaState'
+import { ENE_STATE_KEY, EMPTY_ENE_STATE, type ENERenderState } from '../state/eneState'
+import { createRSIStateKey, EMPTY_RSI_STATE, type RSIRenderState } from '../state/rsiState'
 import type { IndicatorMetadata } from '../indicatorMetadata'
 import type { KLineData } from '@/types/price'
 import type { PluginHost } from '@/plugin'
@@ -131,7 +131,7 @@ describe('IndicatorScheduler', () => {
   let mockHost: PluginHost
 
   beforeEach(() => {
-    scheduler = new IndicatorScheduler()
+    scheduler = new IndicatorScheduler(false)
     mockHost = createMockPluginHost()
     scheduler.setPluginHost(mockHost)
     registerTestIndicators(scheduler)
@@ -144,7 +144,7 @@ describe('IndicatorScheduler', () => {
     })
 
     it('should accept plugin host', () => {
-      const newScheduler = new IndicatorScheduler()
+      const newScheduler = new IndicatorScheduler(false)
       newScheduler.setPluginHost(mockHost)
       // Should not throw
       expect(() => newScheduler.recompute()).not.toThrow()
@@ -426,7 +426,7 @@ describe('BOLL State in scheduler', () => {
   let mockHost: PluginHost
 
   beforeEach(() => {
-    scheduler = new IndicatorScheduler()
+    scheduler = new IndicatorScheduler(false)
     mockHost = createMockPluginHost()
     scheduler.setPluginHost(mockHost)
     registerTestIndicators(scheduler)
@@ -509,7 +509,7 @@ describe('EXPMA State in scheduler', () => {
   let mockHost: PluginHost
 
   beforeEach(() => {
-    scheduler = new IndicatorScheduler()
+    scheduler = new IndicatorScheduler(false)
     mockHost = createMockPluginHost()
     scheduler.setPluginHost(mockHost)
     registerTestIndicators(scheduler)
@@ -563,7 +563,7 @@ describe('ENE State in scheduler', () => {
   let mockHost: PluginHost
 
   beforeEach(() => {
-    scheduler = new IndicatorScheduler()
+    scheduler = new IndicatorScheduler(false)
     mockHost = createMockPluginHost()
     scheduler.setPluginHost(mockHost)
     registerTestIndicators(scheduler)
@@ -620,7 +620,7 @@ describe('Per-indicator dirty flags', () => {
   let mockHost: PluginHost
 
   beforeEach(() => {
-    scheduler = new IndicatorScheduler()
+    scheduler = new IndicatorScheduler(false)
     mockHost = createMockPluginHost()
     scheduler.setPluginHost(mockHost)
     registerTestIndicators(scheduler)
@@ -769,7 +769,7 @@ describe('RSI State in scheduler', () => {
   let mockHost: PluginHost
 
   beforeEach(() => {
-    scheduler = new IndicatorScheduler()
+    scheduler = new IndicatorScheduler(false)
     mockHost = createMockPluginHost()
     scheduler.setPluginHost(mockHost)
     registerTestIndicators(scheduler)

package/src/engine/indicators/__tests__/soa.test.ts

@@ -9,30 +9,64 @@ import {
 } from '../soa'
 import {
   calcBOLLData,
-  calcBOLLDataSoA,
   calcEXPMAData,
-  calcEXPMADataSoA,
   calcENEData,
-  calcENEDataSoA,
   calcMAData,
-  calcMADataSoA,
   calcRSIData,
-  calcRSIDataSoA,
   calcCCIData,
-  calcCCIDataSoA,
   calcSTOCHData,
-  calcSTOCHDataSoA,
   calcMOMData,
-  calcMOMDataSoA,
   calcWMSRData,
-  calcWMSRDataSoA,
   calcKSTData,
-  calcKSTDataSoA,
   calcFASTKData,
-  calcFASTKDataSoA,
   calcMACDData,
-  calcMACDDataSoA,
 } from '../calculators'
+import type {
+  BOLLPoint,
+  EXPMAPoint,
+  ENEPoint,
+  STOCHPoint,
+  KSTPoint,
+  MACDPoint,
+} from '../calculators'
+
+// SoA wrapper functions (test-only — convert SoA → AoS before computing)
+function calcBOLLDataSoA(layout: KLineSoALayout, period: number, multiplier: number): BOLLPoint[] {
+  return calcBOLLData(SharedKLineBuffer.toKLineData(layout), period, multiplier)
+}
+function calcEXPMADataSoA(layout: KLineSoALayout, fastPeriod: number, slowPeriod: number): EXPMAPoint[] {
+  return calcEXPMAData(SharedKLineBuffer.toKLineData(layout), fastPeriod, slowPeriod)
+}
+function calcENEDataSoA(layout: KLineSoALayout, period: number, deviation: number): ENEPoint[] {
+  return calcENEData(SharedKLineBuffer.toKLineData(layout), period, deviation)
+}
+function calcMADataSoA(layout: KLineSoALayout, period: number): (number | undefined)[] {
+  return calcMAData(SharedKLineBuffer.toKLineData(layout), period)
+}
+function calcRSIDataSoA(layout: KLineSoALayout, period: number): (number | undefined)[] {
+  return calcRSIData(SharedKLineBuffer.toKLineData(layout), period)
+}
+function calcCCIDataSoA(layout: KLineSoALayout, period: number): (number | undefined)[] {
+  return calcCCIData(SharedKLineBuffer.toKLineData(layout), period)
+}
+function calcSTOCHDataSoA(layout: KLineSoALayout, n: number, m: number): STOCHPoint[] {
+  return calcSTOCHData(SharedKLineBuffer.toKLineData(layout), n, m)
+}
+function calcMOMDataSoA(layout: KLineSoALayout, period: number): (number | undefined)[] {
+  return calcMOMData(SharedKLineBuffer.toKLineData(layout), period)
+}
+function calcWMSRDataSoA(layout: KLineSoALayout, period: number): (number | undefined)[] {
+  return calcWMSRData(SharedKLineBuffer.toKLineData(layout), period)
+}
+function calcKSTDataSoA(layout: KLineSoALayout, roc1: number, roc2: number, roc3: number, roc4: number, signalPeriod: number): KSTPoint[] {
+  return calcKSTData(SharedKLineBuffer.toKLineData(layout), roc1, roc2, roc3, roc4, signalPeriod)
+}
+function calcFASTKDataSoA(layout: KLineSoALayout, period: number): (number | undefined)[] {
+  return calcFASTKData(SharedKLineBuffer.toKLineData(layout), period)
+}
+function calcMACDDataSoA(layout: KLineSoALayout, fastPeriod: number, slowPeriod: number, signalPeriod: number): MACDPoint[] {
+  return calcMACDData(SharedKLineBuffer.toKLineData(layout), fastPeriod, slowPeriod, signalPeriod)
+}
 
 /**
  * 生成测试用的 K线数据

package/src/engine/indicators/calculators/_shared.ts

@@ -0,0 +1,44 @@
+export function _computeEMASeries(values: (number | undefined)[], period: number): (number | undefined)[] {
+    const n = values.length
+    const result: (number | undefined)[] = new Array(n).fill(undefined)
+    if (n === 0 || period <= 0) return result
+
+    const alpha = 2 / (period + 1)
+
+    let i = 0
+    while (i < n && values[i] === undefined) i++
+    if (i >= n) return result
+
+    let ema = values[i]!
+    result[i] = ema
+    for (let t = i + 1; t < n; t++) {
+        const v = values[t]
+        if (v === undefined) continue
+        ema = v * alpha + ema * (1 - alpha)
+        result[t] = ema
+    }
+    return result
+}
+
+export function _computeWMAOnNumbers(values: (number | undefined)[], period: number): (number | undefined)[] {
+    const n = values.length
+    const result: (number | undefined)[] = new Array(n).fill(undefined)
+    if (n === 0 || period <= 0 || n < period) return result
+
+    const denom = (period * (period + 1)) / 2
+
+    for (let t = period - 1; t < n; t++) {
+        let sw = 0
+        let valid = true
+        for (let k = 0; k < period; k++) {
+            const v = values[t - period + 1 + k]
+            if (v === undefined) {
+                valid = false
+                break
+            }
+            sw += (k + 1) * v
+        }
+        if (valid) result[t] = sw / denom
+    }
+    return result
+}