@363045841yyt/klinechart-core 0.7.3 → 0.7.5

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 (247) hide show
  1. package/README.md +201 -201
  2. package/README.zh-CN.md +201 -201
  3. package/dist/controllers/index.d.ts +1 -0
  4. package/dist/controllers/index.d.ts.map +1 -1
  5. package/dist/controllers/index.js +1 -0
  6. package/dist/controllers/index.js.map +1 -1
  7. package/dist/engine/chart.d.ts +11 -19
  8. package/dist/engine/chart.d.ts.map +1 -1
  9. package/dist/engine/chart.js +92 -109
  10. package/dist/engine/chart.js.map +1 -1
  11. package/dist/engine/renderers/Indicator/indicatorData.d.ts +1 -0
  12. package/dist/engine/renderers/Indicator/indicatorData.d.ts.map +1 -1
  13. package/dist/engine/renderers/Indicator/indicatorData.js +1 -1
  14. package/dist/engine/renderers/Indicator/indicatorData.js.map +1 -1
  15. package/dist/engine/renderers/webgl/candleSurface.js +47 -47
  16. package/dist/engine/subPaneManager.d.ts +4 -0
  17. package/dist/engine/subPaneManager.d.ts.map +1 -1
  18. package/dist/engine/subPaneManager.js +13 -0
  19. package/dist/engine/subPaneManager.js.map +1 -1
  20. package/dist/version.d.ts +1 -1
  21. package/dist/version.d.ts.map +1 -1
  22. package/dist/version.js +1 -2
  23. package/dist/version.js.map +1 -1
  24. package/package.json +129 -122
  25. package/src/__tests__/signal.test.ts +124 -124
  26. package/src/config/chartSettings.ts +66 -66
  27. package/src/controllers/__tests__/drawing.test.ts +214 -214
  28. package/src/controllers/__tests__/indicatorSelector.test.ts +481 -481
  29. package/src/controllers/__tests__/toolbar.test.ts +225 -225
  30. package/src/controllers/createChartController.ts +665 -665
  31. package/src/controllers/createDrawingController.ts +96 -96
  32. package/src/controllers/createIndicatorSelectorController.ts +307 -307
  33. package/src/controllers/createToolbarController.ts +146 -146
  34. package/src/controllers/index.ts +20 -19
  35. package/src/controllers/types.ts +284 -284
  36. package/src/engine/__tests__/chart.dpr.test.ts +401 -401
  37. package/src/engine/__tests__/paneRenderer.resize.test.ts +92 -92
  38. package/src/engine/chart-store.ts +121 -121
  39. package/src/engine/chart.d.ts +617 -617
  40. package/src/engine/chart.ts +2803 -2815
  41. package/src/engine/controller/__tests__/interaction.dpr.test.ts +259 -259
  42. package/src/engine/controller/interaction.ts +722 -722
  43. package/src/engine/controller/markerInteraction.ts +130 -130
  44. package/src/engine/controller/pinchTracker.ts +82 -82
  45. package/src/engine/controller/tooltipPosition.ts +48 -48
  46. package/src/engine/draw/__tests__/pixelAlign.spec.ts +176 -176
  47. package/src/engine/draw/pixelAlign.ts +259 -259
  48. package/src/engine/drawing/index.ts +655 -655
  49. package/src/engine/drawing/interaction.ts +842 -842
  50. package/src/engine/drawing/plugin.ts +343 -343
  51. package/src/engine/indicators/__tests__/__fixtures__/golden/atr.json +38 -38
  52. package/src/engine/indicators/__tests__/__fixtures__/golden/dema.json +14 -14
  53. package/src/engine/indicators/__tests__/__fixtures__/golden/hma.json +14 -14
  54. package/src/engine/indicators/__tests__/__fixtures__/golden/index.ts +55 -55
  55. package/src/engine/indicators/__tests__/__fixtures__/golden/kama.json +14 -14
  56. package/src/engine/indicators/__tests__/__fixtures__/golden/tema.json +14 -14
  57. package/src/engine/indicators/__tests__/__fixtures__/golden/wma.json +40 -40
  58. package/src/engine/indicators/__tests__/__fixtures__/synthetic.ts +65 -65
  59. package/src/engine/indicators/__tests__/_propertyAssertions.ts +76 -76
  60. package/src/engine/indicators/__tests__/atr.test.ts +153 -153
  61. package/src/engine/indicators/__tests__/calculators.test.ts +614 -614
  62. package/src/engine/indicators/__tests__/cmf-mfi.test.ts +100 -100
  63. package/src/engine/indicators/__tests__/dema.test.ts +73 -73
  64. package/src/engine/indicators/__tests__/donchian.test.ts +70 -70
  65. package/src/engine/indicators/__tests__/hma.test.ts +73 -73
  66. package/src/engine/indicators/__tests__/ichimoku.test.ts +105 -105
  67. package/src/engine/indicators/__tests__/kama.test.ts +80 -80
  68. package/src/engine/indicators/__tests__/keltner.test.ts +65 -65
  69. package/src/engine/indicators/__tests__/pivot-fib.test.ts +110 -110
  70. package/src/engine/indicators/__tests__/roc.test.ts +68 -68
  71. package/src/engine/indicators/__tests__/sar.test.ts +86 -86
  72. package/src/engine/indicators/__tests__/scheduler.test.ts +831 -831
  73. package/src/engine/indicators/__tests__/soa.test.ts +533 -533
  74. package/src/engine/indicators/__tests__/structure.test.ts +110 -110
  75. package/src/engine/indicators/__tests__/supertrend.test.ts +65 -65
  76. package/src/engine/indicators/__tests__/tema.test.ts +68 -68
  77. package/src/engine/indicators/__tests__/trix.test.ts +70 -70
  78. package/src/engine/indicators/__tests__/volatility.test.ts +117 -117
  79. package/src/engine/indicators/__tests__/volume.test.ts +115 -115
  80. package/src/engine/indicators/__tests__/volumeProfile.test.ts +74 -74
  81. package/src/engine/indicators/__tests__/vwap.test.ts +69 -69
  82. package/src/engine/indicators/__tests__/wma.test.ts +112 -112
  83. package/src/engine/indicators/__tests__/zones.test.ts +95 -95
  84. package/src/engine/indicators/atrState.ts +27 -27
  85. package/src/engine/indicators/bollState.ts +51 -51
  86. package/src/engine/indicators/calculators.ts +2593 -2593
  87. package/src/engine/indicators/cciState.ts +25 -25
  88. package/src/engine/indicators/chaikinVolState.ts +32 -32
  89. package/src/engine/indicators/cmfState.ts +27 -27
  90. package/src/engine/indicators/demaState.ts +27 -27
  91. package/src/engine/indicators/donchianState.ts +43 -43
  92. package/src/engine/indicators/eneState.ts +43 -43
  93. package/src/engine/indicators/expmaState.ts +43 -43
  94. package/src/engine/indicators/fastkState.ts +25 -25
  95. package/src/engine/indicators/fibState.ts +41 -41
  96. package/src/engine/indicators/hmaState.ts +27 -27
  97. package/src/engine/indicators/hvState.ts +28 -28
  98. package/src/engine/indicators/ichimokuState.ts +70 -70
  99. package/src/engine/indicators/indicator.worker.ts +169 -169
  100. package/src/engine/indicators/indicatorDefinitionRegistry.ts +62 -62
  101. package/src/engine/indicators/indicatorMetadata.ts +110 -110
  102. package/src/engine/indicators/indicatorRegistry.ts +106 -106
  103. package/src/engine/indicators/indicatorRuntime.ts +1548 -1548
  104. package/src/engine/indicators/kamaState.ts +34 -34
  105. package/src/engine/indicators/keltnerState.ts +49 -49
  106. package/src/engine/indicators/kstState.ts +42 -42
  107. package/src/engine/indicators/maState.ts +36 -36
  108. package/src/engine/indicators/macdState.ts +76 -76
  109. package/src/engine/indicators/mfiState.ts +27 -27
  110. package/src/engine/indicators/momState.ts +25 -25
  111. package/src/engine/indicators/obvState.ts +25 -25
  112. package/src/engine/indicators/parkinsonState.ts +28 -28
  113. package/src/engine/indicators/pivotState.ts +51 -51
  114. package/src/engine/indicators/pvtState.ts +25 -25
  115. package/src/engine/indicators/rocState.ts +27 -27
  116. package/src/engine/indicators/rsiState.ts +65 -65
  117. package/src/engine/indicators/sarState.ts +41 -41
  118. package/src/engine/indicators/scheduler.ts +1205 -1205
  119. package/src/engine/indicators/soa.ts +352 -352
  120. package/src/engine/indicators/stateComposer.ts +1262 -1262
  121. package/src/engine/indicators/stochState.ts +26 -26
  122. package/src/engine/indicators/structureState.ts +69 -69
  123. package/src/engine/indicators/supertrendState.ts +37 -37
  124. package/src/engine/indicators/temaState.ts +27 -27
  125. package/src/engine/indicators/trixState.ts +35 -35
  126. package/src/engine/indicators/vmaState.ts +27 -27
  127. package/src/engine/indicators/volumeProfileState.ts +63 -63
  128. package/src/engine/indicators/vwapState.ts +29 -29
  129. package/src/engine/indicators/wmaState.ts +27 -27
  130. package/src/engine/indicators/wmsrState.ts +25 -25
  131. package/src/engine/indicators/workerProtocol.ts +613 -613
  132. package/src/engine/indicators/zonesState.ts +47 -47
  133. package/src/engine/layout/pane.ts +161 -161
  134. package/src/engine/marker/registry.ts +265 -265
  135. package/src/engine/paneRenderer.ts +169 -169
  136. package/src/engine/renderers/Indicator/atr.ts +237 -237
  137. package/src/engine/renderers/Indicator/boll.ts +317 -317
  138. package/src/engine/renderers/Indicator/cci.ts +275 -275
  139. package/src/engine/renderers/Indicator/chaikinVol.ts +138 -138
  140. package/src/engine/renderers/Indicator/cmf.ts +137 -137
  141. package/src/engine/renderers/Indicator/dema.ts +136 -136
  142. package/src/engine/renderers/Indicator/donchian.ts +137 -137
  143. package/src/engine/renderers/Indicator/ene.ts +271 -271
  144. package/src/engine/renderers/Indicator/expma.ts +197 -197
  145. package/src/engine/renderers/Indicator/fastk.ts +316 -316
  146. package/src/engine/renderers/Indicator/fib.ts +141 -141
  147. package/src/engine/renderers/Indicator/hma.ts +136 -136
  148. package/src/engine/renderers/Indicator/hv.ts +124 -124
  149. package/src/engine/renderers/Indicator/ichimoku.ts +181 -181
  150. package/src/engine/renderers/Indicator/index.ts +241 -241
  151. package/src/engine/renderers/Indicator/indicatorData.ts +650 -650
  152. package/src/engine/renderers/Indicator/kama.ts +136 -136
  153. package/src/engine/renderers/Indicator/keltner.ts +137 -137
  154. package/src/engine/renderers/Indicator/kst.ts +302 -302
  155. package/src/engine/renderers/Indicator/ma.ts +200 -200
  156. package/src/engine/renderers/Indicator/macd.ts +477 -477
  157. package/src/engine/renderers/Indicator/macdLegend.ts +141 -141
  158. package/src/engine/renderers/Indicator/mainIndicatorLegend.ts +272 -272
  159. package/src/engine/renderers/Indicator/mfi.ts +142 -142
  160. package/src/engine/renderers/Indicator/mom.ts +311 -311
  161. package/src/engine/renderers/Indicator/obv.ts +123 -123
  162. package/src/engine/renderers/Indicator/parkinson.ts +124 -124
  163. package/src/engine/renderers/Indicator/pivot.ts +131 -131
  164. package/src/engine/renderers/Indicator/pvt.ts +123 -123
  165. package/src/engine/renderers/Indicator/roc.ts +143 -143
  166. package/src/engine/renderers/Indicator/rsi.ts +390 -390
  167. package/src/engine/renderers/Indicator/sar.ts +113 -113
  168. package/src/engine/renderers/Indicator/scale/atr_scale.ts +19 -19
  169. package/src/engine/renderers/Indicator/scale/cci_scale.ts +19 -19
  170. package/src/engine/renderers/Indicator/scale/fastk_scale.ts +19 -19
  171. package/src/engine/renderers/Indicator/scale/indicator_scale.ts +204 -204
  172. package/src/engine/renderers/Indicator/scale/kst_scale.ts +19 -19
  173. package/src/engine/renderers/Indicator/scale/macd_scale.ts +22 -22
  174. package/src/engine/renderers/Indicator/scale/mom_scale.ts +19 -19
  175. package/src/engine/renderers/Indicator/scale/rsi_scale.ts +19 -19
  176. package/src/engine/renderers/Indicator/scale/stoch_scale.ts +19 -19
  177. package/src/engine/renderers/Indicator/scale/volume_scale.ts +26 -26
  178. package/src/engine/renderers/Indicator/scale/wmsr_scale.ts +19 -19
  179. package/src/engine/renderers/Indicator/stoch.ts +359 -359
  180. package/src/engine/renderers/Indicator/structure.ts +126 -126
  181. package/src/engine/renderers/Indicator/subPaneConfig.ts +265 -265
  182. package/src/engine/renderers/Indicator/supertrend.ts +115 -115
  183. package/src/engine/renderers/Indicator/tema.ts +136 -136
  184. package/src/engine/renderers/Indicator/trix.ts +158 -158
  185. package/src/engine/renderers/Indicator/vma.ts +124 -124
  186. package/src/engine/renderers/Indicator/volumeProfile.ts +125 -125
  187. package/src/engine/renderers/Indicator/vwap.ts +123 -123
  188. package/src/engine/renderers/Indicator/wma.ts +136 -136
  189. package/src/engine/renderers/Indicator/wmsr.ts +328 -328
  190. package/src/engine/renderers/Indicator/zones.ts +104 -104
  191. package/src/engine/renderers/__tests__/boll.renderer.test.ts +314 -314
  192. package/src/engine/renderers/__tests__/ene.renderer.test.ts +305 -305
  193. package/src/engine/renderers/__tests__/expma.renderer.test.ts +279 -279
  194. package/src/engine/renderers/__tests__/ma.renderer.test.ts +426 -426
  195. package/src/engine/renderers/__tests__/mainIndicatorLegend.renderer.test.ts +502 -502
  196. package/src/engine/renderers/__tests__/yAxis.renderer.test.ts +173 -173
  197. package/src/engine/renderers/candle.ts +459 -459
  198. package/src/engine/renderers/crosshair.ts +69 -69
  199. package/src/engine/renderers/customMarkers.ts +162 -162
  200. package/src/engine/renderers/extremaMarkers.ts +246 -246
  201. package/src/engine/renderers/gridLines.ts +90 -90
  202. package/src/engine/renderers/lastPrice.ts +97 -97
  203. package/src/engine/renderers/paneTitle.ts +136 -136
  204. package/src/engine/renderers/subVolume.ts +236 -236
  205. package/src/engine/renderers/timeAxis.ts +121 -121
  206. package/src/engine/renderers/webgl/candleSurface.ts +955 -955
  207. package/src/engine/renderers/webgl/sharedWebGLSurface.ts +146 -146
  208. package/src/engine/renderers/yAxis.ts +105 -105
  209. package/src/engine/scale/__tests__/logFormula.spec.ts +148 -148
  210. package/src/engine/scale/logFormula.ts +130 -130
  211. package/src/engine/scale/price.ts +39 -39
  212. package/src/engine/scale/priceScale.ts +264 -264
  213. package/src/engine/subPaneManager.ts +442 -427
  214. package/src/engine/theme/colors.ts +642 -642
  215. package/src/engine/theme/fonts.ts +20 -20
  216. package/src/engine/utils/klineConfig.ts +49 -49
  217. package/src/engine/utils/tickCount.ts +11 -11
  218. package/src/engine/utils/tickPosition.ts +214 -214
  219. package/src/engine/utils/zoom.ts +83 -83
  220. package/src/engine/viewport/viewport.ts +67 -67
  221. package/src/index.ts +3 -3
  222. package/src/plugin/ConfigManager.ts +93 -93
  223. package/src/plugin/EventBus.ts +77 -77
  224. package/src/plugin/HookSystem.ts +106 -106
  225. package/src/plugin/PluginHost.ts +243 -243
  226. package/src/plugin/PluginRegistry.ts +92 -92
  227. package/src/plugin/StateStore.ts +73 -73
  228. package/src/plugin/index.ts +19 -19
  229. package/src/plugin/rendererPluginManager.ts +368 -368
  230. package/src/plugin/stateKeys.ts +8 -8
  231. package/src/plugin/types.ts +526 -526
  232. package/src/reactivity/index.ts +2 -2
  233. package/src/reactivity/signal.ts +119 -119
  234. package/src/semantic/controller.ts +251 -251
  235. package/src/semantic/drawShape.ts +260 -260
  236. package/src/semantic/index.ts +28 -28
  237. package/src/semantic/schema.json +256 -256
  238. package/src/semantic/types.ts +251 -251
  239. package/src/semantic/validator.ts +349 -349
  240. package/src/types/kLine.ts +13 -13
  241. package/src/types/price.ts +56 -56
  242. package/src/types/volumePrice.ts +33 -33
  243. package/src/utils/dateFormat.ts +208 -208
  244. package/src/utils/kLineDraw/axis.ts +562 -562
  245. package/src/utils/priceToY.ts +34 -34
  246. package/src/utils/volumePrice.ts +202 -202
  247. 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
+ }