@363045841yyt/klinechart-core 0.7.3 → 0.7.5-alpha.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (231) hide show
  1. package/README.md +201 -201
  2. package/README.zh-CN.md +201 -201
  3. package/dist/engine/renderers/webgl/candleSurface.js +47 -47
  4. package/dist/version.d.ts +1 -1
  5. package/dist/version.d.ts.map +1 -1
  6. package/dist/version.js +1 -2
  7. package/dist/version.js.map +1 -1
  8. package/package.json +129 -122
  9. package/src/__tests__/signal.test.ts +124 -124
  10. package/src/config/chartSettings.ts +66 -66
  11. package/src/controllers/__tests__/drawing.test.ts +214 -214
  12. package/src/controllers/__tests__/indicatorSelector.test.ts +481 -481
  13. package/src/controllers/__tests__/toolbar.test.ts +225 -225
  14. package/src/controllers/createChartController.ts +665 -665
  15. package/src/controllers/createDrawingController.ts +96 -96
  16. package/src/controllers/createIndicatorSelectorController.ts +307 -307
  17. package/src/controllers/createToolbarController.ts +146 -146
  18. package/src/controllers/index.ts +19 -19
  19. package/src/controllers/types.ts +284 -284
  20. package/src/engine/__tests__/chart.dpr.test.ts +401 -401
  21. package/src/engine/__tests__/paneRenderer.resize.test.ts +92 -92
  22. package/src/engine/chart-store.ts +121 -121
  23. package/src/engine/chart.d.ts +617 -617
  24. package/src/engine/chart.ts +2815 -2815
  25. package/src/engine/controller/__tests__/interaction.dpr.test.ts +259 -259
  26. package/src/engine/controller/interaction.ts +722 -722
  27. package/src/engine/controller/markerInteraction.ts +130 -130
  28. package/src/engine/controller/pinchTracker.ts +82 -82
  29. package/src/engine/controller/tooltipPosition.ts +48 -48
  30. package/src/engine/draw/__tests__/pixelAlign.spec.ts +176 -176
  31. package/src/engine/draw/pixelAlign.ts +259 -259
  32. package/src/engine/drawing/index.ts +655 -655
  33. package/src/engine/drawing/interaction.ts +842 -842
  34. package/src/engine/drawing/plugin.ts +343 -343
  35. package/src/engine/indicators/__tests__/__fixtures__/golden/atr.json +38 -38
  36. package/src/engine/indicators/__tests__/__fixtures__/golden/dema.json +14 -14
  37. package/src/engine/indicators/__tests__/__fixtures__/golden/hma.json +14 -14
  38. package/src/engine/indicators/__tests__/__fixtures__/golden/index.ts +55 -55
  39. package/src/engine/indicators/__tests__/__fixtures__/golden/kama.json +14 -14
  40. package/src/engine/indicators/__tests__/__fixtures__/golden/tema.json +14 -14
  41. package/src/engine/indicators/__tests__/__fixtures__/golden/wma.json +40 -40
  42. package/src/engine/indicators/__tests__/__fixtures__/synthetic.ts +65 -65
  43. package/src/engine/indicators/__tests__/_propertyAssertions.ts +76 -76
  44. package/src/engine/indicators/__tests__/atr.test.ts +153 -153
  45. package/src/engine/indicators/__tests__/calculators.test.ts +614 -614
  46. package/src/engine/indicators/__tests__/cmf-mfi.test.ts +100 -100
  47. package/src/engine/indicators/__tests__/dema.test.ts +73 -73
  48. package/src/engine/indicators/__tests__/donchian.test.ts +70 -70
  49. package/src/engine/indicators/__tests__/hma.test.ts +73 -73
  50. package/src/engine/indicators/__tests__/ichimoku.test.ts +105 -105
  51. package/src/engine/indicators/__tests__/kama.test.ts +80 -80
  52. package/src/engine/indicators/__tests__/keltner.test.ts +65 -65
  53. package/src/engine/indicators/__tests__/pivot-fib.test.ts +110 -110
  54. package/src/engine/indicators/__tests__/roc.test.ts +68 -68
  55. package/src/engine/indicators/__tests__/sar.test.ts +86 -86
  56. package/src/engine/indicators/__tests__/scheduler.test.ts +831 -831
  57. package/src/engine/indicators/__tests__/soa.test.ts +533 -533
  58. package/src/engine/indicators/__tests__/structure.test.ts +110 -110
  59. package/src/engine/indicators/__tests__/supertrend.test.ts +65 -65
  60. package/src/engine/indicators/__tests__/tema.test.ts +68 -68
  61. package/src/engine/indicators/__tests__/trix.test.ts +70 -70
  62. package/src/engine/indicators/__tests__/volatility.test.ts +117 -117
  63. package/src/engine/indicators/__tests__/volume.test.ts +115 -115
  64. package/src/engine/indicators/__tests__/volumeProfile.test.ts +74 -74
  65. package/src/engine/indicators/__tests__/vwap.test.ts +69 -69
  66. package/src/engine/indicators/__tests__/wma.test.ts +112 -112
  67. package/src/engine/indicators/__tests__/zones.test.ts +95 -95
  68. package/src/engine/indicators/atrState.ts +27 -27
  69. package/src/engine/indicators/bollState.ts +51 -51
  70. package/src/engine/indicators/calculators.ts +2593 -2593
  71. package/src/engine/indicators/cciState.ts +25 -25
  72. package/src/engine/indicators/chaikinVolState.ts +32 -32
  73. package/src/engine/indicators/cmfState.ts +27 -27
  74. package/src/engine/indicators/demaState.ts +27 -27
  75. package/src/engine/indicators/donchianState.ts +43 -43
  76. package/src/engine/indicators/eneState.ts +43 -43
  77. package/src/engine/indicators/expmaState.ts +43 -43
  78. package/src/engine/indicators/fastkState.ts +25 -25
  79. package/src/engine/indicators/fibState.ts +41 -41
  80. package/src/engine/indicators/hmaState.ts +27 -27
  81. package/src/engine/indicators/hvState.ts +28 -28
  82. package/src/engine/indicators/ichimokuState.ts +70 -70
  83. package/src/engine/indicators/indicator.worker.ts +169 -169
  84. package/src/engine/indicators/indicatorDefinitionRegistry.ts +62 -62
  85. package/src/engine/indicators/indicatorMetadata.ts +110 -110
  86. package/src/engine/indicators/indicatorRegistry.ts +106 -106
  87. package/src/engine/indicators/indicatorRuntime.ts +1548 -1548
  88. package/src/engine/indicators/kamaState.ts +34 -34
  89. package/src/engine/indicators/keltnerState.ts +49 -49
  90. package/src/engine/indicators/kstState.ts +42 -42
  91. package/src/engine/indicators/maState.ts +36 -36
  92. package/src/engine/indicators/macdState.ts +76 -76
  93. package/src/engine/indicators/mfiState.ts +27 -27
  94. package/src/engine/indicators/momState.ts +25 -25
  95. package/src/engine/indicators/obvState.ts +25 -25
  96. package/src/engine/indicators/parkinsonState.ts +28 -28
  97. package/src/engine/indicators/pivotState.ts +51 -51
  98. package/src/engine/indicators/pvtState.ts +25 -25
  99. package/src/engine/indicators/rocState.ts +27 -27
  100. package/src/engine/indicators/rsiState.ts +65 -65
  101. package/src/engine/indicators/sarState.ts +41 -41
  102. package/src/engine/indicators/scheduler.ts +1205 -1205
  103. package/src/engine/indicators/soa.ts +352 -352
  104. package/src/engine/indicators/stateComposer.ts +1262 -1262
  105. package/src/engine/indicators/stochState.ts +26 -26
  106. package/src/engine/indicators/structureState.ts +69 -69
  107. package/src/engine/indicators/supertrendState.ts +37 -37
  108. package/src/engine/indicators/temaState.ts +27 -27
  109. package/src/engine/indicators/trixState.ts +35 -35
  110. package/src/engine/indicators/vmaState.ts +27 -27
  111. package/src/engine/indicators/volumeProfileState.ts +63 -63
  112. package/src/engine/indicators/vwapState.ts +29 -29
  113. package/src/engine/indicators/wmaState.ts +27 -27
  114. package/src/engine/indicators/wmsrState.ts +25 -25
  115. package/src/engine/indicators/workerProtocol.ts +613 -613
  116. package/src/engine/indicators/zonesState.ts +47 -47
  117. package/src/engine/layout/pane.ts +161 -161
  118. package/src/engine/marker/registry.ts +265 -265
  119. package/src/engine/paneRenderer.ts +169 -169
  120. package/src/engine/renderers/Indicator/atr.ts +237 -237
  121. package/src/engine/renderers/Indicator/boll.ts +317 -317
  122. package/src/engine/renderers/Indicator/cci.ts +275 -275
  123. package/src/engine/renderers/Indicator/chaikinVol.ts +138 -138
  124. package/src/engine/renderers/Indicator/cmf.ts +137 -137
  125. package/src/engine/renderers/Indicator/dema.ts +136 -136
  126. package/src/engine/renderers/Indicator/donchian.ts +137 -137
  127. package/src/engine/renderers/Indicator/ene.ts +271 -271
  128. package/src/engine/renderers/Indicator/expma.ts +197 -197
  129. package/src/engine/renderers/Indicator/fastk.ts +316 -316
  130. package/src/engine/renderers/Indicator/fib.ts +141 -141
  131. package/src/engine/renderers/Indicator/hma.ts +136 -136
  132. package/src/engine/renderers/Indicator/hv.ts +124 -124
  133. package/src/engine/renderers/Indicator/ichimoku.ts +181 -181
  134. package/src/engine/renderers/Indicator/index.ts +241 -241
  135. package/src/engine/renderers/Indicator/indicatorData.ts +650 -650
  136. package/src/engine/renderers/Indicator/kama.ts +136 -136
  137. package/src/engine/renderers/Indicator/keltner.ts +137 -137
  138. package/src/engine/renderers/Indicator/kst.ts +302 -302
  139. package/src/engine/renderers/Indicator/ma.ts +200 -200
  140. package/src/engine/renderers/Indicator/macd.ts +477 -477
  141. package/src/engine/renderers/Indicator/macdLegend.ts +141 -141
  142. package/src/engine/renderers/Indicator/mainIndicatorLegend.ts +272 -272
  143. package/src/engine/renderers/Indicator/mfi.ts +142 -142
  144. package/src/engine/renderers/Indicator/mom.ts +311 -311
  145. package/src/engine/renderers/Indicator/obv.ts +123 -123
  146. package/src/engine/renderers/Indicator/parkinson.ts +124 -124
  147. package/src/engine/renderers/Indicator/pivot.ts +131 -131
  148. package/src/engine/renderers/Indicator/pvt.ts +123 -123
  149. package/src/engine/renderers/Indicator/roc.ts +143 -143
  150. package/src/engine/renderers/Indicator/rsi.ts +390 -390
  151. package/src/engine/renderers/Indicator/sar.ts +113 -113
  152. package/src/engine/renderers/Indicator/scale/atr_scale.ts +19 -19
  153. package/src/engine/renderers/Indicator/scale/cci_scale.ts +19 -19
  154. package/src/engine/renderers/Indicator/scale/fastk_scale.ts +19 -19
  155. package/src/engine/renderers/Indicator/scale/indicator_scale.ts +204 -204
  156. package/src/engine/renderers/Indicator/scale/kst_scale.ts +19 -19
  157. package/src/engine/renderers/Indicator/scale/macd_scale.ts +22 -22
  158. package/src/engine/renderers/Indicator/scale/mom_scale.ts +19 -19
  159. package/src/engine/renderers/Indicator/scale/rsi_scale.ts +19 -19
  160. package/src/engine/renderers/Indicator/scale/stoch_scale.ts +19 -19
  161. package/src/engine/renderers/Indicator/scale/volume_scale.ts +26 -26
  162. package/src/engine/renderers/Indicator/scale/wmsr_scale.ts +19 -19
  163. package/src/engine/renderers/Indicator/stoch.ts +359 -359
  164. package/src/engine/renderers/Indicator/structure.ts +126 -126
  165. package/src/engine/renderers/Indicator/subPaneConfig.ts +265 -265
  166. package/src/engine/renderers/Indicator/supertrend.ts +115 -115
  167. package/src/engine/renderers/Indicator/tema.ts +136 -136
  168. package/src/engine/renderers/Indicator/trix.ts +158 -158
  169. package/src/engine/renderers/Indicator/vma.ts +124 -124
  170. package/src/engine/renderers/Indicator/volumeProfile.ts +125 -125
  171. package/src/engine/renderers/Indicator/vwap.ts +123 -123
  172. package/src/engine/renderers/Indicator/wma.ts +136 -136
  173. package/src/engine/renderers/Indicator/wmsr.ts +328 -328
  174. package/src/engine/renderers/Indicator/zones.ts +104 -104
  175. package/src/engine/renderers/__tests__/boll.renderer.test.ts +314 -314
  176. package/src/engine/renderers/__tests__/ene.renderer.test.ts +305 -305
  177. package/src/engine/renderers/__tests__/expma.renderer.test.ts +279 -279
  178. package/src/engine/renderers/__tests__/ma.renderer.test.ts +426 -426
  179. package/src/engine/renderers/__tests__/mainIndicatorLegend.renderer.test.ts +502 -502
  180. package/src/engine/renderers/__tests__/yAxis.renderer.test.ts +173 -173
  181. package/src/engine/renderers/candle.ts +459 -459
  182. package/src/engine/renderers/crosshair.ts +69 -69
  183. package/src/engine/renderers/customMarkers.ts +162 -162
  184. package/src/engine/renderers/extremaMarkers.ts +246 -246
  185. package/src/engine/renderers/gridLines.ts +90 -90
  186. package/src/engine/renderers/lastPrice.ts +97 -97
  187. package/src/engine/renderers/paneTitle.ts +136 -136
  188. package/src/engine/renderers/subVolume.ts +236 -236
  189. package/src/engine/renderers/timeAxis.ts +121 -121
  190. package/src/engine/renderers/webgl/candleSurface.ts +955 -955
  191. package/src/engine/renderers/webgl/sharedWebGLSurface.ts +146 -146
  192. package/src/engine/renderers/yAxis.ts +105 -105
  193. package/src/engine/scale/__tests__/logFormula.spec.ts +148 -148
  194. package/src/engine/scale/logFormula.ts +130 -130
  195. package/src/engine/scale/price.ts +39 -39
  196. package/src/engine/scale/priceScale.ts +264 -264
  197. package/src/engine/subPaneManager.ts +427 -427
  198. package/src/engine/theme/colors.ts +642 -642
  199. package/src/engine/theme/fonts.ts +20 -20
  200. package/src/engine/utils/klineConfig.ts +49 -49
  201. package/src/engine/utils/tickCount.ts +11 -11
  202. package/src/engine/utils/tickPosition.ts +214 -214
  203. package/src/engine/utils/zoom.ts +83 -83
  204. package/src/engine/viewport/viewport.ts +67 -67
  205. package/src/index.ts +3 -3
  206. package/src/plugin/ConfigManager.ts +93 -93
  207. package/src/plugin/EventBus.ts +77 -77
  208. package/src/plugin/HookSystem.ts +106 -106
  209. package/src/plugin/PluginHost.ts +243 -243
  210. package/src/plugin/PluginRegistry.ts +92 -92
  211. package/src/plugin/StateStore.ts +73 -73
  212. package/src/plugin/index.ts +19 -19
  213. package/src/plugin/rendererPluginManager.ts +368 -368
  214. package/src/plugin/stateKeys.ts +8 -8
  215. package/src/plugin/types.ts +526 -526
  216. package/src/reactivity/index.ts +2 -2
  217. package/src/reactivity/signal.ts +119 -119
  218. package/src/semantic/controller.ts +251 -251
  219. package/src/semantic/drawShape.ts +260 -260
  220. package/src/semantic/index.ts +28 -28
  221. package/src/semantic/schema.json +256 -256
  222. package/src/semantic/types.ts +251 -251
  223. package/src/semantic/validator.ts +349 -349
  224. package/src/types/kLine.ts +13 -13
  225. package/src/types/price.ts +56 -56
  226. package/src/types/volumePrice.ts +33 -33
  227. package/src/utils/dateFormat.ts +208 -208
  228. package/src/utils/kLineDraw/axis.ts +562 -562
  229. package/src/utils/priceToY.ts +34 -34
  230. package/src/utils/volumePrice.ts +202 -202
  231. package/src/version.ts +1 -1
