ArduinoAsync/ThreadController.cpp at master · EmDash00/ArduinoAsync · 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
117
118
119
120
121
122
123
124
125
126
127
128
129
#include"Thread.h"
#include"ThreadController.h"

ThreadController::ThreadController(unsignedlong _interval): Thread(){
clear();
setInterval(_interval);

	#ifdef USE_THREAD_NAMES
// Overrides name
 ThreadName = "ThreadController ";
 ThreadName = ThreadName + ThreadID;
	#endif
}

/*
	ThreadController run() (cool stuf)
*/
voidThreadController::run(){
// Run this thread before
if(_onRun != NULL)
_onRun();

unsignedlong time = millis();
int checks = 0;
for(int i = 0; i < MAX_THREADS && checks < cached_size; i++){
// Object exists? Is enabled? Timeout exceeded?
if(thread[i]){
 checks++;

if(thread[i]->shouldRun(time)){
 thread[i]->run();


// Thread was linked using YIELD_TO or AWAIT
if (thread[i]->linked_thread)
 {
add(thread[i]->linked_thread);

// Don't remove the linked thread if it's AWAITing a result.
if (!thread[i]->awaiting)
 thread[i]->linked_thread = nullptr;

 }
// HALT was called within the thread. Remove it.
if (thread[i]->flag == -1)
remove(thread[i]->ThreadID);
 }
 }
	}

// ThreadController extends Thread, so we should flag as runned thread
runned();
}


/*
	List controller (boring part)
*/
boolThreadController::add(Thread* _thread){
// Check if the Thread already exists on the array
for(int i = 0; i < MAX_THREADS; i++){
if(thread[i] != NULL && thread[i]->ThreadID == _thread->ThreadID)
returntrue;
	}

// Find an empty slot
for(int i = 0; i < MAX_THREADS; i++){
if(!thread[i]){
// Found a empty slot, now add Thread
 thread[i] = _thread;
 cached_size++;
returntrue;
 }
	}

// Array is full
returnfalse;
}

voidThreadController::remove(int id){
// Find Threads with the id, and removes
for(int i = 0; i < MAX_THREADS; i++){
if(thread[i]->ThreadID == id){
 thread[i]->cleanup();
 thread[i] = nullptr;
 cached_size--;
return;
 }
	}
}

voidThreadController::remove(Thread* _thread){
remove(_thread->ThreadID);
}

voidThreadController::clear(){
for(int i = 0; i < MAX_THREADS; i++){
 thread[i] = NULL;
	}
	cached_size = 0;
}

intThreadController::size(bool cached){
if(cached)
return cached_size;

int size = 0;
for(int i = 0; i < MAX_THREADS; i++){
if(thread[i])
 size++;
	}
	cached_size = size;

return cached_size;
}

Thread* ThreadController::get(int index){
int pos = -1;
for(int i = 0; i < MAX_THREADS; i++){
if(thread[i] != NULL){
 pos++;

if(pos == index)
return thread[i];
 }
	}

returnNULL;
}