add pygame menu

auto_user_reg.py

@@ -10,6 +10,14 @@
 import cv2
 import face_recognition
 
+import cv2
+import numpy as np
+import os
+import glob
+from datetime import datetime
+import math
+import time
+
 # To create the file the first time when there are already existing users
 #workers=['Abir','Quang', 'Thomas', 'Prof Hartanto']
 #with open('users.txt', 'wb') as f:
@@ -73,14 +81,15 @@ def save_image(name):
 
 
 
-def add_new_user():
+def add_new_user(name):
     # username entry
-    name = input("Please enter first name: ")
+    #name = input("Please enter first name: ")
 
-    while name in users:
+    if name in users:
         print("Username [%s] already exists!!!" % name)
         print("Please try again!")
-        name = input("Please enter first name: ")
+#        name = input("Please enter first name: ")
+        return None
 
     print("[INFO] No duplicate found, proceeding...")
 
@@ -110,6 +119,138 @@ def add_new_user():
     else:
         print("[INFO] User registration interrupted...")
 
+def sign_in():
+    print("[INFO] Loading user database...")
+    print(known_face_names)
+
+    # Initialization
+    process_this_frame = True
+    count=0
+#    workers=['Abir','Quang', 'Thomas', 'Prof Hartanto'] 
+    name='dummy'
+    video_capture = cv2.VideoCapture(0)
+    video_capture.set(cv2.cv2.CAP_PROP_FPS, 2)
+
+    while True:
+
+        # Grab a single frame of video
+        ret, frame = video_capture.read()
+
+        # Resize frame of video to 1/4 size for faster face recognition processing
+        small_frame = cv2.resize(frame, (0, 0), fx=0.25, fy=0.25)
+
+        # Convert the image from BGR color (which OpenCV uses) to RGB color (which face_recognition uses)
+        rgb_small_frame = small_frame[:, :, ::-1]
+
+        # Only process every other frame of video to save time
+        if process_this_frame:
+            # Find all the faces and face encodings in the current frame of video
+            face_locations = face_recognition.face_locations(rgb_small_frame)
+            face_encodings = face_recognition.face_encodings(rgb_small_frame, face_locations)
+
+            face_names = []
+            for face_encoding in face_encodings:
+                # See if the face is a match for the known face(s)
+                matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
+                name = "Unknown"
+
+                # # If a match was found in known_face_encodings, just use the first one.
+                # if True in matches:
+                #     first_match_index = matches.index(True)
+                #     name = known_face_names[first_match_index]
+
+                # Or instead, use the known face with the smallest distance to the new face
+                face_distances = face_recognition.face_distance(known_face_encodings, face_encoding)
+                best_match_index = np.argmin(face_distances)
+                if matches[best_match_index]:
+                    name = known_face_names[best_match_index]
+
+                face_names.append(name)
+
+        process_this_frame = not process_this_frame
+
+
+        # Display the results
+        for (top, right, bottom, left), name in zip(face_locations, face_names):
+            # Scale back up face locations since the frame we detected in was scaled to 1/4 size
+            top *= 4
+            right *= 4
+            bottom *= 4
+            left *= 4
+
+            # Draw a box around the face
+            cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)
+
+            # Draw a label with a name below the face
+            cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED)
+            font = cv2.FONT_HERSHEY_DUPLEX
+            cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)
+        if name in known_face_names:
+            count+=1
+            if count==5:
+                entry += 'in'
+                break
+        # Display the resulting image
+        cv2.imshow('Video', frame)
+        # Hit 'q' on the keyboard to quit!
+        if cv2.waitKey(1) & 0xFF == ord('q'):
+            break
+
+    # Release handle to the webcam
+    video_capture.release()
+    cv2.destroyAllWindows()
+
+    # Data Logger
+    # TODO: should a person be allowed to log-in multiple times???
+    # 
+
+    now=datetime.now()
+    log=os.listdir('logbook/')
+    month=now.strftime("%B")
+    year=str(now.year)
+    day=str(now.day)
+    hour=str(now.hour)
+    minute=str(now.minute)
+    print(log)
+
+
+    if month+' '+year not in log:
+        os.mkdir('logbook/'+(now.strftime("%B")+' '+year))
+    day_glob=os.listdir("logbook/"+month+' '+year)
+    print(day_glob)
+    file_path="logbook/"+(now.strftime("%B")+' '+year)
+    file_name=day+'-'+month+'-'+year+'.txt'
+    full_path=os.path.join(file_path, file_name)
+    if day+'-'+month+'-'+year+'.txt' not in day_glob: # create new file? Then check out shouldn't happen
+        f=open(full_path,"a+")
+        if 'in' in entry:
+            f.write('[Check In++] Username: '+name+' '+day+'-'+ str(now.month) +'-'+year+' '+ str(now.strftime("%H:%M"))+'\n')
+
+    else:
+        f=open(full_path,"r")                       # 
+        a=f.read()
+        #f=open(full_path,"a+")
+        if 'in' in entry:
+            if '[Check In++] Username: '+name in a and '[--Check Out] Username: '+name not in a:
+                print('You are already in. Logging you out now')
+                g=open(full_path,"a+")
+                g.write('[--Check Out] Username: '+name+' '+day+'-'+ str(now.month) +'-'+year+' '+str(now.strftime("%H:%M"))+'\n')
+                g.close()
+            else:
+                g=open(full_path,"a+")
+                g.write('[Check In++] Username: '+name+' '+day+'-'+ str(now.month) +'-'+year+' '+ str(now.strftime("%H:%M"))+'\n')
+                g.close()
+        else:
+            if '[--Check Out] Username: '+name in a:
+                print('You are already out. Try again with fist bump to come in')
+            else:
+                 g=open(full_path,"a+")
+                 g.write('[--Check Out] Username: '+name+' '+day+'-'+ str(now.month) +'-'+year+' '+str(now.strftime("%H:%M"))+'\n')
+                 g.close()
+
+    f.close()
+
+
 
 #try:    
 #    add_new_user()

