implemented additional functionalities

Author: stitchesnburns

adafruit_fingerprint.py

@@ -62,12 +62,17 @@
 _STORE = const(0x06)
 _LOAD = const(0x07)
 _UPLOAD = const(0x08)
+_DOWNLOAD = const(0x09)
+_UPLOADIMAGE = const(0x0A)
+_DOWNLOADIMAGE = const(0x0B)
 _DELETE = const(0x0C)
 _EMPTY = const(0x0D)
+_READSYSPARA = const(0x0F)
 _HISPEEDSEARCH = const(0x1B)
 _VERIFYPASSWORD = const(0x13)
 _TEMPLATECOUNT = const(0x1D)
 _TEMPLATEREAD = const(0x1F)
+_GETECHO = const(0x53)
 
 # Packet error code
 OK = const(0x0)
@@ -92,6 +97,7 @@
 INVALIDREG = const(0x1A)
 ADDRCODE = const(0x20)
 PASSVERIFY = const(0x21)
+MODULEOK = const(0x55)
 
 class Adafruit_Fingerprint:
     """UART based fingerprint sensor."""
@@ -103,6 +109,11 @@ class Adafruit_Fingerprint:
     confidence = None
     templates = []
     template_count = None
+    library_size = None
+    security_level = None
+    device_address = None
+    data_packet_size = None
+    baudrate = None
 
     def __init__(self, uart, passwd=(0, 0, 0, 0)):
 	# Create object with UART for interface, and default 32-bit password
@@ -111,6 +122,14 @@ def __init__(self, uart, passwd=(0, 0, 0, 0)):
         if self.verify_password() != OK:
             raise RuntimeError('Failed to find sensor, check wiring!')
 
+    def check_module(self):
+        """Checks the state of the fingerprint scanner module.
+        Returns OK or error."""
+        self._send_packet([_GETECHO])
+        if self._get_packet(12)[0] != MODULEOK:
+            raise RuntimeError('Something is wrong with the sensor.')
+        return True
+
     def verify_password(self):
         """Checks if the password/connection is correct, returns True/False"""
         self._send_packet([_VERIFYPASSWORD] + list(self.password))
@@ -124,6 +143,19 @@ def count_templates(self):
         self.template_count = struct.unpack('>H', bytes(r[1:3]))[0]
         return r[0]
 
+    def read_sysparam(self):
+        """Returns the system parameters on success via attributes."""
+        self._send_packet([_READSYSPARA])
+        r = self._get_packet(28)
+        if r[0] != OK:
+            raise RuntimeError('Command failed.')
+        self.library_size = struct.unpack('>H', bytes(r[5:7]))[0]
+        self.security_level = struct.unpack('>H', bytes(r[7:9]))[0]
+        self.device_address = bytes(r[9:13])
+        self.data_packet_size = struct.unpack('>H', bytes(r[13:15]))[0]
+        self.baudrate = struct.unpack('>H', bytes(r[15:17]))[0]
+        return r[0]
+
     def get_image(self):
         """Requests the sensor to take an image and store it memory, returns
         the packet error code or OK success"""
@@ -142,10 +174,10 @@ def create_model(self):
         self._send_packet([_REGMODEL])
         return self._get_packet(12)[0]
 
-    def store_model(self, location):
+    def store_model(self, location, charbuf):
         """Requests the sensor store the model into flash memory and assign
         a location. Returns the packet error code or OK success"""
-        self._send_packet([_STORE, 1, location >> 8, location & 0xFF])
+        self._send_packet([_STORE, charbuf, location >> 8, location & 0xFF])
         return self._get_packet(12)[0]
 
     def delete_model(self, location):
@@ -154,6 +186,48 @@ def delete_model(self, location):
         self._send_packet([_DELETE, location >> 8, location & 0xFF, 0x00, 0x01])
         return self._get_packet(12)[0]
 
