/home/runner/work/DCCEXSimpleThrottle/DCCEXSimpleThrottle/AppOrchestrator.h

Go to the documentation of this file.

/*
 *  © 2024 Peter Cole
 *  © 2023 Peter Cole
 *
 *  This is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  It is distributed in the hope that 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 code.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#ifndef APPORCHESTRATOR_H
#define APPORCHESTRATOR_H
 
#include "CommandStationClient.h"
#include "ConnectionManager.h"
#include "DisplayInterface.h"
#include "ErrorScreen.h"
#include "EventListener.h"
#include "MenuManager.h"
#include "ProgressScreen.h"
#include "StartupScreen.h"
#include "ThrottleScreen.h"
#include "UserConfirmationInterface.h"
#include "UserSelectionInterface.h"
 
enum class AppState {
  Startup,
  SelectCommandStation,
  ConnectCommandStation,
  SelectLoco,
  Throttle,
  SelectAction,
  Error,
  ReadLocoAddress
};
 
class AppOrchestrator : public EventListener {
public:
  AppOrchestrator(DisplayInterface *displayInterface, ConnectionManager *connectionManager, MenuManager *menuManager,
                  CommandStationClient *commandStationClient, UserConfirmationInterface *userConfirmationInterface,
                  UserSelectionInterface *userSelectionInterface, ThrottleScreen *throttleScreen);
 
  void begin();
 
  void update();
 
  void onEvent(Event &event);
 
  void setupSelectLocoMenu();
 
  void setThrottleLoco(Loco *loco);
 
  void updateThrottleLoco(LocoBroadcast locoBroadcast);
 
  void updateThrottleTrackPower(TrackPower trackPower);
 
private:
  DisplayInterface *_displayInterface;
  ConnectionManager *_connectionManager;
  MenuManager *_menuManager;
  CommandStationClient *_commandStationClient;
  UserConfirmationInterface *_userConfirmationInterface;
  UserSelectionInterface *_userSelectionInterface;
  AppState _currentAppState;
 
  StartupScreen *_startupScreen;
  ThrottleScreen *_throttleScreen;
  ErrorScreen *_errorScreen;
  ProgressScreen *_progressScreen;
 
  void _handleStartupState();
 
  void _handleSelectCommandStationState();
 
  void _handleConnectCommandStationState();
 
  void _handleSelectLocoState();
 
  void _handleThrottleState();
 
  void _handleSelectActionState();
 
  void _handleErrorState();
 
  void _switchState(AppState appState);
 
  void _displayMenu(BaseMenu *menu);
 
  void _handleJoinProgTrack();
 
  void _handleSetTrackPower(Event event);
 
  void _readLoco();
 
  void _handleReadLocoState();
 
  void _handleReceivedReadLoco(int address);
};
 
#endif // APPORCHESTRATOR_H