obs-face-tracker/src/face-tracker-base.cpp at ppa · norihiro/obs-face-tracker · 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
#include <obs-module.h>
#include <util/platform.h>
#include <util/threading.h>
#include <util/bmem.h>
#include "plugin-macros.generated.h"
#include "face-tracker-base.h"
#ifndef _WIN32
#include <sys/time.h>
#include <sys/resource.h>
#else // _WIN32
#include <windows.h>
#endif // _WIN32

face_tracker_base::face_tracker_base()
{
	pthread_mutex_init(&mutex, NULL);
	pthread_cond_init(&cond, NULL);
	stop_requested = 0;
	running = 0;
	leak_test = bmalloc(1);
}

face_tracker_base::~face_tracker_base()
{
	bfree(leak_test);
	pthread_cond_destroy(&cond);
	pthread_mutex_destroy(&mutex);
}

void *face_tracker_base::thread_routine(void *p)
{
	face_tracker_base *base = (face_tracker_base*)p;
#ifndef _WIN32
	setpriority(PRIO_PROCESS, 0, 17);
#else // _WIN32
	SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_LOWEST);
#endif // _WIN32
	os_set_thread_name("face-trk");

	base->lock();
	while(!base->stop_requested) {
		if (!base->suspend_requested) {
			try {
				base->track_main();
			}
			catch (std::exception &e) {
				blog(LOG_ERROR, "track_main: exception %s", e.what());
			}
			catch (...) {
				blog(LOG_ERROR, "track_main: unknown exception");
			}
		}
		pthread_cond_wait(&base->cond, &base->mutex);
	}
	base->stopped = 1;
	base->unlock();
	return NULL;
}

void face_tracker_base::start()
{
	stop_requested = 0;
	stopped = 0;
	suspend_requested = 0;
	if (!running) {
		blog(LOG_INFO, "face_tracker_base: starting a new thread.");
		pthread_create(&thread, NULL, thread_routine, (void*)this);
		running = 1;
	}
	else {
		lock();
		signal();
		unlock();
	}
}

void face_tracker_base::stop()
{
	lock();
	stop_requested = 1;
	signal();
	unlock();
	if (running) {
		blog(LOG_INFO, "face_tracker_base: joining the thread...");
		pthread_join(thread, NULL);
		running = 0;
		blog(LOG_INFO, "face_tracker_base: joined the thread.");
	}
}

void face_tracker_base::request_stop()
{
	lock();
	stop_requested = 1;
	signal();
	unlock();
}

bool face_tracker_base::is_stopped()
{
	if (stopped) {
		if (running) {
			blog(LOG_INFO, "face_tracker_base: joining the thread...");
			pthread_join(thread, NULL);
			running = 0;
		}
		return 1;
	}
	else
		return 0;
}

void face_tracker_base::request_suspend()
{
	suspend_requested = true;
}