Basic interface : can add obstacles ; set begin/end position, ...

1 files changed, 171 insertions, 5 deletions

diff --git a/ui.cpp b/ui.cpp
index 23a2b48..9c1fec7 100644
--- a/ui.cpp
+++ b/ui.cpp
@@ -1,19 +1,30 @@
+#include <iostream>
+
 #include "ui.hpp"
 
 /*
 	All modes :
 		- h, j, k, l : move around
 		- i, o : zoom in/out
-	
+	Main mode (command mode) :
+		- s : select start pos
+		- e : select end pos
+		- a : add obstacle
+		- d : delete obstacle under mouse pointer
+	Select modes : no other keys
 */
 
-UI::UI(hilare_a_param *p) {
+using namespace std;
+
+UI::UI(hilare_a_param *p) : _sel_obs(vec(0,0), 0) {
 	_param = p;
 
 	_p.begin_pos.param = _p.end_pos.param = p;
 
 	_view.x0 = _view.y0 = 0;
 	_view.zoom = 1;
+
+	_mode = M_NORMAL;
 }
 
 void UI::run() {
@@ -32,8 +43,24 @@ void UI::run() {
 			if (ev.type == sf::Event::TextEntered) {
 				int k = ev.text.unicode;
 
-				// TODO
+				if (k == 'h') {
+					_view.x0 -= 20 / _view.zoom;
+				} else if (k == 'j') {
+					_view.y0 += 20 / _view.zoom;
+				} else if (k == 'k') {
+					_view.y0 -= 20 / _view.zoom;
+				} else if (k == 'l') {
+					_view.x0 += 20 / _view.zoom;
+				} else if (k == 'i') {
+					_view.zoom *= 1.1;
+				} else if (k == 'o') {
+					_view.zoom /= 1.1;
+				}
 			}
+
+			if (_mode == M_NORMAL) handle_normal(ev);
+			if (_mode == M_INS_OBSTACLE) handle_ins_obs(ev);
+			if (_mode == M_SEL_BEGIN || _mode == M_SEL_END) handle_sel_pos(ev);
 		}
 
 		_win.clear(sf::Color::Black);
@@ -42,10 +69,113 @@ void UI::run() {
 		render_solution();
 		render_problem();
 
+		if (_mode == M_INS_OBSTACLE)
+			render_circle(_sel_obs, sf::Color::Transparent, sf::Color::White, 0);
+		if (_mode == M_SEL_BEGIN || _mode == M_SEL_END)
+			render_pos(_sel_pos, sf::Color::White);
+
 		_win.display();
 	}
 }
 
+void UI::handle_normal(const sf::Event &ev) {
+	if (ev.type == sf::Event::TextEntered) {
+		int k = ev.text.unicode;
+
+		if (k == 's') {
+			_mode = M_SEL_BEGIN;
+			_sel_step = S_XY;
+			_sel_pos = _p.begin_pos;
+			vec p = mouse_coord();
+			_sel_pos.x = p.x;
+			_sel_pos.y = p.y;
+		}
+		if (k == 'e') {
+			_mode = M_SEL_END;
+			_sel_step = S_XY;
+			_sel_pos = _p.end_pos;
+			vec p = mouse_coord();
+			_sel_pos.x = p.x;
+			_sel_pos.y = p.y;
+		}
+		if (k == 'a') {
+			_mode = M_INS_OBSTACLE;
+			_sel_step = S_XY;
+			_sel_obs.c = mouse_coord();
+			_sel_obs.r = 20 / _view.zoom;
+		}
+		if (k == 'd') {
+			vec p = mouse_coord();
+			vector<obstacle> o2;
+			for(auto i: _p.obstacles) {
+				if (!i.c.intersects(p)) o2.push_back(i);
+			}
+			_p.obstacles = o2;
+		}
+	}
+}
+
+void UI::handle_ins_obs(const sf::Event &ev) {
+	if (_sel_step == S_XY) {
+		if (ev.type == sf::Event::MouseMoved) {
+			_sel_obs.c = mouse_coord();
+		} else if (ev.type == sf::Event::MouseButtonReleased
+				&& ev.mouseButton.button == sf::Mouse::Button::Left) {
+			_sel_step = S_RADIUS;
+		}
+	} else {
+		if (ev.type == sf::Event::MouseMoved) {
+			_sel_obs.r = (mouse_coord() - _sel_obs.c).norm();
+		} else if (ev.type == sf::Event::MouseButtonReleased
+				&& ev.mouseButton.button == sf::Mouse::Button::Left) {
+			_p.obstacles.push_back(obstacle(_sel_obs));
+			_mode = M_NORMAL;
+		}
+	}
+
+	if (ev.type == sf::Event::MouseButtonReleased
+			&& ev.mouseButton.button == sf::Mouse::Button::Right) {
+		_mode = M_NORMAL;	// cancel out
+	}
+}
+
+void UI::handle_sel_pos(const sf::Event &ev) {
+	if (_sel_step == S_XY) {
+		if (ev.type == sf::Event::MouseMoved) {
+			vec p = mouse_coord();
+			_sel_pos.x = p.x;
+			_sel_pos.y = p.y;
+		} else if (ev.type == sf::Event::MouseButtonReleased
+				&& ev.mouseButton.button == sf::Mouse::Button::Left) {
+			_sel_step = S_THETA;
+		}
+	} else if (_sel_step == S_THETA) {
+		if (ev.type == sf::Event::MouseMoved) {
+			_sel_pos.theta = (mouse_coord() - _sel_pos.pos()).angle();
+		} else if (ev.type == sf::Event::MouseButtonReleased
+				&& ev.mouseButton.button == sf::Mouse::Button::Left) {
+			_sel_step = S_PHI;
+		}
+	} else {
+		if (ev.type == sf::Event::MouseMoved) {
+			_sel_pos.phi = (mouse_coord() - _sel_pos.pos()).angle() - _sel_pos.theta;
+		} else if (ev.type == sf::Event::MouseButtonReleased
+				&& ev.mouseButton.button == sf::Mouse::Button::Left) {
+			if (_mode == M_SEL_BEGIN) {
+				_p.begin_pos = _sel_pos;
+			} else {
+				_p.end_pos = _sel_pos;
+			}
+			_mode = M_NORMAL;	// cancel out
+		}
+	}
+
+	if (ev.type == sf::Event::MouseButtonReleased
+			&& ev.mouseButton.button == sf::Mouse::Button::Right) {
+		_mode = M_NORMAL;	// cancel out
+	}
+}
+
 void UI::render_circle(const circle &c, sf::Color border, sf::Color inside, int w) {
 	double r = c.r * _view.zoom;
 	sf::CircleShape a(r);
@@ -57,9 +187,35 @@ void UI::render_circle(const circle &c, sf::Color border, sf::Color inside, int
 	_win.draw(a);
 }
 
+void UI::render_seg(vec a, vec b, sf::Color c, int w) {
+	sf::ConvexShape l;
+
+	l.setPointCount(4);
+	l.setPoint(0, to_view(a));
+	l.setPoint(1, to_view(b));
+	l.setPoint(2, to_view(b));
+	l.setPoint(3, to_view(a));
+
+	l.setOutlineColor(c);
+	l.setOutlineThickness(w);
+
+	_win.draw(l);
+}
+
 void UI::render_pos(const hilare_a &pos, sf::Color c) {
 	render_circle(circle(pos.pos(), pos.param->r_c_car), c, sf::Color::Transparent, 2);
 
+	sf::ConvexShape l;
+
+	l.setPointCount(3);
+	l.setPoint(0, to_view(pos.pos() - vec::from_polar(pos.param->r_c_car * 0.8, pos.theta + M_PI / 2)));
+	l.setPoint(1, to_view(pos.pos() + vec::from_polar(pos.param->r_c_car * 0.8, pos.theta + M_PI / 2)));
+	l.setPoint(2, to_view(pos.pos() + vec::from_polar(pos.param->r_c_car * 0.8, pos.theta)));
+
+	l.setFillColor(c);
+
+	_win.draw(l);
+
 	render_circle(circle(pos.pos_trolley(), pos.param->r_c_trolley), c, sf::Color::Transparent, 2);
 }
 
@@ -93,11 +249,21 @@ void UI::render_internal() {
 }
 
 sf::Vector2f UI::to_view(const vec &p) {
-	return sf::Vector2f((p.x - _view.x0) * _view.zoom, (p.y - _view.y0) * _view.zoom);
+	sf::Vector2u s = _win.getSize();
+	return sf::Vector2f((p.x - _view.x0) * _view.zoom + (s.x/2), (p.y - _view.y0) * _view.zoom + (s.y/2));
 }
 
 vec UI::from_view(const sf::Vector2f &p) {
-	return vec(p.x / _view.zoom + _view.x0, p.y / _view.zoom + _view.y0);
+	sf::Vector2u s = _win.getSize();
+	return vec((p.x - s.x/2) / _view.zoom + _view.x0, (p.y - s.y/2) / _view.zoom + _view.y0);
+}
+
+vec UI::from_view(const sf::Vector2i &p) {
+	return from_view(sf::Vector2f(p.x, p.y));
+}
+
+vec UI::mouse_coord() {
+	return from_view(sf::Mouse::getPosition() - _win.getPosition());
 }
 
 /* vim: set ts=4 sw=4 tw=0 noet :*/