mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 22:00:47 +01:00
09ec3cf7c6
TBR: phoglund@webrtc.org No-Try: True Bug: chromium:755660 Change-Id: Iea1ae4a599a644aca18181f68fe155a659163766 Reviewed-on: https://webrtc-review.googlesource.com/77426 Commit-Queue: Oleh Prypin <oprypin@webrtc.org> Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Cr-Commit-Position: refs/heads/master@{#23286}
124 lines
4.1 KiB
Python
Executable file
124 lines
4.1 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
|
|
#
|
|
# Use of this source code is governed by a BSD-style license
|
|
# that can be found in the LICENSE file in the root of the source
|
|
# tree. An additional intellectual property rights grant can be found
|
|
# in the file PATENTS. All contributing project authors may
|
|
# be found in the AUTHORS file in the root of the source tree.
|
|
|
|
"""Checks if a virtual webcam is running and starts it if not.
|
|
|
|
Returns a non-zero return code if the webcam could not be started.
|
|
|
|
Prerequisites:
|
|
* The Python interpreter must have the psutil package installed.
|
|
* Windows: a scheduled task named 'ManyCam' must exist and be configured to
|
|
launch ManyCam preconfigured to auto-play the test clip.
|
|
* Mac: ManyCam must be installed in the default location and be preconfigured
|
|
to auto-play the test clip.
|
|
* Linux: The v4l2loopback kernel module must be compiled and loaded to the
|
|
kernel already and the v4l2_file_player application must be compiled and put
|
|
in the location specified below.
|
|
|
|
NOTICE: When running this script as a buildbot step, make sure to set
|
|
usePTY=False for the build step when adding it, or the subprocess will die as
|
|
soon the step has executed.
|
|
|
|
If any command line arguments are passed to the script, it is executed as a
|
|
command in a subprocess.
|
|
"""
|
|
|
|
import os
|
|
# psutil is not installed on non-Linux machines by default.
|
|
import psutil # pylint: disable=F0401
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
|
|
WEBCAM_WIN = ('schtasks', '/run', '/tn', 'ManyCam')
|
|
WEBCAM_MAC = ('open', '/Applications/ManyCam/ManyCam.app')
|
|
E = os.path.expandvars
|
|
WEBCAM_LINUX = (
|
|
E('$HOME/fake-webcam-driver/linux/v4l2_file_player/v4l2_file_player'),
|
|
E('$HOME/webrtc_video_quality/reference_video.yuv'),
|
|
'640', '480', '/dev/video0',
|
|
)
|
|
|
|
|
|
def IsWebCamRunning():
|
|
if sys.platform == 'win32':
|
|
process_name = 'ManyCam.exe'
|
|
elif sys.platform.startswith('darwin'):
|
|
process_name = 'ManyCam'
|
|
elif sys.platform.startswith('linux'):
|
|
process_name = 'v4l2_file_player'
|
|
else:
|
|
raise Exception('Unsupported platform: %s' % sys.platform)
|
|
for p in psutil.process_iter():
|
|
try:
|
|
if process_name == p.name:
|
|
print 'Found a running virtual webcam (%s with PID %s)' % (p.name,
|
|
p.pid)
|
|
return True
|
|
except psutil.AccessDenied:
|
|
pass # This is normal if we query sys processes, etc.
|
|
return False
|
|
|
|
|
|
def StartWebCam():
|
|
try:
|
|
if sys.platform == 'win32':
|
|
subprocess.check_call(WEBCAM_WIN)
|
|
print 'Successfully launched virtual webcam.'
|
|
elif sys.platform.startswith('darwin'):
|
|
subprocess.check_call(WEBCAM_MAC)
|
|
print 'Successfully launched virtual webcam.'
|
|
elif sys.platform.startswith('linux'):
|
|
|
|
# Must redirect stdout/stderr/stdin to avoid having the subprocess
|
|
# being killed when the parent shell dies (happens on the bots).
|
|
process = subprocess.Popen(WEBCAM_LINUX, stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
stdin=subprocess.PIPE)
|
|
# If the v4l2loopback module is not loaded or incorrectly configured,
|
|
# the process will still launch but will die immediately.
|
|
# Wait for a second and then check for aliveness to catch such errors.
|
|
time.sleep(1)
|
|
if process.poll() is None:
|
|
print 'Successfully launched virtual webcam with PID %s' % process.pid
|
|
else:
|
|
print 'Failed to launch virtual webcam.'
|
|
return False
|
|
|
|
except Exception as e:
|
|
print 'Failed to launch virtual webcam: %s' % e
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def _ForcePythonInterpreter(cmd):
|
|
"""Returns the fixed command line to call the right python executable."""
|
|
out = cmd[:]
|
|
if out[0] == 'python':
|
|
out[0] = sys.executable
|
|
elif out[0].endswith('.py'):
|
|
out.insert(0, sys.executable)
|
|
return out
|
|
|
|
|
|
def Main(argv):
|
|
if not IsWebCamRunning():
|
|
if not StartWebCam():
|
|
return 1
|
|
|
|
if argv:
|
|
return subprocess.call(_ForcePythonInterpreter(argv))
|
|
else:
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(Main(sys.argv[1:]))
|