1
0
mirror of https://github.com/DCC-EX/CommandStation-EX.git synced 2024-12-27 14:21:25 +01:00

Initial Track Manager code

This commit is contained in:
Asbelos 2022-02-22 01:27:27 +00:00
parent 3038f29dac
commit 8db937e985
3 changed files with 201 additions and 0 deletions

View File

@ -37,6 +37,7 @@
#include "CommandDistributor.h"
#include "EEStore.h"
#include "DIAG.h"
#include "TrackManager.h"
#include <avr/wdt.h>
////////////////////////////////////////////////////////////////////////////////
@ -483,6 +484,11 @@ void DCCEXParser::parse(Print *stream, byte *com, RingStream * ringStream)
return;
return;
case 'J': // < >
if (TrackManager::parseJ(stream, params, p))
return;
break;
case '#': // NUMBER OF LOCOSLOTS <#>
StringFormatter::send(stream, F("<# %d>\n"), MAX_LOCOS);
return;

137
TrackManager.cpp Normal file
View File

@ -0,0 +1,137 @@
/*
* © 2022 Chris Harlow
* All rights reserved.
*
* This file is part of Asbelos DCC API
*
* 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/>.
*/
#include "TrackManager.h"
#include "FSH.h"
#include "MotorDriver.h"
#include "DIAG.h"
// Virtualised Motor shield multi-track hardware Interface
#define LOOPMODE(findmode,function) \
for (byte t=0;t<8;t++) \
if (trackMode[t]==findmode) \
track[t]->function;
const int16_t HASH_KEYWORD_PROG = -29718;
const int16_t HASH_KEYWORD_MAIN = 11339;
const int16_t HASH_KEYWORD_OFF = 9191; // TODO
const int16_t HASH_KEYWORD_DC = 9192; // TODO
MotorDriver * TrackManager::track[MAX_TRACKS];
int16_t TrackManager::trackMode[MAX_TRACKS];
void TrackManager::Setup(const FSH * shieldname,
MotorDriver * track0, MotorDriver * track1, MotorDriver * track2,
MotorDriver * track3, MotorDriver * track4, MotorDriver * track5,
MotorDriver * track6, MotorDriver * track7 ) {
(void) shieldname; // TODO
track[0]=track0;
track[1]=track1;
track[2]=track2;
track[3]=track3;
track[4]=track4;
track[5]=track5;
track[6]=track6;
track[7]=track7;
trackMode[0]=TRACK_MODE_MAIN;
trackMode[1]=TRACK_MODE_PROG;
trackMode[2]=TRACK_MODE_OFF;
trackMode[3]=TRACK_MODE_OFF;
trackMode[4]=TRACK_MODE_OFF;
trackMode[5]=TRACK_MODE_OFF;
trackMode[6]=TRACK_MODE_OFF;
trackMode[7]=TRACK_MODE_OFF;
}
void TrackManager::setDCCSignal( bool on) {
LOOPMODE(TRACK_MODE_MAIN,setSignal(on));
}
void TrackManager::setCutout( bool on) {
(void) on;
// TODO LOOPMODE(TRACK_MODE_MAIN,setCutout(on));
}
void TrackManager::setPROGSignal( bool on) {
LOOPMODE(TRACK_MODE_PROG,setSignal(on));
}
void TrackManager::setDCSignal(int16_t cab, byte speedbyte) {
// TODO LOOPMODE(cab,setDC(speedbyte));
}
bool TrackManager::setTrackMode(byte trackToSet, int16_t modeOrAddr) {
if (trackToSet>=8 || track[trackToSet]==NULL) return false;
trackMode[trackToSet]=modeOrAddr;
// re-evaluate HighAccuracy mode
bool canDo=true;
for (byte t=0;t<8;t++)
if (trackMode[t]==TRACK_MODE_MAIN ||trackMode[t]==TRACK_MODE_PROG )
canDo &= track[t]->isPWMCapable();
MotorDriver::usePWM=canDo;
return true;
}
bool TrackManager::parseJ(Print *stream, int16_t params, int16_t p[])
{
int16_t mode;
if (params==0) { // <J> List track assignments
for (byte t=0;t<8;t++) {
if (track[t]==NULL) break;
StringFormatter::send(stream,F("<j %d "),t);
switch(trackMode[t]) {
case TRACK_MODE_MAIN:
StringFormatter::send(stream,F("MAIN"));
break;
case TRACK_MODE_PROG:
StringFormatter::send(stream,F("PROG"));
break;
case TRACK_MODE_OFF:
StringFormatter::send(stream,F("OFF"));
break;
default:
StringFormatter::send(stream,F("DC %d"),trackMode[t]);
}
StringFormatter::send(stream,F(">\n"));
}
return true;
}
if (params>1 && (p[1]<0 || p[1]>=MAX_TRACKS))
return false;
if (params==2 && p[1]==HASH_KEYWORD_MAIN) // <J id MAIN>
return setTrackMode(p[1],TRACK_MODE_MAIN);
if (params==2 && p[1]==HASH_KEYWORD_PROG) // <J id PROG>
return setTrackMode(p[1],TRACK_MODE_PROG);
if (params==2 && p[1]==HASH_KEYWORD_OFF) // <J id OFF>
return setTrackMode(p[1],TRACK_MODE_OFF);
if (params==3 && p[1]==HASH_KEYWORD_DC) // <J id DC cab>
return setTrackMode(p[1],p[2]);
return false;
}

58
TrackManager.h Normal file
View File

@ -0,0 +1,58 @@
/*
* © 2022 Chris Harlow
* All rights reserved.
*
* This file is part of Asbelos DCC API
*
* 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/>.
*/
#ifndef TrackManager_h
#define TrackManager_h
#include "FSH.h"
#include "MotorDriver.h"
// Virtualised Motor shield multi-track hardware Interface
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 const int16_t TRACK_MODE_MAIN=32760;
static const int16_t TRACK_MODE_PROG=32761;
static const int16_t TRACK_MODE_OFF=0;
static const int16_t MAX_TRACKS=8;
static bool setTrackMode(byte track, int16_t DCaddrOrMode);
static bool parseJ(Print * stream, int16_t params, int16_t p[]);
private:
static MotorDriver* track[MAX_TRACKS];
static int16_t trackMode[MAX_TRACKS]; // dc address or TRACK_MODE_DCC, TRACK_MODE_PROG, TRACK_MODE_OFF
};
#endif