1
0
mirror of https://github.com/DCC-EX/CommandStation-EX.git synced 2024-11-23 08:06:13 +01:00
CommandStation-EX/IO_EXIOExpander.h
2022-12-11 19:44:42 +10:00

129 lines
4.4 KiB
C++

/*
* © 2021, Peter Cole. All rights reserved.
*
* This file is part of EX-CommandStation
*
* 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/>.
*/
/*
* The IO_EX-IOExpander.h device driver integrates with one or more EX-IOExpander devices.
* This device driver will configure the device and all I/O ports on startup, along with
* interacting with the device for all input/output duties.
*
* To create EX-IOExpander devices, these are defined in myHal.cpp:
*
* #include "IO_EX-IOExpander.h"
*
* void halSetup() {
* // EXIOExpander::create(vpin, num_vpins, i2c_address);
* EXIOExpander::create(800, 18, 0x90);
}
*/
#ifndef IO_EX_IOEXPANDER_H
#define IO_EX_IOEXPANDER_H
#include "IO_GPIOBase.h"
#include "FSH.h"
#include "EX-IOExpanderPins.h"
// Include user defined pin maps in myEX-IOExpander if defined
#if __has_include ("myEX-IOExpander.h")
#include "myEX-IOExpander.h"
#endif
/////////////////////////////////////////////////////////////////////////////////////////////////////
/*
* IODevice subclass for EX-IOExpander.
*/
class EXIOExpander : public IODevice {
public:
static void create(VPIN vpin, int nPins, uint8_t i2cAddress, uint8_t numDigitalPins, uint8_t numAnaloguePins) {
if (checkNoOverlap(vpin, nPins, i2cAddress)) new EXIOExpander(vpin, nPins, i2cAddress, numDigitalPins, numAnaloguePins);
}
private:
// Constructor
EXIOExpander(VPIN firstVpin, int nPins, uint8_t i2cAddress, uint8_t numDigitalPins, uint8_t numAnaloguePins) {
_firstVpin = firstVpin;
_nPins = nPins;
_i2cAddress = i2cAddress;
_numDigitalPins = numDigitalPins;
_numAnaloguePins = numAnaloguePins;
addDevice(this);
}
void _begin() {
// Initialise EX-IOExander device
if (I2CManager.exists(_i2cAddress)) {
#ifdef DIAG_IO
_display();
#endif
_setupDevice();
} else {
DIAG(F("EX-IOExpander device not found, I2C:%x"), _i2cAddress);
_deviceState = DEVSTATE_FAILED;
}
}
void _setupDevice() {
// Send digital and analogue pin counts
I2CManager.write(_i2cAddress, 3, REG_EXIOINIT, _numDigitalPins, _numAnaloguePins);
// Enable digital ports
_digitalPinBytes = (_numDigitalPins + 7) / 8;
uint8_t enableDigitalPins[_digitalPinBytes];
for (uint8_t byte = 0; byte < _digitalPinBytes; byte++) {
enableDigitalPins[byte] = 0;
}
for (uint8_t pin = 0; pin < _numDigitalPins; pin++) {
int pinByte = pin / 8;
bitSet(enableDigitalPins[pinByte], pin - pinByte * 8);
}
I2CManager.write(_i2cAddress, _digitalPinBytes + 1, REG_EXIODPIN, *enableDigitalPins);
// Enable analogue ports
_analoguePinBytes = (_numAnaloguePins + 7) / 8;
uint8_t enableAnaloguePins[_analoguePinBytes];
for (uint8_t byte = 0; byte < _analoguePinBytes; byte++) {
enableAnaloguePins[byte] = 0;
}
for (uint8_t pin = 0; pin < _numAnaloguePins; pin++) {
int pinByte = pin / 8;
bitSet(enableAnaloguePins[pinByte], pin - pinByte * 8);
}
I2CManager.write(_i2cAddress, _analoguePinBytes + 1, REG_EXIOAPIN, *enableAnaloguePins);
}
void _display() override {
DIAG(F("EX-IOExpander I2C:x%x Configured on Vpins:%d-%d %S"), _i2cAddress, _firstVpin, _firstVpin+_nPins-1,
_deviceState == DEVSTATE_FAILED ? F("OFFLINE") : F(""));
}
uint8_t _i2cAddress;
uint8_t _numDigitalPins;
uint8_t _numAnaloguePins;
int _digitalPinBytes;
int _analoguePinBytes;
enum {
REG_EXIOINIT = 0xE0, // Flag to initialise setup procedure
REG_EXIODPIN = 0xE1, // Flag we're sending digital pin assignments
REG_EXIOAPIN = 0xE2, // Flag we're sending analogue pin assignments
REG_EXIORDY = 0xE3, // Flag we have completed setup procedure, also for EX-IO to ACK setup
REG_EXIODDIR = 0xE4, // Flag we're sending digital pin direction configuration
REG_EXIODPUP = 0xE5, // Flag we're sending digital pin pullup configuration
};
};
#endif