VisualNet/SVG_Bbox_Script.js at master · weiyx16/VisualNet · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
//
//     Fetch bbox of svg elements using webdriver + javascript
//     @ Yixuan Wei
//     @ weiyx16@mails.tsinghua.edu.cn
//


var bboxes = {};
var figure_type = '%s'; // need to assign from outside
var drawing_obj_id = '%s'; // need to assign from outside
figure_show_bbox = document.documentElement.getBoundingClientRect();
bboxes['figure_show'] = [figure_show_bbox['x'], figure_show_bbox['y'], figure_show_bbox['width'], figure_show_bbox['height']];
var layer_nodes = [];
var next_layer_nodes = [];
var bar_num = -1;
var bar_group = drawing_obj_id + 0;
// document.documentElement == svg
// document.documentElement.children[1] == ('the_figure')
layer_nodes.push(document.documentElement.children[1]);
while (layer_nodes.length != 0) {
    for (var node_idx = 0; node_idx<layer_nodes.length; node_idx++){
        var current_node = layer_nodes[node_idx];
        var cur_SVGRect = current_node.getBBox();

        // fetch path for line/dot element in path of a drawing object
        if (current_node.id.startsWith(drawing_obj_id)){
            if (figure_type == 'Line_chart' || figure_type == 'Area_chart'){
                bboxes[current_node.id] = [cur_SVGRect['x'], cur_SVGRect['y'],
                        cur_SVGRect['width']+cur_SVGRect['x'], cur_SVGRect['height']+cur_SVGRect['y']];
                try {
                    bboxes[current_node.id + '_path'] = current_node.getElementsByTagName('path')[0].getAttribute('d');
                } catch(error) {
                    console.error(error);  // expected output: TypeError: Cannot read property 'getAttribute' of undefined
                }
            }
            if (figure_type === 'Scatter_chart') {
                bboxes[current_node.id] = [cur_SVGRect['x'], cur_SVGRect['y'],
                        cur_SVGRect['width']+cur_SVGRect['x'], cur_SVGRect['height']+cur_SVGRect['y']];
                /*
                // get path for line
                try {
                    bboxes[current_node.id + '_path'] = current_node.getElementsByTagName('path')[0].getAttribute('d');
                } catch(error) {
                    console.error(error);  // expected output: TypeError: Cannot read property 'getAttribute' of undefined
                }
                */
                // get path for scatter dot
                try {
                    var dots = current_node.getElementsByTagName('g')[0];
                    var tmp_bbox = {};
                    for (var dot_num=0; dot_num<dots.childElementCount; dot_num++){
                        var tmp_SVGRect = dots.children[dot_num].getBBox();
                        tmp_bbox[current_node.id + '_Dot_' + dot_num] = [tmp_SVGRect['x'], tmp_SVGRect['y'],
                            tmp_SVGRect['width']+tmp_SVGRect['x'], tmp_SVGRect['height']+tmp_SVGRect['y']];
                    }
                    bboxes[current_node.id + '_path'] = tmp_bbox;
                } catch (error) {
                    console.error(error);  // expected output: TypeError: Cannot read property 'getAttribute' of undefined
                }
            }
            if (figure_type == 'Bar_chart'){
                if (current_node.id == bar_group){
                    bar_num ++;
                }
                else{
                    bar_num = 0;
                    bar_group = current_node.id;
                }
                bboxes[current_node.id + '_bbox_' + bar_num] = [cur_SVGRect['x'], cur_SVGRect['y'],
                        cur_SVGRect['width']+cur_SVGRect['x'], cur_SVGRect['height']+cur_SVGRect['y']];
            }
            if (figure_type == 'Pie_chart'){
                bboxes[current_node.id] = [cur_SVGRect['x'], cur_SVGRect['y'],
                    cur_SVGRect['width']+cur_SVGRect['x'], cur_SVGRect['height']+cur_SVGRect['y']];
            }
        }
        else{
            bboxes[current_node.id] = [cur_SVGRect['x'], cur_SVGRect['y'],
                    cur_SVGRect['width']+cur_SVGRect['x'], cur_SVGRect['height']+cur_SVGRect['y']];
        }
        for (var next_node_idx = 0;
            next_node_idx<current_node.childElementCount;
            next_node_idx++){
            var next_node = layer_nodes[node_idx].children[next_node_idx];
            if (next_node.id.startsWith('the')){
                next_layer_nodes.push(next_node);
            }
        }
    }
    layer_nodes = next_layer_nodes;
    next_layer_nodes = [];
}
// console.log(bboxes);
// return bboxes;  // for use in python through web driver