1765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang#!/usr/bin/python
2765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# Author: Zion Orent <zorent@ics.com>
3765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# Copyright (c) 2015 Intel Corporation.
4765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang#
5765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# Permission is hereby granted, free of charge, to any person obtaining
6765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# a copy of this software and associated documentation files (the
7765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# "Software"), to deal in the Software without restriction, including
8765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# without limitation the rights to use, copy, modify, merge, publish,
9765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# distribute, sublicense, and/or sell copies of the Software, and to
10765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# permit persons to whom the Software is furnished to do so, subject to
11765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# the following conditions:
12765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang#
13765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# The above copyright notice and this permission notice shall be
14765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# included in all copies or substantial portions of the Software.
15765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang#
16765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
20765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
21765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
22765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
23765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
24765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangimport time, sys, signal, atexit
25765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangimport pyupm_zfm20 as upmZfm20
26765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
27765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# Instantiate a ZFM20 Fingerprint reader on UART 0
28765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun ZhangmyFingerprintSensor = upmZfm20.ZFM20(0)
29765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
30765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
31765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang## Exit handlers ##
32765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# This stops python from printing a stacktrace when you hit control-C
33765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangdef SIGINTHandler(signum, frame):
34765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	raise SystemExit
35765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
36765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# This function lets you run code on exit,
37765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# including functions from myFingerprintSensor
38765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangdef exitHandler():
39765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	print "Exiting"
40765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	sys.exit(0)
41765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
42765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# Register exit handlers
43765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangatexit.register(exitHandler)
44765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangsignal.signal(signal.SIGINT, SIGINTHandler)
45765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
46765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
47765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# make sure port is initialized properly.  57600 baud is the default.
48765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangif (not myFingerprintSensor.setupTty(upmZfm20.cvar.int_B57600)):
49765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	print "Failed to setup tty port parameters"
50765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	sys.exit(1)
51765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
52765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# how many valid stored templates (fingerprints) do we have?
53765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint "Total stored templates: %d" % myFingerprintSensor.getNumTemplates()
54765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint " "
55765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
56765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# now spin waiting for a fingerprint to successfully image
57765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint "Waiting for finger print..."
58765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
59765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangwhile (myFingerprintSensor.generateImage() == upmZfm20.ZFM20.ERR_NO_FINGER):
60765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	pass
61765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
62765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# in theory, we have an image
63765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint "Image captured, converting..."
64765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
65765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangrv = myFingerprintSensor.image2Tz(1)
66765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangif (rv != upmZfm20.ZFM20.ERR_OK):
67765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	print "Image conversion failed with error code %d" % rv
68765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	sys.exit(1)
69765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
70765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint "Image conversion succeeded."
71765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint "Searching database..."
72765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
73765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangmyid = upmZfm20.uint16Array(0)
74765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangmyid.__setitem__(0, 0)
75765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangmyscore = upmZfm20.uint16Array(0)
76765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangmyscore.__setitem__(0, 0)
77765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
78765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# we search for a print matching slot 1, where we stored our last
79765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang# converted fingerprint
80765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangrv = myFingerprintSensor.search(1, myid, myscore)
81765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangif (rv != upmZfm20.ZFM20.ERR_OK):
82765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	if (rv == upmZfm20.ZFM20.ERR_FP_NOTFOUND):
83765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang		print "Finger Print not found"
84765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang		sys.exit(0)
85765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang	else:
86765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang		print "Search failed with error code %d" % rv
87765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang		sys.exit(1)
88765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhang
89765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint "Fingerprint found!"
90765adb95dc941c32690d6c43bc08b9d07d197fcbJianxun Zhangprint "ID: %d, Score: %d" % (myid.__getitem__(0), myscore.__getitem__(0))
91