elementHighlight.js 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
  1. var __rest = (this && this.__rest) || function (s, e) {
  2. var t = {};
  3. for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
  4. t[p] = s[p];
  5. if (s != null && typeof Object.getOwnPropertySymbols === "function")
  6. for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
  7. if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
  8. t[p[i]] = s[p[i]];
  9. }
  10. return t;
  11. };
  12. import { deepMix } from '@antv/util';
  13. import { group } from 'd3-array';
  14. import { subObject } from '../utils/helper';
  15. import { createDatumof, createValueof, mergeState, offsetTransform, renderBackground, renderLink, selectElementByData, selectG2Elements, selectPlotArea, useState, } from './utils';
  16. /**
  17. * highlight a group of elements.
  18. */
  19. export function elementHighlight(root, { elements: elementsof, // given the root of chart returns elements to be manipulated
  20. datum, // given each element returns the datum of it
  21. groupKey = (d) => d, // group elements by specified key
  22. link = false, // draw link or not
  23. background = false, // draw background or not
  24. delay = 60, // delay to unhighlighted element
  25. scale, coordinate, emitter, state = {}, }) {
  26. var _a;
  27. const elements = elementsof(root);
  28. const elementSet = new Set(elements);
  29. const keyGroup = group(elements, groupKey);
  30. const valueof = createValueof(elements, datum);
  31. const [appendLink, removeLink] = renderLink(Object.assign({ elements,
  32. valueof,
  33. link,
  34. coordinate }, subObject(state.active, 'link')));
  35. const [appendBackground, removeBackground, isBackground] = renderBackground(Object.assign({ scale,
  36. coordinate,
  37. background,
  38. valueof }, subObject(state.active, 'background')));
  39. const elementStyle = deepMix(state, {
  40. active: Object.assign({}, (((_a = state.active) === null || _a === void 0 ? void 0 : _a.offset) && {
  41. //Apply translate to mock slice out.
  42. transform: (...params) => {
  43. const value = state.active.offset(...params);
  44. const [, i] = params;
  45. return offsetTransform(elements[i], value, coordinate);
  46. },
  47. })),
  48. });
  49. const { setState, removeState, hasState } = useState(elementStyle, valueof);
  50. let out; // Timer for delaying unhighlighted.
  51. const pointerover = (event) => {
  52. const { target: element, nativeEvent = true } = event;
  53. if (!elementSet.has(element))
  54. return;
  55. if (out)
  56. clearTimeout(out);
  57. const k = groupKey(element);
  58. const group = keyGroup.get(k);
  59. const groupSet = new Set(group);
  60. for (const e of elements) {
  61. if (groupSet.has(e)) {
  62. if (!hasState(e, 'active'))
  63. setState(e, 'active');
  64. }
  65. else {
  66. setState(e, 'inactive');
  67. removeLink(e);
  68. }
  69. if (e !== element)
  70. removeBackground(e);
  71. }
  72. appendBackground(element);
  73. appendLink(group);
  74. // Emit events.
  75. if (!nativeEvent)
  76. return;
  77. emitter.emit('element:highlight', {
  78. nativeEvent,
  79. data: {
  80. data: datum(element),
  81. group: group.map(datum),
  82. },
  83. });
  84. };
  85. const delayUnhighlighted = () => {
  86. if (out)
  87. clearTimeout(out);
  88. out = setTimeout(() => {
  89. unhighlighted();
  90. out = null;
  91. }, delay);
  92. };
  93. const unhighlighted = (nativeEvent = true) => {
  94. for (const e of elements) {
  95. removeState(e, 'active', 'inactive');
  96. removeBackground(e);
  97. removeLink(e);
  98. }
  99. if (nativeEvent) {
  100. emitter.emit('element:unhighlight', { nativeEvent });
  101. }
  102. };
  103. const pointerout = (event) => {
  104. const { target: element } = event;
  105. if (background && !isBackground(element))
  106. return;
  107. if (!background && !elementSet.has(element))
  108. return;
  109. if (delay > 0)
  110. delayUnhighlighted();
  111. else
  112. unhighlighted();
  113. };
  114. const pointerleave = () => {
  115. unhighlighted();
  116. };
  117. root.addEventListener('pointerover', pointerover);
  118. root.addEventListener('pointerout', pointerout);
  119. root.addEventListener('pointerleave', pointerleave);
  120. const onRest = (e) => {
  121. const { nativeEvent } = e;
  122. if (nativeEvent)
  123. return;
  124. unhighlighted(false);
  125. };
  126. const onHighlight = (e) => {
  127. const { nativeEvent } = e;
  128. if (nativeEvent)
  129. return;
  130. const { data } = e.data;
  131. const element = selectElementByData(elements, data, datum);
  132. if (!element)
  133. return;
  134. pointerover({ target: element, nativeEvent: false });
  135. };
  136. emitter.on('element:highlight', onHighlight);
  137. emitter.on('element:unhighlight', onRest);
  138. return () => {
  139. root.removeEventListener('pointerover', pointerover);
  140. root.removeEventListener('pointerout', pointerout);
  141. root.removeEventListener('pointerleave', pointerleave);
  142. emitter.off('element:highlight', onHighlight);
  143. emitter.off('element:unhighlight', onRest);
  144. for (const e of elements) {
  145. removeBackground(e);
  146. removeLink(e);
  147. }
  148. };
  149. }
  150. export function ElementHighlight(_a) {
  151. var { delay, createGroup, background = false, link = false } = _a, rest = __rest(_a, ["delay", "createGroup", "background", "link"]);
  152. return (context, _, emitter) => {
  153. const { container, view, options } = context;
  154. const { scale, coordinate } = view;
  155. const plotArea = selectPlotArea(container);
  156. return elementHighlight(plotArea, Object.assign({ elements: selectG2Elements, datum: createDatumof(view), groupKey: createGroup ? createGroup(view) : undefined, coordinate,
  157. scale, state: mergeState(options, [
  158. ['active', background ? {} : { lineWidth: '1', stroke: '#000' }],
  159. 'inactive',
  160. ]), background,
  161. link,
  162. delay,
  163. emitter }, rest));
  164. };
  165. }
  166. //# sourceMappingURL=elementHighlight.js.map