marlin/Marlin/twibus.cpp

/*
 * Marlin 3D Printer Firmware
 * Copyright (C) 2016 MarlinFirmware [https://github.com/MarlinFirmware/Marlin]
 *
 * Based on Sprinter and grbl.
 * 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/>.
 *
 */

#include "Marlin.h"

#if ENABLED(EXPERIMENTAL_I2CBUS)

#include "twibus.h"

#include <Wire.h>

TWIBus::TWIBus() {
  #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
  this->reset();
}

void TWIBus::reset() {
  this->buffer_s = 0;
  this->buffer[0] = 0x00;
}

void TWIBus::address(const uint8_t adr) {
  this->addr = adr;

  #if ENABLED(DEBUG_TWIBUS)
    debug(PSTR("address"), adr);
  #endif
}

void TWIBus::addbyte(const char c) {
  if (this->buffer_s >= COUNT(this->buffer)) return;
  this->buffer[this->buffer_s++] = c;
  #if ENABLED(DEBUG_TWIBUS)
    debug(PSTR("addbyte"), c);
  #endif
}

void TWIBus::addbytes(char src[], uint8_t bytes) {
  #if ENABLED(DEBUG_TWIBUS)
    debug(PSTR("addbytes"), bytes);
  #endif
  while (bytes--) this->addbyte(*src++);
}

void TWIBus::addstring(char str[]) {
  #if ENABLED(DEBUG_TWIBUS)
    debug(PSTR("addstring"), str);
  #endif
  while (char c = *str++) this->addbyte(c);
}

void TWIBus::send() {
  if (!this->addr) return;

  #if ENABLED(DEBUG_TWIBUS)
    debug(PSTR("send"), this->addr);
  #endif

  Wire.beginTransmission(this->addr);
  Wire.write(this->buffer, this->buffer_s);
  Wire.endTransmission();

  this->reset();
}

void TWIBus::echodata(uint8_t bytes, const char prefix[], uint8_t adr) {
  SERIAL_ECHO_START;
  serialprintPGM(prefix);
  SERIAL_ECHOPAIR(": from:", adr);
  SERIAL_ECHOPAIR(" bytes:", bytes);
  SERIAL_ECHOPGM (" data:");
  while (bytes-- && Wire.available()) SERIAL_CHAR(Wire.read());
  SERIAL_EOL;
}

void TWIBus::reqbytes(const uint8_t bytes) {
  if (!this->addr) return;

  #if ENABLED(DEBUG_TWIBUS)
    debug(PSTR("reqbytes"), bytes);
  #endif

  // requestFrom() is a blocking function
  Wire.requestFrom(this->addr, bytes);

  // Wait until all bytes arrive, or timeout
  millis_t t = millis() + this->timeout;
  while (Wire.available() < bytes && PENDING(millis(), t)) { /*nada*/ }

  // Simply echo the data to the bus
  this->echodata(bytes, PSTR("i2c-reply"), this->addr);
}

#if I2C_SLAVE_ADDRESS > 0

  void TWIBus::receive(uint8_t bytes) {
    #if ENABLED(DEBUG_TWIBUS)
      debug(PSTR("receive"), bytes);
    #endif
    this->echodata(bytes, PSTR("i2c-receive"), 0);
  }

  void TWIBus::reply(char str[]/*=NULL*/) {
    #if ENABLED(DEBUG_TWIBUS)
      debug(PSTR("reply"), str);
    #endif

    if (str) {
      this->reset();
      this->addstring(str);
    }

    Wire.write(this->buffer, this->buffer_s);

    this->reset();
  }

#endif

#if ENABLED(DEBUG_TWIBUS)

  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

#endif //EXPERIMENTAL_I2CBUS