fix: provide a mock function to measure text width

antvis · Nov 10, 2023 · f0b4c13 · f0b4c13
1 parent 74abd22
commit f0b4c13
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@antv/component",
-  "version": "1.0.0-beta.1",
+  "version": "1.0.0-beta.2",
   "description": "Visualization components for AntV, based on G.",
   "license": "MIT",
   "main": "lib/index.js",

diff --git a/src/util/text.ts b/src/util/text.ts
@@ -2,6 +2,11 @@ import { isString, memoize } from '@antv/util';
 import type { DisplayObject, Text } from '../shapes';
 
 let ctx: CanvasRenderingContext2D;
+let mockMeasureTextWidth: ((text: string, fontSize: number) => number) | undefined;
+
+export function setMockMeasureTextWidth(mock: (text: string, fontSize: number) => number) {
+  mockMeasureTextWidth = mock;
+}
 
 /**
  * 计算文本在画布中的宽度
@@ -10,6 +15,11 @@ export const measureTextWidth = memoize(
   (text: string | Text, font?: any): number => {
     const content = isString(text) ? text : text.style.text.toString();
     const { fontSize, fontFamily, fontWeight, fontStyle, fontVariant } = font || getFont(text as Text);
+
+    if (mockMeasureTextWidth) {
+      return mockMeasureTextWidth(content, fontSize);
+    }
+
     if (!ctx) {
       ctx = document.createElement('canvas').getContext('2d') as CanvasRenderingContext2D;
     }