Prusa-Firmware/Firmware/twi.cpp
2022-02-17 00:05:31 +01:00

148 lines
3.1 KiB
C++

/*
twi.c - Stripped-down TWI/I2C library
Copyright (c) 2006 Nicholas Zambetti. All right reserved.
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library 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
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
Modified 2012 by Todd Krein (todd@krein.org) to implement repeated starts
*/
#include <math.h>
#include "config.h"
#include "fastio.h"
#include "twi.h"
#include "Timer.h"
void twi_init(void)
{
// activate internal pullups for twi.
WRITE(SDA_PIN, 1);
WRITE(SCL_PIN, 1);
// initialize twi prescaler and bit rate
TWSR &= ~(_BV(TWPS0) | _BV(TWPS1));
TWBR = ((F_CPU / TWI_FREQ) - 16) / 2;
/* twi bit rate formula from atmega128 manual pg 204
SCL Frequency = CPU Clock Frequency / (16 + (2 * TWBR))
note: TWBR should be 10 or higher for master mode
It is 72 for a 16mhz Wiring board with 100kHz TWI */
}
void twi_disable(void)
{
// disable TWI hardware.
TWCR = 0;
// deactivate internal pullups for twi.
WRITE(SDA_PIN, 0);
WRITE(SCL_PIN, 0);
}
static void twi_stop()
{
TWCR = _BV(TWEN) | _BV(TWINT) | _BV(TWSTO);
}
static uint8_t twi_wait(uint8_t status)
{
static ShortTimer timmy;
timmy.start();
while(!(TWCR & _BV(TWINT))) {
if (timmy.expired(TWI_TIMEOUT_MS)) {
return 2;
}
}
if(TW_STATUS != status)
{
twi_stop();
return 1;
}
return 0;
}
static uint8_t twi_start(uint8_t address, uint8_t reg)
{
// send start condition
TWCR = _BV(TWEN) | _BV(TWINT) | _BV(TWSTA);
if(twi_wait(TW_START))
return 1;
// send address
TWDR = TW_WRITE | (address << 1);
TWCR = _BV(TWEN) | _BV(TWINT);
if(twi_wait(TW_MT_SLA_ACK))
return 2;
// send register
TWDR = reg;
TWCR = _BV(TWEN) | _BV(TWINT);
if(twi_wait(TW_MT_DATA_ACK))
return 3;
return 0;
}
uint8_t twi_r8(uint8_t address, uint8_t reg, uint8_t* data)
{
if(twi_start(address, reg))
return 1;
// repeat start
TWCR = _BV(TWEN) | _BV(TWINT) | _BV(TWSTA);
if(twi_wait(TW_REP_START))
return 2;
// start receiving
TWDR = TW_READ | (address << 1);
TWCR = _BV(TWEN) | _BV(TWINT);
if(twi_wait(TW_MR_SLA_ACK))
return 3;
// receive data
TWCR = _BV(TWEN) | _BV(TWINT);
if(twi_wait(TW_MR_DATA_NACK))
return 4;
*data = TWDR;
// send stop
twi_stop();
return 0;
}
uint8_t twi_w8(uint8_t address, uint8_t reg, uint8_t data)
{
if(twi_start(address, reg))
return 1;
// send data
TWDR = data;
TWCR = _BV(TWEN) | _BV(TWINT);
if(twi_wait(TW_MT_DATA_ACK))
return 2;
// send stop
twi_stop();
return 0;
}