mirror of
https://github.com/DCC-EX/CommandStation-EX.git
synced 2024-11-22 23:56:13 +01:00
ba9b363058
Following on from the change to I2CManager_SAMD.h, the capability of deferring a request to change the speed of the I2C has been removed from the SAMD driver and put into the common NonBlocking code, so that all native drivers benefit from it.
297 lines
11 KiB
C
297 lines
11 KiB
C
/*
|
|
* © 2023, Neil McKechnie
|
|
* © 2022 Paul M Antoine
|
|
* All rights reserved.
|
|
*
|
|
* This file is part of CommandStation-EX
|
|
*
|
|
* This 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.
|
|
*
|
|
* It 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 CommandStation. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef I2CMANAGER_NONBLOCKING_H
|
|
#define I2CMANAGER_NONBLOCKING_H
|
|
|
|
#include <Arduino.h>
|
|
#include "I2CManager.h"
|
|
#if defined(I2C_USE_INTERRUPTS)
|
|
// atomic.h isn't available on SAMD, and likely others too...
|
|
#if defined(__AVR__)
|
|
#include <util/atomic.h>
|
|
#elif defined(__arm__)
|
|
// Helper assembly language functions
|
|
static __inline__ uint8_t my_iSeiRetVal(void)
|
|
{
|
|
__asm__ __volatile__ ("cpsie i" ::);
|
|
return 1;
|
|
}
|
|
|
|
static __inline__ uint8_t my_iCliRetVal(void)
|
|
{
|
|
__asm__ __volatile__ ("cpsid i" ::);
|
|
return 1;
|
|
}
|
|
|
|
static __inline__ void my_iRestore(const uint32_t *__s)
|
|
{
|
|
uint32_t res = *__s;
|
|
__asm__ __volatile__ ("MSR primask, %0" : : "r" (res) );
|
|
}
|
|
|
|
static __inline__ uint32_t my_iGetIReg( void )
|
|
{
|
|
uint32_t reg;
|
|
__asm__ __volatile__ ("MRS %0, primask" : "=r" (reg) );
|
|
return reg;
|
|
}
|
|
// Macros for atomic isolation
|
|
#define MY_ATOMIC_RESTORESTATE uint32_t _sa_saved \
|
|
__attribute__((__cleanup__(my_iRestore))) = my_iGetIReg()
|
|
|
|
#define ATOMIC() \
|
|
for ( MY_ATOMIC_RESTORESTATE, _done = my_iCliRetVal(); \
|
|
_done; _done = 0 )
|
|
|
|
#define ATOMIC_BLOCK(x) ATOMIC()
|
|
#define ATOMIC_RESTORESTATE
|
|
#endif
|
|
#else
|
|
#define ATOMIC_BLOCK(x)
|
|
#define ATOMIC_RESTORESTATE
|
|
#endif
|
|
|
|
// This module is only compiled if I2C_USE_WIRE is not defined, so undefine it here
|
|
// to get intellisense to work correctly.
|
|
#if defined(I2C_USE_WIRE)
|
|
#undef I2C_USE_WIRE
|
|
#endif
|
|
|
|
/***************************************************************************
|
|
* Initialise the I2CManagerAsync class.
|
|
***************************************************************************/
|
|
void I2CManagerClass::_initialise()
|
|
{
|
|
queueHead = queueTail = NULL;
|
|
state = I2C_STATE_FREE;
|
|
I2C_init();
|
|
_setClock(_clockSpeed);
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Set I2C clock speed. Normally 100000 (Standard) or 400000 (Fast)
|
|
* on Arduino. Mega4809 supports 1000000 (Fast+) too.
|
|
* This function saves the desired clock speed and the startTransaction
|
|
* function acts on it before a new transaction, to avoid speed changes
|
|
* during an I2C transaction.
|
|
***************************************************************************/
|
|
void I2CManagerClass::_setClock(unsigned long i2cClockSpeed) {
|
|
pendingClockSpeed = i2cClockSpeed;
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Helper function to start operations, if the I2C interface is free and
|
|
* there is a queued request to be processed.
|
|
* If there's an I2C clock speed change pending, then implement it before
|
|
* starting the operation.
|
|
***************************************************************************/
|
|
void I2CManagerClass::startTransaction() {
|
|
ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
|
|
if ((state == I2C_STATE_FREE) && (queueHead != NULL)) {
|
|
state = I2C_STATE_ACTIVE;
|
|
// Check for pending clock speed change
|
|
if (pendingClockSpeed) {
|
|
// We're about to start a new I2C transaction, so set clock now.
|
|
I2C_setClock(pendingClockSpeed);
|
|
pendingClockSpeed = 0;
|
|
}
|
|
startTime = micros();
|
|
currentRequest = queueHead;
|
|
rxCount = txCount = 0;
|
|
// Copy key fields to static data for speed.
|
|
operation = currentRequest->operation & OPERATION_MASK;
|
|
// Start the I2C process going.
|
|
I2C_sendStart();
|
|
}
|
|
}
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Function to queue a request block and initiate operations.
|
|
***************************************************************************/
|
|
void I2CManagerClass::queueRequest(I2CRB *req) {
|
|
req->status = I2C_STATUS_PENDING;
|
|
req->nextRequest = NULL;
|
|
|
|
ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
|
|
if (!queueTail)
|
|
queueHead = queueTail = req; // Only item on queue
|
|
else
|
|
queueTail = queueTail->nextRequest = req; // Add to end
|
|
startTransaction();
|
|
}
|
|
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Initiate a write to an I2C device (non-blocking operation)
|
|
***************************************************************************/
|
|
uint8_t I2CManagerClass::write(uint8_t i2cAddress, const uint8_t *writeBuffer, uint8_t writeLen, I2CRB *req) {
|
|
// Make sure previous request has completed.
|
|
req->wait();
|
|
req->setWriteParams(i2cAddress, writeBuffer, writeLen);
|
|
queueRequest(req);
|
|
return I2C_STATUS_OK;
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Initiate a write from PROGMEM (flash) to an I2C device (non-blocking operation)
|
|
***************************************************************************/
|
|
uint8_t I2CManagerClass::write_P(uint8_t i2cAddress, const uint8_t * writeBuffer, uint8_t writeLen, I2CRB *req) {
|
|
// Make sure previous request has completed.
|
|
req->wait();
|
|
req->setWriteParams(i2cAddress, writeBuffer, writeLen);
|
|
req->operation = OPERATION_SEND_P;
|
|
queueRequest(req);
|
|
return I2C_STATUS_OK;
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Initiate a read from the I2C device, optionally preceded by a write
|
|
* (non-blocking operation)
|
|
***************************************************************************/
|
|
uint8_t I2CManagerClass::read(uint8_t i2cAddress, uint8_t *readBuffer, uint8_t readLen,
|
|
const uint8_t *writeBuffer, uint8_t writeLen, I2CRB *req)
|
|
{
|
|
// Make sure previous request has completed.
|
|
req->wait();
|
|
req->setRequestParams(i2cAddress, readBuffer, readLen, writeBuffer, writeLen);
|
|
queueRequest(req);
|
|
return I2C_STATUS_OK;
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Set I2C timeout value in microseconds. The timeout applies to the entire
|
|
* I2CRB request, e.g. where a write+read is performed, the timer is not
|
|
* reset before the read.
|
|
***************************************************************************/
|
|
void I2CManagerClass::setTimeout(unsigned long value) {
|
|
timeout = value;
|
|
};
|
|
|
|
/***************************************************************************
|
|
* checkForTimeout() function, called from isBusy() and wait() to cancel
|
|
* requests that are taking too long to complete. Such faults
|
|
* may be caused by an I2C wire short for example.
|
|
***************************************************************************/
|
|
void I2CManagerClass::checkForTimeout() {
|
|
ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
|
|
I2CRB *t = queueHead;
|
|
if (state==I2C_STATE_ACTIVE && t!=0 && t==currentRequest && timeout > 0) {
|
|
// Check for timeout
|
|
unsigned long elapsed = micros() - startTime;
|
|
if (elapsed > timeout) {
|
|
#ifdef DIAG_IO
|
|
//DIAG(F("I2CManager Timeout on x%x, I2CRB=x%x"), t->i2cAddress, currentRequest);
|
|
#endif
|
|
// Excessive time. Dequeue request
|
|
queueHead = t->nextRequest;
|
|
if (!queueHead) queueTail = NULL;
|
|
currentRequest = NULL;
|
|
bytesToReceive = bytesToSend = 0;
|
|
// Post request as timed out.
|
|
t->status = I2C_STATUS_TIMEOUT;
|
|
// Reset TWI interface so it is able to continue
|
|
// Try close and init, not entirely satisfactory but sort of works...
|
|
I2C_close(); // Shutdown and restart twi interface
|
|
I2C_init();
|
|
_setClock(_clockSpeed);
|
|
state = I2C_STATE_FREE;
|
|
// I2C_sendStop(); // in case device is waiting for a stop condition
|
|
|
|
// Initiate next queued request if any.
|
|
startTransaction();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Loop function, for general background work
|
|
***************************************************************************/
|
|
void I2CManagerClass::loop() {
|
|
#if !defined(I2C_USE_INTERRUPTS)
|
|
handleInterrupt();
|
|
#endif
|
|
// Call function to monitor for stuch I2C operations.
|
|
checkForTimeout();
|
|
}
|
|
|
|
/***************************************************************************
|
|
* Interupt handler. Call I2C state machine, and dequeue request
|
|
* if completed.
|
|
***************************************************************************/
|
|
void I2CManagerClass::handleInterrupt() {
|
|
|
|
// Update hardware state machine
|
|
I2C_handleInterrupt();
|
|
|
|
// Check if current request has completed. If there's a current request
|
|
// and state isn't active then state contains the completion status of the request.
|
|
if (state != I2C_STATE_ACTIVE && currentRequest != NULL) {
|
|
// Operation has completed.
|
|
if (state == I2C_STATUS_OK || ++retryCounter > MAX_I2C_RETRIES
|
|
|| currentRequest->operation & OPERATION_NORETRY)
|
|
{
|
|
// Status is OK, or has failed and retry count exceeded, or retries disabled.
|
|
// Remove completed request from head of queue
|
|
I2CRB * t = queueHead;
|
|
if (t == currentRequest) {
|
|
queueHead = t->nextRequest;
|
|
if (!queueHead) queueTail = queueHead;
|
|
t->nBytes = rxCount;
|
|
t->status = state;
|
|
|
|
// I2C state machine is now free for next request
|
|
currentRequest = NULL;
|
|
state = I2C_STATE_FREE;
|
|
}
|
|
retryCounter = 0;
|
|
} else {
|
|
// Status is failed and retry permitted.
|
|
// Retry previous request.
|
|
state = I2C_STATE_FREE;
|
|
}
|
|
}
|
|
|
|
if (state == I2C_STATE_FREE && queueHead != NULL) {
|
|
// Allow any pending interrupts before starting the next request.
|
|
interrupts();
|
|
// Start next request
|
|
I2CManager.startTransaction();
|
|
}
|
|
}
|
|
|
|
// Fields in I2CManager class specific to Non-blocking implementation.
|
|
I2CRB * volatile I2CManagerClass::queueHead = NULL;
|
|
I2CRB * volatile I2CManagerClass::queueTail = NULL;
|
|
I2CRB * volatile I2CManagerClass::currentRequest = NULL;
|
|
volatile uint8_t I2CManagerClass::state = I2C_STATE_FREE;
|
|
volatile uint8_t I2CManagerClass::txCount;
|
|
volatile uint8_t I2CManagerClass::rxCount;
|
|
volatile uint8_t I2CManagerClass::operation;
|
|
volatile uint8_t I2CManagerClass::bytesToSend;
|
|
volatile uint8_t I2CManagerClass::bytesToReceive;
|
|
volatile unsigned long I2CManagerClass::startTime;
|
|
uint8_t I2CManagerClass::retryCounter = 0;
|
|
|
|
#endif |