Integrated class structure

PythonCode/UltimateSecurityCam.py

@@ -8,6 +8,7 @@
 
 #if you get error while importing the google how to install <Package Name> in python 3.6
 
+#initial values set
 THRESHOLD = 40
 camera = cv2.VideoCapture(0)
 
@@ -22,121 +23,148 @@
 size = (int(camera.get(cv2.CAP_PROP_FRAME_WIDTH)),
 		int(camera.get(cv2.CAP_PROP_FRAME_HEIGHT)))
 
+
 dir_path = os.path.dirname(os.path.realpath(__file__))
+#video file name
 videofile = "basic_motion_detection.avi"
 
 videoWriter = cv2.VideoWriter(os.path.join(str(dir_path),videofile),
 				  cv2.VideoWriter_fourcc('D', 'I', 'V', 'X'),
 				  fps, size)
 
-initial = int(time.time())
-final = initial + 4
-
-while (final-initial):
-	#start timer on the frames
-	ret, frame = camera.read()
-	initailiztion_text = "Starting in " + str(final-initial) + "..."
-	cv2.putText(frame,initailiztion_text,(60,30),cv2.FONT_HERSHEY_TRIPLEX,1,(0,100,255),2)
-	cv2.imshow("Ultimate Security Camera",frame)
-
-
-	if cv2.waitKey(int(45)) &0xff == ord('q'):
-		break
-
-	elif int(time.time()) == (initial + 1):
-		initial = initial + 1
-		#print(str(final-initial) + "...")
-
-maxcnts=0		
-start = time.time()	
-while (True):
-	ret, frame = camera.read()
-	# The first frame as the background
-	if background is None:
-		background = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-		background = cv2.GaussianBlur(background, (21,21), 0)
-		continue
-
-	gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-	gray_frame = cv2.GaussianBlur(gray_frame, (21,21), 0)
-
-	# Compare the difference between each frame of image and the background
-	#print(background.shape, gray_frame.shape)
-	diff = cv2.absdiff(background, gray_frame)
-	diff = cv2.threshold(diff, THRESHOLD, 255, cv2.THRESH_BINARY)[1]
-	diff = cv2.dilate(diff, es, iterations=2)
-	# Calculate the outline of the target in the image
-	image, cnts, hierarchy = cv2.findContours(diff.copy(),
-						  cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-	detection_text = ("Detecting " + str(len(cnts)) + " Moving Objects")
-	detection_text_colour = (0,255,0) 	#set as green
-	if len(cnts) > 0:
-		#if breach detected
-		detection_text_colour = (0,0,255)   #set to red
-		cameraSound.play()
-
-	for c in cnts:
-		if cv2.contourArea(c) < (background.shape[0]*background.shape[1])/204:
-			#minimum area to be calculated based on image size and camera megapixels
-			continue
-		# Calculate the bounding box
-		(x, y, w, h) = cv2.boundingRect(c)
-		cv2.rectangle(frame, (x,y), (x+w, y+h), (0,255,0), 2)
-
-	#maximum object detected
-	if len(cnts)>maxcnts: maxcnts=len(cnts)
-
-	#print(detection_text)
-	cv2.putText(frame,detection_text,(60,30),cv2.FONT_HERSHEY_DUPLEX,1,detection_text_colour,2)
-
-	diff = cv2.cvtColor(diff, cv2.COLOR_GRAY2BGR)			#3 channel gray scaled image
-
-	horizontal_stack = np.hstack((frame, diff))				#proper way to stack 3D arrays
-	merged_windows = np.concatenate((frame, diff), axis=1)
-
-	cv2.imshow("Ultimate Security Camera", merged_windows)
-
-	#cv2.imshow("contours", frame)
-	videoWriter.write(frame)
-	#cv2.imshow("dif", diff)
-	#cv2.imwrite('didff.jpg', diff)
-
-	keypress = cv2.waitKey(45)
-	if keypress:
-		if keypress &0xff == ord('q'):
-			break
-		elif keypress &0xff == ord('r'):			
-			#reset the camera
-			background = None
-
-cv2.destroyAllWindows()
-camera.release()
-
-end = time.time() 
-
-confirm = input("Do you wish to save the current configs? [Y/N]: ")
-#if run with python2 use raw_input
-
-data={"Date and Time":time.asctime(time.localtime(time.time())),
-	 "Camera FPS":fps,
-	 "Threshold":THRESHOLD,
-	 "Max Objects":maxcnts,
-	 "Video File":"basic_motion_detection.avi",
-	 "Path":dir_path,
-	 "Duration": '%0.2f' %(end-start)}
-
-configfile = "config.txt"
-
-if confirm.startswith('y' or 'Y'):
-	print("\nUpdating config file...")
-	with open(configfile,'w') as jfile:
-		json.dump(data, jfile, indent = 4)
-	print("Data updated to " + configfile + " successfully!")
+
+class UltimateSecurityCam:
+	"""	UltimateSecurityCam class identifies object movements and 
+		detection of any kind of undesirable movement in the 
+		surroundings
+	"""
+
+	def __init__(self):
+		pass
+
+	def initial_window(self):
+		#initial window starts 
+		initial = int(time.time())
+		final = initial + 4
+		while (final-initial):
+			#start timer on the frames
+			ret, frame = camera.read()
+			initailiztion_text = ("Starting in " + str(final-initial) + "...")
+			cv2.putText(frame,initailiztion_text,(60,30),
+						cv2.FONT_HERSHEY_TRIPLEX,1,(0,100,255),2)
+			cv2.imshow("Ultimate Security Camera",frame)
+
+			if cv2.waitKey(int(45)) &0xff == ord('q'):
+				break
+
+			elif int(time.time()) == (initial + 1):
+				initial = initial + 1
+				#print(str(final-initial) + "...")
+
+	def usc(self):
+		#main window opens and opject movement detection starts
+		maxcnts = 0		
+		global background
+		start = time.time()	
+		while (True):
+			ret, frame = camera.read()
+			# The first frame as the background
+			if background is None:
+				background = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+				background = cv2.GaussianBlur(background, (21,21), 0)
+				continue
+
+			gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+			gray_frame = cv2.GaussianBlur(gray_frame, (21,21), 0)
+
+			# Compare the difference between each frame of image and the background
+			#print(background.shape, gray_frame.shape)
+			diff = cv2.absdiff(background, gray_frame)
+			diff = cv2.threshold(diff, THRESHOLD, 255, cv2.THRESH_BINARY)[1]
+			diff = cv2.dilate(diff, es, iterations=2)
+			# Calculate the outline of the target in the image
+			image, cnts, hierarchy = cv2.findContours(diff.copy(),
+								  cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+			detection_text = ("Detecting " + str(len(cnts)) + " Moving Objects")
+			detection_text_colour = (0,255,0) 	#set as green
+			if len(cnts) > 0:
+				#if breach detected
+				detection_text_colour = (0,0,255)   #set to red
+				cameraSound.play()
+
+			for c in cnts:
+				if cv2.contourArea(c) < (background.shape[0]*background.shape[1])/204:
+					#minimum area to be calculated based on image size and camera megapixels
+					continue
+				# Calculate the bounding box
+				(x, y, w, h) = cv2.boundingRect(c)
+				cv2.rectangle(frame, (x,y), (x+w, y+h), (0,255,0), 2)
+
+			#maximum object detected
+			if len(cnts)>maxcnts: maxcnts=len(cnts)
+
+			#print(detection_text)
+			cv2.putText(frame,detection_text,(60,30),cv2.FONT_HERSHEY_DUPLEX,1,detection_text_colour,2)
+
+			diff = cv2.cvtColor(diff, cv2.COLOR_GRAY2BGR)			#3 channel gray scaled image
+
+			horizontal_stack = np.hstack((frame, diff))				#proper way to stack 3D arrays
+			merged_windows = np.concatenate((frame, diff), axis=1)
+
+			cv2.imshow("Ultimate Security Camera", merged_windows)
+
+			#cv2.imshow("contours", frame)
+			videoWriter.write(frame)
+			#cv2.imshow("dif", diff)
+			#cv2.imwrite('didff.jpg', diff)
+
+			keypress = cv2.waitKey(45)
+			if keypress:
+				if keypress &0xff == ord('q'):
+					break
+				elif keypress &0xff == ord('r'):			
+					#reset the camera
+					background = None
+
+		cv2.destroyAllWindows()
+		camera.release()
+
+		end = time.time() 
+		duration = end-start
+
+		data={"Date and Time":time.asctime(time.localtime(time.time())),
+			 "Camera FPS":fps,
+			 "Threshold":THRESHOLD,
+			 "Max Objects recorded":maxcnts,
+			 "Video File":videofile,
+			 "Path":dir_path,
+			 "Duration": '%0.2f' %(duration) + ' seconds'}
+		return data
+
+	def config(self,data):
+		#saves all necessary configurations
 
-elif confirm.startswith('n' or 'N'):
-	pass
+		confirm = input("Do you wish to save the current configs? [Y/N]: ")
+		#if run with python2 use raw_input
+
+		configfile = "config.txt"
+
+		if confirm.startswith('y' or 'Y'):
+			print("\nUpdating config file...")
+			with open(configfile,'w') as jfile:
+				json.dump(data, jfile, indent = 4)
+			print("Data updated to " + configfile + " successfully!")
+
+		elif confirm.startswith('n' or 'N'):
+			pass
+
+		else:
+			print("Invalid input!")
+
 
-else:
-	print("Invalid input!")
-
+#Sequence of main program execution
+start = UltimateSecurityCam()
+start.initial_window()
+data = start.usc()
+start.config(data)
 print("Program succesfully terminated!")

PythonCode/config.txt

@@ -1,9 +1,9 @@
 {
-    "Date and Time": "Sun Dec 23 03:47:01 2018",
+    "Date and Time": "Wed Dec 26 05:17:21 2018",
     "Camera FPS": 2,
     "Threshold": 40,
-    "Max Objects": 17,
+    "Max Objects recorded": 3,
     "Video File": "basic_motion_detection.avi",
-    "Path": "/home/johri/github/UltimateSecurityCam/latest 23-12/UltimateSecurityCam/PythonCode",
-    "Duration": "7.88"
+    "Path": "/home/johri/github/UltimateSecurityCam/latest 23-12/26-12-18/PythonCode",
+    "Duration": "13.76 seconds"
 }