整理代码

hardware/SPI.cpp

@@ -0,0 +1,73 @@
+/*
+ *******************************************************************************
+ *                                   INCLUDE
+ *******************************************************************************
+ */
+#include "SPI.h"
+#include <Poco/Format.h>
+#include <Poco/Exception.h>
+#include <linux/spi/spidev.h>
+#include <sys/stat.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+#include <fcntl.h>
+
+namespace Hardware
+{
+/*
+ *******************************************************************************
+ *                                CLASS DEFINE
+ *******************************************************************************
+ */
+const uint8_t SPI::MODE_0 = SPI_MODE_0;
+const uint8_t SPI::MODE_1 = SPI_MODE_1;
+const uint8_t SPI::MODE_2 = SPI_MODE_2;
+const uint8_t SPI::MODE_3 = SPI_MODE_3;
+
+SPI::SPI(int bus, int cs)
+{
+    _fd = open(Poco::format("/dev/spidev%d.%d", bus, cs).c_str(), O_RDWR);
+    if (_fd < 0)
+        throw Poco::IOException("SPI: Open device FAIL");
+}
+
+SPI::~SPI()
+{
+    close(_fd);
+}
+
+int SPI::setSpeed(uint32_t speed)
+{
+    return ioctl(_fd, SPI_IOC_WR_MAX_SPEED_HZ, &speed);
+}
+
+int SPI::setMode(uint8_t mode)
+{
+    return ioctl(_fd, SPI_IOC_WR_MODE, &mode);
+}
+
+int SPI::setBitsPerWord(uint8_t bits)
+{
+    return ioctl(_fd, SPI_IOC_WR_BITS_PER_WORD, &bits);
+}
+
+int SPI::setLSBFirst(bool enable)
+{
+    uint8_t lsb = enable ? 1 : 0;
+    return ioctl(_fd, SPI_IOC_WR_LSB_FIRST, &lsb);
+}
+
+int SPI::writeThenRead(const uint8_t* tx, int txlen, uint8_t* rx, int rxlen)
+{
+    struct spi_ioc_transfer trans[2] = {{0}, {0}};
+
+    trans[0].tx_buf = (unsigned long)tx;
+    trans[0].rx_buf = 0;
+    trans[0].len    = txlen;
+
+    trans[1].tx_buf = 0;
+    trans[1].rx_buf = (unsigned long)rx;
+    trans[1].len    = rxlen;
+    return ioctl(_fd, SPI_IOC_MESSAGE(2), trans);
+}
+}