mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-25 01:41:43 +00:00
119 lines
3.0 KiB
C++
119 lines
3.0 KiB
C++
/*----------------------------------------------------------------------------*/
|
|
/* Copyright (c) 2011-2018 FIRST. All Rights Reserved. */
|
|
/* Open Source Software - may be modified and shared by FRC teams. The code */
|
|
/* must be accompanied by the FIRST BSD license file in the root directory of */
|
|
/* the project. */
|
|
/*----------------------------------------------------------------------------*/
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
#include <set>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <networktables/NetworkTableEntry.h>
|
|
#include <wpi/mutex.h>
|
|
|
|
#include "Commands/Command.h"
|
|
#include "ErrorBase.h"
|
|
#include "SmartDashboard/SendableBase.h"
|
|
|
|
namespace frc {
|
|
|
|
class ButtonScheduler;
|
|
class Subsystem;
|
|
|
|
class Scheduler : public ErrorBase, public SendableBase {
|
|
public:
|
|
/**
|
|
* Returns the Scheduler, creating it if one does not exist.
|
|
*
|
|
* @return the Scheduler
|
|
*/
|
|
static Scheduler* GetInstance();
|
|
|
|
/**
|
|
* Add a command to be scheduled later.
|
|
*
|
|
* In any pass through the scheduler, all commands are added to the additions
|
|
* list, then at the end of the pass, they are all scheduled.
|
|
*
|
|
* @param command The command to be scheduled
|
|
*/
|
|
void AddCommand(Command* command);
|
|
|
|
void AddButton(ButtonScheduler* button);
|
|
|
|
/**
|
|
* Registers a Subsystem to this Scheduler, so that the Scheduler might know
|
|
* if a default Command needs to be run.
|
|
*
|
|
* All Subsystems should call this.
|
|
*
|
|
* @param system the system
|
|
*/
|
|
void RegisterSubsystem(Subsystem* subsystem);
|
|
|
|
/**
|
|
* Runs a single iteration of the loop.
|
|
*
|
|
* This method should be called often in order to have a functioning
|
|
* Command system. The loop has five stages:
|
|
*
|
|
* <ol>
|
|
* <li>Poll the Buttons</li>
|
|
* <li>Execute/Remove the Commands</li>
|
|
* <li>Send values to SmartDashboard</li>
|
|
* <li>Add Commands</li>
|
|
* <li>Add Defaults</li>
|
|
* </ol>
|
|
*/
|
|
void Run();
|
|
|
|
/**
|
|
* Removes the Command from the Scheduler.
|
|
*
|
|
* @param command the command to remove
|
|
*/
|
|
void Remove(Command* command);
|
|
|
|
void RemoveAll();
|
|
|
|
/**
|
|
* Completely resets the scheduler. Undefined behavior if running.
|
|
*/
|
|
void ResetAll();
|
|
|
|
void SetEnabled(bool enabled);
|
|
|
|
void InitSendable(SendableBuilder& builder) override;
|
|
|
|
private:
|
|
Scheduler();
|
|
~Scheduler() override = default;
|
|
|
|
void ProcessCommandAddition(Command* command);
|
|
|
|
Command::SubsystemSet m_subsystems;
|
|
wpi::mutex m_buttonsMutex;
|
|
typedef std::vector<ButtonScheduler*> ButtonVector;
|
|
ButtonVector m_buttons;
|
|
typedef std::vector<Command*> CommandVector;
|
|
wpi::mutex m_additionsMutex;
|
|
CommandVector m_additions;
|
|
typedef std::set<Command*> CommandSet;
|
|
CommandSet m_commands;
|
|
bool m_adding = false;
|
|
bool m_enabled = true;
|
|
std::vector<std::string> commands;
|
|
std::vector<double> ids;
|
|
std::vector<double> toCancel;
|
|
nt::NetworkTableEntry m_namesEntry;
|
|
nt::NetworkTableEntry m_idsEntry;
|
|
nt::NetworkTableEntry m_cancelEntry;
|
|
bool m_runningCommandsChanged = false;
|
|
};
|
|
|
|
} // namespace frc
|