mirror of https://github.com/ArduPilot/ardupilot
117 lines
3.4 KiB
C++
117 lines
3.4 KiB
C++
#pragma once
|
|
|
|
#include <AP_HAL/AP_HAL.h>
|
|
#if CONFIG_HAL_BOARD == HAL_BOARD_SITL
|
|
#include "AP_HAL_SITL_Namespace.h"
|
|
#include <sys/time.h>
|
|
#include <pthread.h>
|
|
|
|
#define SITL_SCHEDULER_MAX_TIMER_PROCS 8
|
|
|
|
/* Scheduler implementation: */
|
|
class HALSITL::Scheduler : public AP_HAL::Scheduler {
|
|
public:
|
|
explicit Scheduler(SITL_State *sitlState);
|
|
static Scheduler *from(AP_HAL::Scheduler *scheduler) {
|
|
return static_cast<HALSITL::Scheduler*>(scheduler);
|
|
}
|
|
|
|
/* AP_HAL::Scheduler methods */
|
|
|
|
void init() override;
|
|
void delay(uint16_t ms) override;
|
|
void delay_microseconds(uint16_t us) override;
|
|
|
|
void register_timer_process(AP_HAL::MemberProc) override;
|
|
void register_io_process(AP_HAL::MemberProc) override;
|
|
|
|
void register_timer_failsafe(AP_HAL::Proc, uint32_t period_us) override;
|
|
|
|
bool in_main_thread() const override;
|
|
bool is_system_initialized() override { return _initialized; };
|
|
void set_system_initialized() override;
|
|
|
|
void reboot(bool hold_in_bootloader) override;
|
|
|
|
bool interrupts_are_blocked(void) const {
|
|
return _nested_atomic_ctr != 0;
|
|
}
|
|
|
|
void sitl_begin_atomic() {
|
|
_nested_atomic_ctr++;
|
|
}
|
|
void sitl_end_atomic();
|
|
|
|
static void timer_event() {
|
|
_run_timer_procs();
|
|
_run_io_procs();
|
|
}
|
|
|
|
uint64_t stopped_clock_usec() const { return _stopped_clock_usec; }
|
|
|
|
static void _run_io_procs();
|
|
static bool _should_exit;
|
|
|
|
/*
|
|
create a new thread
|
|
*/
|
|
bool thread_create(AP_HAL::MemberProc, const char *name,
|
|
uint32_t stack_size, priority_base base, int8_t priority) override;
|
|
|
|
void set_in_semaphore_take_wait(bool value) { _in_semaphore_take_wait = value; }
|
|
/*
|
|
* semaphore_wait_hack_required - possibly move time input step
|
|
* forward even if we are currently pretending to be the IO or timer
|
|
* threads.
|
|
*/
|
|
// a couple of helper functions to cope with SITL's time stepping
|
|
bool semaphore_wait_hack_required() const;
|
|
|
|
// get the name of the current thread, or nullptr if not known
|
|
const char *get_current_thread_name(void) const;
|
|
|
|
private:
|
|
SITL_State *_sitlState;
|
|
uint8_t _nested_atomic_ctr;
|
|
static AP_HAL::Proc _failsafe;
|
|
|
|
static void _run_timer_procs();
|
|
|
|
static volatile bool _timer_event_missed;
|
|
static AP_HAL::MemberProc _timer_proc[SITL_SCHEDULER_MAX_TIMER_PROCS];
|
|
static AP_HAL::MemberProc _io_proc[SITL_SCHEDULER_MAX_TIMER_PROCS];
|
|
static uint8_t _num_timer_procs;
|
|
static uint8_t _num_io_procs;
|
|
static bool _in_timer_proc;
|
|
static bool _in_io_proc;
|
|
|
|
// boolean set by the Semaphore code to indicate it's currently
|
|
// waiting for a take-timeout to occur.
|
|
static bool _in_semaphore_take_wait;
|
|
|
|
void stop_clock(uint64_t time_usec) override;
|
|
|
|
static void *thread_create_trampoline(void *ctx);
|
|
static void check_thread_stacks(void);
|
|
|
|
bool _initialized;
|
|
uint64_t _stopped_clock_usec;
|
|
uint64_t _last_io_run;
|
|
pthread_t _main_ctx;
|
|
|
|
static HAL_Semaphore _thread_sem;
|
|
struct thread_attr {
|
|
struct thread_attr *next;
|
|
AP_HAL::MemberProc *f;
|
|
pthread_attr_t attr;
|
|
uint32_t stack_size;
|
|
void *stack;
|
|
const uint8_t *stack_min;
|
|
const char *name;
|
|
pthread_t thread;
|
|
};
|
|
static struct thread_attr *threads;
|
|
static const uint8_t stackfill = 0xEB;
|
|
};
|
|
#endif // CONFIG_HAL_BOARD
|