summaryrefslogtreecommitdiffhomepage
path: root/tracker-s2bot/ftnoir_tracker_s2bot.cpp
blob: 48220c5ae9fbd9123e6f9d701894888934e58c51 (plain)
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
#include "ftnoir_tracker_s2bot.h"
#include "api/plugin-api.hpp"

#include <cinttypes>
#include <algorithm>
#include <cmath>
#include <QNetworkRequest>
#include <QNetworkReply>

tracker_s2bot::tracker_s2bot() : pose { 0,0,0, 0,0,0 }, m_nam (std::make_unique<QNetworkAccessManager>())
{
}

tracker_s2bot::~tracker_s2bot()
{
    QThread::exit(0); // stop event loop
    wait();
}

template<typename t>
static const t bound(t datum, t least, t max)
{
    if (datum < least)
        return least;
    if (datum > max)
        return max;
    return datum;
}

void tracker_s2bot::run() {
	if (s.freq == 0) s.freq = 10;
	timer.setInterval(1000.0/s.freq);
	timer.setSingleShot(false);
	connect(&timer, &QTimer::timeout, [this]() {
		auto reply = m_nam->get(QNetworkRequest(QUrl("http://localhost:17317/poll")));
		connect(reply, &QNetworkReply::finished, [this, reply]() {
			if (reply->error() == QNetworkReply::NoError) {
				qDebug() << "Request submitted OK";
			}
			else {
				qWarning() << "Request bounced:" << reply->attribute(QNetworkRequest::HttpStatusCodeAttribute) << reply->errorString();
				return;
			}
			QByteArray ba = reply->readAll();
			QStringList slist = QString(ba).split(QRegExp("[\r\n]"), QString::SkipEmptyParts);
			int order[] = {
				bound<int>(s.idx_x, 0, 3),
				bound<int>(s.idx_y, 0, 3),
				bound<int>(s.idx_z, 0, 3)
			};
			double orient[4] = { 0, 0, 0, 0 };
			static const int add_cbx[] = {
				0,
				90,
				-90,
				180,
				-180,
			};
			int indices[] = { s.add_yaw, s.add_pitch, s.add_roll };
			for (auto line : slist) {
				QStringList keyval = line.split(" ");
				if (keyval.count() < 2) continue;
				if (keyval[0].startsWith("accelerometerZ")) orient[0] = keyval[1].toInt();
				else if (keyval[0].startsWith("accelerometerY")) orient[1] = keyval[1].toInt();
				else if (keyval[0].startsWith("accelerometerX")) orient[2] = keyval[1].toInt();
				else if (keyval[0].startsWith("bearing")) orient[3] = keyval[1].toInt();
			}
			QMutexLocker foo(&mtx);
			static constexpr double r2d = 180 / M_PI;
                        (void)r2d;
			for (int i = 0; i < 3; i++)
			{
				int val = 0;
				int idx = indices[order[i]];
				if (idx >= 0 && idx < (int)(sizeof(add_cbx) / sizeof(*add_cbx)))
					val = add_cbx[idx];
				pose[Yaw + i] = orient[order[i]] + val; // * r2d if it was radians
			}
			reply->close();
			reply->deleteLater();
		});
	});

	timer.start();
	exec();
        timer.stop();
}

void tracker_s2bot::start_tracker(QFrame*)
{
    start();
    timer.moveToThread(this);
    m_nam->moveToThread(this);
}

void tracker_s2bot::data(double *data)
{
    QMutexLocker foo(&mtx);

    data[Yaw] = pose[Yaw];
    data[Pitch] = pose[Pitch];
    data[Roll] = pose[Roll];
}

OPENTRACK_DECLARE_TRACKER(tracker_s2bot, dialog_s2bot, meta_s2bot)