@363045841yyt/klinechart-core 0.8.12 → 0.8.13-alpha.0

package/src/engine/renderers/Indicator/wmsr.ts

@@ -15,302 +15,326 @@ import { createSingleLineTitleInfo } from './shared/titleInfo'
 type LinePoint = { x: number; y: number }
 
 interface WMSRRendererOptions {
-    /** 目标 pane ID（默认 'sub'） */
-    paneId?: string
+  /** 目标 pane ID（默认 'sub'） */
+  paneId?: string
 }
 
 function getWMSRStateKey(host: PluginHost | null, paneId: string): string | null {
-    const scheduler = host?.getService<IndicatorScheduler>('indicatorScheduler')
-    if (!scheduler) {
-        console.warn('[WMSRRenderer] Scheduler not available via service locator')
-        return null
-    }
-    const meta = scheduler.getIndicatorMetadata('wmsr')
-    if (!meta) {
-        console.warn("[WMSRRenderer] Indicator metadata for 'wmsr' not found, skip rendering")
-        return null
-    }
-    return resolveStateKey(meta.stateKey, paneId)
+  const scheduler = host?.getService<IndicatorScheduler>('indicatorScheduler')
+  if (!scheduler) {
+    console.warn('[WMSRRenderer] Scheduler not available via service locator')
+    return null
+  }
+  const meta = scheduler.getIndicatorMetadata('wmsr')
+  if (!meta) {
+    console.warn("[WMSRRenderer] Indicator metadata for 'wmsr' not found, skip rendering")
+    return null
+  }
+  return resolveStateKey(meta.stateKey, paneId)
 }
 
 /**
  * 创建 WMSR 渲染器插件
  */
 function createWMSRRendererPlugin(options: WMSRRendererOptions = {}): RendererPluginWithHost {
-    const { paneId = 'sub' } = options
-    let pluginHost: PluginHost | null = null
-
-    function resolveKey(): string | null {
-        return getWMSRStateKey(pluginHost, paneId)
+  const { paneId = 'sub' } = options
+  let pluginHost: PluginHost | null = null
+
+  function resolveKey(): string | null {
+    return getWMSRStateKey(pluginHost, paneId)
+  }
+
+  // 线条点缓存
+  let cachedKey = ''
+  let cachedWMSRPoints: LinePoint[] = []
+
+  // 离屏 Canvas 缓存虚线背景线 (-20 / -50 / -80)
+  let offscreenCanvas: HTMLCanvasElement | null = null
+  let offscreenCtx: CanvasRenderingContext2D | null = null
+  let cachedDashedLinesKey = ''
+
+  function clearLineCache() {
+    cachedKey = ''
+    cachedWMSRPoints = []
+  }
+
+  function getOffscreenCanvas(
+    width: number,
+    height: number,
+  ): { canvas: HTMLCanvasElement; ctx: CanvasRenderingContext2D } {
+    if (!offscreenCanvas || offscreenCanvas.width !== width || offscreenCanvas.height !== height) {
+      offscreenCanvas = document.createElement('canvas')
+      offscreenCanvas.width = width
+      offscreenCanvas.height = height
+      offscreenCtx = offscreenCanvas.getContext('2d')!
+      cachedDashedLinesKey = ''
     }
+    return { canvas: offscreenCanvas, ctx: offscreenCtx! }
+  }
+
+  function buildDashedLinesKey(
+    paneWidth: number,
+    paneHeight: number,
+    displayMin: number,
+    displayMax: number,
+    dpr: number,
+  ): string {
+    return `${paneWidth}|${paneHeight}|${displayMin.toFixed(4)}|${displayMax.toFixed(4)}|${dpr}`
+  }
+
+  function renderDashedLinesToOffscreen(
+    ctx: CanvasRenderingContext2D,
+    paneWidth: number,
+    paneHeight: number,
+    displayMin: number,
+    displayMax: number,
+    dpr: number,
+    colors: { wmsr: { overbought: string; oversold: string }; wmsrGrid: string },
+  ): void {
+    const displayValueRange = displayMax - displayMin || 1
+    const y20 = alignToPhysicalPixelCenter(
+      paneHeight - ((-20 - displayMin) / displayValueRange) * paneHeight,
+      dpr,
+    )
+    const y80 = alignToPhysicalPixelCenter(
+      paneHeight - ((-80 - displayMin) / displayValueRange) * paneHeight,
+      dpr,
+    )
+    const y50 = alignToPhysicalPixelCenter(
+      paneHeight - ((-50 - displayMin) / displayValueRange) * paneHeight,
+      dpr,
+    )
+
+    ctx.clearRect(0, 0, ctx.canvas.width, ctx.canvas.height)
+    ctx.save()
+    ctx.scale(dpr, dpr)
+    ctx.lineWidth = 1
+    ctx.setLineDash([4, 4])
 
-    // 线条点缓存
-    let cachedKey = ''
-    let cachedWMSRPoints: LinePoint[] = []
+    ctx.strokeStyle = colors.wmsr.overbought
+    ctx.beginPath()
+    ctx.moveTo(0, y20)
+    ctx.lineTo(paneWidth, y20)
+    ctx.stroke()
 
-    // 离屏 Canvas 缓存虚线背景线 (-20 / -50 / -80)
-    let offscreenCanvas: HTMLCanvasElement | null = null
-    let offscreenCtx: CanvasRenderingContext2D | null = null
-    let cachedDashedLinesKey = ''
+    ctx.strokeStyle = colors.wmsr.oversold
+    ctx.beginPath()
+    ctx.moveTo(0, y80)
+    ctx.lineTo(paneWidth, y80)
+    ctx.stroke()
 
-    function clearLineCache() {
-        cachedKey = ''
-        cachedWMSRPoints = []
-    }
+    ctx.strokeStyle = colors.wmsrGrid
+    ctx.beginPath()
+    ctx.moveTo(0, y50)
+    ctx.lineTo(paneWidth, y50)
+    ctx.stroke()
 
-    function getOffscreenCanvas(width: number, height: number): { canvas: HTMLCanvasElement; ctx: CanvasRenderingContext2D } {
-        if (!offscreenCanvas || offscreenCanvas.width !== width || offscreenCanvas.height !== height) {
-            offscreenCanvas = document.createElement('canvas')
-            offscreenCanvas.width = width
-            offscreenCanvas.height = height
-            offscreenCtx = offscreenCanvas.getContext('2d')!
-            cachedDashedLinesKey = ''
-        }
-        return { canvas: offscreenCanvas, ctx: offscreenCtx! }
-    }
+    ctx.restore()
+  }
+
+  function buildWMSRCacheKey(
+    range: { start: number; end: number },
+    kLineCenters: number[],
+    pane: RenderContext['pane'],
+    params: WMSRRenderState['params'],
+    stateTimestamp: number,
+  ): string {
+    const dr = pane.yAxis.getDisplayRange()
+    return [
+      stateTimestamp,
+      range.start,
+      range.end,
+      kLineCenters.length,
+      kLineCenters[0]?.toFixed(2) ?? 'n',
+      kLineCenters[kLineCenters.length - 1]?.toFixed(2) ?? 'n',
+      dr.maxPrice.toFixed(6),
+      dr.minPrice.toFixed(6),
+      pane.yAxis.getPriceOffset().toFixed(6),
+      pane.yAxis.getScaleType(),
+      pane.height.toFixed(2),
+      params.showWMSR,
+      params.period,
+    ].join('|')
+  }
+
+  return {
+    name: `wmsr_${paneId}`,
+    version: '2.1.0',
+    description: 'WMSR 威廉指标渲染器（WebGL + Canvas2D 回退）',
+    debugName: 'WMSR',
+    paneId: paneId,
+    priority: RENDERER_PRIORITY.MAIN,
+
+    onInstall(host: PluginHost) {
+      pluginHost = host
+    },
 
-    function buildDashedLinesKey(
-        paneWidth: number,
-        paneHeight: number,
-        displayMin: number,
-        displayMax: number,
-        dpr: number
-    ): string {
-        return `${paneWidth}|${paneHeight}|${displayMin.toFixed(4)}|${displayMax.toFixed(4)}|${dpr}`
-    }
+    getDeclaredNamespaces() {
+      const key = resolveKey()
+      return key ? [key] : []
+    },
 
-    function renderDashedLinesToOffscreen(
-        ctx: CanvasRenderingContext2D,
-        paneWidth: number,
-        paneHeight: number,
-        displayMin: number,
-        displayMax: number,
-        dpr: number,
-        colors: { wmsr: { overbought: string; oversold: string }; wmsrGrid: string }
-    ): void {
-        const displayValueRange = displayMax - displayMin || 1
-        const y20 = alignToPhysicalPixelCenter(paneHeight - (-20 - displayMin) / displayValueRange * paneHeight, dpr)
-        const y80 = alignToPhysicalPixelCenter(paneHeight - (-80 - displayMin) / displayValueRange * paneHeight, dpr)
-        const y50 = alignToPhysicalPixelCenter(paneHeight - (-50 - displayMin) / displayValueRange * paneHeight, dpr)
-
-        ctx.clearRect(0, 0, ctx.canvas.width, ctx.canvas.height)
-        ctx.save()
-        ctx.scale(dpr, dpr)
-        ctx.lineWidth = 1
-        ctx.setLineDash([4, 4])
-
-        ctx.strokeStyle = colors.wmsr.overbought
-        ctx.beginPath()
-        ctx.moveTo(0, y20)
-        ctx.lineTo(paneWidth, y20)
-        ctx.stroke()
-
-        ctx.strokeStyle = colors.wmsr.oversold
-        ctx.beginPath()
-        ctx.moveTo(0, y80)
-        ctx.lineTo(paneWidth, y80)
-        ctx.stroke()
-
-        ctx.strokeStyle = colors.wmsrGrid
-        ctx.beginPath()
-        ctx.moveTo(0, y50)
-        ctx.lineTo(paneWidth, y50)
-        ctx.stroke()
-
-        ctx.restore()
-    }
+    draw(context: RenderContext) {
+      const { ctx, pane, range, scrollLeft, dpr, kLineCenters, lineWebGLSurface } = context
+      const colors = resolveThemeColors(
+        context.theme,
+        context.isAsiaMarket,
+        context.colorPresetSettings,
+      )
+
+      const stateKey = resolveKey()
+      if (!stateKey) return
+      const state = pluginHost?.getSharedState<WMSRRenderState>(stateKey)
+      if (!state || state.visibleMin > state.visibleMax) {
+        clearLineCache()
+        return
+      }
+
+      const { valueMin, valueMax, params, series } = state
+
+      const displayRange = pane.yAxis.getDisplayRange({ minPrice: valueMin, maxPrice: valueMax })
+      const displayMin = displayRange.minPrice
+      const displayMax = displayRange.maxPrice
+      const displayValueRange = displayMax - displayMin || 1
+
+      const paneWidth = context.paneWidth
+      const paneHeight = pane.height
+      const dashedLinesKey = buildDashedLinesKey(paneWidth, paneHeight, displayMin, displayMax, dpr)
+
+      if (cachedDashedLinesKey !== dashedLinesKey) {
+        cachedDashedLinesKey = dashedLinesKey
+        const { ctx: offCtx } = getOffscreenCanvas(
+          Math.ceil(paneWidth * dpr),
+          Math.ceil(paneHeight * dpr),
+        )
+        renderDashedLinesToOffscreen(
+          offCtx,
+          paneWidth,
+          paneHeight,
+          displayMin,
+          displayMax,
+          dpr,
+          colors,
+        )
+      }
+
+      if (offscreenCanvas) {
+        ctx.drawImage(offscreenCanvas, 0, 0, paneWidth, paneHeight)
+      }
+
+      // 确定绘制范围
+      const drawStart = Math.max(range.start, params.period - 1)
+      const drawEnd = Math.min(range.end, series.length)
+
+      // 更新线条缓存
+      const cacheKey = buildWMSRCacheKey(range, kLineCenters, pane, params, state.timestamp)
+      if (cachedKey !== cacheKey) {
+        cachedKey = cacheKey
+
+        const paneH = paneHeight
+        const invRange = paneH / displayValueRange
+        const rangeStart = range.start
+
+        if (params.showWMSR) {
+          const points: LinePoint[] = []
+          for (let i = drawStart; i < drawEnd; i++) {
+            const value = series[i]
+            if (value === undefined) continue
+
+            const centerX = kLineCenters[i - rangeStart]
+            if (centerX === undefined) continue
+
+            points.push({ x: centerX, y: paneH - (value - displayMin) * invRange })
+          }
+          cachedWMSRPoints = points
+        } else {
+          cachedWMSRPoints = []
+        }
+      }
+
+      // 绘制 WMSR 线（WebGL 优先，Canvas2D 回退）
+      const enableWebGL = context.settings?.enableWebGLRendering !== false
+      let usedWebGL = false
+      if (enableWebGL && lineWebGLSurface?.isAvailable()) {
+        if (params.showWMSR && cachedWMSRPoints.length >= 2) {
+          const ok = lineWebGLSurface.drawLineStrips(
+            [{ points: cachedWMSRPoints, width: 1, color: colors.wmsr.wmsr }],
+            scrollLeft,
+          )
+          if (ok) {
+            usedWebGL = true
+            lineWebGLSurface.compositeTo(ctx, { imageSmoothingEnabled: false })
+          }
+        }
+      }
 
-    function buildWMSRCacheKey(
-        range: { start: number; end: number },
-        kLineCenters: number[],
-        pane: RenderContext['pane'],
-        params: WMSRRenderState['params'],
-        stateTimestamp: number
-    ): string {
-        const dr = pane.yAxis.getDisplayRange()
-        return [
-            stateTimestamp,
-            range.start,
-            range.end,
-            kLineCenters.length,
-            kLineCenters[0]?.toFixed(2) ?? 'n',
-            kLineCenters[kLineCenters.length - 1]?.toFixed(2) ?? 'n',
-            dr.maxPrice.toFixed(6),
-            dr.minPrice.toFixed(6),
-            pane.yAxis.getPriceOffset().toFixed(6),
-            pane.yAxis.getScaleType(),
-            pane.height.toFixed(2),
-            params.showWMSR,
-            params.period,
-        ].join('|')
-    }
+      if (!usedWebGL) {
+        drawWMSRLineWithCanvas2D(ctx, scrollLeft, cachedWMSRPoints, params, colors)
+      }
+    },
 
-    return {
-        name: `wmsr_${paneId}`,
-        version: '2.1.0',
-        description: 'WMSR 威廉指标渲染器（WebGL + Canvas2D 回退）',
-        debugName: 'WMSR',
-        paneId: paneId,
-        priority: RENDERER_PRIORITY.MAIN,
-
-        onInstall(host: PluginHost) {
-            pluginHost = host
-        },
-
-        getDeclaredNamespaces() {
-            const key = resolveKey()
-            return key ? [key] : []
-        },
-
-        draw(context: RenderContext) {
-            const { ctx, pane, range, scrollLeft, dpr, kLineCenters, lineWebGLSurface } = context
-            const colors = resolveThemeColors(context.theme, context.isAsiaMarket, context.colorPresetSettings)
-
-            const stateKey = resolveKey()
-            if (!stateKey) return
-            const state = pluginHost?.getSharedState<WMSRRenderState>(stateKey)
-            if (!state || state.visibleMin > state.visibleMax) {
-                clearLineCache()
-                return
-            }
-
-            const { valueMin, valueMax, params, series } = state
-
-            const displayRange = pane.yAxis.getDisplayRange({ minPrice: valueMin, maxPrice: valueMax })
-            const displayMin = displayRange.minPrice
-            const displayMax = displayRange.maxPrice
-            const displayValueRange = displayMax - displayMin || 1
-
-            const paneWidth = context.paneWidth
-            const paneHeight = pane.height
-            const dashedLinesKey = buildDashedLinesKey(paneWidth, paneHeight, displayMin, displayMax, dpr)
-
-            if (cachedDashedLinesKey !== dashedLinesKey) {
-                cachedDashedLinesKey = dashedLinesKey
-                const { ctx: offCtx } = getOffscreenCanvas(
-                    Math.ceil(paneWidth * dpr),
-                    Math.ceil(paneHeight * dpr)
-                )
-                renderDashedLinesToOffscreen(offCtx, paneWidth, paneHeight, displayMin, displayMax, dpr, colors)
-            }
-
-            if (offscreenCanvas) {
-                ctx.drawImage(offscreenCanvas, 0, 0, paneWidth, paneHeight)
-            }
-
-            // 确定绘制范围
-            const drawStart = Math.max(range.start, params.period - 1)
-            const drawEnd = Math.min(range.end, series.length)
-
-            // 更新线条缓存
-            const cacheKey = buildWMSRCacheKey(range, kLineCenters, pane, params, state.timestamp)
-            if (cachedKey !== cacheKey) {
-                cachedKey = cacheKey
-
-                const paneH = paneHeight
-                const invRange = paneH / displayValueRange
-                const rangeStart = range.start
-
-                if (params.showWMSR) {
-                    const points: LinePoint[] = []
-                    for (let i = drawStart; i < drawEnd; i++) {
-                        const value = series[i]
-                        if (value === undefined) continue
-
-                        const centerX = kLineCenters[i - rangeStart]
-                        if (centerX === undefined) continue
-
-                        points.push({ x: centerX, y: paneH - (value - displayMin) * invRange })
-                    }
-                    cachedWMSRPoints = points
-                } else {
-                    cachedWMSRPoints = []
-                }
-            }
-
-            // 绘制 WMSR 线（WebGL 优先，Canvas2D 回退）
-            const enableWebGL = context.settings?.enableWebGLRendering !== false
-            let usedWebGL = false
-            if (enableWebGL && lineWebGLSurface?.isAvailable()) {
-                if (params.showWMSR && cachedWMSRPoints.length >= 2) {
-                    const ok = lineWebGLSurface.drawLineStrips(
-                        [{ points: cachedWMSRPoints, width: 1, color: colors.wmsr.wmsr }],
-                        scrollLeft
-                    )
-                    if (ok) {
-                        usedWebGL = true
-                        lineWebGLSurface.compositeTo(ctx, { imageSmoothingEnabled: false })
-                    }
-                }
-            }
-
-            if (!usedWebGL) {
-                drawWMSRLineWithCanvas2D(ctx, scrollLeft, cachedWMSRPoints, params, colors)
-            }
-        },
-
-        getConfig() {
-            const stateKey = resolveKey()
-            if (!stateKey) return {}
-            const state = pluginHost?.getSharedState<WMSRRenderState>(stateKey)
-            return state?.params ?? {}
-        },
-
-        setConfig() {
-            // no-op: 配置通过 scheduler.updateIndicatorConfig() 更新
-        },
-    }
+    getConfig() {
+      const stateKey = resolveKey()
+      if (!stateKey) return {}
+      const state = pluginHost?.getSharedState<WMSRRenderState>(stateKey)
+      return state?.params ?? {}
+    },
+
+    setConfig() {
+      // no-op: 配置通过 scheduler.updateIndicatorConfig() 更新
+    },
+  }
 }
 
 /**
  * 使用 Canvas 2D 绘制 WMSR 线（WebGL 回退）
  */
 function drawWMSRLineWithCanvas2D(
-    ctx: CanvasRenderingContext2D,
-    scrollLeft: number,
-    wmsrPoints: LinePoint[],
-    params: { showWMSR: boolean },
-    colors: { wmsr: { wmsr: string } }
+  ctx: CanvasRenderingContext2D,
+  scrollLeft: number,
+  wmsrPoints: LinePoint[],
+  params: { showWMSR: boolean },
+  colors: { wmsr: { wmsr: string } },
 ): void {
-    if (!params.showWMSR || wmsrPoints.length < 2) return
-
-    ctx.save()
-    ctx.translate(-scrollLeft, 0)
-    ctx.strokeStyle = colors.wmsr.wmsr
-    ctx.lineWidth = 1
-    ctx.lineJoin = 'round'
-    ctx.lineCap = 'round'
-    ctx.beginPath()
-    ctx.moveTo(wmsrPoints[0]!.x, wmsrPoints[0]!.y)
-    for (let i = 1; i < wmsrPoints.length; i++) {
-        const point = wmsrPoints[i]!
-        ctx.lineTo(point.x, point.y)
-    }
-    ctx.stroke()
-    ctx.restore()
+  if (!params.showWMSR || wmsrPoints.length < 2) return
+
+  ctx.save()
+  ctx.translate(-scrollLeft, 0)
+  ctx.strokeStyle = colors.wmsr.wmsr
+  ctx.lineWidth = 1
+  ctx.lineJoin = 'round'
+  ctx.lineCap = 'round'
+  ctx.beginPath()
+  ctx.moveTo(wmsrPoints[0]!.x, wmsrPoints[0]!.y)
+  for (let i = 1; i < wmsrPoints.length; i++) {
+    const point = wmsrPoints[i]!
+    ctx.lineTo(point.x, point.y)
+  }
+  ctx.stroke()
+  ctx.restore()
 }
 
 const getWMSRTitleInfo = createSingleLineTitleInfo({
-    createStateKey: createWMSRStateKey,
-    name: 'WMSR',
-    defaultPeriod: 14,
-    getColor: (colors) => colors.wmsr.wmsr,
+  createStateKey: createWMSRStateKey,
+  name: 'WMSR',
+  defaultPeriod: 14,
+  getColor: (colors) => colors.wmsr.wmsr,
 })
 
 @Indicator({
-    name: 'wmsr',
-    displayName: 'WMSR',
-    category: 'oscillator',
-    defaultPaneId: 'sub_WMSR',
-    visibleState: { compose: createFixedRangeSparseVisibleStateComposer('wmsr', EMPTY_WMSR_STATE) },
-    scaleRendererFactory: createWmsrScaleRendererPlugin,
-    getTitleInfo: getWMSRTitleInfo,
-    runtime: {
-        defaultConfig: { period: 14, showWMSR: true },
-        computeKey: 'calcWMSRData',
-        compute: (data, c) => calcWMSRData(data, c.period),
-    },
+  name: 'wmsr',
+  displayName: 'WMSR',
+  category: 'oscillator',
+  defaultPaneId: 'sub_WMSR',
+  visibleState: { compose: createFixedRangeSparseVisibleStateComposer('wmsr', EMPTY_WMSR_STATE) },
+  scaleRendererFactory: createWmsrScaleRendererPlugin,
+  getTitleInfo: getWMSRTitleInfo,
+  runtime: {
+    defaultConfig: { period: 14, showWMSR: true },
+    computeKey: 'calcWMSRData',
+    compute: (data, c) => calcWMSRData(data, c.period),
+  },
 })
 class WMSRIndicatorDefinition {
-    static rendererFactory = createWMSRRendererPlugin
+  static rendererFactory = createWMSRRendererPlugin
 }