|
|
|
|
|
|
|
|
|
|
|
import subprocess |
|
|
import shlex |
|
|
import re |
|
|
import platform |
|
|
import tempfile |
|
|
import os |
|
|
import sys |
|
|
|
|
|
def pci_records(): |
|
|
records = [] |
|
|
command = shlex.split('lspci -vmm') |
|
|
output = subprocess.check_output(command).decode() |
|
|
|
|
|
for devices in output.strip().split("\n\n"): |
|
|
record = {} |
|
|
records.append(record) |
|
|
for row in devices.split("\n"): |
|
|
key, value = row.split("\t") |
|
|
record[key.split(':')[0]] = value |
|
|
|
|
|
return records |
|
|
|
|
|
def generate_xorg_conf(devices): |
|
|
xorg_conf = [] |
|
|
|
|
|
device_section = """ |
|
|
Section "Device" |
|
|
Identifier "Device{device_id}" |
|
|
Driver "nvidia" |
|
|
VendorName "NVIDIA Corporation" |
|
|
BusID "{bus_id}" |
|
|
EndSection |
|
|
""" |
|
|
server_layout_section = """ |
|
|
Section "ServerLayout" |
|
|
Identifier "Layout0" |
|
|
{screen_records} |
|
|
EndSection |
|
|
""" |
|
|
screen_section = """ |
|
|
Section "Screen" |
|
|
Identifier "Screen{screen_id}" |
|
|
Device "Device{device_id}" |
|
|
DefaultDepth 24 |
|
|
Option "AllowEmptyInitialConfiguration" "True" |
|
|
SubSection "Display" |
|
|
Depth 24 |
|
|
Virtual 1024 768 |
|
|
EndSubSection |
|
|
EndSection |
|
|
""" |
|
|
screen_records = [] |
|
|
for i, bus_id in enumerate(devices): |
|
|
xorg_conf.append(device_section.format(device_id=i, bus_id=bus_id)) |
|
|
xorg_conf.append(screen_section.format(device_id=i, screen_id=i)) |
|
|
screen_records.append('Screen {screen_id} "Screen{screen_id}" 0 0'.format(screen_id=i)) |
|
|
|
|
|
xorg_conf.append(server_layout_section.format(screen_records="\n ".join(screen_records))) |
|
|
|
|
|
output = "\n".join(xorg_conf) |
|
|
print(output) |
|
|
return output |
|
|
|
|
|
def startx(display): |
|
|
if platform.system() != 'Linux': |
|
|
raise Exception("Can only run startx on linux") |
|
|
|
|
|
devices = [] |
|
|
for r in pci_records(): |
|
|
if r.get('Vendor', '') == 'NVIDIA Corporation' \ |
|
|
and r['Class'] in ['VGA compatible controller', '3D controller']: |
|
|
bus_id = 'PCI:' + ':'.join(map(lambda x: str(int(x, 16)), re.split(r'[:\.]', r['Slot']))) |
|
|
devices.append(bus_id) |
|
|
|
|
|
if not devices: |
|
|
raise Exception("no nvidia cards found") |
|
|
|
|
|
try: |
|
|
fd, path = tempfile.mkstemp() |
|
|
with open(path, "w") as f: |
|
|
f.write(generate_xorg_conf(devices)) |
|
|
command = shlex.split("Xorg -noreset +extension GLX +extension RANDR +extension RENDER -config %s :%s" % (path, display)) |
|
|
subprocess.call(command) |
|
|
finally: |
|
|
os.close(fd) |
|
|
os.unlink(path) |
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
display = 1 |
|
|
if len(sys.argv) > 1: |
|
|
display = int(sys.argv[1]) |
|
|
print("Starting X on DISPLAY=:%s" % display) |
|
|
startx(display) |
|
|
|
