ImgGraph/dot.cpp at master · shikbupt/ImgGraph · 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
#include "dot.h"
#include "graph.h"

void Dot::Edges2Dot() {
	for (Graph::ConstSuccessorsIter successors_iter = graph_->Nbegin();
		successors_iter != graph_->Nend(); ++successors_iter) {
			for (Graph::ConstNodesIter nodes_iter = successors_iter->nodes_list->begin();
				nodes_iter != successors_iter->nodes_list->end(); ++nodes_iter
				) {
					if (nodes_iter->stat.is_undot) {
						string_buffer_ << successors_iter->node_value << "--" <<nodes_iter->node_value
							<< "[" << "label=\"" << nodes_iter->edge_value<< "\"];" <<'\n';
						graph_->SetNodeStat(successors_iter->node_value, nodes_iter->node_value, false);
					}

			}
	}
}

void Dot::Nodes2Dot() {
	for (Graph::ConstSuccessorsIter successors_iter = graph_->Nbegin();
		successors_iter != graph_->Nend(); ++successors_iter) {
			string_buffer_ << successors_iter->node_value << ";" << '\n';
	}
}

void Dot::CreatDotFile(const string& file_path) {
	Nodes2Dot();
	Edges2Dot();
	fstream out_file(file_path, std::ios::out);
	out_file << "graph main {\n" << string_buffer_.str() << '}';
}

void Dot::SetStat() {
	for (Graph::ConstSuccessorsIter successors_iter = graph_->Nbegin();
		successors_iter != graph_->Nend(); ++successors_iter) {
			for (Graph::ConstNodesIter nodes_iter = successors_iter->nodes_list->begin();
				nodes_iter != successors_iter->nodes_list->end(); ++nodes_iter) {
					graph_->SetNodeStat(successors_iter->node_value, nodes_iter->node_value, true);
				}
	}
}