parent
d745c95939
commit
a0a88e6285
@ -1,61 +1,61 @@
|
|||||||
#!/usr/bin/python3
|
#!/usr/bin/python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
#
|
#
|
||||||
# Copyright (C) 2016 Richard Hughes <richard@hughsie.com>
|
# Copyright (C) 2021 Simone Caronni <negativo17@gmail.com>
|
||||||
# Licensed under the GNU General Public License Version or later
|
# Licensed under the GNU General Public License Version or later
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
if len(sys.argv) != 3:
|
if len(sys.argv) != 2:
|
||||||
print("usage: %s README.txt \"header to match\"" % sys.argv[0])
|
print("usage: %s README.txt" % sys.argv[0])
|
||||||
return 1
|
return 1
|
||||||
|
|
||||||
# open file
|
|
||||||
f = open(sys.argv[1])
|
f = open(sys.argv[1])
|
||||||
in_section = False
|
in_section = False
|
||||||
in_table = False
|
in_table = False
|
||||||
pids = []
|
pids = []
|
||||||
|
sections = ["NVIDIA GEFORCE GPUS", "NVIDIA RTX/QUADRO GPUS", "NVIDIA NVS GPUS", "NVIDIA TESLA GPUS", "NVIDIA GRID GPUS"]
|
||||||
|
for section in sections:
|
||||||
|
|
||||||
for line in f.readlines():
|
for line in f.readlines():
|
||||||
|
|
||||||
# find the right data table
|
# Find the right data tables
|
||||||
if line.find(sys.argv[2]) != -1:
|
if line.find(section) != -1:
|
||||||
in_section = True
|
in_section = True
|
||||||
continue
|
continue
|
||||||
if not in_section:
|
if not in_section:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# remove Windows and Linux line endings
|
# Remove Windows and Linux line endings
|
||||||
line = line.replace('\r', '')
|
line = line.replace('\r', '')
|
||||||
line = line.replace('\n', '')
|
line = line.replace('\n', '')
|
||||||
|
|
||||||
# end of section
|
# End of section
|
||||||
if len(line) > 0 and not line.startswith(' '):
|
if len(line) > 0 and not line.startswith(' '):
|
||||||
in_section = False
|
in_section = False
|
||||||
in_table = False
|
in_table = False
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# empty line
|
|
||||||
if len(line) == 0:
|
if len(line) == 0:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# skip the header
|
# Skip the header
|
||||||
if line.startswith(' ---'):
|
if line.startswith(' ---'):
|
||||||
in_table = True
|
in_table = True
|
||||||
continue
|
continue
|
||||||
if not in_table:
|
if not in_table:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# get name
|
# PCI ID
|
||||||
pid = int(line[42:46], 16)
|
pid = int(line[50:54], 16)
|
||||||
if not pid in pids:
|
if not pid in pids:
|
||||||
pids.append(pid)
|
pids.append(pid)
|
||||||
|
|
||||||
# output
|
|
||||||
for pid in pids:
|
for pid in pids:
|
||||||
vid = 0x10de
|
vid = 0x10de
|
||||||
print("pci:v%08Xd%08Xsv*sd*bc*sc*i*" % (vid, pid))
|
print("pci:v%08Xd%08Xsv*sd*bc*sc*i*" % (vid, pid))
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|
||||||
|
Loading…
Reference in new issue