Avr/MarklinController
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

working marklin digital support

Browse Source
This commit is contained in:
IMback
2018-10-23 14:42:03 +02:00
parent 6a679f71ad
commit 18dc36e928
3 changed files with 178 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

188
main.cpp
View File

@ -12,57 +12,46 @@
#define MAX_TRAINS 16
#define COMMAND_BUFFER_SIZE 64
#define SNPRINTF_BUFFER_SIZE 64
#define EPPROM_SIZE 1024
char buffer[SNPRINTF_BUFFER_SIZE];
uint16_t storedTrains = 0;
Train trains[MAX_TRAINS];
#define TICK_MAX 255
uint8_t tick = 0;
bool autoff = true;
bool powerIsOn = true;
#define TICK_MAX 255
volatile uint8_t tick = 0;
volatile bool resendEvent = false;
ISR(TIMER0_OVF_vect)
{
if(powerIsOn)
{
++tick;
if(tick == TICK_MAX)
{
for(uint16_t i = 0; i < storedTrains; i++) trains[i].resendSpeed();
if(autoff)
{
bool trainsRunning = false;
for(uint16_t i = 0; i < storedTrains; i++) trainsRunning = trainsRunning || trains[i].getSpeed();
if(!trainsRunning)
{
powerIsOn = false;
Train::setOutput(Train::OFF);
}
}
tick = 0;
}
else
{
Train::setOutput(Train::HIGH);
_delay_us(200);
Train::setOutput(Train::LOW);
}
}
++tick;
if(tick == 0) resendEvent = true;
if((tick & 0b00000111) < 1) Train::setOutput(Train::HIGH);
else Train::setOutput(Train::LOW);
}
void timer0InterruptEnable(const bool enable)
{
if(enable) TIMSK0 = 0b00000001;
else TIMSK0 = 0b00000000;
}
void save_state()
{
cli();
EEPROM_write_char( 0, storedTrains );
EEPROM_write_char( 1, autoff );
for(uint16_t i = 0; i < storedTrains; i++)
{
EEPROM_write_char( i+32, trains[i].getAddress());
EEPROM_write_char( i*2+32, trains[i].getAddress());
EEPROM_write_char( i*2+32+1, trains[i].getProtocol());
}
sei();
}
void restore_state()
@ -71,13 +60,13 @@ void restore_state()
autoff = EEPROM_read_char(1);
if(storedTrains > MAX_TRAINS )
{
for(uint16_t i = 0; i < MAX_TRAINS+32; i++) EEPROM_write_char(i, 0);
for(uint16_t i = 0; i < EPPROM_SIZE; i++) EEPROM_write_char(i, 0);
storedTrains = 0;
}
else for(uint8_t i = 0; i <= storedTrains; i++)
else for(uint8_t i = 0; i < storedTrains; i++)
{
uint8_t address = EEPROM_read_char(32+i);
trains[i].setAddress(address);
trains[i].setAddress(EEPROM_read_char(32+i*2));
trains[i].setProtocol(EEPROM_read_char(32+1+i*2));
}
}
@ -104,27 +93,26 @@ inline static void printHelp(Serial* serial)
void trainDispatch(char* inBuffer, Serial* serial)
{
cli();
if(powerIsOn == false)
{
powerIsOn = true;
Train::setOutput(Train::LOW);
_delay_ms(100);
}
if( strcmp(inBuffer, "add") == 0 )
{
char* token = strtok(NULL, " ");
uint8_t address = strtol(token, nullptr, 2 );
uint8_t address = 0;
if(token != NULL) address = strtol(token, nullptr, 2 );
if(address != 0 && storedTrains < MAX_TRAINS)
{
trains[storedTrains].setAddress(address);
uint8_t size = snprintf(buffer, SNPRINTF_BUFFER_SIZE, "TRAIN saved! NUMBER: %u ADRESS: %s%s\n", storedTrains, bit_rep[address >> 4], bit_rep[address & 0x0F]);
token = strtok(NULL, " ");
if(token != NULL && strcmp(token, "delta") == 0 ) trains[storedTrains].setProtocol(Train::M_DELTA);
else trains[storedTrains].setProtocol(Train::M_DIGITAL);
uint8_t size = snprintf(buffer, SNPRINTF_BUFFER_SIZE, "TRAIN saved! NUMBER: %u ADRESS: %s%s PROTCOL: %s\n", storedTrains, bit_rep[address >> 4], bit_rep[address & 0x0F], trains[storedTrains].getProtocol() == Train::M_DELTA ? "delta" : "digital" );
serial->write(buffer, size);
storedTrains++;
save_state();
}
else serial->write_p(PSTR("Usage: train add [address]"));
serial->write_p(PSTR("Usage: train add [address]"));
}
else if( strcmp(inBuffer, "delete") == 0)
{
@ -138,8 +126,11 @@ void trainDispatch(char* inBuffer, Serial* serial)
serial->write_p(PSTR("Trains:\n"));
for(uint8_t i = 0; i < storedTrains; i++)
{
snprintf(buffer, SNPRINTF_BUFFER_SIZE, "NUMBER: %u ID: %s%s CURRENT SPD: %u\n", i, bit_rep[trains[i].getAddress() >> 4], bit_rep[trains[i].getAddress() & 0x0F], trains[i].getSpeed());
snprintf(buffer, SNPRINTF_BUFFER_SIZE, "NUMBER: %u ID: %s%s CURRENT PACKET: %x", i, bit_rep[trains[i].getAddress() >> 4], bit_rep[trains[i].getAddress() & 0x0F], trains[i].getLastPacket());
serial->write(buffer, SNPRINTF_BUFFER_SIZE);
serial->write_p(PSTR(" PROTOCOL: "));
if(trains[i].getProtocol() == Train::M_DELTA)serial->write_p(PSTR("delta\n"));
else serial->write_p(PSTR("digital\n"));
}
serial->putChar('\n');
}
@ -148,6 +139,13 @@ void trainDispatch(char* inBuffer, Serial* serial)
uint8_t id = strtol(inBuffer, nullptr, 10);
if(id < storedTrains )
{
if(powerIsOn == false)
{
powerIsOn = true;
Train::setOutput(Train::LOW);
_delay_ms(100);
timer0InterruptEnable(true);
}
char* token = strtok(NULL, " ");
if( token != NULL && strcmp(token, "speed") == 0 )
{
@ -156,32 +154,56 @@ void trainDispatch(char* inBuffer, Serial* serial)
serial->write_p(PSTR("Set Train speed\n"));
}
else if(token != NULL && strcmp(token, "function") == 0 )
{
token = strtok(NULL, " ");
char* boolToken = strtok(NULL, " ");
if(token != NULL && boolToken != NULL)
{
uint8_t functionId = atoi(token);
bool on = (strcmp(boolToken, "on") == 0);
trains[id].sendFunction(functionId, on );
serial->write_p(PSTR("Set Train function "));
serial->write(functionId);
serial->write(on ? " on\n" : " off\n");
}
}
else if(token != NULL && strcmp(token, "probe") == 0 )
{
for(uint16_t j = 0; j < 1024; j++)
{
trains[id].sendRaw(j);
snprintf(buffer, SNPRINTF_BUFFER_SIZE, "TRYING: %x\n", j);
serial->write(buffer, strlen(buffer));
sei();
_delay_ms(250);
cli();
}
}
else if(token != NULL && strcmp(token, "raw") == 0 )
{
token = strtok(NULL, " ");
if(token != NULL)
{
for(uint16_t j = 0; j < 8; j++)
cli();
uint16_t i = strtol(token, nullptr, 16 );
snprintf(buffer, SNPRINTF_BUFFER_SIZE, "SENDING: %x\n", i);
serial->write(buffer, strlen(buffer));
for(uint8_t j = 0; j < 100; j++)
{
uint16_t i = strtol(token, nullptr, 16 );
trains[id].sendFunction(i);
snprintf(buffer, SNPRINTF_BUFFER_SIZE, "TRYING: %x\n", i);
serial->write(buffer, strlen(buffer));
sei();
_delay_ms(250);
cli();
trains[id].sendRaw(i);
_delay_ms(20);
}
serial->write_p(PSTR("Finished\n"));
sei();
}
/*
trains[id].sendFunction(atoi(token));
serial->write_p(PSTR("Set Train function\n"));*/
}
else if( token != NULL && strcmp(token, "reverse") == 0 )
{
trains[id].reverse();
serial->write_p(PSTR("Reversed Train\n"));
}
else if( token != NULL && strcmp(token, "stop") )trains[id].setSpeed(0);
else if( token != NULL && strcmp(token, "stop") == 0 )trains[id].stop();
else serial->write_p(PSTR("Not a valid command\n"));
}
else serial->write_p(PSTR("Id out of range.\n"));
@ -193,10 +215,7 @@ void powerDispatch(char* token, Serial* serial)
{
if(strcmp(token, "off") == 0)
{
for(uint16_t i = 0; i < storedTrains; i++)
{
trains[i].setSpeed(0);
}
timer0InterruptEnable(false);
Train::setOutput(Train::OFF);
powerIsOn = false;
}
@ -207,6 +226,7 @@ void powerDispatch(char* token, Serial* serial)
trains[i].setSpeed(0);
}
Train::setOutput(Train::LOW);
timer0InterruptEnable(true);
}
else if(strcmp(token, "auto") == 0)
{
@ -252,13 +272,13 @@ void serialDispatch(Serial* serial)
}
else if(length > 4 && strncmp(token, "erase", 4) == 0)
{
for(uint16_t i = 0; i < MAX_TRAINS+1; i++) EEPROM_write_char(i, 0);
for(uint16_t i = 0; i < EPPROM_SIZE; i++) EEPROM_write_char(i, 0);
serial->write_p(PSTR("EEPROM erased\n"));
storedTrains = 0;
}
else if(length > 3 && strcmp(token, "dump") == 0)
{
for(uint16_t i = 0; i < MAX_TRAINS+32; i++)
for(uint16_t i = 0; i < EPPROM_SIZE; i++)
{
if(i != 0) serial->putChar(',');
serial->write((uint16_t)EEPROM_read_char(i));
@ -270,7 +290,7 @@ void serialDispatch(Serial* serial)
for(uint16_t i = 0; i < storedTrains; i++)
{
cli();
trains[i].setSpeed(0);
trains[i].stop();
sei();
}
}
@ -297,22 +317,54 @@ void serialDispatch(Serial* serial)
int main()
{
TCNT0 = 0;
TIMSK0 = 0b00000001; //turn on timer0 interupt on OCIEB
TCCR0B = (1<<CS02) /*| (1<<CS00)*/; // run timer0 with /1024 scaler
TCCR0B = (1<<CS01) /*| (1<<CS00)*/; // run timer0 with /64 scaler
DDRD = (1 << PD2) | (1 << PD3) | (1 << PD4) | (1 << PD5);
restore_state();
autoff ? Train::setOutput(Train::OFF) : Train::setOutput(Train::LOW);
if(autoff)
{
timer0InterruptEnable(false);
Train::setOutput(Train::OFF);
}
else
{
timer0InterruptEnable(true);
Train::setOutput(Train::LOW);
}
sei();
Serial serial;
serial.write_p(PSTR("TrainController v0.1 starting\n"));
uint8_t trainToResend = 0;
while(true)
{
if(resendEvent && storedTrains > 0)
{
timer0InterruptEnable(false);
trains[trainToResend].resendData();
trainToResend++;
if(storedTrains <= trainToResend) trainToResend = 0;
resendEvent = false;
timer0InterruptEnable(true);
}
if(autoff)
{
bool trainsRunning = false;
for(uint16_t i = 0; i < storedTrains; i++) trainsRunning = trainsRunning || trains[i].isActive();
if(!trainsRunning)
{
powerIsOn = false;
timer0InterruptEnable(false);
Train::setOutput(Train::OFF);
}
}
serialDispatch(&serial);
}
return 0;