'use strict';

const DirectedGraphTracer = require('./directed_graph');

class UndirectedGraphTracer extends DirectedGraphTracer {

static getClassName() {

return 'UndirectedGraphTracer';

}

constructor(name) {

super(name);

if (this.isNew) initView(this);

}

setTreeData(G, root) {

return super.setTreeData(G, root, true);

}

setData(G) {

return super.setData(G, true);

}

e(v1, v2) {

if (v1 > v2) {

var temp = v1;

v1 = v2;

v2 = temp;

}

return 'e' + v1 + '_' + v2;

}

drawOnHover(node, context, settings, next) {

var tracer = this;

context.setLineDash([5, 5]);

var nodeIdx = node.id.substring(1);

this.graph.edges().forEach(function (edge) {

var ends = edge.id.substring(1).split("_");

if (ends[0] == nodeIdx) {

var color = '#0ff';

var source = node;

var target = tracer.graph.nodes('n' + ends[1]);

tracer.drawEdge(edge, source, target, color, context, settings);

if (next) next(edge, source, target, color, context, settings);

} else if (ends[1] == nodeIdx) {

var color = '#0ff';

var source = tracer.graph.nodes('n' + ends[0]);

var target = node;

tracer.drawEdge(edge, source, target, color, context, settings);

if (next) next(edge, source, target, color, context, settings);

}

});

}

drawEdge(edge, source, target, color, context, settings) {

var prefix = settings('prefix') || '',

size = edge[prefix + 'size'] || 1;

context.strokeStyle = color;

context.lineWidth = size;

context.beginPath();

context.moveTo(

source[prefix + 'x'],

source[prefix + 'y']

);

context.lineTo(

target[prefix + 'x'],

target[prefix + 'y']

);

context.stroke();

}

}

const initView = (tracer) => {

tracer.s.settings({

defaultEdgeType: 'def',

funcEdgesDef(edge, source, target, context, settings) {

var color = tracer.getColor(edge, source, target, settings);

tracer.drawEdge(edge, source, target, color, context, settings);

}

});

};

module.exports = UndirectedGraphTracer;