ref
This commit is contained in:
parent
d749216c84
commit
6b1b820dad
|
|
@ -1,243 +0,0 @@
|
||||||
|
|
||||||
import cv2
|
|
||||||
import threading
|
|
||||||
import time
|
|
||||||
import face_processing as fp
|
|
||||||
from access_control_mqtt import access_control
|
|
||||||
from database import Database, Student, Parent
|
|
||||||
from display import Display
|
|
||||||
from stm32 import stm32_distance
|
|
||||||
|
|
||||||
SERIAL_PORT_DISPLAY = "COM11"
|
|
||||||
CAMERA_INDEX = 0
|
|
||||||
|
|
||||||
display = Display(SERIAL_PORT_DISPLAY)
|
|
||||||
cam = cv2.VideoCapture(CAMERA_INDEX)
|
|
||||||
actrl = access_control()
|
|
||||||
db = Database()
|
|
||||||
|
|
||||||
global img
|
|
||||||
global frame_ready
|
|
||||||
|
|
||||||
frame_ready = False
|
|
||||||
|
|
||||||
def read_webcam():
|
|
||||||
global img
|
|
||||||
global frame_ready
|
|
||||||
while True:
|
|
||||||
ret, img = cam.read()
|
|
||||||
frame_ready = True
|
|
||||||
|
|
||||||
'''
|
|
||||||
display.set_page("student")
|
|
||||||
display.set_string("msg.txt","Door Open")
|
|
||||||
'''
|
|
||||||
|
|
||||||
student_info = None
|
|
||||||
parent_info = None
|
|
||||||
faces = None
|
|
||||||
|
|
||||||
|
|
||||||
threading.Thread(target=read_webcam).start()
|
|
||||||
|
|
||||||
|
|
||||||
def door_open():
|
|
||||||
actrl.unlock_door()
|
|
||||||
for i in range(10):
|
|
||||||
display.set_string("msg.txt",f'Door will close in {10-i}')
|
|
||||||
time.sleep(1)
|
|
||||||
|
|
||||||
i = 5
|
|
||||||
#TODO Check if door close, lock
|
|
||||||
alarm_active: bool = False
|
|
||||||
display.set_page("leftopen")
|
|
||||||
while(i>=0):
|
|
||||||
if (not actrl.get_door_state() and not alarm_active): # False door is left open
|
|
||||||
|
|
||||||
display.set_string("msg.txt","Door is left open, Alram in {} sec".format(i))
|
|
||||||
time.sleep(1)
|
|
||||||
i -= 1
|
|
||||||
if i == 0:
|
|
||||||
actrl.activate_alarm()
|
|
||||||
display.set_string("msg.txt","Alarm Active, Please Close Door.")
|
|
||||||
#actrl.activate_LED_alarm()
|
|
||||||
|
|
||||||
elif(actrl.get_door_state()):
|
|
||||||
actrl.lock_door()
|
|
||||||
display.set_page("scan")
|
|
||||||
should_restart = True
|
|
||||||
break
|
|
||||||
#display.set_page("scan")
|
|
||||||
|
|
||||||
|
|
||||||
SD = stm32_distance("COM13")
|
|
||||||
#start_time = time.perf_counter()
|
|
||||||
|
|
||||||
flag: bool = False
|
|
||||||
|
|
||||||
"""
|
|
||||||
while True:
|
|
||||||
state = round((perf_counter())%1)
|
|
||||||
if(flag!=state):
|
|
||||||
flag = state
|
|
||||||
print(f"change to : {state}")
|
|
||||||
print(state)
|
|
||||||
delay(0.05)
|
|
||||||
"""
|
|
||||||
|
|
||||||
while True:
|
|
||||||
|
|
||||||
while not frame_ready:
|
|
||||||
time.sleep(1)
|
|
||||||
|
|
||||||
should_restart = True
|
|
||||||
distance = SD.get_distance()
|
|
||||||
print(distance)
|
|
||||||
state = distance < 800
|
|
||||||
if state!= flag:
|
|
||||||
flag = state
|
|
||||||
if(state):
|
|
||||||
display.set_string("msg.txt","Scanning")
|
|
||||||
else:
|
|
||||||
display.set_page("scan")
|
|
||||||
|
|
||||||
if (state): # Scan state
|
|
||||||
while should_restart:
|
|
||||||
#Try to identify face
|
|
||||||
should_restart = False
|
|
||||||
#if actrl.get_scan_state():
|
|
||||||
if(True):
|
|
||||||
#print("SCAN ACTIVE!")
|
|
||||||
#display.set_string("msg.txt","Scanning")
|
|
||||||
faces = fp.identify_face(img, target_condidence=0.6) ## Scan student face
|
|
||||||
if(len(faces)==1):
|
|
||||||
#print("faces == 1")
|
|
||||||
facefile = faces[0]['name'] ## File name
|
|
||||||
print(facefile)
|
|
||||||
student_info = db.get_student_info(facefile)## Check is it student?
|
|
||||||
parent_info = db.get_parent_info(facefile)
|
|
||||||
if(student_info!=None): #is student, request another scan for parent
|
|
||||||
print("Find student info")
|
|
||||||
#db.log_access_student_file()
|
|
||||||
parent_info = None
|
|
||||||
display.set_page("student")
|
|
||||||
#display.set_string("msg.txt",f'Scan parent face ({10-i})')
|
|
||||||
display.set_string("name_std.txt", f'Name: {student_info.name}')
|
|
||||||
display.set_string("surname_std.txt",f'Surname: {student_info.surname}')
|
|
||||||
display.set_string("id_std.txt", f'ID: {str(student_info.id)}')
|
|
||||||
"""
|
|
||||||
parent_face = fp.identify_face(img, target_condidence=0.6)
|
|
||||||
parent_info = db.get_parent_info(parent_face)"""
|
|
||||||
#DONE TODO Fill in student info Resolve
|
|
||||||
|
|
||||||
print(student_info)
|
|
||||||
|
|
||||||
i = 0
|
|
||||||
start_time = time.perf_counter()
|
|
||||||
while(time.perf_counter()-start_time < 10):
|
|
||||||
parent_face = fp.identify_face(img, target_condidence=0.6)
|
|
||||||
|
|
||||||
if (len(parent_face)!= 0):
|
|
||||||
parent_facefile = parent_face[0]['name']
|
|
||||||
parent_info = db.get_parent_info(parent_facefile)
|
|
||||||
if(parent_info != None): # Identified parent
|
|
||||||
if(db.check_relationship(student_info,parent_info)): # Check if the detected parent is right for the detected student.
|
|
||||||
print("Relation is OKAY")
|
|
||||||
#DONE TODO Log Access STDw/P
|
|
||||||
db.log_access_student_with_parent(student_info, student_info)
|
|
||||||
#DONE TODO write parrent info to display
|
|
||||||
|
|
||||||
display.set_string("name_pt.txt", f'Name: {parent_info.name}')
|
|
||||||
display.set_string("surname_pt.txt", f'Surname: {parent_info.surname}')
|
|
||||||
display.set_string("id_pt.txt", f'ID: {parent_info.id}')
|
|
||||||
actrl.unlock_door() # Door open
|
|
||||||
display.set_string("msg.txt","Get your kid")
|
|
||||||
#DONE TODO Send timer to display
|
|
||||||
door_open()
|
|
||||||
"""
|
|
||||||
for i in range(10):
|
|
||||||
display.set_string("msg.txt",f'Door will close in {10-i}')
|
|
||||||
time.sleep(1)
|
|
||||||
|
|
||||||
i = 5
|
|
||||||
#TODO Check if door close, lock
|
|
||||||
alarm_active: bool = False
|
|
||||||
while(i>=0):
|
|
||||||
if (not actrl.get_door_state() and not alarm_active): # False door is left open
|
|
||||||
display.set_string("msg.txt","Door is left open, Alram in {} sec".format(i))
|
|
||||||
time.sleep(1)
|
|
||||||
i -= 1
|
|
||||||
if i == 0:
|
|
||||||
actrl.activate_LED_alarm()
|
|
||||||
alarm_active = True
|
|
||||||
|
|
||||||
elif(actrl.get_door_state()):
|
|
||||||
actrl.lock_door()
|
|
||||||
display.set_page("scan")
|
|
||||||
should_restart = True
|
|
||||||
break
|
|
||||||
display.set_page("scan")
|
|
||||||
|
|
||||||
"""
|
|
||||||
|
|
||||||
|
|
||||||
else:
|
|
||||||
display.set_page("scan")
|
|
||||||
display.set_string("msg.txt","Wrong parent. \t Try Again")
|
|
||||||
time.sleep(1)
|
|
||||||
should_restart = True
|
|
||||||
break
|
|
||||||
#TODO Try again does not try again
|
|
||||||
|
|
||||||
elif(len(parent_face) == 0):
|
|
||||||
time.sleep(1)
|
|
||||||
display.set_string("msg.txt",f'Scan parent face ({10-i})')
|
|
||||||
i += 1
|
|
||||||
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
display.set_page("scan")
|
|
||||||
|
|
||||||
elif(parent_info!=None): #is parent, open door
|
|
||||||
#parent_facefile = faces[0]['name']
|
|
||||||
#parent_info = db.get_parent_info(parent_facefile)
|
|
||||||
#if(parent_info!= None):
|
|
||||||
print("Find parent info")
|
|
||||||
display.set_page("prt_wo_stu")
|
|
||||||
display.set_string("name_pt.txt", f'Name: {parent_info.name}')
|
|
||||||
display.set_string("surname_pt.txt", f'Surname: {parent_info.surname}')
|
|
||||||
display.set_string("id_pt.txt", f'ID: {parent_info.id}')
|
|
||||||
|
|
||||||
door_open()
|
|
||||||
pass
|
|
||||||
|
|
||||||
elif (len(faces)>1):
|
|
||||||
#More than one people, error.
|
|
||||||
print("more than one people. \t try again")
|
|
||||||
display.set_string("msg.txt","More than one people. \t try again")
|
|
||||||
time.sleep(2)
|
|
||||||
pass
|
|
||||||
#print("DONE")
|
|
||||||
|
|
||||||
else:
|
|
||||||
display.set_string("msg.txt","Move your face closer to the scanner")
|
|
||||||
print("Scan inactivate!!")
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
'''
|
|
||||||
#is student, request another scan for parent
|
|
||||||
parent_face = fp.identify_face(img, target_condidence=0.6)
|
|
||||||
if(len(faces)==1):
|
|
||||||
parent_info = db.get_parent_info(faces[0]['name'])
|
|
||||||
if(parent_info!=None):
|
|
||||||
if(db.check_relationship(student_info,parent_info)):
|
|
||||||
#Student is under parent, open door.
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
#Wrong Parent, retry.
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
#Not a parent, retry
|
|
||||||
pass
|
|
||||||
'''
|
|
||||||
|
|
@ -1,3 +1,4 @@
|
||||||
|
|
||||||
import cv2
|
import cv2
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
|
|
@ -5,17 +6,19 @@ import face_processing as fp
|
||||||
from access_control_mqtt import access_control
|
from access_control_mqtt import access_control
|
||||||
from database import Database, Student, Parent
|
from database import Database, Student, Parent
|
||||||
from display import Display
|
from display import Display
|
||||||
|
from stm32 import stm32_distance
|
||||||
|
|
||||||
SERIAL_PORT_DISPLAY = "/dev/serial0"
|
SERIAL_PORT_DISPLAY = "COM11"
|
||||||
CAMERA_INDEX = 0
|
CAMERA_INDEX = 0
|
||||||
|
|
||||||
|
display = Display(SERIAL_PORT_DISPLAY)
|
||||||
cam = cv2.VideoCapture(CAMERA_INDEX)
|
cam = cv2.VideoCapture(CAMERA_INDEX)
|
||||||
actrl = access_control()
|
actrl = access_control()
|
||||||
display = Display(SERIAL_PORT_DISPLAY)
|
|
||||||
db = Database()
|
db = Database()
|
||||||
|
|
||||||
global img
|
global img
|
||||||
global frame_ready
|
global frame_ready
|
||||||
|
|
||||||
frame_ready = False
|
frame_ready = False
|
||||||
|
|
||||||
def read_webcam():
|
def read_webcam():
|
||||||
|
|
@ -30,111 +33,211 @@ display.set_page("student")
|
||||||
display.set_string("msg.txt","Door Open")
|
display.set_string("msg.txt","Door Open")
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
student_info = None
|
||||||
|
parent_info = None
|
||||||
|
faces = None
|
||||||
|
|
||||||
|
|
||||||
threading.Thread(target=read_webcam).start()
|
threading.Thread(target=read_webcam).start()
|
||||||
|
|
||||||
|
|
||||||
|
def door_open():
|
||||||
|
actrl.unlock_door()
|
||||||
|
for i in range(10):
|
||||||
|
display.set_string("msg.txt",f'Door will close in {10-i}')
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
i = 5
|
||||||
|
#TODO Check if door close, lock
|
||||||
|
alarm_active: bool = False
|
||||||
|
display.set_page("leftopen")
|
||||||
|
while(i>=0):
|
||||||
|
if (not actrl.get_door_state() and not alarm_active): # False door is left open
|
||||||
|
|
||||||
|
display.set_string("msg.txt","Door is left open, Alram in {} sec".format(i))
|
||||||
|
time.sleep(1)
|
||||||
|
i -= 1
|
||||||
|
if i == 0:
|
||||||
|
actrl.activate_alarm()
|
||||||
|
display.set_string("msg.txt","Alarm Active, Please Close Door.")
|
||||||
|
#actrl.activate_LED_alarm()
|
||||||
|
|
||||||
|
elif(actrl.get_door_state()):
|
||||||
|
actrl.lock_door()
|
||||||
|
display.set_page("scan")
|
||||||
|
should_restart = True
|
||||||
|
break
|
||||||
|
#display.set_page("scan")
|
||||||
|
|
||||||
|
|
||||||
|
SD = stm32_distance("COM13")
|
||||||
|
#start_time = time.perf_counter()
|
||||||
|
|
||||||
|
flag: bool = False
|
||||||
|
|
||||||
|
"""
|
||||||
while True:
|
while True:
|
||||||
|
state = round((perf_counter())%1)
|
||||||
|
if(flag!=state):
|
||||||
|
flag = state
|
||||||
|
print(f"change to : {state}")
|
||||||
|
print(state)
|
||||||
|
delay(0.05)
|
||||||
|
"""
|
||||||
|
|
||||||
|
while True:
|
||||||
|
|
||||||
while not frame_ready:
|
while not frame_ready:
|
||||||
time.sleep(1)
|
time.sleep(1)
|
||||||
|
|
||||||
|
should_restart = True
|
||||||
|
distance = SD.get_distance()
|
||||||
|
print(distance)
|
||||||
|
state = distance < 800
|
||||||
|
if state!= flag:
|
||||||
|
flag = state
|
||||||
|
if(state):
|
||||||
|
display.set_string("msg.txt","Scanning")
|
||||||
|
else:
|
||||||
|
display.set_page("scan")
|
||||||
|
|
||||||
#Try to identify face
|
if (state): # Scan state
|
||||||
#if actrl.get_scan_state():
|
while should_restart:
|
||||||
if True:
|
#Try to identify face
|
||||||
print("SCAN ACTIVE!")
|
should_restart = False
|
||||||
faces = fp.identify_face(img, target_condidence=0.6) ## Scan student face
|
#if actrl.get_scan_state():
|
||||||
if(len(faces)==1):
|
if(True):
|
||||||
facefile = faces[0]['name'] ## File name
|
#print("SCAN ACTIVE!")
|
||||||
student_info = db.get_student_info(facefile)## Check is it student?
|
#display.set_string("msg.txt","Scanning")
|
||||||
if(student_info!=None): #is student, request another scan for parent
|
faces = fp.identify_face(img, target_condidence=0.6) ## Scan student face
|
||||||
time_count = time.perf_counter()
|
if(len(faces)==1):
|
||||||
endtime = time_count + 10
|
#print("faces == 1")
|
||||||
while (time_count != endtime):
|
facefile = faces[0]['name'] ## File name
|
||||||
#db.log_access_student_file()
|
print(facefile)
|
||||||
time_count = time.perf_counter()
|
student_info = db.get_student_info(facefile)## Check is it student?
|
||||||
time.sleep(1)
|
parent_info = db.get_parent_info(facefile)
|
||||||
display.set_page("student")
|
if(student_info!=None): #is student, request another scan for parent
|
||||||
display.set_string("msg.txt","Scan parent face")
|
print("Find student info")
|
||||||
parent_face = fp.identify_face(img, target_condidence=0.6)
|
#db.log_access_student_file()
|
||||||
parent_info = db.get_parent_info(parent_face)
|
parent_info = None
|
||||||
#DONE TODO Fill in student info Resolve
|
|
||||||
display.set_page("student")
|
|
||||||
display.set_string("name_std.txt", student_info.name)
|
|
||||||
display.set_string("surname_std.txt", student_info.surname)
|
|
||||||
display.set_string("id_std.txt", student_info.id)
|
|
||||||
time_count = time.perf_counter()
|
|
||||||
|
|
||||||
if(parent_info != None): # Identified parent
|
|
||||||
if(db.check_relationship(student_info,parent_info)): # Check if the detected parent is right for the detected student.
|
|
||||||
#DONE TODO Log Access STDw/P
|
|
||||||
db.log_access_student_with_parent(student_info, student_info)
|
|
||||||
#DONE TODO write parrent info to display
|
|
||||||
display.set_page("student")
|
|
||||||
display.set_string("name_std.txt", parent_info.name)
|
|
||||||
display.set_string("surname_std.txt", parent_info.surname)
|
|
||||||
display.set_string("id_std.txt", parent_info.id)
|
|
||||||
access_control.unlock_door() # Door open
|
|
||||||
display.set_string("msg.txt","Get your kid")
|
|
||||||
#TODO Send timer to display
|
|
||||||
#display.set_page("student")
|
|
||||||
i = 10
|
|
||||||
while(i>=1):
|
|
||||||
display.set_string("msg.txt","Door will close in {}".format(i))
|
|
||||||
time.sleep(1)
|
|
||||||
i -= 1
|
|
||||||
actrl.lock_door()
|
|
||||||
|
|
||||||
i = 10
|
|
||||||
#TODO Check if door close, lock
|
|
||||||
while(i>=0):
|
|
||||||
if (not actrl.get_door_state()): # False door is left open
|
|
||||||
display.set_string("msg.txt","Door is left open, Alram in {} sec".format(i))
|
|
||||||
time.sleep(1)
|
|
||||||
i -= 1
|
|
||||||
if i == 0:
|
|
||||||
display.set_string("msg.txt", "Alarm Active!! Please close the door")
|
|
||||||
actrl.activate_alarm()
|
|
||||||
else:
|
|
||||||
actrl.lock_door()
|
|
||||||
break
|
|
||||||
|
|
||||||
else:
|
|
||||||
display.set_page("scan")
|
|
||||||
display.set_string("msg.txt","Wrong parent. \t Try Again")
|
|
||||||
#TODO Try again does not try again
|
|
||||||
else:
|
|
||||||
display.set_page("student")
|
display.set_page("student")
|
||||||
display.set_string("msg.txt","Cannot find parent data. \t Try Again")
|
#display.set_string("msg.txt",f'Scan parent face ({10-i})')
|
||||||
should_start = True
|
display.set_string("name_std.txt", f'Name: {student_info.name}')
|
||||||
|
display.set_string("surname_std.txt",f'Surname: {student_info.surname}')
|
||||||
|
display.set_string("id_std.txt", f'ID: {str(student_info.id)}')
|
||||||
|
"""
|
||||||
|
parent_face = fp.identify_face(img, target_condidence=0.6)
|
||||||
|
parent_info = db.get_parent_info(parent_face)"""
|
||||||
|
#DONE TODO Fill in student info Resolve
|
||||||
|
|
||||||
|
print(student_info)
|
||||||
|
|
||||||
|
i = 0
|
||||||
|
start_time = time.perf_counter()
|
||||||
|
while(time.perf_counter()-start_time < 10):
|
||||||
|
parent_face = fp.identify_face(img, target_condidence=0.6)
|
||||||
|
|
||||||
|
if (len(parent_face)!= 0):
|
||||||
|
parent_facefile = parent_face[0]['name']
|
||||||
|
parent_info = db.get_parent_info(parent_facefile)
|
||||||
|
if(parent_info != None): # Identified parent
|
||||||
|
if(db.check_relationship(student_info,parent_info)): # Check if the detected parent is right for the detected student.
|
||||||
|
print("Relation is OKAY")
|
||||||
|
#DONE TODO Log Access STDw/P
|
||||||
|
db.log_access_student_with_parent(student_info, student_info)
|
||||||
|
#DONE TODO write parrent info to display
|
||||||
|
|
||||||
|
display.set_string("name_pt.txt", f'Name: {parent_info.name}')
|
||||||
|
display.set_string("surname_pt.txt", f'Surname: {parent_info.surname}')
|
||||||
|
display.set_string("id_pt.txt", f'ID: {parent_info.id}')
|
||||||
|
actrl.unlock_door() # Door open
|
||||||
|
display.set_string("msg.txt","Get your kid")
|
||||||
|
#DONE TODO Send timer to display
|
||||||
|
door_open()
|
||||||
|
"""
|
||||||
|
for i in range(10):
|
||||||
|
display.set_string("msg.txt",f'Door will close in {10-i}')
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
i = 5
|
||||||
|
#TODO Check if door close, lock
|
||||||
|
alarm_active: bool = False
|
||||||
|
while(i>=0):
|
||||||
|
if (not actrl.get_door_state() and not alarm_active): # False door is left open
|
||||||
|
display.set_string("msg.txt","Door is left open, Alram in {} sec".format(i))
|
||||||
|
time.sleep(1)
|
||||||
|
i -= 1
|
||||||
|
if i == 0:
|
||||||
|
actrl.activate_LED_alarm()
|
||||||
|
alarm_active = True
|
||||||
|
|
||||||
|
elif(actrl.get_door_state()):
|
||||||
|
actrl.lock_door()
|
||||||
|
display.set_page("scan")
|
||||||
|
should_restart = True
|
||||||
|
break
|
||||||
|
display.set_page("scan")
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
else:
|
||||||
|
display.set_page("scan")
|
||||||
|
display.set_string("msg.txt","Wrong parent. \t Try Again")
|
||||||
|
time.sleep(1)
|
||||||
|
should_restart = True
|
||||||
|
break
|
||||||
|
#TODO Try again does not try again
|
||||||
|
|
||||||
|
elif(len(parent_face) == 0):
|
||||||
|
time.sleep(1)
|
||||||
|
display.set_string("msg.txt",f'Scan parent face ({10-i})')
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
else:
|
||||||
|
pass
|
||||||
|
display.set_page("scan")
|
||||||
|
|
||||||
|
elif(parent_info!=None): #is parent, open door
|
||||||
|
#parent_facefile = faces[0]['name']
|
||||||
|
#parent_info = db.get_parent_info(parent_facefile)
|
||||||
|
#if(parent_info!= None):
|
||||||
|
print("Find parent info")
|
||||||
|
display.set_page("prt_wo_stu")
|
||||||
|
display.set_string("name_pt.txt", f'Name: {parent_info.name}')
|
||||||
|
display.set_string("surname_pt.txt", f'Surname: {parent_info.surname}')
|
||||||
|
display.set_string("id_pt.txt", f'ID: {parent_info.id}')
|
||||||
|
|
||||||
|
door_open()
|
||||||
|
pass
|
||||||
|
|
||||||
|
elif (len(faces)>1):
|
||||||
|
#More than one people, error.
|
||||||
|
print("more than one people. \t try again")
|
||||||
|
display.set_string("msg.txt","More than one people. \t try again")
|
||||||
|
time.sleep(2)
|
||||||
|
pass
|
||||||
|
#print("DONE")
|
||||||
|
|
||||||
else:
|
else:
|
||||||
display.set_page("student")
|
display.set_string("msg.txt","Move your face closer to the scanner")
|
||||||
display.set_string("msg.txt","Cannot find student data. \t Try Again")
|
print("Scan inactivate!!")
|
||||||
pass
|
|
||||||
|
|
||||||
elif (len(faces)>1):
|
|
||||||
#More than one people, error.
|
|
||||||
print("more than one people. \t try again")
|
|
||||||
display.set_string("msg.txt","More than one people. \t try again")
|
|
||||||
time.sleep(2)
|
|
||||||
pass
|
|
||||||
print("DONE")
|
|
||||||
|
|
||||||
else:
|
|
||||||
print("Scan inactivate!!")
|
|
||||||
|
|
||||||
|
|
||||||
'''
|
|
||||||
#is student, request another scan for parent
|
'''
|
||||||
parent_face = fp.identify_face(img, target_condidence=0.6)
|
#is student, request another scan for parent
|
||||||
if(len(faces)==1):
|
parent_face = fp.identify_face(img, target_condidence=0.6)
|
||||||
parent_info = db.get_parent_info(faces[0]['name'])
|
if(len(faces)==1):
|
||||||
if(parent_info!=None):
|
parent_info = db.get_parent_info(faces[0]['name'])
|
||||||
if(db.check_relationship(student_info,parent_info)):
|
if(parent_info!=None):
|
||||||
#Student is under parent, open door.
|
if(db.check_relationship(student_info,parent_info)):
|
||||||
pass
|
#Student is under parent, open door.
|
||||||
else:
|
|
||||||
#Wrong Parent, retry.
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
#Not a parent, retry
|
|
||||||
pass
|
pass
|
||||||
|
else:
|
||||||
|
#Wrong Parent, retry.
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
#Not a parent, retry
|
||||||
|
pass
|
||||||
'''
|
'''
|
||||||
|
|
@ -0,0 +1,140 @@
|
||||||
|
import cv2
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
import face_processing as fp
|
||||||
|
from access_control_mqtt import access_control
|
||||||
|
from database import Database, Student, Parent
|
||||||
|
from display import Display
|
||||||
|
|
||||||
|
SERIAL_PORT_DISPLAY = "/dev/serial0"
|
||||||
|
CAMERA_INDEX = 0
|
||||||
|
|
||||||
|
cam = cv2.VideoCapture(CAMERA_INDEX)
|
||||||
|
actrl = access_control()
|
||||||
|
display = Display(SERIAL_PORT_DISPLAY)
|
||||||
|
db = Database()
|
||||||
|
|
||||||
|
global img
|
||||||
|
global frame_ready
|
||||||
|
frame_ready = False
|
||||||
|
|
||||||
|
def read_webcam():
|
||||||
|
global img
|
||||||
|
global frame_ready
|
||||||
|
while True:
|
||||||
|
ret, img = cam.read()
|
||||||
|
frame_ready = True
|
||||||
|
|
||||||
|
'''
|
||||||
|
display.set_page("student")
|
||||||
|
display.set_string("msg.txt","Door Open")
|
||||||
|
'''
|
||||||
|
|
||||||
|
threading.Thread(target=read_webcam).start()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
while not frame_ready:
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
#Try to identify face
|
||||||
|
#if actrl.get_scan_state():
|
||||||
|
if True:
|
||||||
|
print("SCAN ACTIVE!")
|
||||||
|
faces = fp.identify_face(img, target_condidence=0.6) ## Scan student face
|
||||||
|
if(len(faces)==1):
|
||||||
|
facefile = faces[0]['name'] ## File name
|
||||||
|
student_info = db.get_student_info(facefile)## Check is it student?
|
||||||
|
if(student_info!=None): #is student, request another scan for parent
|
||||||
|
time_count = time.perf_counter()
|
||||||
|
endtime = time_count + 10
|
||||||
|
while (time_count != endtime):
|
||||||
|
#db.log_access_student_file()
|
||||||
|
time_count = time.perf_counter()
|
||||||
|
time.sleep(1)
|
||||||
|
display.set_page("student")
|
||||||
|
display.set_string("msg.txt","Scan parent face")
|
||||||
|
parent_face = fp.identify_face(img, target_condidence=0.6)
|
||||||
|
parent_info = db.get_parent_info(parent_face)
|
||||||
|
#DONE TODO Fill in student info Resolve
|
||||||
|
display.set_page("student")
|
||||||
|
display.set_string("name_std.txt", student_info.name)
|
||||||
|
display.set_string("surname_std.txt", student_info.surname)
|
||||||
|
display.set_string("id_std.txt", student_info.id)
|
||||||
|
time_count = time.perf_counter()
|
||||||
|
|
||||||
|
if(parent_info != None): # Identified parent
|
||||||
|
if(db.check_relationship(student_info,parent_info)): # Check if the detected parent is right for the detected student.
|
||||||
|
#DONE TODO Log Access STDw/P
|
||||||
|
db.log_access_student_with_parent(student_info, student_info)
|
||||||
|
#DONE TODO write parrent info to display
|
||||||
|
display.set_page("student")
|
||||||
|
display.set_string("name_std.txt", parent_info.name)
|
||||||
|
display.set_string("surname_std.txt", parent_info.surname)
|
||||||
|
display.set_string("id_std.txt", parent_info.id)
|
||||||
|
access_control.unlock_door() # Door open
|
||||||
|
display.set_string("msg.txt","Get your kid")
|
||||||
|
#TODO Send timer to display
|
||||||
|
#display.set_page("student")
|
||||||
|
i = 10
|
||||||
|
while(i>=1):
|
||||||
|
display.set_string("msg.txt","Door will close in {}".format(i))
|
||||||
|
time.sleep(1)
|
||||||
|
i -= 1
|
||||||
|
actrl.lock_door()
|
||||||
|
|
||||||
|
i = 10
|
||||||
|
#TODO Check if door close, lock
|
||||||
|
while(i>=0):
|
||||||
|
if (not actrl.get_door_state()): # False door is left open
|
||||||
|
display.set_string("msg.txt","Door is left open, Alram in {} sec".format(i))
|
||||||
|
time.sleep(1)
|
||||||
|
i -= 1
|
||||||
|
if i == 0:
|
||||||
|
display.set_string("msg.txt", "Alarm Active!! Please close the door")
|
||||||
|
actrl.activate_alarm()
|
||||||
|
else:
|
||||||
|
actrl.lock_door()
|
||||||
|
break
|
||||||
|
|
||||||
|
else:
|
||||||
|
display.set_page("scan")
|
||||||
|
display.set_string("msg.txt","Wrong parent. \t Try Again")
|
||||||
|
#TODO Try again does not try again
|
||||||
|
else:
|
||||||
|
display.set_page("student")
|
||||||
|
display.set_string("msg.txt","Cannot find parent data. \t Try Again")
|
||||||
|
should_start = True
|
||||||
|
|
||||||
|
else:
|
||||||
|
display.set_page("student")
|
||||||
|
display.set_string("msg.txt","Cannot find student data. \t Try Again")
|
||||||
|
pass
|
||||||
|
|
||||||
|
elif (len(faces)>1):
|
||||||
|
#More than one people, error.
|
||||||
|
print("more than one people. \t try again")
|
||||||
|
display.set_string("msg.txt","More than one people. \t try again")
|
||||||
|
time.sleep(2)
|
||||||
|
pass
|
||||||
|
print("DONE")
|
||||||
|
|
||||||
|
else:
|
||||||
|
print("Scan inactivate!!")
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
#is student, request another scan for parent
|
||||||
|
parent_face = fp.identify_face(img, target_condidence=0.6)
|
||||||
|
if(len(faces)==1):
|
||||||
|
parent_info = db.get_parent_info(faces[0]['name'])
|
||||||
|
if(parent_info!=None):
|
||||||
|
if(db.check_relationship(student_info,parent_info)):
|
||||||
|
#Student is under parent, open door.
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
#Wrong Parent, retry.
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
#Not a parent, retry
|
||||||
|
pass
|
||||||
|
'''
|
||||||
Loading…
Reference in New Issue