+    def load_model(self, location, charbuf):
+        """
+        """
+        self._send_packet([_LOAD, charbuf, location >> 8, location & 0xFF])
+        return self._get_packet(12)[0]
+
+    def get_fpdata(self, buffer='char', slot=1):
+        """Requests the sensor to transfer the fingerprint image or
+        template.  Returns the data payload only."""
+        if slot != 1 or slot != 2:
+            # raise error or use default value?
+            slot = 1
+        if buffer == 'image':
+            self._send_packet([_UPLOADIMAGE])
+        elif buffer == 'char':
+            self._send_packet([_UPLOAD, slot])
+        else:
+            raise RuntimeError('Uknown buffer type')
+        if self._get_packet(12)[0] == 0:
+            res = self._get_data(9)
+            #print('datasize: ' + str(len(res)))
+        #print(res)
+        return res
+
+    def send_fpdata(self, data, buffer='char', slot=1):
+        """ONGOING"""
+        print(slot)
+        if slot != 1 or slot != 2:
+            # raise error or use default value?
+            slot = 2
+        if buffer == 'image':
+            self._send_packet([_DOWNLOADIMAGE])
+        elif buffer == 'char':
+            self._send_packet([_DOWNLOAD, slot])
+        else:
+            raise RuntimeError('Uknown buffer type')
+        if self._get_packet(12)[0] == 0:
+            self._send_data(data) 
+            #print('datasize: ' + str(len(res)))
+        #print(res)
+        return True
+
     def empty_library(self):
         """Requests the sensor to delete all models from flash memory.
         Returns the packet error code or OK success"""
@@ -164,16 +238,17 @@ def read_templates(self):
         """Requests the sensor to list of all template locations in use and
         stores them in self.templates. Returns the packet error code or OK success"""
         self.templates = []
+        self.read_sysparam()
         temp_r = [0x0c, ]
-        for j in range(4):
+        for j in range(int(self.library_size/250)):
             self._send_packet([_TEMPLATEREAD, j])
             r = self._get_packet(44)
             if r[0] == OK:
                 for i in range(32):
                     byte = r[i+1]
                     for bit in range(8):
                         if byte & (1 << bit):
-                            self.templates.append(i * 8 + bit + j * 256)
+                            self.templates.append((i * 8) + bit + (j * 256))
                 temp_r = r
             else:
                 r = temp_r
