summaryrefslogtreecommitdiff
path: root/src/thread.cpp
blob: 401b38855c5d2b43b5dc205b4adf00d3f409d057 (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
/*******************************************************************************
 * Copyright (c) 2006 International Business Machines Corporation.             *
 * All rights reserved. This program and the accompanying materials            *
 * are made available under the terms of the Common Public License v1.0        *
 * which accompanies this distribution, and is available at                    *
 * http://www.opensource.org/licenses/cpl1.0.php                               *
 *                                                                             *
 * Contributors:                                                               *
 *    Douglas M. Pase - initial API and implementation                         *
 *    Tim Besard - prefetching, JIT compilation                                *
 *******************************************************************************/

//
// Configuration
//

// Implementation header
#include "thread.h"

// System includes
#include <cstdio>
#include <pthread.h>
#include <unistd.h>

Lock Thread::_global_lock;
int Thread::count = 0;


//
// Implementation
//

Thread::Thread() {
	Thread::global_lock();
	this->id = Thread::count;
	Thread::count += 1;
	Thread::global_unlock();
}

Thread::~Thread() {
}

int Thread::start() {
	return pthread_create(&this->thread, NULL, Thread::start_routine, this);
}

void*
Thread::start_routine(void* p) {
	// get the current affinity
	cpu_set_t cs;
	CPU_ZERO(&cs);
	sched_getaffinity(0, sizeof(cs), &cs);

	// deduce the amount of CPUs
	int count = 0;
	for (int i = 0; i < 8; i++)
	{
		if (CPU_ISSET(i, &cs))
				count++;
	}

	// restrict to a single CPU
	CPU_ZERO(&cs);
	size_t size = CPU_ALLOC_SIZE(1);
	CPU_SET_S(((Thread*) p)->id % count, size, &cs);
	sched_setaffinity(pthread_self(), size, &cs);

	// run
	((Thread*) p)->run();

	return NULL;
}

void Thread::exit() {
	pthread_exit(NULL);
}

int Thread::wait() {
	pthread_join(this->thread, NULL);

	return 0;
}

void Thread::lock() {
	this->object_lock.lock();
}

void Thread::unlock() {
	this->object_lock.unlock();
}

void Thread::global_lock() {
	Thread::_global_lock.lock();
}

void Thread::global_unlock() {
	Thread::_global_lock.unlock();
}