@363045841yyt/klinechart-core 0.8.1-alpha.3 → 0.8.1

package/src/engine/render/chartRenderer.ts

@@ -0,0 +1,579 @@
+import type { KLineData } from '../../types/price'
+import type { ChartSettings } from '../../config/chartSettings'
+import type { SymbolSpec } from '../../controllers/types'
+import { getVisibleRange } from '../viewport/viewport'
+import type { RendererPlugin, RendererPluginWithHost, PluginHostImpl, RenderContext, YAxisLabel, XAxisLabel, YAxisRange, XAxisRange } from '../../plugin'
+import { RendererPluginManager, wrapPaneInfo } from '../../plugin'
+import type { ChartDom, PaneSpec, ChartOptions, KLinePositions, Viewport, ViewportState } from '../chartTypes'
+import { PaneRenderer } from '../paneRenderer'
+import { SharedWebGLSurface } from '../renderers/webgl/sharedWebGLSurface'
+import { MarkerManager, type CustomMarkerEntity } from '../marker/registry'
+import { getPhysicalKLineConfig } from '../utils/klineConfig'
+import { ChartViewportManager } from '../viewport/chartViewportManager'
+import { ChartDataManager } from '../data/chartDataManager'
+import { ChartIndicatorManager } from '../indicators/chartIndicatorManager'
+import { InteractionController } from '../controller/interaction'
+import { UpdateLevel } from '../layout/pane'
+import type { VisibleRange } from '../layout/pane'
+import { DrawingStore } from '../drawing'
+import { createMainIndicatorLegendRendererPlugin } from '../renderers/Indicator/mainIndicatorLegend'
+import { createDrawingRendererPlugin, createDrawingLabelOverlayPlugin } from '../drawing/plugin'
+import { createGridLinesRendererPlugin } from '../renderers/gridLines'
+import { createCandleRenderer } from '../renderers/candle'
+import { createComparisonLineRenderer } from '../renderers/comparisonLine'
+import { createLastPriceLineRendererPlugin, createLastPriceLabelRegistrarPlugin } from '../renderers/lastPrice'
+import { createCustomMarkersRenderer } from '../renderers/customMarkers'
+import { createExtremaMarkersRendererPlugin } from '../renderers/extremaMarkers'
+import { createYAxisRendererPlugin } from '../renderers/yAxis'
+import { createCrosshairRendererPlugin } from '../renderers/crosshair'
+import { createTimeAxisRendererPlugin } from '../renderers/timeAxis'
+
+type ResolvedChartOptions = Omit<ChartOptions, 'kWidth' | 'kGap'> & {
+  kWidth: number
+  kGap: number
+}
+
+export type FrameContext = {
+  vp: Viewport
+  range: VisibleRange
+  kLinePositions: KLinePositions
+  kLineCenters: number[]
+  kBarRects: Array<{ x: number; width: number }>
+  kWidthPx: number
+  useCachedFrame: boolean
+  data: KLineData[]
+  mainIndicatorRange: { min: number; max: number } | null
+  hasCrosshair: boolean
+  zoomLevel: number
+  zoomLevelCount: number
+}
+
+export interface RendererDependencies {
+  getDom: () => ChartDom
+  getOption: () => ResolvedChartOptions
+  getPaneRenderers: () => PaneRenderer[]
+  getInteraction: () => InteractionController
+  getSharedWebGLSurface: () => SharedWebGLSurface
+  getPluginHost: () => PluginHostImpl
+  getRendererPluginManager: () => RendererPluginManager
+  getTheme: () => 'light' | 'dark'
+  getCurrentZoomLevel: () => number
+  getZoomLevelCount: () => number
+  getViewportManager: () => ChartViewportManager
+  getDataManager: () => ChartDataManager
+  getIndicatorManager: () => ChartIndicatorManager
+}
+
+export class ChartRenderer {
+  private deps: RendererDependencies
+
+  private raf: number | null = null
+  private pendingUpdateLevel: UpdateLevel = UpdateLevel.All
+
+  readonly markerManager: MarkerManager
+  readonly drawingStore: DrawingStore
+  private settings: ChartSettings = {}
+  private overlayHadCrosshair = false
+  private xAxisCtx: CanvasRenderingContext2D | null = null
+
+  private cachedDrawFrame: {
+    viewport: Viewport
+    range: VisibleRange
+    kLinePositions: KLinePositions
+    kLineCenters: number[]
+    kBarRects: Array<{ x: number; width: number }>
+    kWidthPx: number
+  } | null = null
+
+  constructor(deps: RendererDependencies) {
+    this.deps = deps
+    this.markerManager = new MarkerManager()
+    this.drawingStore = new DrawingStore()
+  }
+
+  initCoreRenderers(): void {
+    const opt = this.deps.getOption()
+    const axisWidth = opt.rightAxisWidth + (opt.priceLabelWidth ?? 0)
+    const interaction = this.deps.getInteraction()
+
+    this.useDrawingPlugin(createGridLinesRendererPlugin())
+    this.useDrawingPlugin(createCandleRenderer())
+    this.useDrawingPlugin(createComparisonLineRenderer())
+    this.useDrawingPlugin(createLastPriceLineRendererPlugin())
+    this.useDrawingPlugin(createLastPriceLabelRegistrarPlugin())
+    this.useDrawingPlugin(createCustomMarkersRenderer())
+    this.useDrawingPlugin(createExtremaMarkersRendererPlugin())
+    this.useDrawingPlugin(createMainIndicatorLegendRendererPlugin({
+      yPaddingPx: opt.yPaddingPx,
+    }))
+    this.useDrawingPlugin(createYAxisRendererPlugin({
+      axisWidth,
+      yPaddingPx: opt.yPaddingPx,
+      getCrosshair: () => {
+        const pos = interaction.crosshairPos
+        const price = interaction.crosshairPrice
+        const activePaneId = interaction.activePaneId
+        if (pos && price !== null) {
+          return { y: pos.y, price, activePaneId }
+        }
+        return null
+      },
+    }))
+    this.useDrawingPlugin(createCrosshairRendererPlugin({
+      getCrosshairState: () => ({
+        pos: interaction.crosshairPos,
+        activePaneId: interaction.activePaneId,
+        isDragging: interaction.isDraggingState(),
+        price: interaction.crosshairPrice,
+      }),
+    }))
+    this.useDrawingPlugin(createTimeAxisRendererPlugin({
+      height: opt.bottomAxisHeight,
+      getCrosshair: () => {
+        const pos = interaction.crosshairPos
+        const idx = interaction.crosshairIndex
+        if (pos && idx !== null) {
+          return { x: pos.x, index: idx }
+        }
+        return null
+      },
+    }))
+  }
+
+  registerDrawingPlugins(): void {
+    this.useDrawingPlugin(createDrawingRendererPlugin({ store: this.drawingStore }))
+    this.useDrawingPlugin(createDrawingLabelOverlayPlugin({ store: this.drawingStore }))
+  }
+
+  private useDrawingPlugin(plugin: RendererPlugin | RendererPluginWithHost, config?: Record<string, unknown>): void {
+    this.deps.getRendererPluginManager().register(plugin)
+    if (config && plugin.setConfig) {
+      plugin.setConfig(config)
+    }
+  }
+
+  getMarkerManager(): MarkerManager {
+    return this.markerManager
+  }
+
+  getDrawingStore(): DrawingStore {
+    return this.drawingStore
+  }
+
+  getSettings(): ChartSettings {
+    return this.settings
+  }
+
+  updateSettings(settings: ChartSettings): void {
+    this.settings = { ...settings }
+  }
+
+  scheduleDraw(level: UpdateLevel = UpdateLevel.All): void {
+    if (this.raf !== null) {
+      if (this.pendingUpdateLevel === UpdateLevel.All) return
+      if (level === UpdateLevel.All) {
+        this.pendingUpdateLevel = UpdateLevel.All
+        return
+      }
+      if (
+        (this.pendingUpdateLevel === UpdateLevel.Main && level === UpdateLevel.Overlay) ||
+        (this.pendingUpdateLevel === UpdateLevel.Overlay && level === UpdateLevel.Main)
+      ) {
+        this.pendingUpdateLevel = UpdateLevel.All
+        return
+      }
+      return
+    }
+
+    this.pendingUpdateLevel = level
+    this.raf = requestAnimationFrame(() => {
+      this.raf = null
+      const levelToDraw = this.pendingUpdateLevel
+      this.pendingUpdateLevel = UpdateLevel.All
+      this.draw(levelToDraw)
+      const c = this.deps.getDom().container
+      if (c) {
+        this.deps.getViewportManager().applyPendingScrollLeft(c)
+      }
+    })
+  }
+
+  draw(level: UpdateLevel = UpdateLevel.All): void {
+    this.markerManager.clear()
+
+    const frame = this.prepareFrameData(level)
+    if (!frame) {
+      if (this.deps.getDataManager().getInternalData().length === 0) this.clearAllCanvases()
+      return
+    }
+
+    const { vp, range, kLinePositions, kLineCenters, kBarRects, kWidthPx, useCachedFrame } = frame
+
+    this.deps.getInteraction().setKLinePositions(kLinePositions, range, kWidthPx)
+
+    const indicatorManager = this.deps.getIndicatorManager()
+    indicatorManager.indicatorSchedulerAccessor.setActiveMainIndicators(
+      [...indicatorManager.mainIndicatorsSignalPeek.entries()].map(([id, entry]) => ({ id, params: entry.params })),
+    )
+    const mainIndicatorRange = useCachedFrame ? null : indicatorManager.indicatorSchedulerAccessor.getMainIndicatorPriceRange()
+    const hasCrosshair = this.deps.getInteraction().getCrosshairIndex() !== null
+
+    const { sharedXAxisLabels, sharedXAxisRanges } = this.renderPanes(
+      vp, range, kLinePositions, kLineCenters, kBarRects, kWidthPx,
+      mainIndicatorRange, hasCrosshair, useCachedFrame, level,
+    )
+
+    this.overlayHadCrosshair = hasCrosshair
+    this.renderXAxis(vp, range, kLinePositions, kLineCenters, kBarRects, kWidthPx, sharedXAxisLabels, sharedXAxisRanges)
+  }
+
+  private prepareFrameData(level: UpdateLevel): FrameContext | null {
+    const useCachedFrame = level === UpdateLevel.Overlay && this.cachedDrawFrame !== null
+
+    const vp = useCachedFrame ? this.cachedDrawFrame!.viewport : this.deps.getViewportManager().computeViewport()
+    if (!vp) return null
+
+    const internalData = this.deps.getDataManager().getInternalData()
+    if (internalData.length === 0) return null
+
+    const opt = this.deps.getOption()
+    const rawRange = useCachedFrame
+      ? this.cachedDrawFrame!.range
+      : (() => {
+        const { start, end } = getVisibleRange(
+          vp.scrollLeft,
+          vp.plotWidth,
+          opt.kWidth,
+          opt.kGap,
+          internalData.length,
+          vp.dpr,
+        )
+        return { start, end }
+      })()
+    const range = { start: Math.max(0, rawRange.start), end: rawRange.end }
+
+    const dataManager = this.deps.getDataManager()
+    if (!useCachedFrame && (
+      range.start !== dataManager.lastVisibleRange.start
+      || range.end !== dataManager.lastVisibleRange.end
+      || rawRange.start !== dataManager.lastRawVisibleRange.start
+      || rawRange.end !== dataManager.lastRawVisibleRange.end
+    )) {
+      this.deps.getIndicatorManager().indicatorSchedulerAccessor.updateVisibleRange(range)
+      dataManager.lastVisibleRange = range
+      dataManager.lastRawVisibleRange = rawRange
+      this.checkVisibleRangeGapWhenIdle()
+    }
+
+    const kLinePositions = useCachedFrame
+      ? this.cachedDrawFrame!.kLinePositions
+      : this.calcKLinePositions(range)
+
+    let kLineCenters: number[]
+    let kBarRects: Array<{ x: number; width: number }>
+    let kWidthPx: number
+
+    if (useCachedFrame) {
+      kLineCenters = this.cachedDrawFrame!.kLineCenters
+      kBarRects = this.cachedDrawFrame!.kBarRects
+      kWidthPx = this.cachedDrawFrame!.kWidthPx
+    } else {
+      const physConfig = getPhysicalKLineConfig(opt.kWidth, opt.kGap, vp.dpr)
+      let barWidthPx = Math.max(1, physConfig.unitPx - 1)
+      if (barWidthPx % 2 === 0) barWidthPx -= 1
+
+      kLineCenters = new Array(kLinePositions.length)
+      kBarRects = new Array(kLinePositions.length)
+
+      for (let i = 0; i < kLinePositions.length; i++) {
+        const x = kLinePositions[i]!
+        const leftPx = Math.round(x * vp.dpr)
+        const wickXPx = leftPx + (physConfig.kWidthPx - 1) / 2
+        kLineCenters[i] = wickXPx / vp.dpr
+
+        const barLeftPx = wickXPx - (barWidthPx - 1) / 2
+        kBarRects[i] = { x: barLeftPx / vp.dpr, width: barWidthPx / vp.dpr }
+      }
+
+      kWidthPx = getPhysicalKLineConfig(opt.kWidth, opt.kGap, vp.dpr).kWidthPx
+      this.cachedDrawFrame = {
+        viewport: { ...vp },
+        range: { ...range },
+        kLinePositions,
+        kLineCenters,
+        kBarRects,
+        kWidthPx,
+      }
+    }
+
+    return {
+      vp, range, kLinePositions, kLineCenters, kBarRects, kWidthPx, useCachedFrame,
+      data: internalData,
+      mainIndicatorRange: null,
+      hasCrosshair: false,
+      zoomLevel: this.deps.getCurrentZoomLevel(),
+      zoomLevelCount: this.deps.getZoomLevelCount(),
+    }
+  }
+
+  clearAllCanvases(): void {
+    const vp = this.deps.getViewportManager().computeViewport()
+    if (!vp) return
+    for (const r of this.deps.getPaneRenderers()) {
+      const { mainCtx, overlayCtx, yAxisCtx } = r.getContexts()
+      const pane = r.getPane()
+      mainCtx?.clearRect(0, 0, vp.plotWidth + 1, pane.height + 2 / vp.dpr)
+      overlayCtx?.clearRect(0, 0, vp.plotWidth + 1, pane.height + 2 / vp.dpr)
+      yAxisCtx?.clearRect(0, 0, vp.plotWidth + 1, pane.height + 2 / vp.dpr)
+    }
+    const xCtx = this.xAxisCtx
+    if (xCtx) {
+      const xW = xCtx.canvas.width
+      const xH = xCtx.canvas.height
+      xCtx.clearRect(0, 0, xW, xH)
+    }
+  }
+
+  private renderPanes(
+    vp: Viewport,
+    range: VisibleRange,
+    kLinePositions: KLinePositions,
+    kLineCenters: number[],
+    kBarRects: Array<{ x: number; width: number }>,
+    kWidthPx: number,
+    mainIndicatorRange: { min: number; max: number } | null,
+    hasCrosshair: boolean,
+    useCachedFrame: boolean,
+    level: UpdateLevel,
+  ): { sharedXAxisLabels: XAxisLabel[]; sharedXAxisRanges: XAxisRange[] } {
+    const sharedYAxisLabels: YAxisLabel[] = []
+    const sharedXAxisLabels: XAxisLabel[] = []
+    const sharedYAxisRanges: YAxisRange[] = []
+    const sharedXAxisRanges: XAxisRange[] = []
+
+    const dataManager = this.deps.getDataManager()
+    const rendererPluginManager = this.deps.getRendererPluginManager()
+    const pluginHost = this.deps.getPluginHost()
+
+    for (const renderer of this.deps.getPaneRenderers()) {
+      const pane = renderer.getPane()
+      const { mainCtx, overlayCtx, yAxisCtx } = renderer.getContexts()
+      const { candleSurface, lineSurface } = renderer.getWebGL()
+
+      if (!useCachedFrame) {
+        const indicatorRange = pane.role === 'price' ? mainIndicatorRange : null
+        const comparisonRange = pane.id === 'main' ? dataManager.getComparisonEquivalentPriceRange(range) : null
+        const mergedRange = this.mergeNumericRanges(indicatorRange, comparisonRange)
+        pane.updateRange(dataManager.getInternalData(), range, mergedRange)
+        if (pane.id === 'main' && this.settings.disableMainPaneVerticalScroll) {
+          pane.yAxis.resetTransform()
+        }
+      }
+
+      const shouldUpdateMain = level === UpdateLevel.Main || level === UpdateLevel.All
+      const shouldUpdateOverlay = level === UpdateLevel.All || (level === UpdateLevel.Overlay && (hasCrosshair || this.overlayHadCrosshair))
+
+      if (shouldUpdateMain && mainCtx) {
+        mainCtx.setTransform(1, 0, 0, 1, 0, 0)
+        mainCtx.scale(vp.dpr, vp.dpr)
+        mainCtx.clearRect(0, 0, vp.plotWidth + 1, pane.height + 2 / vp.dpr)
+        candleSurface?.clear()
+        lineSurface?.clear()
+      }
+
+      if (shouldUpdateOverlay && overlayCtx) {
+        const overlayWidth = overlayCtx.canvas.width / vp.dpr
+        overlayCtx.setTransform(1, 0, 0, 1, 0, 0)
+        overlayCtx.scale(vp.dpr, vp.dpr)
+        overlayCtx.clearRect(0, 0, overlayWidth + 1, pane.height + 2 / vp.dpr)
+      }
+
+      if (yAxisCtx && !useCachedFrame) {
+        const yAxisWidth = yAxisCtx.canvas.width / vp.dpr
+        yAxisCtx.setTransform(1, 0, 0, 1, 0, 0)
+        yAxisCtx.scale(vp.dpr, vp.dpr)
+        yAxisCtx.clearRect(0, 0, yAxisWidth, pane.height + 2 / vp.dpr)
+      }
+
+      const opt = this.deps.getOption()
+      const context: RenderContext = {
+        ctx: mainCtx!,
+        overlayCtx: overlayCtx ?? undefined,
+        pane: wrapPaneInfo(pane),
+        data: dataManager.getInternalData(),
+        comparisonData: dataManager.getComparisonData(),
+        comparisonSymbols: dataManager.getComparisonSpecs(),
+        comparisonColors: dataManager.getComparisonColors(),
+        range,
+        scrollLeft: vp.scrollLeft,
+        kWidth: opt.kWidth,
+        kGap: opt.kGap,
+        dpr: vp.dpr,
+        paneWidth: vp.plotWidth,
+        kLinePositions,
+        kLineCenters,
+        kBarRects,
+        markerManager: this.markerManager,
+        crosshairIndex: this.deps.getInteraction().getCrosshairIndex(),
+        yAxisCtx: yAxisCtx ?? undefined,
+        candleWebGLSurface: candleSurface ?? undefined,
+        lineWebGLSurface: lineSurface ?? undefined,
+        zoomLevel: this.deps.getCurrentZoomLevel(),
+        zoomLevelCount: this.deps.getZoomLevelCount(),
+        viewport: {
+          scrollLeft: vp.scrollLeft,
+          plotWidth: vp.plotWidth,
+          plotHeight: vp.plotHeight,
+        },
+        settings: this.settings,
+        yAxisLabels: sharedYAxisLabels,
+        xAxisLabels: sharedXAxisLabels,
+        yAxisRanges: sharedYAxisRanges,
+        xAxisRanges: sharedXAxisRanges,
+        theme: this.deps.getTheme(),
+        isAsiaMarket: this.settings.isAsiaMarket as boolean,
+        colorPresetSettings: this.settings.colorPresetSettings,
+      }
+
+      if (shouldUpdateMain || shouldUpdateOverlay) {
+        const errors = rendererPluginManager.render(pane.id, context, level)
+        if (errors.length > 0) {
+          pluginHost.events.emit('renderer:error', { paneId: pane.id, errors })
+        }
+
+        const yAxisErrors = rendererPluginManager.renderPlugin('yAxis', context)
+        if (yAxisErrors.length > 0) {
+          pluginHost.events.emit('renderer:error', { paneId: pane.id, errors: yAxisErrors })
+        }
+      }
+    }
+
+    return { sharedXAxisLabels, sharedXAxisRanges }
+  }
+
+  private renderXAxis(
+    vp: Viewport,
+    range: VisibleRange,
+    kLinePositions: KLinePositions,
+    kLineCenters: number[],
+    kBarRects: Array<{ x: number; width: number }>,
+    kWidthPx: number,
+    sharedXAxisLabels: XAxisLabel[],
+    sharedXAxisRanges: XAxisRange[],
+  ): void {
+    const dom = this.deps.getDom()
+    const xAxisCtx = this.xAxisCtx ?? dom.xAxisCanvas.getContext('2d')
+    if (!this.xAxisCtx) {
+      this.xAxisCtx = xAxisCtx
+    }
+    if (xAxisCtx) {
+      const opt = this.deps.getOption()
+      const timeAxisContext: RenderContext = {
+        ctx: xAxisCtx,
+        pane: {
+          id: 'xAxis',
+          role: 'auxiliary',
+          capabilities: {
+            showPriceAxisTicks: false,
+            showCrosshairPriceLabel: false,
+            candleHitTest: false,
+            supportsPriceTranslate: false,
+          },
+          top: 0,
+          height: opt.bottomAxisHeight,
+          yAxis: {
+            priceToY: () => 0,
+            yToPrice: () => 0,
+            getPaddingTop: () => 0,
+            getPaddingBottom: () => 0,
+            getPriceOffset: () => 0,
+            getDisplayRange: (baseRange) => baseRange ?? { maxPrice: 0, minPrice: 0 },
+            getScaleType: () => 'linear' as const,
+            getBasePrice: () => null,
+            toPercent: () => 0,
+            fromPercent: () => 0,
+            getDisplayPercentRange: () => ({ minPct: 0, maxPct: 0 }),
+          },
+          priceRange: { maxPrice: 0, minPrice: 0 },
+        },
+        data: this.deps.getDataManager().getInternalData(),
+        range,
+        scrollLeft: vp.scrollLeft,
+        kWidth: opt.kWidth,
+        kGap: opt.kGap,
+        dpr: vp.dpr,
+        paneWidth: vp.plotWidth,
+        kLinePositions,
+        kLineCenters,
+        kBarRects,
+        xAxisCtx,
+        viewport: {
+          scrollLeft: vp.scrollLeft,
+          plotWidth: vp.plotWidth,
+          plotHeight: vp.plotHeight,
+        },
+        yAxisLabels: [],
+        xAxisLabels: sharedXAxisLabels,
+        xAxisRanges: sharedXAxisRanges,
+        theme: this.deps.getTheme(),
+        isAsiaMarket: this.settings.isAsiaMarket as boolean,
+        colorPresetSettings: this.settings.colorPresetSettings,
+      }
+      const errors = this.deps.getRendererPluginManager().renderPlugin('timeAxis', timeAxisContext)
+      if (errors.length > 0) {
+        this.deps.getPluginHost().events.emit('renderer:error', { paneId: 'timeAxis', errors })
+      }
+    }
+  }
+
+  private calcKLinePositions(range: VisibleRange): KLinePositions {
+    const { start, end } = range
+    const count = end - start
+
+    if (count <= 0) return []
+
+    const dpr = this.deps.getViewportManager().getEffectiveDpr()
+    const opt = this.deps.getOption()
+    const { unitPx, startXPx } = getPhysicalKLineConfig(opt.kWidth, opt.kGap, dpr)
+
+    const positions: number[] = new Array(count)
+
+    for (let i = 0; i < count; i++) {
+      const dataIndex = start + i
+      const leftPx = startXPx + dataIndex * unitPx
+      positions[i] = leftPx / dpr
+    }
+
+    return positions
+  }
+
+  private checkVisibleRangeGapWhenIdle(): void {
+    if (this.deps.getInteraction().isPointerDown()) return
+    this.deps.getDataManager().checkVisibleRangeGap()
+  }
+
+  private mergeNumericRanges(
+    left: { min: number; max: number } | null | undefined,
+    right: { min: number; max: number } | null | undefined,
+  ): { min: number; max: number } | null {
+    if (!left) return right ?? null
+    if (!right) return left
+    return {
+      min: Math.min(left.min, right.min),
+      max: Math.max(left.max, right.max),
+    }
+  }
+
+  clearCachedFrame(): void {
+    this.cachedDrawFrame = null
+  }
+
+  destroy(): void {
+    if (this.raf !== null) {
+      cancelAnimationFrame(this.raf)
+      this.raf = null
+    }
+    this.cachedDrawFrame = null
+    this.xAxisCtx = null
+  }
+}