/* *
|
* @Author: jianglei
|
* @Date: 2017-10-12 12:06:49
|
*/
|
"use strict";
|
import Vue from "vue";
|
export default function treeToArray(data, expandAll, parent, level, item) {
|
const marLTemp = [];
|
let tmp = [];
|
Array.from(data).forEach(function(record) {
|
if (record._expanded === undefined) {
|
Vue.set(record, "_expanded", expandAll);
|
}
|
let _level = 1;
|
if (level !== undefined && level !== null) {
|
_level = level + 1;
|
}
|
Vue.set(record, "_level", _level);
|
// 如果有父元素
|
if (parent) {
|
Vue.set(record, "parent", parent);
|
// 如果父元素有偏移量，需要计算在this的偏移量中
|
// 偏移量还与前面同级元素有关，需要加上前面所有元素的长度和
|
if (!marLTemp[_level]) {
|
marLTemp[_level] = 0;
|
}
|
Vue.set(record, "_marginLeft", marLTemp[_level] + parent._marginLeft);
|
Vue.set(record, "_width", record[item] / parent[item] * parent._width);
|
// 在本次计算过偏移量后加上自己长度，以供下一个元素使用
|
marLTemp[_level] += record._width;
|
} else {
|
// 如果为根
|
// 初始化偏移量存储map
|
marLTemp[record.id] = [];
|
// map中是一个数组，存储的是每级的长度和
|
// 初始情况下为0
|
marLTemp[record.id][_level] = 0;
|
Vue.set(record, "_marginLeft", 0);
|
Vue.set(record, "_width", 1);
|
}
|
tmp.push(record);
|
if (record.children && record.children.length > 0) {
|
const children = treeToArray(record.children, expandAll, record, _level, item);
|
tmp = tmp.concat(children);
|
}
|
});
|
return tmp;
|
}
|
