traversalHelper.js 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. /*
  2. * Licensed to the Apache Software Foundation (ASF) under one
  3. * or more contributor license agreements. See the NOTICE file
  4. * distributed with this work for additional information
  5. * regarding copyright ownership. The ASF licenses this file
  6. * to you under the Apache License, Version 2.0 (the
  7. * "License"); you may not use this file except in compliance
  8. * with the License. You may obtain a copy of the License at
  9. *
  10. * http://www.apache.org/licenses/LICENSE-2.0
  11. *
  12. * Unless required by applicable law or agreed to in writing,
  13. * software distributed under the License is distributed on an
  14. * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15. * KIND, either express or implied. See the License for the
  16. * specific language governing permissions and limitations
  17. * under the License.
  18. */
  19. /*
  20. * Licensed to the Apache Software Foundation (ASF) under one
  21. * or more contributor license agreements. See the NOTICE file
  22. * distributed with this work for additional information
  23. * regarding copyright ownership. The ASF licenses this file
  24. * to you under the Apache License, Version 2.0 (the
  25. * "License"); you may not use this file except in compliance
  26. * with the License. You may obtain a copy of the License at
  27. *
  28. * http://www.apache.org/licenses/LICENSE-2.0
  29. *
  30. * Unless required by applicable law or agreed to in writing,
  31. * software distributed under the License is distributed on an
  32. * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  33. * KIND, either express or implied. See the License for the
  34. * specific language governing permissions and limitations
  35. * under the License.
  36. */
  37. /**
  38. * Traverse the tree from bottom to top and do something
  39. * @param {module:echarts/data/Tree~TreeNode} root The real root of the tree
  40. * @param {Function} callback
  41. */
  42. function eachAfter(root, callback, separation) {
  43. var nodes = [root];
  44. var next = [];
  45. var node;
  46. while (node = nodes.pop()) {
  47. // jshint ignore:line
  48. next.push(node);
  49. if (node.isExpand) {
  50. var children = node.children;
  51. if (children.length) {
  52. for (var i = 0; i < children.length; i++) {
  53. nodes.push(children[i]);
  54. }
  55. }
  56. }
  57. }
  58. while (node = next.pop()) {
  59. // jshint ignore:line
  60. callback(node, separation);
  61. }
  62. }
  63. /**
  64. * Traverse the tree from top to bottom and do something
  65. * @param {module:echarts/data/Tree~TreeNode} root The real root of the tree
  66. * @param {Function} callback
  67. */
  68. function eachBefore(root, callback) {
  69. var nodes = [root];
  70. var node;
  71. while (node = nodes.pop()) {
  72. // jshint ignore:line
  73. callback(node);
  74. if (node.isExpand) {
  75. var children = node.children;
  76. if (children.length) {
  77. for (var i = children.length - 1; i >= 0; i--) {
  78. nodes.push(children[i]);
  79. }
  80. }
  81. }
  82. }
  83. }
  84. exports.eachAfter = eachAfter;
  85. exports.eachBefore = eachBefore;