@@ -1,343 +1,343 @@
1
- import type { RendererPlugin, RenderContext, DrawingStyle, DrawingPrimitive, YAxisLabel, XAxisLabel, YAxisRange, XAxisRange } from '../../plugin'
2
- import { RENDERER_PRIORITY } from '../../plugin'
3
- import {
4
- DrawingStore,
5
- DrawingDefinitionRegistry,
6
- createDefaultPrimitiveRendererSet,
7
- registerDefaultDrawingDefinitions,
8
- } from './index'
9
- import type { PrimitiveRendererSet } from './index'
10
- import type { KLineData } from '../../types/price'
11
- import { getPhysicalKLineConfig } from '../utils/klineConfig'
12
-
13
- /**
14
- * 为选中的绘图推送锚点轴标签
15
- * 提取为独立函数供主插件和 overlay 插件共用
16
- */
17
- function pushSelectedDrawingLabels(
18
- store: DrawingStore,
19
- definitions: DrawingDefinitionRegistry,
20
- context: RenderContext,
21
- pane: RenderContext['pane'],
22
- seriesData: KLineData[],
23
- range: RenderContext['range'],
24
- kWidth: number,
25
- kGap: number,
26
- dpr: number,
27
- viewport: NonNullable<RenderContext['viewport']>,
28
- startXPx: number,
29
- unitPx: number
30
- ): void {
31
- const selectedId = store.getSelectedId()
32
- if (!selectedId) return
33
-
34
- const selectedDrawing = store.getAll().find(d => d.id === selectedId)
35
- if (!selectedDrawing || !selectedDrawing.visible || selectedDrawing.paneId !== pane.id) return
36
-
37
- const geometry = definitions.compute(selectedDrawing, {
38
- pane,
39
- visibleData: seriesData.slice(range.start, range.end),
40
- seriesData,
41
- range,
42
- kLinePositions: context.kLinePositions,
43
- kLineCenters: context.kLineCenters,
44
- kBarRects: context.kBarRects,
45
- kWidth,
46
- kGap,
47
- dpr,
48
- paneWidth: context.paneWidth,
49
- viewport,
50
- toScreen(anchor) {
51
- if (!Number.isFinite(anchor.index) || anchor.index < 0) {
52
- return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
53
- }
54
- const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
55
- return { x, y: pane.yAxis.priceToY(anchor.price) }
56
- },
57
- })
58
-
59
- if (!geometry) return
60
-
61
- // 只在价格 pane 时添加标签
62
- if (pane.role !== 'price') return
63
-
64
- // 合并用户锚点和计算锚点
65
- const allAnchors = [...selectedDrawing.anchors, ...(geometry.computedAnchors ?? [])]
66
- if (allAnchors.length === 0) return
67
-
68
- // 计算锚点价格范围,用于Y轴价格范围带
69
- if (allAnchors.length >= 2) {
70
- let minP = Infinity, maxP = -Infinity
71
- for (const a of allAnchors) {
72
- if (!Number.isFinite(a.price)) continue
73
- if (a.price < minP) minP = a.price
74
- if (a.price > maxP) maxP = a.price
75
- }
76
- if (Number.isFinite(minP) && Number.isFinite(maxP) && minP !== maxP) {
77
- if (!context.yAxisRanges) context.yAxisRanges = []
78
- context.yAxisRanges.push({
79
- topY: pane.yAxis.priceToY(maxP),
80
- bottomY: pane.yAxis.priceToY(minP),
81
- color: selectedDrawing.style?.stroke ?? '#2962ff',
82
- opacity: 0.15,
83
- })
84
- }
85
- }
86
-
87
- // 计算锚点X坐标范围,用于X轴时间范围带
88
- if (allAnchors.length >= 2) {
89
- let minIdx = Infinity, maxIdx = -Infinity
90
- for (const a of allAnchors) {
91
- if (!Number.isFinite(a.index) || a.index < 0) continue
92
- if (a.index < minIdx) minIdx = a.index
93
- if (a.index > maxIdx) maxIdx = a.index
94
- }
95
- if (Number.isFinite(minIdx) && Number.isFinite(maxIdx) && minIdx !== maxIdx) {
96
- if (!context.xAxisRanges) context.xAxisRanges = []
97
- const leftX = (startXPx + minIdx * unitPx + (unitPx - 1) / 2) / dpr
98
- const rightX = (startXPx + maxIdx * unitPx + (unitPx - 1) / 2) / dpr
99
- context.xAxisRanges.push({
100
- leftX,
101
- rightX,
102
- color: selectedDrawing.style?.stroke ?? '#2962ff',
103
- opacity: 0.15,
104
- })
105
- }
106
- }
107
-
108
- // 辅助函数:根据index获取时间戳
109
- const getTimestampForIndex = (idx: number): number | null => {
110
- if (idx >= 0 && idx < seriesData.length) {
111
- return seriesData[idx]?.timestamp ?? null
112
- }
113
- if (seriesData.length >= 2 && idx >= seriesData.length) {
114
- const lastIdx = seriesData.length - 1
115
- const secondLastIdx = seriesData.length - 2
116
- const lastTs = seriesData[lastIdx]!.timestamp
117
- const secondLastTs = seriesData[secondLastIdx]!.timestamp
118
- const timeStep = lastTs - secondLastTs
119
- return lastTs + (idx - lastIdx) * timeStep
120
- }
121
- return null
122
- }
123
-
124
- // 推送每个锚点的标签
125
- for (const anchor of allAnchors) {
126
- if (!Number.isFinite(anchor.index) || !Number.isFinite(anchor.price)) continue
127
-
128
- const screenPoint = (() => {
129
- if (anchor.index < 0) {
130
- return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
131
- }
132
- const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
133
- return { x, y: pane.yAxis.priceToY(anchor.price) }
134
- })()
135
-
136
- // Y轴标签
137
- if (screenPoint.y >= 0 && screenPoint.y <= pane.height) {
138
- if (!context.yAxisLabels) context.yAxisLabels = []
139
- context.yAxisLabels.push({
140
- dataIndex: Math.round(anchor.index),
141
- price: anchor.price,
142
- y: screenPoint.y,
143
- style: {
144
- bgColor: selectedDrawing.style?.stroke ?? '#2962ff',
145
- borderColor: selectedDrawing.style?.stroke ?? '#2962ff',
146
- textColor: '#ffffff',
147
- }
148
- })
149
- }
150
-
151
- // X轴标签
152
- if (screenPoint.x >= -kWidth && screenPoint.x <= viewport.plotWidth + kWidth) {
153
- const timestamp = anchor.time
154
- ? (typeof anchor.time === 'string' ? new Date(anchor.time).getTime() : anchor.time)
155
- : getTimestampForIndex(Math.round(anchor.index))
156
- if (!timestamp) continue
157
-
158
- if (!context.xAxisLabels) context.xAxisLabels = []
159
- context.xAxisLabels.push({
160
- dataIndex: Math.round(anchor.index),
161
- timestamp,
162
- x: screenPoint.x + viewport.scrollLeft,
163
- style: {
164
- bgColor: selectedDrawing.style?.stroke ?? '#2962ff',
165
- textColor: '#ffffff',
166
- }
167
- })
168
- }
169
- }
170
- }
171
-
172
- /**
173
- * 创建绘图渲染器插件(主层,负责绘制 shape)
174
- * 注意:此插件不再推送轴标签,标签由 createDrawingLabelOverlayPlugin 负责
175
- */
176
- export function createDrawingRendererPlugin(options: {
177
- store: DrawingStore
178
- paneId?: string
179
- definitions?: DrawingDefinitionRegistry
180
- renderers?: PrimitiveRendererSet
181
- }): RendererPlugin {
182
- const store = options.store
183
- const definitions = options.definitions ?? new DrawingDefinitionRegistry()
184
- const renderers = options.renderers ?? createDefaultPrimitiveRendererSet()
185
- registerDefaultDrawingDefinitions(definitions)
186
-
187
- return {
188
- name: 'drawingRenderer',
189
- version: '0.1.0',
190
- description: '绘图渲染器(仅负责绘制形状)',
191
- debugName: '绘图层',
192
- paneId: options.paneId ?? 'main',
193
- priority: -25,
194
- draw(context: RenderContext) {
195
- const { ctx, pane, data, range, dpr, paneWidth, kLinePositions, kLineCenters, kBarRects, kWidth, kGap } = context
196
- const viewport = context.viewport ?? {
197
- scrollLeft: context.scrollLeft,
198
- plotWidth: paneWidth,
199
- plotHeight: pane.height,
200
- }
201
- const seriesData = data as KLineData[]
202
- const visibleData = seriesData.slice(range.start, range.end)
203
- const drawings = store.getVisibleByPane(pane.id)
204
- if (drawings.length === 0) return
205
-
206
- const viewportClip = {
207
- left: 0,
208
- top: 0,
209
- right: viewport.plotWidth,
210
- bottom: pane.height,
211
- }
212
-
213
- ctx.save()
214
- ctx.beginPath()
215
- ctx.rect(0, 0, viewport.plotWidth, pane.height)
216
- ctx.clip()
217
-
218
- for (const drawing of drawings) {
219
- const geometry = definitions.compute(drawing, {
220
- pane,
221
- visibleData,
222
- seriesData,
223
- range,
224
- kLinePositions,
225
- kLineCenters,
226
- kBarRects,
227
- kWidth,
228
- kGap,
229
- dpr,
230
- paneWidth,
231
- viewport,
232
- toScreen(anchor) {
233
- const { startXPx, unitPx } = getPhysicalKLineConfig(kWidth, kGap, dpr)
234
- if (!Number.isFinite(anchor.index) || anchor.index < 0) {
235
- return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
236
- }
237
- const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
238
- return { x, y: pane.yAxis.priceToY(anchor.price) }
239
- },
240
- })
241
- if (!geometry) continue
242
-
243
- const isSelected = store.getSelectedId() === drawing.id
244
- const primitives = isSelected
245
- ? geometry.primitives.map((p) => applySelectedStyle(p, drawing.style))
246
- : geometry.primitives
247
-
248
- // 绘制形状(不再推送轴标签,标签由 overlay 插件负责)
249
- for (const primitive of primitives) {
250
- if (primitive.kind === 'point') {
251
- renderers.point(ctx, primitive, dpr)
252
- continue
253
- }
254
- if (primitive.kind === 'line') {
255
- renderers.line(ctx, primitive, viewportClip, dpr)
256
- continue
257
- }
258
- if (primitive.kind === 'area') {
259
- renderers.area(ctx, primitive, dpr)
260
- continue
261
- }
262
- renderers.text(ctx, primitive, dpr)
263
- }
264
- }
265
-
266
- ctx.restore()
267
- },
268
- }
269
- }
270
-
271
- /**
272
- * 创建绘图标签 Overlay 插件
273
- *
274
- * ⚠️ 警告:此插件必须在 All 和 Overlay 更新级别运行
275
- * 当前代码库中没有 UpdateLevel.Main 的触发点,因此此插件设置为 layer: 'overlay' 是安全的
276
- * 如果将来添加 Main 级别的更新,此插件会被跳过,导致选中绘图的轴标签消失
277
- *
278
- * 解决方案:如果将来需要使用 Main 级别,请将此插件改为同时在 main 和 overlay 层注册,
279
- * 或移除分层过滤让此插件在所有级别运行
280
- */
281
- export function createDrawingLabelOverlayPlugin(options: {
282
- store: DrawingStore
283
- paneId?: string
284
- definitions?: DrawingDefinitionRegistry
285
- }): RendererPlugin {
286
- const store = options.store
287
- const definitions = options.definitions ?? new DrawingDefinitionRegistry()
288
- registerDefaultDrawingDefinitions(definitions)
289
-
290
- return {
291
- name: 'drawingLabelOverlay',
292
- version: '0.1.0',
293
- description: '绘图轴标签渲染器(overlay 层,负责推送选中绘图的轴标签)',
294
- debugName: '绘图标签层',
295
- paneId: options.paneId ?? 'main',
296
- layer: 'overlay',
297
- priority: -24, // 比主绘图层稍晚,确保在其他 overlay 插件之后
298
- draw(context: RenderContext) {
299
- const { pane, data, range, dpr, paneWidth, kWidth, kGap } = context
300
- const viewport = context.viewport ?? {
301
- scrollLeft: context.scrollLeft,
302
- plotWidth: paneWidth,
303
- plotHeight: pane.height,
304
- }
305
- const seriesData = data as KLineData[]
306
- const { startXPx, unitPx } = getPhysicalKLineConfig(kWidth, kGap, dpr)
307
-
308
- // 推送选中绘图的锚点标签
309
- pushSelectedDrawingLabels(
310
- store,
311
- definitions,
312
- context,
313
- pane,
314
- seriesData,
315
- range,
316
- kWidth,
317
- kGap,
318
- dpr,
319
- viewport,
320
- startXPx,
321
- unitPx
322
- )
323
- },
324
- }
325
- }
326
-
327
- function applySelectedStyle(primitive: DrawingPrimitive, baseStyle: DrawingStyle): DrawingPrimitive {
328
- const selectedStroke = baseStyle.stroke ?? '#2962ff'
329
- const selectedWidth = (baseStyle.strokeWidth ?? 1) + 1
330
- const selectedPointRadius = (baseStyle.pointRadius ?? 4) + 2
331
-
332
- if (primitive.kind === 'point') {
333
- return { ...primitive, style: { ...primitive.style, stroke: selectedStroke, pointRadius: selectedPointRadius } }
334
- }
335
- if (primitive.kind === 'line') {
336
- return { ...primitive, style: { ...primitive.style, stroke: selectedStroke, strokeWidth: selectedWidth } }
337
- }
338
- if (primitive.kind === 'area') {
339
- return { ...primitive, style: { ...primitive.style, stroke: selectedStroke } }
340
- }
341
- // text
342
- return { ...primitive, style: { ...primitive.style, textColor: selectedStroke, fontSize: (primitive.style?.fontSize ?? 12) + 1 } }
343
- }
1
+ import type { RendererPlugin, RenderContext, DrawingStyle, DrawingPrimitive, YAxisLabel, XAxisLabel, YAxisRange, XAxisRange } from '../../plugin'
2
+ import { RENDERER_PRIORITY } from '../../plugin'
3
+ import {
4
+ DrawingStore,
5
+ DrawingDefinitionRegistry,
6
+ createDefaultPrimitiveRendererSet,
7
+ registerDefaultDrawingDefinitions,
8
+ } from './index'
9
+ import type { PrimitiveRendererSet } from './index'
10
+ import type { KLineData } from '../../types/price'
11
+ import { getPhysicalKLineConfig } from '../utils/klineConfig'
12
+
13
+ /**
14
+ * 为选中的绘图推送锚点轴标签
15
+ * 提取为独立函数供主插件和 overlay 插件共用
16
+ */
17
+ function pushSelectedDrawingLabels(
18
+ store: DrawingStore,
19
+ definitions: DrawingDefinitionRegistry,
20
+ context: RenderContext,
21
+ pane: RenderContext['pane'],
22
+ seriesData: KLineData[],
23
+ range: RenderContext['range'],
24
+ kWidth: number,
25
+ kGap: number,
26
+ dpr: number,
27
+ viewport: NonNullable<RenderContext['viewport']>,
28
+ startXPx: number,
29
+ unitPx: number
30
+ ): void {
31
+ const selectedId = store.getSelectedId()
32
+ if (!selectedId) return
33
+
34
+ const selectedDrawing = store.getAll().find(d => d.id === selectedId)
35
+ if (!selectedDrawing || !selectedDrawing.visible || selectedDrawing.paneId !== pane.id) return
36
+
37
+ const geometry = definitions.compute(selectedDrawing, {
38
+ pane,
39
+ visibleData: seriesData.slice(range.start, range.end),
40
+ seriesData,
41
+ range,
42
+ kLinePositions: context.kLinePositions,
43
+ kLineCenters: context.kLineCenters,
44
+ kBarRects: context.kBarRects,
45
+ kWidth,
46
+ kGap,
47
+ dpr,
48
+ paneWidth: context.paneWidth,
49
+ viewport,
50
+ toScreen(anchor) {
51
+ if (!Number.isFinite(anchor.index) || anchor.index < 0) {
52
+ return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
53
+ }
54
+ const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
55
+ return { x, y: pane.yAxis.priceToY(anchor.price) }
56
+ },
57
+ })
58
+
59
+ if (!geometry) return
60
+
61
+ // 只在价格 pane 时添加标签
62
+ if (pane.role !== 'price') return
63
+
64
+ // 合并用户锚点和计算锚点
65
+ const allAnchors = [...selectedDrawing.anchors, ...(geometry.computedAnchors ?? [])]
66
+ if (allAnchors.length === 0) return
67
+
68
+ // 计算锚点价格范围,用于Y轴价格范围带
69
+ if (allAnchors.length >= 2) {
70
+ let minP = Infinity, maxP = -Infinity
71
+ for (const a of allAnchors) {
72
+ if (!Number.isFinite(a.price)) continue
73
+ if (a.price < minP) minP = a.price
74
+ if (a.price > maxP) maxP = a.price
75
+ }
76
+ if (Number.isFinite(minP) && Number.isFinite(maxP) && minP !== maxP) {
77
+ if (!context.yAxisRanges) context.yAxisRanges = []
78
+ context.yAxisRanges.push({
79
+ topY: pane.yAxis.priceToY(maxP),
80
+ bottomY: pane.yAxis.priceToY(minP),
81
+ color: selectedDrawing.style?.stroke ?? '#2962ff',
82
+ opacity: 0.15,
83
+ })
84
+ }
85
+ }
86
+
87
+ // 计算锚点X坐标范围,用于X轴时间范围带
88
+ if (allAnchors.length >= 2) {
89
+ let minIdx = Infinity, maxIdx = -Infinity
90
+ for (const a of allAnchors) {
91
+ if (!Number.isFinite(a.index) || a.index < 0) continue
92
+ if (a.index < minIdx) minIdx = a.index
93
+ if (a.index > maxIdx) maxIdx = a.index
94
+ }
95
+ if (Number.isFinite(minIdx) && Number.isFinite(maxIdx) && minIdx !== maxIdx) {
96
+ if (!context.xAxisRanges) context.xAxisRanges = []
97
+ const leftX = (startXPx + minIdx * unitPx + (unitPx - 1) / 2) / dpr
98
+ const rightX = (startXPx + maxIdx * unitPx + (unitPx - 1) / 2) / dpr
99
+ context.xAxisRanges.push({
100
+ leftX,
101
+ rightX,
102
+ color: selectedDrawing.style?.stroke ?? '#2962ff',
103
+ opacity: 0.15,
104
+ })
105
+ }
106
+ }
107
+
108
+ // 辅助函数:根据index获取时间戳
109
+ const getTimestampForIndex = (idx: number): number | null => {
110
+ if (idx >= 0 && idx < seriesData.length) {
111
+ return seriesData[idx]?.timestamp ?? null
112
+ }
113
+ if (seriesData.length >= 2 && idx >= seriesData.length) {
114
+ const lastIdx = seriesData.length - 1
115
+ const secondLastIdx = seriesData.length - 2
116
+ const lastTs = seriesData[lastIdx]!.timestamp
117
+ const secondLastTs = seriesData[secondLastIdx]!.timestamp
118
+ const timeStep = lastTs - secondLastTs
119
+ return lastTs + (idx - lastIdx) * timeStep
120
+ }
121
+ return null
122
+ }
123
+
124
+ // 推送每个锚点的标签
125
+ for (const anchor of allAnchors) {
126
+ if (!Number.isFinite(anchor.index) || !Number.isFinite(anchor.price)) continue
127
+
128
+ const screenPoint = (() => {
129
+ if (anchor.index < 0) {
130
+ return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
131
+ }
132
+ const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
133
+ return { x, y: pane.yAxis.priceToY(anchor.price) }
134
+ })()
135
+
136
+ // Y轴标签
137
+ if (screenPoint.y >= 0 && screenPoint.y <= pane.height) {
138
+ if (!context.yAxisLabels) context.yAxisLabels = []
139
+ context.yAxisLabels.push({
140
+ dataIndex: Math.round(anchor.index),
141
+ price: anchor.price,
142
+ y: screenPoint.y,
143
+ style: {
144
+ bgColor: selectedDrawing.style?.stroke ?? '#2962ff',
145
+ borderColor: selectedDrawing.style?.stroke ?? '#2962ff',
146
+ textColor: '#ffffff',
147
+ }
148
+ })
149
+ }
150
+
151
+ // X轴标签
152
+ if (screenPoint.x >= -kWidth && screenPoint.x <= viewport.plotWidth + kWidth) {
153
+ const timestamp = anchor.time
154
+ ? (typeof anchor.time === 'string' ? new Date(anchor.time).getTime() : anchor.time)
155
+ : getTimestampForIndex(Math.round(anchor.index))
156
+ if (!timestamp) continue
157
+
158
+ if (!context.xAxisLabels) context.xAxisLabels = []
159
+ context.xAxisLabels.push({
160
+ dataIndex: Math.round(anchor.index),
161
+ timestamp,
162
+ x: screenPoint.x + viewport.scrollLeft,
163
+ style: {
164
+ bgColor: selectedDrawing.style?.stroke ?? '#2962ff',
165
+ textColor: '#ffffff',
166
+ }
167
+ })
168
+ }
169
+ }
170
+ }
171
+
172
+ /**
173
+ * 创建绘图渲染器插件(主层,负责绘制 shape)
174
+ * 注意:此插件不再推送轴标签,标签由 createDrawingLabelOverlayPlugin 负责
175
+ */
176
+ export function createDrawingRendererPlugin(options: {
177
+ store: DrawingStore
178
+ paneId?: string
179
+ definitions?: DrawingDefinitionRegistry
180
+ renderers?: PrimitiveRendererSet
181
+ }): RendererPlugin {
182
+ const store = options.store
183
+ const definitions = options.definitions ?? new DrawingDefinitionRegistry()
184
+ const renderers = options.renderers ?? createDefaultPrimitiveRendererSet()
185
+ registerDefaultDrawingDefinitions(definitions)
186
+
187
+ return {
188
+ name: 'drawingRenderer',
189
+ version: '0.1.0',
190
+ description: '绘图渲染器(仅负责绘制形状)',
191
+ debugName: '绘图层',
192
+ paneId: options.paneId ?? 'main',
193
+ priority: -25,
194
+ draw(context: RenderContext) {
195
+ const { ctx, pane, data, range, dpr, paneWidth, kLinePositions, kLineCenters, kBarRects, kWidth, kGap } = context
196
+ const viewport = context.viewport ?? {
197
+ scrollLeft: context.scrollLeft,
198
+ plotWidth: paneWidth,
199
+ plotHeight: pane.height,
200
+ }
201
+ const seriesData = data as KLineData[]
202
+ const visibleData = seriesData.slice(range.start, range.end)
203
+ const drawings = store.getVisibleByPane(pane.id)
204
+ if (drawings.length === 0) return
205
+
206
+ const viewportClip = {
207
+ left: 0,
208
+ top: 0,
209
+ right: viewport.plotWidth,
210
+ bottom: pane.height,
211
+ }
212
+
213
+ ctx.save()
214
+ ctx.beginPath()
215
+ ctx.rect(0, 0, viewport.plotWidth, pane.height)
216
+ ctx.clip()
217
+
218
+ for (const drawing of drawings) {
219
+ const geometry = definitions.compute(drawing, {
220
+ pane,
221
+ visibleData,
222
+ seriesData,
223
+ range,
224
+ kLinePositions,
225
+ kLineCenters,
226
+ kBarRects,
227
+ kWidth,
228
+ kGap,
229
+ dpr,
230
+ paneWidth,
231
+ viewport,
232
+ toScreen(anchor) {
233
+ const { startXPx, unitPx } = getPhysicalKLineConfig(kWidth, kGap, dpr)
234
+ if (!Number.isFinite(anchor.index) || anchor.index < 0) {
235
+ return { x: -kWidth, y: pane.yAxis.priceToY(anchor.price) }
236
+ }
237
+ const x = (startXPx + anchor.index * unitPx + (unitPx - 1) / 2) / dpr - viewport.scrollLeft
238
+ return { x, y: pane.yAxis.priceToY(anchor.price) }
239
+ },
240
+ })
241
+ if (!geometry) continue
242
+
243
+ const isSelected = store.getSelectedId() === drawing.id
244
+ const primitives = isSelected
245
+ ? geometry.primitives.map((p) => applySelectedStyle(p, drawing.style))
246
+ : geometry.primitives
247
+
248
+ // 绘制形状(不再推送轴标签,标签由 overlay 插件负责)
249
+ for (const primitive of primitives) {
250
+ if (primitive.kind === 'point') {
251
+ renderers.point(ctx, primitive, dpr)
252
+ continue
253
+ }
254
+ if (primitive.kind === 'line') {
255
+ renderers.line(ctx, primitive, viewportClip, dpr)
256
+ continue
257
+ }
258
+ if (primitive.kind === 'area') {
259
+ renderers.area(ctx, primitive, dpr)
260
+ continue
261
+ }
262
+ renderers.text(ctx, primitive, dpr)
263
+ }
264
+ }
265
+
266
+ ctx.restore()
267
+ },
268
+ }
269
+ }
270
+
271
+ /**
272
+ * 创建绘图标签 Overlay 插件
273
+ *
274
+ * ⚠️ 警告:此插件必须在 All 和 Overlay 更新级别运行
275
+ * 当前代码库中没有 UpdateLevel.Main 的触发点,因此此插件设置为 layer: 'overlay' 是安全的
276
+ * 如果将来添加 Main 级别的更新,此插件会被跳过,导致选中绘图的轴标签消失
277
+ *
278
+ * 解决方案:如果将来需要使用 Main 级别,请将此插件改为同时在 main 和 overlay 层注册,
279
+ * 或移除分层过滤让此插件在所有级别运行
280
+ */
281
+ export function createDrawingLabelOverlayPlugin(options: {
282
+ store: DrawingStore
283
+ paneId?: string
284
+ definitions?: DrawingDefinitionRegistry
285
+ }): RendererPlugin {
286
+ const store = options.store
287
+ const definitions = options.definitions ?? new DrawingDefinitionRegistry()
288
+ registerDefaultDrawingDefinitions(definitions)
289
+
290
+ return {
291
+ name: 'drawingLabelOverlay',
292
+ version: '0.1.0',
293
+ description: '绘图轴标签渲染器(overlay 层,负责推送选中绘图的轴标签)',
294
+ debugName: '绘图标签层',
295
+ paneId: options.paneId ?? 'main',
296
+ layer: 'overlay',
297
+ priority: -24, // 比主绘图层稍晚,确保在其他 overlay 插件之后
298
+ draw(context: RenderContext) {
299
+ const { pane, data, range, dpr, paneWidth, kWidth, kGap } = context
300
+ const viewport = context.viewport ?? {
301
+ scrollLeft: context.scrollLeft,
302
+ plotWidth: paneWidth,
303
+ plotHeight: pane.height,
304
+ }
305
+ const seriesData = data as KLineData[]
306
+ const { startXPx, unitPx } = getPhysicalKLineConfig(kWidth, kGap, dpr)
307
+
308
+ // 推送选中绘图的锚点标签
309
+ pushSelectedDrawingLabels(
310
+ store,
311
+ definitions,
312
+ context,
313
+ pane,
314
+ seriesData,
315
+ range,
316
+ kWidth,
317
+ kGap,
318
+ dpr,
319
+ viewport,
320
+ startXPx,
321
+ unitPx
322
+ )
323
+ },
324
+ }
325
+ }
326
+
327
+ function applySelectedStyle(primitive: DrawingPrimitive, baseStyle: DrawingStyle): DrawingPrimitive {
328
+ const selectedStroke = baseStyle.stroke ?? '#2962ff'
329
+ const selectedWidth = (baseStyle.strokeWidth ?? 1) + 1
330
+ const selectedPointRadius = (baseStyle.pointRadius ?? 4) + 2
331
+
332
+ if (primitive.kind === 'point') {
333
+ return { ...primitive, style: { ...primitive.style, stroke: selectedStroke, pointRadius: selectedPointRadius } }
334
+ }
335
+ if (primitive.kind === 'line') {
336
+ return { ...primitive, style: { ...primitive.style, stroke: selectedStroke, strokeWidth: selectedWidth } }
337
+ }
338
+ if (primitive.kind === 'area') {
339
+ return { ...primitive, style: { ...primitive.style, stroke: selectedStroke } }
340
+ }
341
+ // text
342
+ return { ...primitive, style: { ...primitive.style, textColor: selectedStroke, fontSize: (primitive.style?.fontSize ?? 12) + 1 } }
343
+ }