Prusa-Firmware-MMU/src/main.cpp

408 lines
13 KiB
C++

/// @file main.cpp
#include "hal/cpu.h"
#include "hal/adc.h"
#include "hal/gpio.h"
#include "hal/shr16.h"
#include "hal/spi.h"
#include "hal/usart.h"
#include "hal/watchdog.h"
#include "pins.h"
#include <avr/interrupt.h>
#include <util/delay.h>
#include "modules/buttons.h"
#include "modules/finda.h"
#include "modules/fsensor.h"
#include "modules/globals.h"
#include "modules/idler.h"
#include "modules/leds.h"
#include "modules/protocol.h"
#include "modules/pulley.h"
#include "modules/selector.h"
#include "modules/user_input.h"
#include "modules/timebase.h"
#include "modules/motion.h"
#include "modules/usb_cdc.h"
#include "logic/command_base.h"
#include "logic/cut_filament.h"
#include "logic/eject_filament.h"
#include "logic/home.h"
#include "logic/load_filament.h"
#include "logic/move_selector.h"
#include "logic/no_command.h"
#include "logic/set_mode.h"
#include "logic/tool_change.h"
#include "logic/unload_filament.h"
#include "version.h"
#include "panic.h"
/// Global instance of the protocol codec
static mp::Protocol protocol;
/// A command that resulted in the currently on-going operation
static logic::CommandBase *currentCommand = &logic::noCommand;
/// Remember the request message that started the currently running command
/// For the start we report "Reset finished" which in fact corresponds with the MMU state pretty closely
/// and plays nicely even with the protocol implementation.
/// And, since the default startup command is the noCommand, which always returns "Finished"
/// the implementation is clean and straightforward - the response to the first Q0 messages
/// will look like "X0 F" until a command (T, L, U ...) has been issued.
static mp::RequestMsg currentCommandRq(mp::RequestMsgCodes::Reset, 0);
static uint16_t lastCommandProcessedMs = 0;
/// One-time setup of HW and SW components
/// Called before entering the loop() function
/// Green LEDs signalize the progress of initialization. If anything goes wrong we shall turn on a red LED
void setup() {
hal::cpu::Init();
mt::timebase.Init();
// watchdog init
hwd::Enable(hwd::configuration::compute(8000)); //set 8s timeout
mg::globals.Init();
hal::shr16::shr16.Init();
ml::leds.SetMode(4, ml::green, ml::on);
ml::leds.Step();
// if the shift register doesn't work we really can't signalize anything, only internal variables will be accessible if the UART works
hu::USART::USART_InitTypeDef usart_conf = {
.rx_pin = USART_RX,
.tx_pin = USART_TX,
.baudrate = 115200,
};
hu::usart1.Init(&usart_conf);
ml::leds.SetMode(3, ml::green, ml::on);
ml::leds.Step();
// @@TODO if both shift register and the UART are dead, we are sitting ducks :(
hal::spi::SPI_InitTypeDef spi_conf = {
.miso_pin = TMC2130_SPI_MISO_PIN,
.mosi_pin = TMC2130_SPI_MOSI_PIN,
.sck_pin = TMC2130_SPI_SCK_PIN,
.ss_pin = TMC2130_SPI_SS_PIN,
.prescaler = 2, //4mhz
.cpha = 1,
.cpol = 1,
};
hal::spi::Init(SPI0, &spi_conf);
ml::leds.SetMode(2, ml::green, ml::on);
ml::leds.Step();
mm::Init();
ml::leds.SetMode(1, ml::green, ml::on);
ml::leds.Step();
ha::Init();
ml::leds.SetMode(0, ml::green, ml::on);
ml::leds.Step();
mu::cdc.Init();
// waits at least finda debounce period
// which is abused to let the LEDs shine for ~100ms
mf::finda.BlockingInit();
/// Turn off all leds
for (uint8_t i = 0; i < config::toolCount; i++) {
ml::leds.SetMode(i, ml::green, ml::off);
ml::leds.SetMode(i, ml::red, ml::off);
}
ml::leds.Step();
// Idler and Selector decide whether homing is possible/safe
mi::idler.Init();
ms::selector.Init();
// activate the correct LED if filament is present
if (mg::globals.FilamentLoaded() > mg::FilamentLoadState::AtPulley) {
ml::leds.SetMode(mg::globals.ActiveSlot(), ml::green, ml::on);
}
}
static constexpr const uint8_t maxMsgLen = 10;
bool WriteToUSART(const uint8_t *src, uint8_t len) {
// How to properly enqueue the message? Especially in case of a full buffer.
// We neither can stay here in an endless loop until the buffer drains.
// Nor can we save the message elsewhere ... it must be just skipped and the protocol must handle it.
// Under normal circumstances, such a situation should not happen.
// The MMU cannot produce response messages on its own - it only responds to requests from the printer.
// That means there is only one message in the output buffer at once as long as the printer waits for the response before sending another request.
for (uint8_t i = 0; i < len; ++i) {
if (hu::usart1.CanWrite()) {
// should not block waiting for the TX buffer to drain as there was an empty spot for at least 1 byte
hu::usart1.Write(src[i]);
} else {
//buffer full - must skip the rest of the message - the communication will drop out anyway
return false;
}
}
return true; // not sure if we can actually leverage the knowledge of success while sending the message
}
void ReportCommandAccepted(const mp::RequestMsg &rq, mp::ResponseMsgParamCodes status) {
uint8_t tmp[maxMsgLen];
uint8_t len = protocol.EncodeResponseCmdAR(rq, status, tmp);
WriteToUSART(tmp, len);
}
void ReportFINDA(const mp::RequestMsg &rq) {
#ifdef DEBUG_FINDA
using namespace hal;
hu::usart1.puts("FINDA:");
if (hal::gpio::ReadPin(FINDA_PIN) == hal::gpio::Level::high) {
hu::usart1.puts(" TIRGGERED\n");
} else {
hu::usart1.puts(" NOT TRIGGERED\n");
}
#endif //DEBUG_FINDA
uint8_t rsp[maxMsgLen];
uint8_t len = protocol.EncodeResponseReadFINDA(rq, mf::finda.Pressed(), rsp);
WriteToUSART(rsp, len);
}
void ReportVersion(const mp::RequestMsg &rq) {
uint8_t v = 0;
switch (rq.value) {
case 0:
v = project_version_major;
break;
case 1:
v = project_version_minor;
break;
case 2:
v = project_version_revision;
break;
case 3:
// @@TODO may be allow reporting uint16_t number of errors,
// but anything beyond 255 errors means there is something seriously wrong with the MMU
v = mg::globals.DriveErrors();
break;
default:
v = 0;
break;
}
uint8_t rsp[10];
uint8_t len = protocol.EncodeResponseVersion(rq, v, rsp);
WriteToUSART(rsp, len);
}
void ReportRunningCommand() {
mp::ResponseMsgParamCodes commandStatus;
uint16_t value = 0;
switch (currentCommand->Error()) {
case ErrorCode::RUNNING:
commandStatus = mp::ResponseMsgParamCodes::Processing;
value = (uint16_t)currentCommand->State();
break;
case ErrorCode::OK:
commandStatus = mp::ResponseMsgParamCodes::Finished;
lastCommandProcessedMs = mt::timebase.Millis();
break;
default:
commandStatus = mp::ResponseMsgParamCodes::Error;
value = (uint16_t)currentCommand->Error();
break;
}
uint8_t rsp[maxMsgLen];
uint8_t len = protocol.EncodeResponseQueryOperation(currentCommandRq, commandStatus, value, rsp);
WriteToUSART(rsp, len);
}
void PlanCommand(const mp::RequestMsg &rq) {
if (currentCommand->State() == ProgressCode::OK) {
// We are allowed to start a new command as the previous one is in the OK finished state
// The previous command may be in an error state, but as long as it is in ProgressCode::OK (aka finished)
// we are safe here. It is the responsibility of the printer to ask for a command error code
// before issuing another one - if needed.
switch (rq.code) {
case mp::RequestMsgCodes::Cut:
currentCommand = &logic::cutFilament;
break;
case mp::RequestMsgCodes::Eject:
currentCommand = &logic::ejectFilament;
break;
case mp::RequestMsgCodes::Home:
currentCommand = &logic::home;
break;
case mp::RequestMsgCodes::Load:
currentCommand = &logic::loadFilament;
break;
case mp::RequestMsgCodes::Tool:
currentCommand = &logic::toolChange;
break;
case mp::RequestMsgCodes::Unload:
currentCommand = &logic::unloadFilament;
break;
case mp::RequestMsgCodes::Mode:
currentCommand = &logic::setMode;
break;
default:
currentCommand = &logic::noCommand;
break;
}
currentCommandRq = rq; // save the Current Command Request for indentification of responses
currentCommand->Reset(rq.value);
ReportCommandAccepted(rq, mp::ResponseMsgParamCodes::Accepted);
} else {
ReportCommandAccepted(rq, mp::ResponseMsgParamCodes::Rejected);
}
}
void ProcessRequestMsg(const mp::RequestMsg &rq) {
switch (rq.code) {
case mp::RequestMsgCodes::Button:
// behave just like if the user pressed a button
mui::userInput.ProcessMessage(rq.value);
break;
case mp::RequestMsgCodes::Finda:
// immediately report FINDA status
ReportFINDA(rq);
break;
case mp::RequestMsgCodes::Query:
// immediately report progress of currently running command
ReportRunningCommand();
break;
case mp::RequestMsgCodes::Reset:
// immediately reset the board - there is no response in this case
hal::cpu::Reset();
break;
case mp::RequestMsgCodes::Version:
ReportVersion(rq);
break;
case mp::RequestMsgCodes::Wait:
break; // @@TODO - not used anywhere yet
case mp::RequestMsgCodes::Cut:
case mp::RequestMsgCodes::Eject:
case mp::RequestMsgCodes::Home:
case mp::RequestMsgCodes::Load:
case mp::RequestMsgCodes::Tool:
case mp::RequestMsgCodes::Unload:
PlanCommand(rq);
break;
case mp::RequestMsgCodes::FilamentSensor: // set filament sensor state in the printer
mfs::fsensor.ProcessMessage(rq.value != 0);
ReportCommandAccepted(rq, mp::ResponseMsgParamCodes::Accepted);
break;
default:
// respond with an error message
break;
}
}
/// @returns true if a request was successfully finished
bool CheckMsgs() {
using mpd = mp::DecodeStatus;
while (!hu::usart1.ReadEmpty()) {
switch (protocol.DecodeRequest(hu::usart1.Read())) {
case mpd::MessageCompleted:
// process the input message
return true;
break;
case mpd::NeedMoreData:
// just continue reading
break;
case mpd::Error:
// @@TODO what shall we do? Start some watchdog? We cannot send anything spontaneously
break;
}
}
return false;
}
void Panic(ErrorCode ec) {
currentCommand->Panic(ec);
}
bool CheckManualOperation() {
if (currentCommand == &logic::noCommand
&& mg::globals.FilamentLoaded() <= mg::FilamentLoadState::AtPulley
&& lastCommandProcessedMs + 5000 < mt::timebase.Millis()) {
lastCommandProcessedMs = mt::timebase.Millis() - 5000; // @@TODO prevent future overflows - must be called every time
if (mui::userInput.AnyEvent()) {
switch (mui::userInput.ConsumeEvent()) {
case mui::Event::Left:
// move selector left if possible
if (mg::globals.ActiveSlot() > 0) {
currentCommand = &logic::moveSelector;
currentCommand->Reset(mg::globals.ActiveSlot() - 1);
}
break;
case mui::Event::Middle:
// plan load
if (mg::globals.ActiveSlot() < config::toolCount) { // do we have a meaningful selector position?
logic::loadFilament.ResetUnlimited(mg::globals.ActiveSlot());
currentCommand = &logic::loadFilament;
}
break;
case mui::Event::Right:
// move selector right if possible (including the park position)
if (mg::globals.ActiveSlot() < config::toolCount) {
currentCommand = &logic::moveSelector;
currentCommand->Reset(mg::globals.ActiveSlot() + 1); // we allow also the park position
}
break;
default:
break;
}
}
}
}
/// Main loop of the firmware
/// Proposed architecture
/// checkMsgs();
/// if(msg is command){
/// activate command handling
/// } else if(msg is query){
/// format response to query
/// }
/// StepCurrentCommand();
/// StepMotors();
/// StepLED();
/// StepWhateverElseNeedsStepping();
/// The idea behind the Step* routines is to keep each automaton non-blocking allowing for some “concurrency”.
/// Some FW components will leverage ISR to do their stuff (UART, motor stepping?, etc.)
void loop() {
CheckManualOperation();
if (CheckMsgs()) {
ProcessRequestMsg(protocol.GetRequestMsg());
}
mb::buttons.Step();
ml::leds.Step();
mf::finda.Step();
mfs::fsensor.Step();
mi::idler.Step();
ms::selector.Step();
mpu::pulley.Step();
mui::userInput.Step();
currentCommand->Step();
hal::cpu::Step();
mu::cdc.Step();
hal::watchdog::Reset();
}
int main() {
setup();
sei(); ///enable interrupts
for (;;) {
loop();
}
return 0;
}