mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2025-01-25 16:32:55 +00:00
58 lines
No EOL
2.2 KiB
C++
58 lines
No EOL
2.2 KiB
C++
/*
|
|
* Copyright (c) 2018 Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#pragma once
|
|
#include <switch.h>
|
|
#include <vector>
|
|
|
|
#include "waitablemanager.hpp"
|
|
#include "systemevent.hpp"
|
|
|
|
class MultiThreadedWaitableManager : public WaitableManager {
|
|
protected:
|
|
u32 num_threads;
|
|
Thread *threads;
|
|
HosMutex get_waitable_lock;
|
|
SystemEvent *new_waitable_event;
|
|
public:
|
|
MultiThreadedWaitableManager(u32 n, u64 t, u32 ss = 0x8000) : WaitableManager(t), num_threads(n-1) {
|
|
u32 prio;
|
|
u32 cpuid = svcGetCurrentProcessorNumber();
|
|
Result rc;
|
|
threads = new Thread[num_threads];
|
|
if (R_FAILED((rc = svcGetThreadPriority(&prio, CUR_THREAD_HANDLE)))) {
|
|
fatalSimple(rc);
|
|
}
|
|
for (unsigned int i = 0; i < num_threads; i++) {
|
|
threads[i] = {0};
|
|
threadCreate(&threads[i], &MultiThreadedWaitableManager::thread_func, this, ss, prio, cpuid);
|
|
}
|
|
new_waitable_event = new SystemEvent(this, &MultiThreadedWaitableManager::add_waitable_callback);
|
|
this->waitables.push_back(new_waitable_event);
|
|
}
|
|
~MultiThreadedWaitableManager() override {
|
|
/* TODO: Exit the threads? */
|
|
}
|
|
|
|
IWaitable *get_waitable();
|
|
|
|
void add_waitable(IWaitable *waitable) override;
|
|
void process() override;
|
|
void process_until_timeout() override;
|
|
|
|
static Result add_waitable_callback(void *this_ptr, Handle *handles, size_t num_handles, u64 timeout);
|
|
static void thread_func(void *this_ptr);
|
|
}; |