menu_test.py

@@ -0,0 +1,104 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Created on Thu Dec 19 09:34:19 2019
+
+@author: mtc-20
+"""
+
+import pygameMenu
+import pygame
+from auto_user_reg import *
+
+# SETTTINGS
+BLACK = (0,0,0)
+WHITE = (255,255,255)
+ORANGE = (255, 150, 10)
+MENU_BACKGROUND_COLOR = (228, 55, 36)
+COLOR_BACKGROUND = (20, 50, 30)
+
+FPS = 30.0
+WINDOW_SIZE = (640,480)
+ABOUT = ['Autologger v0.1',
+         'Author: @bathonSpidey, @mtc-20',
+         pygameMenu.locals.TEXT_NEWLINE,
+         'Email: [email protected]'.format(pygameMenu.__email__)]
+
+clock = None
+main_menu = None
+screen = None
+logo_rect = None
+logo = None
+
+def stets():
+    pass
+
+def main_bg():
+    global screen
+    screen.fill(BLACK)
+
+
+def about_bg():
+    global screen
+    global logo
+    global logo_rect
+    #screen.fill(BLACK)
+    screen.blit(logo, logo_rect)    
+
+
+def main(test=False):
+    # GLOBALS
+    global standby
+    global screen
+    global clock
+    global logo
+
+    # INITIALIZATION
+    pygame.init()
+    screen = pygame.display.set_mode(WINDOW_SIZE)
+    #screen.fill(BLACK)
+    pygame.display.set_caption('Auto Logger')
+    clock = pygame.time.Clock()
+
+    logo = pygame.image.load('robotics_logo.png').convert()
+    #img_lg = pygame.transform.scale(img, (75,75))
+    logo_rect = logo.get_rect()
+#    logo_rect.centerx = int(WINDOW_SIZE[0]/2)
+#    logo_rect.centery = int(WINDOW_SIZE[1]/2)
+
+    # MENUS
+    new_user_menu = pygameMenu.Menu(surface=screen,bgfun=main_bg, color_selected=WHITE, font=pygameMenu.font.FONT_HELVETICA, window_width=int(WINDOW_SIZE[0]), window_height=int(WINDOW_SIZE[1]),menu_width=int(WINDOW_SIZE[0]), menu_height=int(WINDOW_SIZE[1]*0.5), onclose=pygameMenu.events.DISABLE_CLOSE, title='New User', font_color=BLACK, menu_color_title=ORANGE, menu_color=BLACK, menu_alpha=100, widget_alignment=pygameMenu.locals.ALIGN_LEFT, font_size=20)
+    wid = new_user_menu.add_text_input('First Name: ', textinput_id='first_name', onreturn=add_new_user)
+
+    about_menu = pygameMenu.TextMenu(surface=screen, bgfun=about_bg, color_selected=ORANGE, font=pygameMenu.font.FONT_BEBAS, window_width=WINDOW_SIZE[0], window_height=int(WINDOW_SIZE[1]), menu_color_title=ORANGE, menu_width=int(WINDOW_SIZE[0]*0.7), menu_height=int(WINDOW_SIZE[1]*0.7), text_fontsize=20, menu_alpha=0, onclose=pygameMenu.events.DISABLE_CLOSE, title='ABOUT', font_color=WHITE, menu_color=BLACK)
+    for m in ABOUT:
+        about_menu.add_line(m)
+    about_menu.add_line(pygameMenu.locals.TEXT_NEWLINE)
+    about_menu.add_option('Return to Menu', pygameMenu.events.BACK)
+
+    standby = pygameMenu.Menu(surface=screen,bgfun=main_bg, color_selected=WHITE, font=pygameMenu.font.FONT_BEBAS, window_width=int(WINDOW_SIZE[0]), window_height=int(WINDOW_SIZE[1]),menu_width=int(WINDOW_SIZE[0]*0.8), menu_height=int(WINDOW_SIZE[1]*0.8), onclose=pygameMenu.events.DISABLE_CLOSE, title='Standby', font_color=BLACK, menu_color_title=WHITE, menu_color=ORANGE, menu_alpha=100)
+    standby.add_option('Sign in', sign_in)
+    standby.add_option('New user', new_user_menu)
+    standby.add_option('About', about_menu)
+
+    standby.set_fps(FPS)
+
+    while True:
+        clock.tick(FPS)
+        main_bg()
+        events = pygame.event.get()
+        for event in events:
+            if event.type == pygame.QUIT:
+                exit()
+#        screen.blit()        
+        screen.blit(logo, logo_rect)
+        standby.mainloop(events, disable_loop=test)
+        screen.blit(logo, logo_rect)
+        pygame.display.flip()
+
+        if test:
+            break
+
+if __name__ == '__main__':
+    main()
+