vectorix/v_image.cpp at master · hadrava/vectorix · 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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*
 * Vectorix -- line-based image vectorizer
 * (c) 2016 Jan Hadrava <had@atrey.karlin.mff.cuni.cz>
 */
#include "v_image.h"
#include "parameters.h"
#include <list>
#include <cmath>
#include <cstdlib>
#include <cstdarg>
#include <cstdio>

namespace vectorix {

void v_image::clean() { // Remove all lines
	for (v_line _line: line) {
		_line.segment.erase(_line.segment.begin(), _line.segment.end());
	}
	line.erase(line.begin(), line.end());
	for (v_line _line: debug_line) {
		_line.segment.erase(_line.segment.begin(), _line.segment.end());
	}
	debug_line.erase(debug_line.begin(), debug_line.end());
}

v_smooth_type v_point::get_smooth() {
	v_pt prev = control_prev - main;
	v_pt next = control_next - main;

	p pl = prev.len();
	p nl = next.len();

	if ((pl > epsilon) && (nl > epsilon)) {
		//Normalize:
		prev /= pl;
		next /= nl;

		prev += next; // vector will be zero iff prev and next are heading in opposite direction
		if (prev.len() <= epsilon) {
			if ((pl - nl > epsilon) || (nl - pl > epsilon))
				return v_smooth_type::smooth;
			else
				return v_smooth_type::symetric;
		}
	}
	return v_smooth_type::corner;
}

v_line::v_line(p x0, p y0, p x1, p y1, p x2, p y2, p x3, p y3) { // New line with control points: main(0), control(1) -- control(2), main(3)
	segment.emplace_back(v_point(v_pt(x0,y0), v_pt(x0,y0), v_pt(x1,y1)));
	segment.emplace_back(v_point(v_pt(x2,y2), v_pt(x3,y3), v_pt(x3,y3)));
	type_ = v_line_type::stroke;
}

void v_line::add_point(v_pt _p_control_next, v_pt _control_prev, v_pt _main) { // Add segment to end of a line (control_prev is added to previous point)
	segment.back().control_next = _p_control_next;
	segment.emplace_back(v_point(_control_prev, _main, _main));
}

void v_line::add_point(v_pt _p_control_next, v_pt _control_prev, v_pt _main, v_co _color, p _width) { // Add segment with specified color and width
	segment.back().control_next = _p_control_next;
	segment.emplace_back(v_point(_control_prev, _main, _main, _color, _width));
}

void v_line::add_point(v_pt _main) { // Add point (straight continuation)
	segment.emplace_back(v_point(_main, _main, _main));
}

void v_line::add_point(v_pt _main, v_co _color, p _width) { // Add point with color and width
	segment.emplace_back(v_point(_main, _main, _main, _color, _width));
}

void v_line::reverse() { // Reverse line
	segment.reverse();
	for (auto control = segment.begin(); control != segment.end(); control++) {
		std::swap(control->control_next, control->control_prev); // Swap control points
	}
};

void v_image::add_line(v_line _line) { // Add bezier curve to image
	line.push_back(_line);
}

void v_line::set_color(v_co color) { // Set color to each point of line
	for (auto pt = segment.begin(); pt != segment.end(); pt++) {
		pt->color = color;
	}
}

void v_image::false_colors(p hue_step) { // Colorize each line with saturated color
	p hue = 0;
	for (auto l = line.begin(); l != line.end(); l++) {
		v_co col;
		col.saturated_hue(hue); // Calculate color from hue
		l->set_color(col); // Set color for current line
		hue += hue_step; // Increase hue angle
		hue = fmodf (hue, 360);
	}
}

void v_image::force_black() {
	v_co col(0, 0, 0);
	for (auto l = line.begin(); l != line.end(); l++) {
		l->set_color(col); // Set color for current line
	}
}

void v_image::add_debug_line(v_pt a, v_pt b) {
	v_line lin;
	lin.add_point(a);
	lin.add_point(b);
	lin.set_color(v_co(255, 0, 0));
	debug_line.push_back(lin);
}

void v_image::show_debug_lines() {
	line.splice(line.end(), debug_line);
}

}; // namespace