<template>
<div class="penetrate-chart">
<div id="penetrateChart">
</div>
</div>
</template>
<!--关联图谱图-->
<script>
import html2canvas from 'html2canvas'
// 过渡时间
const DURATION = 0
// 加减符号半径
const SYMBOLA_S_R = 9
// 公司
const COMPANY = '0'
// 人
const PERSON = '1'
export default {
props: {},
components: {},
data() {
return {
layoutTree: '',
diamonds: '',
d3: this.$d3,
i: 0,
hasChildNodeArr: [],
originDiamonds: '',
diagonalUp: '',
diagonalDown: '',
tree: {
'children': [{
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'children': [{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
'children': [{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}]
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}],
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小米金融科技有限公司',
'type': '0',
'Holding':0,
'children': [{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
},{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}]
}, {
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'Holding':0,
'children': [{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}]
}, {
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'Holding':0,
'children': []
}, {
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'Holding':0,
'children': []
}, {
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'Holding':0,
'children': []
}, {
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'Holding':0,
'children': [{
'scale': 100,
'money': '30000',
'Holding':0,
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
}, {
'scale': 100,
'money': '30000',
'Holding':0,
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
}, {
'scale': 100,
'money': '30000',
'Holding':0,
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
}, {
'scale': 100,
'money': '30000',
'Holding':0,
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
}]
}, {
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'Holding':0,
'children': [{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}]
}, {
'money': '20000',
'name': '重庆小米创业投资有限公司',
'scale': 100,
'type': '0',
'Holding':0,
'children': [{
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}, {
'scale': 100,
'money': '30000',
'name': '[注销]珠海小额贷款有限公司',
'type': '0',
'Holding':0,
}]
}],
'name': '小米科技有限责任公司',
'parents': [{
'money': '143934.04',
'name': '雷军',
'scale': 77.80,
'type': '1',
'Holding':0,
'children': [],
}, {
'money': '18724.35',
'parentMoney': 3000,
'name': '黎万强',
'scale': 10.12,
'type': '1',
'Holding':'',
'children': [],
}, {
'money': '18623',
'parentMoney': 3000,
'name': '洪峰',
'scale': 10.07,
'type': '1',
'Holding':'',
'children': [],
}, {
'money': '3718.4963',
'name': '刘德',
'scale': 2.01,
'type': '1',
'Holding':'',
'children': [],
}]
},
rootUp: '',
rootDown: '',
svg: '',
svgW: document.body.clientHeight,
svgH: document.body.clientWidth
}
},
beforeCreate() {
document.body.style.overflow = 'hidden'
},
beforeDestroy() {
document.body.style.overflow = 'auto'
},
mounted() {
this.init()
},
methods: {
init() {
let d3 = this.d3
// 强制横屏 所以取反
let svgW = this.svgW
let svgH = this.svgH
console.log(svgW + ':' + svgH)
// 方块形状
this.diamonds = {
w: 145,
h: 68,
intervalW: 200,
intervalH: 150
}
// 源头对象
this.originDiamonds = {
w: 190
}
this.layoutTree = d3.tree().nodeSize([this.diamonds.intervalW, this.diamonds.intervalH]).separation(() => 1);
// 主图
this.svg = d3.select('#penetrateChart').append('svg').attr('width', svgW).attr('height', svgH).attr('id',
'treesvg')
.call(d3.zoom().scaleExtent([0, 5]).on('zoom', () => {
// 设置缩放位置以及平移初始位置
this.svg.attr('transform', d3.event.transform.translate(svgW / 1, svgH / 5));
}))
.attr('style', 'position: relative;z-index: 2;')
.append('g').attr('id', 'g').attr('transform', 'translate(' + (svgW / 1) + ',' + (svgH / 5) + ')');
let upTree = null
let downTree = null
// 拷贝树的数据
Object.keys(this.tree).map(item => {
if (item === 'parents') {
upTree = JSON.parse(JSON.stringify(this.tree))
upTree.children = this.tree[item]
upTree.parents = null
} else if (item === 'children') {
downTree = JSON.parse(JSON.stringify(this.tree))
downTree.children = this.tree[item]
downTree.parents = null
}
})
// hierarchy 返回新的结构 x0,y0初始化起点坐标
this.rootUp = d3.hierarchy(upTree, d => d.children);
this.rootUp.x0 = 0
this.rootUp.y0 = 0
this.rootDown = d3.hierarchy(downTree, d => d.children);
this.rootDown.x0 = 0
this.rootDown.y0 = 0;
// 上 和 下 结构
let treeArr = [{
data: this.rootUp,
type: 'up'
},
{
data: this.rootDown,
type: 'down'
}
]
treeArr.map(item => {
if (item.data.children) {
item.data.children.forEach(this.collapse);
this.update(item.data, item.type, item.data)
}
})
},
/*
*[update 函数描述], [click 函数描述]
* @param {[Object]} source 第一次是初始源对象，后面是点击的对象
* @param {[String]} showtype up表示向上 down表示向下
* @param {[Object]} sourceTree 初始源对象
*/
update(source, showtype, sourceTree) {
let _this = this
if (source.parents === null) {
source.isOpen = !source.isOpen
}
let nodes
if (showtype === 'up') {
nodes = this.layoutTree(this.rootUp).descendants()
} else {
nodes = this.layoutTree(this.rootDown).descendants()
}
let links = nodes.slice(1);
nodes.forEach(d => {
d.y = d.depth * this.diamonds.intervalH;
});
let node = this.svg.selectAll('g.node' + showtype)
.data(nodes, d => d.id || (d.id = showtype + ++this.i));
let nodeEnter = node.enter().append('g')
.attr('class', d => showtype === 'up' && !d.depth ? 'hide-node' : 'node' + showtype)
.attr('transform', d => showtype === 'up' ? 'translate(' + d.x + ',' + -(d.y) + ')' : 'translate(' + d.x +
',' + d.y + ')')
.attr('opacity', d => showtype === 'up' && !d.depth ? (this.rootDown.data.children.length ? 0 : 1) : 1); // 拥有下部分则隐藏初始块
// 创建矩形
nodeEnter.append('rect')
.attr('type', d => d.id)
.attr('width', d => d.depth ? this.diamonds.w : this.originDiamonds.w)
.attr('height', d => d.depth ? (d.type === COMPANY ? this.diamonds.h : this.diamonds.h - 10) : 30)
.attr('x', d => d.depth ? -this.diamonds.w / 2 : -this.originDiamonds.w / 2)
.attr('y', d => d.depth ? showtype === 'up' ? -this.diamonds.h / 2 : 0 : -15)
.attr('stroke', d => d.data.type === COMPANY || !d.depth ? '#7A9EFF' : '#7A9EFF')
.attr('stroke-width', 1)
.attr('rx', 5)
.attr('ry', 5)
.style('fill', d => {
if (d.data.type === COMPANY || !d.depth) {
return d._children ? '#fff' : (d.depth ? '#fff' : '#7A9EFF')
} else if (d.data.type === PERSON || !d.depth) {
return d._children ? '#fff' : (d.depth ? '#F3F9FE' : '#7A9EFF')
}
});
// 创建圆 加减
nodeEnter.append('circle')
.attr('type', d => d.id || (d.id = showtype + 'text' + ++this.i))
.attr('r', (d) => d.depth ? (this.hasChildNodeArr.indexOf(d) === -1 ? 0 : SYMBOLA_S_R) : 0)
.attr('cy', d => d.depth ? showtype === 'up' ? -(SYMBOLA_S_R + this.diamonds.h / 2) : this.diamonds.h : 0)
.attr('cx', 0)
.attr('fill', d => d.children ? '#fff' : '#7A9EFF')
.attr('stroke', d => d._children || d.children ? '#7A9EFF' : '')
.on('click', function(d) {
_this.click(d, showtype, sourceTree)
setTimeout(() => {
if (document.querySelector(`text[type="${d.id}"]`).innerHTML === '-') {
d.isOpen = false
this.innerHTML = '+'
this.setAttribute('fill', '#7A9EFF')
document.querySelector(`text[type="${d.id}"]`).setAttribute('fill', '#fff')
document.querySelector(`rect[type="${d.id}"]`).setAttribute('style', 'fill:#fff')
document.querySelector(`text[type="${d.id}"]`).innerHTML = '+'
} else {
d.isOpen = true
this.setAttribute('fill', '#fff')
document.querySelector(`text[type="${d.id}"]`).setAttribute('fill', '#7A9EFF')
document.querySelector(`rect[type="${d.id}"]`).setAttribute('style', 'fill:#fff')
document.querySelector(`text[type="${d.id}"]`).innerHTML = '-'
}
}, DURATION)
});
// 持股比例
nodeEnter.append('g')
.attr('transform', () => 'translate(0,0)')
.append('text')
.attr('x', d => d.x > 0 ? (showtype === 'up' ? -30 : 30) : 30)
.attr('y', showtype === 'up' ? this.diamonds.h : -20)
.attr('text-anchor', 'middle')
.attr('fill', d => d.data.type === COMPANY ? '#7A9EFF' : '#7A9EFF')
.attr('opacity', d => !d.depth ? 0 : 1)
.text(d => showtype === 'up' ? d.data.scale === 0 ? '非公示' :d.data.Holding===2? '':d.data.scale + '%':'')
.style('font-size', '10px')
.style('font-family', 'Microsoft YaHei')
.style('font-weight', '400');
nodeEnter.append('g')
.attr('transform', () => 'translate(-27,-75)')
.append('text')
.attr('x', d => d.x > 0 ? (showtype === 'down' ? 35 : 100) : 30)
.attr('y', showtype === 'down' ? this.diamonds.h : -80)
.attr('text-anchor', 'middle')
.attr('fill', d => d.data.type === COMPANY ? '#7A9EFF' : '#7A9EFF')
.attr('opacity', d => !d.depth ? 0 : 1)
.text(d => showtype === 'down' ? d.data.scale === undefined ? '' : d.data.Holding === 0 ? '控股' +
'\xa0\xa0\xa0\xa0\xa0\xa0\xa0' + d.data.scale + '%' : '\xa0\xa0\xa0\xa0\xa0\xa0\xa0' + d.data.scale + '%' :
'')
.style('font-size', '10px')
.style('font-family', 'Microsoft YaHei')
.style('font-weight', '400');
// 公司名称
// y轴 否表源头的字体距离
nodeEnter.append('text')
.attr('x', 0)
.attr('y', d => {
// 如果是上半部分
if (showtype === 'up') {
// 如果是1层以上
if (d.depth) {
return -this.diamonds.h / 2
} else {
// 如果名字长度大于9个
if (d.data.name.length > 10) {
return -5
}
return 0
}
} else {
if (d.depth) {
return 0
} else {
if (d.data.name.length > 10) {
return -5
}
return 0
}
}
})
.attr('dy', d => d.depth ? (d.data.name.length > 10 ? '1.5em' : '2em') : '.3em')
.attr('text-anchor', 'middle')
.attr('fill', d => d.depth ? '#465166' : '#fff')
.text(d => (d.data.name.length > 10) ? d.data.name.substr(0, 10) : d.data.name)
.style('font-size', '12px')
.style('font-family', 'PingFangSC-Medium')
.style('font-weight', '800');
// 名称过长 第二段
nodeEnter.append('text')
.attr('x', 0)
.attr('y', d => {
// ? (d.depth ? -this.diamonds.h / 2 : 0) : 0
if (showtype === 'up') {
if (d.depth) {
return -this.diamonds.h / 2
}
return 9
} else {
if (!d.depth) {
return 8
}
return 0
}
})
.attr('dy', d => d.depth ? '3em' : '.3em')
.attr('text-anchor', 'middle')
.attr('fill', d => d.depth ? '#465166' : '#fff')
.text(d => {
// 索引从第19个开始截取有表示超出
if (d.data.name.substr(19, 1)) {
return d.data.name.substr(10, 11) + '...'
} else if (d.data.name.substr(19, 1)) {
}
return d.data.name.substr(10, 11)
})
.style('font-size', '12px')
.style('font-family', 'PingFangSC-Medium')
.style('font-weight', '800');
// 认缴金额
nodeEnter.append('text')
.attr('x', 0)
.attr('y', showtype === 'up' ? -this.diamonds.h / 2 : 0)
.attr('dy', d => d.data.name.substr(9, d.data.name.length).length ? '5em' : '4em')
.attr('text-anchor', 'middle')
.attr('fill', d => d.depth ? '#465166' : '#fff')
.text(d => d.data.money ? d.data.money === '' ? '认缴金额：非公示' :`认缴金额：${d.data.money}万元` : '')
.style('font-size', '10px')
.style('font-family', 'PingFangSC-Regular')
.style('font-weight', '500')
.style('color', 'rgba(70,81,102,1)');
/*
* 绘制箭头
* @param {string} markerUnits [设置为strokeWidth箭头会随着线的粗细发生变化]
* @param {string} viewBox 坐标系的区域
* @param {number} markerWidth,markerHeight 标识的大小
* @param {string} orient 绘制方向，可设定为：auto（自动确认方向）和 角度值
* @param {number} stroke-width 箭头宽度
* @param {string} d 箭头的路径
* @param {string} fill 箭头颜色
* @param {string} id resolved0表示公司 resolved1表示个人
* 直接用一个marker达不到两种颜色都展示的效果
*/
nodeEnter.append('marker')
.attr('id', showtype + 'resolved0')
.attr('markerUnits', 'strokeWidth')
.attr('markerUnits', 'userSpaceOnUse')
.attr('viewBox', '0 -5 10 10')
.attr('markerWidth', 12)
.attr('markerHeight', 12)
.attr('orient', '90')
.attr('refX', () => showtype === 'up' ? '-5' : '15')
.attr('stroke-width', 2)
.attr('fill', '#7a9eff')
.append('path')
.attr('d', 'M0,-5L10,0L0,5')
.attr('fill', '#7a9eff');
nodeEnter.append('marker')
.attr('id', showtype + 'resolved1')
.attr('markerUnits', 'strokeWidth')
.attr('markerUnits', 'userSpaceOnUse')
.attr('viewBox', '0 -5 10 10')
.attr('markerWidth', 12)
.attr('markerHeight', 12)
.attr('orient', '90')
.attr('refX', () => showtype === 'up' ? '-5' : '15')
.attr('stroke-width', 2)
.attr('fill', '#7a9eff')
.append('path')
.attr('d', 'M0,-5L10,0L0,5')
.attr('fill', '#7A9EFF');
// 将节点转换到它们的新位置。
let nodeUpdate = node.transition()
.duration(DURATION)
.attr('transform', d => showtype === 'up' ? 'translate(' + d.x + ',' + -(d.y) + ')' : 'translate(' + d.x +
',' + (d.y) + ')');
// 代表是否展开的+-号,function this指向当前dom
nodeEnter.append('svg:text')
.attr('type', d => d.id || (d.id = showtype + 'text' + ++this.i))
.on('click', function(d) {
_this.click(d, showtype, sourceTree)
setTimeout(() => {
if (this.innerHTML === '-') {
d.isOpen = false
this.innerHTML = '+'
this.setAttribute('fill', '#fff')
document.querySelector(`circle[type="${d.id}"]`).setAttribute('fill', '#7A9EFF')
document.querySelector(`rect[type="${d.id}"]`).setAttribute('style', 'fill:#fff')
} else {
d.isOpen = true
this.innerHTML = '-'
this.setAttribute('fill', '#7A9EFF')
document.querySelector(`circle[type="${d.id}"]`).setAttribute('fill', '#fff')
document.querySelector(`rect[type="${d.id}"]`).setAttribute('style', 'fill:#fff')
}
}, DURATION)
})
.attr('x', 0)
.attr('dy', d => d.depth ? (showtype === 'up' ? -(SYMBOLA_S_R / 2 + this.diamonds.h / 2) : this.diamonds.h +
4) : 0)
.attr('text-anchor', 'middle')
.attr('fill', d => d._children ? '#fff' : '#7A9EFF')
.text(d => this.hasChildNodeArr.indexOf(d) !== -1 ? (source.depth && d.isOpen ? '-' : '+') : '');
// 将退出节点转换到父节点的新位置.
let nodeExit = node.exit().transition()
.duration(DURATION)
.attr('transform', () => showtype === 'up' ? 'translate(' + source.x + ',' + -(source.y) + ')' : 'translate(' +
source.x + ',' + (parseInt(source.y)) + ')')
.remove();
nodeExit.select('rect')
.attr('width', this.diamonds.w)
.attr('height', this.diamonds.h)
.attr('stroke', 'black')
.attr('stroke-width', 1);
// 修改线条
let link = this.svg.selectAll('path.link' + showtype)
.data(links, d => d.id);
// 在父级前的位置画线。
let linkEnter = link.enter().insert('path', 'g')
.attr('class', 'link' + showtype)
.attr('marker-start', d => `url(#${showtype}resolved${d.data.type})`) // 根据箭头标记的id号标记箭头
.attr('stroke', d => d.data.type === COMPANY ? '#7A9EFF' : '#7A9EFF')
.style('fill-opacity', 1)
.attr('fill', 'none')
.attr('stroke-width', '1px')
.attr('d', () => {
let o = {
x: source.x0,
y: source.y0
};
return _this.diagonal(o, o, showtype)
});
let linkUpdate = linkEnter.merge(link);
// 过渡更新位置.
linkUpdate.transition()
.duration(DURATION)
.attr('d', d => _this.diagonal(d, d.parent, showtype));
// 将退出节点转换到父节点的新位置
link.exit().transition()
.duration(DURATION)
.attr('d', () => {
let o = {
x: source.x,
y: source.y
};
return _this.diagonal(o, o, showtype)
}).remove();
// 隐藏旧位置方面过渡.
nodes.forEach(d => {
d.x0 = d.x;
d.y0 = d.y
});
},
// 拷贝到_children 隐藏1排以后的树
collapse(source) {
if (source.children) {
source._children = source.children;
source._children.forEach(this.collapse);
source.children = null;
this.hasChildNodeArr.push(source);
}
},
click(source, showType, sourceTree) {
// 不是起点才能点
if (source.depth) {
if (source.children) {
source._children = source.children;
source.children = null;
} else {
source.children = source._children;
source._children = null;
}
this.update(source, showType, sourceTree)
}
},
diagonal(s, d, showtype) {
let path
if (showtype === 'up') {
path =
`M ${s.x} ${-s.y + 24}
C${s.x} -${(s.y + d.y) * 0.45},
${s.x} -${(s.y + d.y) * 0.45},
${d.x} -${d.y}`;
} else {
path =
`M ${s.x} ${s.y}
C${s.x} ${(s.y + d.y) * 0.45},
${s.x} ${(s.y + d.y) * 0.45},
${d.x} ${d.y}`;
}
return path;
},
saveImg() {
html2canvas(document.getElementById('penetrateChart')).then((canvas) => {
const context = canvas.getContext('2d');
context.mozImageSmoothingEnabled = false;
context.webkitImageSmoothingEnabled = false;
context.msImageSmoothingEnabled = false;
context.imageSmoothingEnabled = false;
let base64File = canvas.toDataURL('image/jpg', 1.0);
base64File = base64File.replace(/^.*?,/, '')
console.log(base64File)
});
},
resetSvg() {
this.d3.select('#treesvg').remove()
this.init()
}
}
}
</script>
<style lang="less">
#treesvg {
display: block;
margin: auto;
width: 100%;
height: 100%;
}
</style>
效果图：