feat: 词云图重写changeData方法 && 添加单测 (#2214)

Co-authored-by: 沈建斌 <[email protected]>

Author: yp0413150120

__tests__/unit/plots/word-cloud/change-data-spec.ts

@@ -0,0 +1,44 @@
+import { WordCloud } from '../../../../src';
+import { CountryEconomy } from '../../../data/country-economy';
+import { createDiv } from '../../../utils/dom';
+
+describe('word-cloud changeData', () => {
+  it('changeData: normal', () => {
+    const cloud = new WordCloud(createDiv(), {
+      width: 1024,
+      height: 1024,
+      data: CountryEconomy,
+      wordField: 'Country',
+      weightField: 'GDP',
+    });
+
+    cloud.render();
+
+    expect(cloud.chart.geometries[0].elements.length).toEqual(CountryEconomy.length + 2);
+
+    const newData = CountryEconomy.slice(0, 20);
+    cloud.changeData(newData);
+    expect(cloud.chart.geometries[0].elements.length).toEqual(newData.length + 2);
+    expect(cloud.options.data).toEqual(newData);
+
+    cloud.destroy();
+  });
+
+  it('changeData: from empty to have data', () => {
+    const cloud = new WordCloud(createDiv(), {
+      width: 1024,
+      height: 1024,
+      data: [],
+      wordField: 'Country',
+      weightField: 'GDP',
+    });
+
+    cloud.render();
+
+    cloud.changeData(CountryEconomy);
+    expect(cloud.chart.geometries[0].elements.length).toEqual(CountryEconomy.length + 2);
+    expect(cloud.options.data).toEqual(CountryEconomy);
+
+    cloud.destroy();
+  });
+});

src/plots/word-cloud/index.ts

@@ -4,7 +4,7 @@ import { Datum } from '../../types';
 import { deepAssign } from '../../utils';
 import { WordCloudOptions } from './types';
 import { adaptor } from './adaptor';
-import { processImageMask } from './utils';
+import { processImageMask, transform } from './utils';
 // 注册的shape
 import './shapes/word-cloud';
 
@@ -14,6 +14,15 @@ export class WordCloud extends Plot<WordCloudOptions> {
   /** 词云图 */
   public type: string = 'word-cloud';
 
+  /**
+   * @override
+   * @param data
+   */
+  public changeData(data) {
+    this.updateOption({ data });
+    this.chart.changeData(transform({ chart: this.chart, options: this.options }));
+  }
+
   /**
    * 获取默认的 options 配置项
    */