You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.2 KiB
106 lines
3.2 KiB
4 months ago
|
#!/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: Not implemented
|
||
|
|
|
||
|
|
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.
|
||
|
|
"""
|
||
|
|
|
||
|
|
# psutil is not installed on non-Linux machines by default.
|
||
|
|
import psutil # pylint: disable=F0401
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
WEBCAM_WIN = ('schtasks', '/run', '/tn', 'ManyCam')
|
||
|
|
WEBCAM_MAC = ('open', '/Applications/ManyCam/ManyCam.app')
|
||
|
|
|
||
|
|
|
||
|
|
def IsWebCamRunning():
|
||
|
|
if sys.platform == 'win32':
|
||
|
|
process_name = 'ManyCam.exe'
|
||
|
|
elif sys.platform.startswith('darwin'):
|
||
|
|
process_name = 'ManyCam'
|
||
|
|
elif sys.platform.startswith('linux'):
|
||
|
|
# TODO(bugs.webrtc.org/9636): Currently a no-op on Linux: sw webcams no
|
||
|
|
# longer in use.
|
||
|
|
print 'Virtual webcam: no-op on Linux'
|
||
|
|
return True
|
||
|
|
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'):
|
||
|
|
# TODO(bugs.webrtc.org/9636): Currently a no-op on Linux: sw webcams no
|
||
|
|
# longer in use.
|
||
|
|
print 'Not implemented on Linux'
|
||
|
|
|
||
|
|
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:]))
|