- scheduler->set_current_thread(thread);
- Thread::swap(&system_context, thread);
+ Thread *nextThread = nullptr;
+ for (unsigned int i = curr_thread_num;i < get_num_threads();i++) {
+ thread_id_t tid = int_to_id(i);
+ Thread *thr = get_thread(tid);
+
+ if (!thr->is_complete()) {
+ if (!thr->get_pending()) {
+ curr_thread_num = i;
+ nextThread = thr;
+ break;
+ }
+ } else if (thr != old && !thr->is_freed()) {
+ thr->freeResources();
+ }
+
+ ModelAction *act = thr->get_pending();
+ if (act && execution->is_enabled(tid)){
+ /* Don't schedule threads which should be disabled */
+ if (!execution->check_action_enabled(act)) {
+ scheduler->sleep(thr);
+ }
+
+ /* Allow pending relaxed/release stores or thread actions to perform first */
+ else if (!chosen_thread) {
+ if (act->is_write()) {
+ std::memory_order order = act->get_mo();
+ if (order == std::memory_order_relaxed || \
+ order == std::memory_order_release) {
+ chosen_thread = thr;
+ }
+ } else if (act->get_type() == THREAD_CREATE || \
+ act->get_type() == PTHREAD_CREATE || \
+ act->get_type() == THREAD_START || \
+ act->get_type() == THREAD_FINISH) {
+ chosen_thread = thr;
+ }
+ }
+ }
+ }
+ return nextThread;