summaryrefslogtreecommitdiffhomepage
path: root/facetracknoir/tracker.h
blob: 49e7f30212e0401a97f50615d84bc2b0a0182e80 (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
#ifndef __TRACKER_H__
#define __TRACKER_H__

#include <QThread>
#include <QMessageBox>
#include <QLineEdit>
#include <QPoint>
#include <QWaitCondition>
#include <QList>
#include <QPainterPath>
#include <QDebug>
#include <QMutex>
#include "plugin-support.h"
#include <ftnoir_tracker_base/ftnoir_tracker_types.h>
#include <vector>

#include <qfunctionconfigurator/functionconfig.h>
#include "tracker_types.h"
#include "facetracknoir/main-settings.hpp"
#include "facetracknoir/options.h"
#include "facetracknoir/timer.hpp"
using namespace options;

class FaceTrackNoIR;				// pre-define parent-class to avoid circular includes

class THeadPoseDOF {
private:
    THeadPoseDOF(const THeadPoseDOF &) = delete;
    THeadPoseDOF& operator=(const THeadPoseDOF&) = delete;
public:
    THeadPoseDOF(QString primary,
                 QString secondary,
                 int maxInput1,
                 int maxOutput1,
                 int maxInput2,
                 int maxOutput2,
                 axis_opts* opts) :
        headPos(0),
        curve(primary, maxInput1, maxOutput1),
        curveAlt(secondary, maxInput2, maxOutput2),
        opts(*opts)
    {
        QSettings settings("opentrack");
        QString currentFile = settings.value ( "SettingsFile", QCoreApplication::applicationDirPath() + "/settings/default.ini" ).toString();
        QSettings iniFile( currentFile, QSettings::IniFormat );
        curve.loadSettings(iniFile);
        curveAlt.loadSettings(iniFile);
    }
    volatile double headPos;
    FunctionConfig curve;
	FunctionConfig curveAlt;
    axis_opts& opts;
};

class Tracker : protected QThread {
	Q_OBJECT

private:
    FaceTrackNoIR *mainApp;
    QMutex mtx;
    main_settings& s;
    volatile bool should_quit;
    Timer t;
protected:
	void run();

public:
    Tracker( FaceTrackNoIR *parent, main_settings& s);
    ~Tracker();

    void getHeadPose(double *data);
    void getOutputHeadPose(double *data);
    volatile bool do_center;
    volatile bool enabled;
    
    T6DOF output_camera;

    void start() { QThread::start(); }
};

class HeadPoseData {
public:
    THeadPoseDOF* axes[6];
    HeadPoseData(std::vector<axis_opts*> opts)
    {
        axes[TX] = new THeadPoseDOF("tx","tx_alt", 100, 100, 100, 100, opts[TX]);
        axes[TY] = new THeadPoseDOF("ty","ty_alt", 100, 100, 100, 100, opts[TY]);
        axes[TZ] = new THeadPoseDOF("tz","tz_alt", 100, 100, 100, 100, opts[TZ]);
        axes[Yaw] = new THeadPoseDOF("rx", "rx_alt", 180, 180, 180, 180, opts[Yaw]);
        axes[Pitch] = new THeadPoseDOF("ry", "ry_alt", 90, 90, 90, 90, opts[Pitch]);
        axes[Roll] = new THeadPoseDOF("rz", "rz_alt", 180, 180, 180, 180, opts[Roll]);
    }
    ~HeadPoseData()
    {
        for (int i = 0; i < 6; i++)
        {
            delete axes[i];
        }
    }
};

#endif