flywithu/ntel-gpu-tools
1
0
Fork 0
mirror of https://github.com/tiagovignatti/intel-gpu-tools.git synced 2025-06-25 00:36:16 +00:00
Code Issues Releases Wiki Activity

quick_dump: Refactor to allow future code

Browse Source 
Extract everything possible, and create a main function which is pretty
darn simple. The goal is to make the various paths the program can take
based on command line options (one more is coming) more decipherable.

Signed-off-by: Ben Widawsky <ben@bwidawsk.net>
This commit is contained in:
Ben Widawsky 2014-01-31 04:42:50 +00:00
parent 48ad03ca0c
commit 8ce7c366d7
1 changed files with 40 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
tools/quick_dump/quick_dump.py
View File

@ -44,47 +44,60 @@ def parse_file(file):
print('{0:#010x} | {1:<28} | {2:#010x}'.format(intreg, register[0], val)) print('{0:#010x} | {1:<28} | {2:#010x}'.format(intreg, register[0], val))
print('') print('')
def walk_base_files():
parser = argparse.ArgumentParser(description='Dumb register dumper.')
parser.add_argument('-b', '--baseless', action='store_true', default=False, help='baseless mode, ignore files starting with base_')
parser.add_argument('-a', '--autodetect', action='store_true', default=False, help='autodetect chipset')
parser.add_argument('profile', nargs='?', type=argparse.FileType('r'), default=None)
args = parser.parse_args()
if reg.init() == False:
print("Register initialization failed")
sys.exit()
# Put us where the script is
os.chdir(os.path.dirname(sys.argv[0]))
#parse anything named base_ these are assumed to apply for all gens.
if args.baseless == False:
for root, dirs, files in os.walk('.'): for root, dirs, files in os.walk('.'):
for name in files: for name in files:
if name.startswith(("base_")): if name.startswith(("base_")):
file = open(name.rstrip(), 'r') file = open(name.rstrip(), 'r')
parse_file(file) parse_file(file)
if args.autodetect: def autodetect_chipset():
pci_dev = chipset.intel_get_pci_device() pci_dev = chipset.intel_get_pci_device()
devid = chipset.pcidev_to_devid(pci_dev) devid = chipset.pcidev_to_devid(pci_dev)
if chipset.is_sandybridge(devid): if chipset.is_sandybridge(devid):
args.profile = open('sandybridge', 'r') return open('sandybridge', 'r')
elif chipset.is_ivybridge(devid): elif chipset.is_ivybridge(devid):
args.profile = open('ivybridge', 'r') return open('ivybridge', 'r')
elif chipset.is_valleyview(devid): elif chipset.is_valleyview(devid):
args.profile = open('valleyview', 'r') return open('valleyview', 'r')
elif chipset.is_haswell(devid): elif chipset.is_haswell(devid):
args.profile = open('haswell', 'r') return open('haswell', 'r')
elif chipset.is_broadwell(devid): elif chipset.is_broadwell(devid):
args.profile = open('broadwell', 'r') return open('broadwell', 'r')
else: else:
print("Autodetect of devid " + hex(devid) + " failed") print("Autodetect of devid " + hex(devid) + " failed")
return None
if args.profile == None: if __name__ == "__main__":
sys.exit() parser = argparse.ArgumentParser(description='Dumb register dumper.')
parser.add_argument('-b', '--baseless',
action='store_true', default=False,
help='baseless mode, ignore files starting with base_')
parser.add_argument('-a', '--autodetect',
action='store_true', default=False,
help='autodetect chipset')
parser.add_argument('profile', nargs='?',
type=argparse.FileType('r'), default=None)
for extra in args.profile: args = parser.parse_args()
extra_file = open(extra.rstrip(), 'r')
parse_file(extra_file) if reg.init() == False:
print("Register initialization failed")
sys.exit()
# Put us where the script is
os.chdir(os.path.dirname(sys.argv[0]))
#parse anything named base_ these are assumed to apply for all gens.
if args.baseless == False:
walk_base_files()
if args.autodetect:
args.profile = autodetect_chipset()
if args.profile == None:
sys.exit()
for extra in args.profile:
extra_file = open(extra.rstrip(), 'r')
parse_file(extra_file)