@@ -185,7 +260,8 @@ def finger_fast_search(self):
         and self.confidence. Returns the packet error code or OK success"""
         # high speed search of slot #1 starting at page 0x0000 and page #0x00A3
         #self._send_packet([_HISPEEDSEARCH, 0x01, 0x00, 0x00, 0x00, 0xA3])
-        self._send_packet([_HISPEEDSEARCH, 0x01, 0x00, 0x00, 0x03, 0xE8])
+        # or page #0x03E9 to accommodate modules with up to 1000 capacity
+        self._send_packet([_HISPEEDSEARCH, 0x01, 0x00, 0x00, 0x03, 0xE9])
         r = self._get_packet(16)
         self.finger_id, self.confidence = struct.unpack('>HH', bytes(r[1:5]))
         return r[0]
@@ -214,10 +290,61 @@ def _get_packet(self, expected):
         if packet_type != _ACKPACKET:
             raise RuntimeError('Incorrect packet data')
 
+        # we should check the checksum
+        # but i don't know how
+        # not yet anyway
+        #packet_sum = struct.unpack('>H', res[9+(length-2):9+length])[0]
+        #print(packet_sum)
+        #print(packet_type + length + struct.unpack('>HHHH', res[9:9+(length-2)]))
+
         reply = [i for i in res[9:9+(length-2)]]
         #print(reply)
         return reply
 
+    def _get_data(self, expected):
+        """ Gets packet from serial and checks structure for _DATAPACKET
+        and _ENDDATAPACKET.  Alternate method for getting data such
+        as fingerprint image, etc.  Returns the data payload."""
+        res = self._uart.read(expected)
+        #print("Got", res)
+        if (not res) or (len(res) != expected):
+            raise RuntimeError('Failed to read data from sensor')
+
+        # first two bytes are start code
+        start = struct.unpack('>H', res[0:2])[0]
+        #print(start)
+        if start != _STARTCODE:
+            raise RuntimeError('Incorrect packet data')
+        # next 4 bytes are address
+        addr = [i for i in res[2:6]]
+        #print(addr)
+        if addr != self.address:
+            raise RuntimeError('Incorrect address')
+
+        packet_type, length = struct.unpack('>BH', res[6:9])
+        #print(str(packet_type) + ' ' + str(length))
+
+        # todo: check checksum
+
+        if packet_type != _DATAPACKET:
+            if packet_type != _ENDDATAPACKET:
+                raise RuntimeError('Incorrect packet data')
+
+        if packet_type == _DATAPACKET:
+            res = self._uart.read(length-2)
+            # todo: we should really inspect the headers and checksum
+            reply = [i for i in res[0:length]]
+            self._uart.read(2) # disregard checksum but we really shouldn't
+            reply += self._get_data(9)
+        elif packet_type == _ENDDATAPACKET:
+            res = self._uart.read(length-2)
+            # todo: we should really inspect the headers and checksum
+            reply = [i for i in res[0:length]]
+            self._uart.read(2) # disregard checksum but we really shouldn't
+        print(len(reply))
+        #print(reply)
+        return reply
+
     def _send_packet(self, data):
         packet = [_STARTCODE >> 8, _STARTCODE & 0xFF]
         packet = packet + self.address
@@ -235,3 +362,74 @@ def _send_packet(self, data):
 
         #print("Sending: ", [hex(i) for i in packet])
         self._uart.write(bytearray(packet))
+
+    def _send_data(self, data):
+        """ONGOING"""
+        print(len(data))
+        self.read_sysparam()
+        if self.data_packet_size == 0:
+            data_length = 32
+        elif self.data_packet_size == 1:
+            data_length = 64
+        elif self.data_packet_size == 2:
+            data_length = 128
+        elif self.data_packet_size == 3:
+            data_length = 256
+
+        for i in range(int(len(data) / (data_length - 2))):
+            start = i * (data_length - 2)
+            end = (i + 1) * (data_length - 2)
+            #print(start)
+            #print(end)
+            #print(i)
+
+            packet = [_STARTCODE >> 8, _STARTCODE & 0xFF]
+            packet = packet + self.address
+            packet.append(_DATAPACKET)
+            length = len(data[start:end]) + 2
+            #print(length)
+            packet.append(length >> 8)
+            packet.append(length & 0xFF)
+            checksum = _DATAPACKET + (length >> 8) + (length & 0xFF)
+
+            for j in range(len(data[start:end])):
+                packet.append(data[j])
+                #packet.append(struct.pack('@B', data[j]))
+                checksum += data[j]
+
+            packet.append(checksum >> 8)
+            packet.append(checksum & 0xFF)
+
+            #print("Sending: ", [hex(i) for i in packet])
+            #self._uart.write(bytearray(packet))
+            self._uart.write(packet)
+            #print(i)
+
+        i += 1
+        start = i * (data_length - 2)
+        end = (i + 1) * (data_length - 2)
+        #print(start)
+        #print(end)
+        #print(i)
+
+        packet = [_STARTCODE >> 8, _STARTCODE & 0xFF]
+        packet = packet + self.address
+        packet.append(_ENDDATAPACKET)
+        length = len(data[start:end]) + 2
+        #print(length)
+        packet.append(length >> 8)
+        packet.append(length & 0xFF)
+        checksum = _DATAPACKET + (length >> 8) + (length & 0xFF)
+
+        for j in range(len(data[start:end])):
+            packet.append(data[j])
+            #packet.append(struct.pack('@B', data[j]))
+            checksum += data[j]
+
+        packet.append(checksum >> 8)
+        packet.append(checksum & 0xFF)
+
+        #print("Sending: ", [hex(i) for i in packet])
+        #self._uart.write(bytearray(packet))
+        self._uart.write(packet)
+        #print(i)