Firmware2/Marlin/src/feature/twibus.cpp

181 lines
4.0 KiB
C++
Raw Normal View History

2017-02-25 11:15:18 +01:00
/**
* Marlin 3D Printer Firmware
2020-02-03 15:00:57 +01:00
* Copyright (c) 2020 MarlinFirmware [https://github.com/MarlinFirmware/Marlin]
*
* Based on Sprinter and grbl.
2019-06-28 06:57:50 +02:00
* Copyright (c) 2011 Camiel Gubbels / Erik van der Zalm
*
* This program 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.
*
* This program 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 program. If not, see <http://www.gnu.org/licenses/>.
*
*/
2017-09-06 13:28:32 +02:00
#include "../inc/MarlinConfig.h"
#if ENABLED(EXPERIMENTAL_I2CBUS)
#include "twibus.h"
2017-09-06 13:28:32 +02:00
#include <Wire.h>
2016-04-20 21:34:55 +02:00
TWIBus::TWIBus() {
2016-08-11 08:04:18 +02:00
#if I2C_SLAVE_ADDRESS == 0
Wire.begin(); // No address joins the BUS as the master
#else
Wire.begin(I2C_SLAVE_ADDRESS); // Join the bus as a slave
#endif
2019-09-25 15:29:59 +02:00
reset();
}
void TWIBus::reset() {
2019-09-25 15:29:59 +02:00
buffer_s = 0;
buffer[0] = 0x00;
}
2016-08-14 03:06:10 +02:00
void TWIBus::address(const uint8_t adr) {
2017-03-31 16:00:49 +02:00
if (!WITHIN(adr, 8, 127)) {
SERIAL_ECHO_MSG("Bad I2C address (8-127)");
}
2019-09-25 15:29:59 +02:00
addr = adr;
2020-03-28 07:19:43 +01:00
debug(PSTR("address"), adr);
}
2016-08-11 08:04:18 +02:00
void TWIBus::addbyte(const char c) {
2019-09-25 15:29:59 +02:00
if (buffer_s >= COUNT(buffer)) return;
buffer[buffer_s++] = c;
2020-03-28 07:19:43 +01:00
debug(PSTR("addbyte"), c);
2016-08-14 03:06:10 +02:00
}
void TWIBus::addbytes(char src[], uint8_t bytes) {
2020-03-28 07:19:43 +01:00
debug(PSTR("addbytes"), bytes);
2019-09-25 15:29:59 +02:00
while (bytes--) addbyte(*src++);
2016-08-14 03:06:10 +02:00
}
2016-08-14 03:06:10 +02:00
void TWIBus::addstring(char str[]) {
2020-03-28 07:19:43 +01:00
debug(PSTR("addstring"), str);
2019-09-25 15:29:59 +02:00
while (char c = *str++) addbyte(c);
}
void TWIBus::send() {
2020-03-28 07:19:43 +01:00
debug(PSTR("send"), addr);
2019-09-25 15:29:59 +02:00
Wire.beginTransmission(I2C_ADDRESS(addr));
Wire.write(buffer, buffer_s);
Wire.endTransmission();
2019-09-25 15:29:59 +02:00
reset();
}
2016-08-15 06:45:28 +02:00
// static
2020-03-28 07:19:43 +01:00
void TWIBus::echoprefix(uint8_t bytes, const char pref[], uint8_t adr) {
2017-06-09 17:51:23 +02:00
SERIAL_ECHO_START();
2020-03-28 07:19:43 +01:00
serialprintPGM(pref);
SERIAL_ECHOPAIR(": from:", adr, " bytes:", bytes, " data:");
2016-08-15 06:45:28 +02:00
}
// static
2020-03-28 07:19:43 +01:00
void TWIBus::echodata(uint8_t bytes, const char pref[], uint8_t adr) {
echoprefix(bytes, pref, adr);
2016-08-14 03:06:10 +02:00
while (bytes-- && Wire.available()) SERIAL_CHAR(Wire.read());
2017-06-09 17:51:23 +02:00
SERIAL_EOL();
2016-08-14 03:06:10 +02:00
}
2020-03-28 07:19:43 +01:00
void TWIBus::echobuffer(const char pref[], uint8_t adr) {
echoprefix(buffer_s, pref, adr);
2020-03-14 05:18:16 +01:00
LOOP_L_N(i, buffer_s) SERIAL_CHAR(buffer[i]);
2017-06-09 17:51:23 +02:00
SERIAL_EOL();
2016-08-15 06:45:28 +02:00
}
bool TWIBus::request(const uint8_t bytes) {
2019-09-25 15:29:59 +02:00
if (!addr) return false;
2020-03-28 07:19:43 +01:00
debug(PSTR("request"), bytes);
2016-08-11 08:04:18 +02:00
// requestFrom() is a blocking function
2019-09-25 15:29:59 +02:00
if (Wire.requestFrom(addr, bytes) == 0) {
2020-03-28 07:19:43 +01:00
debug("request fail", addr);
return false;
}
2016-08-15 06:45:28 +02:00
return true;
}
void TWIBus::relay(const uint8_t bytes) {
2020-03-28 07:19:43 +01:00
debug(PSTR("relay"), bytes);
2016-08-15 06:45:28 +02:00
2019-09-25 15:29:59 +02:00
if (request(bytes))
echodata(bytes, PSTR("i2c-reply"), addr);
2016-08-15 06:45:28 +02:00
}
uint8_t TWIBus::capture(char *dst, const uint8_t bytes) {
2019-09-25 15:29:59 +02:00
reset();
2016-08-15 06:45:28 +02:00
uint8_t count = 0;
while (count < bytes && Wire.available())
dst[count++] = Wire.read();
2020-03-28 07:19:43 +01:00
debug(PSTR("capture"), count);
2016-08-15 06:45:28 +02:00
return count;
}
2016-08-15 06:45:28 +02:00
// static
void TWIBus::flush() {
while (Wire.available()) Wire.read();
2016-08-11 08:04:18 +02:00
}
2016-08-14 03:06:10 +02:00
#if I2C_SLAVE_ADDRESS > 0
2016-08-14 03:06:10 +02:00
void TWIBus::receive(uint8_t bytes) {
2020-03-28 07:19:43 +01:00
debug(PSTR("receive"), bytes);
2016-08-15 06:45:28 +02:00
echodata(bytes, PSTR("i2c-receive"), 0);
2016-08-14 03:06:10 +02:00
}
void TWIBus::reply(char str[]/*=nullptr*/) {
2020-03-28 07:19:43 +01:00
debug(PSTR("reply"), str);
2016-08-14 03:06:10 +02:00
if (str) {
2019-09-25 15:29:59 +02:00
reset();
addstring(str);
}
2016-08-14 03:06:10 +02:00
2019-09-25 15:29:59 +02:00
Wire.write(buffer, buffer_s);
2016-08-14 03:06:10 +02:00
2019-09-25 15:29:59 +02:00
reset();
2016-08-14 03:06:10 +02:00
}
#endif
#if ENABLED(DEBUG_TWIBUS)
2016-08-15 06:45:28 +02:00
// static
2016-08-14 03:06:10 +02:00
void TWIBus::prefix(const char func[]) {
SERIAL_ECHOPGM("TWIBus::");
serialprintPGM(func);
SERIAL_ECHOPGM(": ");
}
void TWIBus::debug(const char func[], uint32_t adr) {
if (DEBUGGING(INFO)) { prefix(func); SERIAL_ECHOLN(adr); }
}
void TWIBus::debug(const char func[], char c) {
if (DEBUGGING(INFO)) { prefix(func); SERIAL_ECHOLN(c); }
}
void TWIBus::debug(const char func[], char str[]) {
if (DEBUGGING(INFO)) { prefix(func); SERIAL_ECHOLN(str); }
}
#endif
2017-05-09 19:35:43 +02:00
#endif // EXPERIMENTAL_I2CBUS