Merge pull request #276 from makermelissa/master

[Adafruit_Blinka-hackapet.git] / src / adafruit_blinka / microcontroller / nova / spi.py

diff --git a/src/adafruit_blinka/microcontroller/nova/spi.py b/src/adafruit_blinka/microcontroller/nova/spi.py
index 39269229db72879e7546cd6a268a9bf9bc66499a..ce61fd631d48f5594830052ce8f0b828c7bcb44b 100644 (file)
--- a/src/adafruit_blinka/microcontroller/nova/spi.py
+++ b/src/adafruit_blinka/microcontroller/nova/spi.py
@@ -1,16 +1,21 @@
+"""SPI Class for Binho Nova"""
+from adafruit_blinka.microcontroller.nova import Connection
+
+

 class SPI:
 class SPI:

+    """Custom SPI Class for Binho Nova"""
+

     MSB = 0
     PAYLOAD_MAX_LENGTH = 64
 
     def __init__(self, clock):
     MSB = 0
     PAYLOAD_MAX_LENGTH = 64
 
     def __init__(self, clock):

-        from adafruit_blinka.microcontroller.nova import Connection

         self._nova = Connection.getInstance()
         self._nova.setNumericalBase(10)
         self._nova = Connection.getInstance()
         self._nova.setNumericalBase(10)

-        self._nova.setOperationMode(0, 'SPI')
+        self._nova.setOperationMode(0, "SPI")

         self._nova.setClockSPI(0, clock)
         self._nova.setModeSPI(0, 0)
         self._nova.setClockSPI(0, clock)
         self._nova.setModeSPI(0, 0)

-        self._nova.setIOpinMode(0, 'DOUT')
-        self._nova.setIOpinMode(1, 'DOUT')
+        self._nova.setIOpinMode(0, "DOUT")
+        self._nova.setIOpinMode(1, "DOUT")

         self._nova.beginSPI(0)
 
         # Cpol and Cpha set by mode
         self._nova.beginSPI(0)
 
         # Cpol and Cpha set by mode


@@ -20,22 +25,38 @@ class SPI:
         #  2     1    0
         #  3     1    1
 
         #  2     1    0
         #  3     1    1
 

-    def init(self, baudrate=100000, polarity=0, phase=0, bits=8,
-             firstbit=MSB, sck=None, mosi=None, miso=None):
-        #print("baudrate: " + str(baudrate))
-        #print("mode: " + str((polarity<<1) | (phase)))
+    # pylint: disable=too-many-arguments,unused-argument
+    def init(
+        self,
+        baudrate=100000,
+        polarity=0,
+        phase=0,
+        bits=8,
+        firstbit=MSB,
+        sck=None,
+        mosi=None,
+        miso=None,
+    ):
+        """Initialize the Port"""
+        # print("baudrate: " + str(baudrate))
+        # print("mode: " + str((polarity<<1) | (phase)))

         self._nova.setClockSPI(0, baudrate)
         self._nova.setClockSPI(0, baudrate)

-        self._nova.setModeSPI(0, (polarity<<1) | (phase))
+        self._nova.setModeSPI(0, (polarity << 1) | (phase))
+
+    # pylint: enable=too-many-arguments,unused-argument

 
     @staticmethod
     def get_received_data(lineOutput):
 
     @staticmethod
     def get_received_data(lineOutput):

-        return (lineOutput.split('RXD ')[1])
+        """Return any received data"""
+        return lineOutput.split("RXD ")[1]

 
     @property
     def frequency(self):
 
     @property
     def frequency(self):

-        return self._nova.getClockSPI(0).split('CLK ')[1]
+        """Return the current frequency"""
+        return self._nova.getClockSPI(0).split("CLK ")[1]

 
     def write(self, buf, start=0, end=None):
 
     def write(self, buf, start=0, end=None):

+        """Write data from the buffer to SPI"""

         end = end if end else len(buf)
         chunks, rest = divmod(end - start, self.PAYLOAD_MAX_LENGTH)
         for i in range(chunks):
         end = end if end else len(buf)
         chunks, rest = divmod(end - start, self.PAYLOAD_MAX_LENGTH)
         for i in range(chunks):


@@ -46,21 +67,30 @@ class SPI:
             self._nova.writeToBuffer(0, 0, buffer_data)
             self._nova.transferBufferSPI(0, chunk_end - chunk_start + 1)
         if rest:
             self._nova.writeToBuffer(0, 0, buffer_data)
             self._nova.transferBufferSPI(0, chunk_end - chunk_start + 1)
         if rest:

-            buffer_data = buf[-1*rest:]
+            buffer_data = buf[-1 * rest :]

             self._nova.clearBuffer(0)
             self._nova.writeToBuffer(0, 0, buffer_data)
             self._nova.transferBufferSPI(0, rest)
 
     def readinto(self, buf, start=0, end=None, write_value=0):
             self._nova.clearBuffer(0)
             self._nova.writeToBuffer(0, 0, buffer_data)
             self._nova.transferBufferSPI(0, rest)
 
     def readinto(self, buf, start=0, end=None, write_value=0):

+        """Read data from SPI and into the buffer"""

         end = end if end else len(buf)
         for i in range(start, end):
         end = end if end else len(buf)
         for i in range(start, end):

-            buf[start+i] = int(self.get_received_data(self._nova.transferSPI(0, write_value)))
+            buf[start + i] = int(
+                self.get_received_data(self._nova.transferSPI(0, write_value))
+            )

 
 

-    def write_readinto(self, buffer_out, buffer_in, out_start=0, out_end=None, in_start=0, in_end=None):
+    # pylint: disable=too-many-arguments
+    def write_readinto(
+        self, buffer_out, buffer_in, out_start=0, out_end=None, in_start=0, in_end=None
+    ):
+        """Perform a half-duplex write from buffer_out and then
+        read data into buffer_in
+        """

         out_end = out_end if out_end else len(buffer_out)
         in_end = in_end if in_end else len(buffer_in)
         out_end = out_end if out_end else len(buffer_out)
         in_end = in_end if in_end else len(buffer_in)

-        readlen = in_end-in_start
-        writelen = out_end-out_start
+        readlen = in_end - in_start
+        writelen = out_end - out_start

         if readlen > writelen:
             # resize out and pad with 0's
             tmp = bytearray(buffer_out)
         if readlen > writelen:
             # resize out and pad with 0's
             tmp = bytearray(buffer_out)


@@ -70,5 +100,7 @@ class SPI:
         for data_out in buffer_out:
             data_in = int(self.get_received_data(self._nova.transferSPI(0, data_out)))
             if i < readlen:
         for data_out in buffer_out:
             data_in = int(self.get_received_data(self._nova.transferSPI(0, data_out)))
             if i < readlen:

-                buffer_in[in_start+i] = data_in
+                buffer_in[in_start + i] = data_in

             i += 1
             i += 1

+
+    # pylint: enable=too-many-arguments