capSTONE/visualization.cpp at main · yi-jongwoo/capSTONE · 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
#include "stone_internal_header.h"

void make_svg_undirected(string filename,int w,int h,int n,vector<int>* arr,int* x,int* y){
	ofstream fout(filename);
	fout<<"<html><body>\n";

	fout<<"<svg width=\""<<w<<"\" height=\""<<h<<"\">\n";

	for(int i=0;i<n;i++)for(int j:arr[i])
		fout<<"<line x1=\""<<x[i]<<"\" y1=\""<<y[i]
			<<"\" x2=\""<<x[j]<<"\" y2=\""<<y[j]
			<<"\" style=\"stroke:red;stroke-width:2\" />\n";

	for(int i=0;i<n;i++)
		fout<<"<circle cx=\""<<x[i]<<"\" cy=\""<<y[i]
			<<"\" r=\"4\" fill=\"blue\" />\n";
	for(int i=0;i<n;i++)
		fout<<"<text x=\""<<x[i]+2<<"\" y=\""<<y[i]
			<<"\" fill=\"white\" stroke=\"red\" font-weight=\"bold\">"<<i<<"</text>\n";
	fout<<"</svg></body></html>";
}

void make_svg(string filename,int w,int h,int n,vector<int>* arr,int* x,int* y){
    ofstream fout(filename);
    fout<<"<html><body>\n";

    fout<<"<svg width=\""<<w<<"\" height=\""<<h<<"\">\n";

    fout<<"<defs><marker id='head' orient=\"auto\"markerWidth='1.5' markerHeight='6'\
refX='3' refY='2' viewBox=\"0 0 2 1\"><path d='M0,0 V4 L2,2 Z' fill=\"black\"/></marker></defs>\n";

    for(int i=0;i<n;i++)
        for(int j:arr[i])
            fout<<"<path id='arrow-line' marker-end='url(#head)' stroke-width='4' \
fill='none' stroke='#404040' d='M"<<x[i]<<","<<y[i]<<", "<<x[j]<<" "<<y[j]<<",120'/>\n";
    for(int i=0;i<n;i++)
        fout<<"<circle cx=\""<<x[i]<<"\" cy=\""<<y[i]
            <<"\" r=\"4\" fill=\"blue\" />\n";
    for(int i=0;i<n;i++)
        fout<<"<text x=\""<<x[i]+2<<"\" y=\""<<y[i]
            <<"\" fill=\"white\" font-size=\"24\" stroke=\"red\" font-weight=\"bold\">" <<i<<"</text>\n";
    fout<<"</svg></body></html>";
}