add recording specific window tutorial

x4nth055 · x4nth055 · commit 7e4a5677e6f7 · 2021-12-17T10:31:09.000+01:00
diff --git a/README.md b/README.md
@@ -181,6 +181,7 @@ This is a repository of all the tutorials of [The Python Code](https://www.thepy
     - [How to Extract Frames from Video in Python](https://www.thepythoncode.com/article/extract-frames-from-videos-in-python). ([code](python-for-multimedia/extract-frames-from-video))
     - [How to Reverse Videos in Python](https://www.thepythoncode.com/article/reverse-video-in-python). ([code](python-for-multimedia/reverse-video))
     - [How to Extract Video Metadata in Python](https://www.thepythoncode.com/article/extract-media-metadata-in-python). ([code](python-for-multimedia/extract-video-metadata))
+    - [How to Record a Specific Window in Python](https://www.thepythoncode.com/article/record-a-specific-window-in-python). ([code](python-for-multimedia/record-specific-window))
     
 
 For any feedback, please consider pulling requests.
diff --git a/python-for-multimedia/record-specific-window/README.md b/python-for-multimedia/record-specific-window/README.md
@@ -0,0 +1,4 @@
+# [How to Record a Specific Window in Python](https://www.thepythoncode.com/article/record-a-specific-window-in-python)
+To run this:
+- `pip3 install -r requirements.txt`
+- Example: for recording Chrome: `python record_specific_window.py Chrome`
diff --git a/python-for-multimedia/record-specific-window/record_specific_window.py b/python-for-multimedia/record-specific-window/record_specific_window.py
@@ -0,0 +1,41 @@
+import cv2
+import numpy as np
+import pyautogui
+import pygetwindow as gw
+import sys
+
+# the window name, e.g "notepad", "Chrome", etc.
+window_name = sys.argv[1]
+
+# define the codec
+fourcc = cv2.VideoWriter_fourcc(*"XVID")
+# frames per second
+fps = 12.0
+# the time you want to record in seconds
+record_seconds = 10
+# search for the window, getting the first matched window with the title
+w = gw.getWindowsWithTitle(window_name)[0]
+# activate the window
+w.activate()
+# create the video write object
+out = cv2.VideoWriter("output.avi", fourcc, fps, tuple(w.size))
+
+for i in range(int(record_seconds * fps)):
+    # make a screenshot
+    img = pyautogui.screenshot(region=(w.left, w.top, w.width, w.height))
+    # convert these pixels to a proper numpy array to work with OpenCV
+    frame = np.array(img)
+    # convert colors from BGR to RGB
+    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    # write the frame
+    out.write(frame)
+    # show the frame
+    cv2.imshow("screenshot", frame)
+    # if the user clicks q, it exits
+    if cv2.waitKey(1) == ord("q"):
+        break
+
+# make sure everything is closed when exited
+cv2.destroyAllWindows()
+out.release()
+
diff --git a/python-for-multimedia/record-specific-window/requirements.txt b/python-for-multimedia/record-specific-window/requirements.txt
@@ -0,0 +1,4 @@
+numpy
+opencv-python
+pyautogui
+pygetwindow
