mirror of
https://github.com/DCC-EX/CommandStation-EX.git
synced 2024-11-30 03:26:13 +01:00
99 lines
3.7 KiB
C++
99 lines
3.7 KiB
C++
/*
|
|
* © 2022 Chris Harlow
|
|
* © 2022 Harald Barth
|
|
* All rights reserved.
|
|
*
|
|
* This file is part of CommandStation-EX
|
|
*
|
|
* 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 CommandStation. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
#ifdef ARDUINO_ARCH_ESP32
|
|
#include <vector>
|
|
#endif
|
|
#ifndef TrackManager_h
|
|
#define TrackManager_h
|
|
#include "FSH.h"
|
|
#include "MotorDriver.h"
|
|
// Virtualised Motor shield multi-track hardware Interface
|
|
|
|
enum TRACK_MODE : byte {TRACK_MODE_OFF, TRACK_MODE_MAIN, TRACK_MODE_PROG,
|
|
TRACK_MODE_DC, TRACK_MODE_DCX, TRACK_MODE_EXT};
|
|
|
|
// These constants help EXRAIL macros say SET_TRACK(2,mode) OR SET_TRACK(C,mode) etc.
|
|
const byte TRACK_NUMBER_0=0, TRACK_NUMBER_A=0;
|
|
const byte TRACK_NUMBER_1=1, TRACK_NUMBER_B=1;
|
|
const byte TRACK_NUMBER_2=2, TRACK_NUMBER_C=2;
|
|
const byte TRACK_NUMBER_3=3, TRACK_NUMBER_D=3;
|
|
const byte TRACK_NUMBER_4=4, TRACK_NUMBER_E=4;
|
|
const byte TRACK_NUMBER_5=5, TRACK_NUMBER_F=5;
|
|
const byte TRACK_NUMBER_6=6, TRACK_NUMBER_G=6;
|
|
const byte TRACK_NUMBER_7=7, TRACK_NUMBER_H=7;
|
|
|
|
class TrackManager {
|
|
public:
|
|
static void Setup(const FSH * shieldName,
|
|
MotorDriver * track0,
|
|
MotorDriver * track1=NULL,
|
|
MotorDriver * track2=NULL,
|
|
MotorDriver * track3=NULL,
|
|
MotorDriver * track4=NULL,
|
|
MotorDriver * track5=NULL,
|
|
MotorDriver * track6=NULL,
|
|
MotorDriver * track7=NULL
|
|
);
|
|
|
|
static void setDCCSignal( bool on);
|
|
static void setCutout( bool on);
|
|
static void setPROGSignal( bool on);
|
|
static void setDCSignal(int16_t cab, byte speedbyte);
|
|
static MotorDriver * getProgDriver();
|
|
#ifdef ARDUINO_ARCH_ESP32
|
|
static std::vector<MotorDriver *>getMainDrivers();
|
|
#endif
|
|
static void setPower2(bool progTrack,POWERMODE mode);
|
|
static void setPower(POWERMODE mode) {setMainPower(mode); setProgPower(mode);}
|
|
static void setMainPower(POWERMODE mode) {setPower2(false,mode);}
|
|
static void setProgPower(POWERMODE mode) {setPower2(true,mode);}
|
|
|
|
static const int16_t MAX_TRACKS=8;
|
|
static bool setTrackMode(byte track, TRACK_MODE mode, int16_t DCaddr=0);
|
|
static bool parseJ(Print * stream, int16_t params, int16_t p[]);
|
|
static void loop();
|
|
static POWERMODE getMainPower() {return mainPowerGuess;}
|
|
static POWERMODE getProgPower();
|
|
static void setJoin(bool join);
|
|
static bool isJoined() { return progTrackSyncMain;}
|
|
static void setJoinRelayPin(byte joinRelayPin);
|
|
static int16_t joinRelay;
|
|
static bool progTrackSyncMain; // true when prog track is a siding switched to main
|
|
static bool progTrackBoosted; // true when prog track is not current limited
|
|
|
|
|
|
private:
|
|
static void addTrack(byte t, MotorDriver* driver);
|
|
static byte lastTrack;
|
|
static byte nextCycleTrack;
|
|
static POWERMODE mainPowerGuess;
|
|
static void applyDCSpeed(byte t);
|
|
|
|
static MotorDriver* track[MAX_TRACKS];
|
|
static TRACK_MODE trackMode[MAX_TRACKS];
|
|
static int16_t trackDCAddr[MAX_TRACKS]; // dc address if TRACK_MODE_DC or TRACK_MODE_DCX
|
|
#ifdef ARDUINO_ARCH_ESP32
|
|
static byte tempProgTrack; // holds the prog track number during join
|
|
#endif
|
|
};
|
|
|
|
#endif
|