RaspberryPiSamples/TestingthePlayback.py at main · PaulThe4/RaspberryPiSamples · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import cv2
import os
import datetime

# Function to save video with timestamped filename
def save_video_with_timestamp(video_capture1, video_capture2, camera_number1, camera_number2):

    frame_count = 0
    # Get current timestamp
    timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
    # Define filename with camera number and timestamp
    filename1 = f"video{camera_number1}_{timestamp}.avi"
    #filename2 = f"video{camera_number2}_{timestamp}.avi"
    # Define video writer object
    result1 = cv2.VideoWriter(filename1, cv2.VideoWriter_fourcc(*'MJPG'), 30, size)
    #result2 = cv2.VideoWriter(filename2, cv2.VideoWriter_fourcc(*'MJPG'), 10, size)

    #start_time = datetime.datetime.now()

    # while (datetime.datetime.now() - start_time).total_seconds() < 30:
    #     ret1, frame1 = video_capture1.read()
    #     if ret1:
    #         frame_count += 1
    #         result1.write(frame1)  # Write frame to video
    #         resized_frame1 = cv2.resize(frame1, (640, 480))  # Resize for display
    #         cv2.imshow(f"Camera {camera_number1}", resized_frame1)  # Display frame
    #         if cv2.waitKey(10) & 0xFF == ord('s'):  # Adjust delay if needed
    #             break
    #     else:
    #         break
    # print(f"Total frames captured: {frame_count}")
    # fps = video1.get(cv2.CAP_PROP_FPS)
    # print(f"Camera FPS: {fps}")


    # Record video for 10 seconds
    start_time = datetime.datetime.now()
    while (datetime.datetime.now() - start_time).total_seconds() < 30:
        ret1, frame1 = video_capture1.read()
        #ret2, frame2 = video_capture2.read()
        if ret1: #and ret2:
             # Get the current time for the timestamp
            current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")

            # Put the timestamp text on the frame
            cv2.putText(frame1, current_time, (10, frame_height1 - 10),
                        cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA)
			# Resize frame for display
            resized_frame1 = cv2.resize(frame1, (640, 480))
            #resized_frame2 = cv2.resize(frame2, (640, 480))
            result1.write(frame1)
            #result2.write(frame2)
            cv2.imshow(f"Camera {camera_number1}", resized_frame1)
            #cv2.imshow(f"Camera {camera_number2}", resized_frame2)
            if cv2.waitKey(10) & 0xFF == ord('s'):
                break
        else:
            break


    # Release resources
    result1.release()
    #result2.release()
    cv2.destroyAllWindows()
    print(f"Camera {camera_number1}: Video saved as {filename1}")
    #print(f"Camera {camera_number2}: Video saved as {filename2}")


# Create video capture objects for each camera
video1 = cv2.VideoCapture(1)
#video2 = cv2.VideoCapture(2)

# Check if cameras are opened successfully
if not video1.isOpened(): #or not video2.isOpened():
    print("Error: One or both cameras couldn't be opened.")
    exit()

# Set resolutions for video capture objects
frame_width1 = int(video1.get(3))
frame_height1 = int(video1.get(4))
#frame_width2 = int(video2.get(3))
#frame_height2 = int(video2.get(4))

size = (frame_width1, frame_height1)  # Assuming both cameras have the same resolution

# Save videos from both cameras
save_video_with_timestamp(video1, video1, 1, 2) #change second video1 to video2
#save_video_with_timestamp(video2, 2)

# Release video capture objects
video1.release()
#video2.release()