mirror of
https://github.com/LTTLabsOSS/markbench-tests.git
synced 2026-04-24 03:01:09 -04:00
2d99156818
- changes to how 3d mark harness handles errors and adds main function - renaming ambiguous file names - ordering imports
122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
"""Utility functions supporting Hitman World of Assassination test script."""
|
|
|
|
import os
|
|
import re
|
|
import winreg
|
|
from argparse import ArgumentParser
|
|
|
|
|
|
def get_args() -> any:
|
|
"""Get command line arguments"""
|
|
parser = ArgumentParser()
|
|
parser.add_argument(
|
|
"--kerasHost",
|
|
dest="keras_host",
|
|
help="Host for Keras OCR service",
|
|
required=True,
|
|
)
|
|
parser.add_argument(
|
|
"--kerasPort",
|
|
dest="keras_port",
|
|
help="Port for Keras OCR service",
|
|
required=True,
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
def export_registry_key(hive, subkey, input_file):
|
|
"""Exports a registry key for interpretation."""
|
|
try:
|
|
if not os.path.exists(input_file):
|
|
with open(input_file, "w", encoding="utf-8") as file:
|
|
file.write("")
|
|
with winreg.OpenKey(hive, subkey) as reg_key:
|
|
with open(input_file, "w", encoding="utf-8") as reg_file:
|
|
reg_file.write("Windows Registry Editor Version 5.00\n\n")
|
|
reg_file.write(f"[{subkey}]\n")
|
|
try:
|
|
index = 0
|
|
while True:
|
|
value_name, value_data, value_type = winreg.EnumValue(
|
|
reg_key, index
|
|
)
|
|
if value_type == winreg.REG_DWORD:
|
|
value_data = f"dword:{value_data:08x}"
|
|
elif value_type == winreg.REG_SZ:
|
|
value_data = f'"{value_data}"'
|
|
elif value_type == winreg.REG_QWORD:
|
|
value_data = f"qword:{value_data:0x16x}"
|
|
else:
|
|
value_data = f'"{value_data}"'
|
|
reg_file.write(f'"{value_name}"={value_data}\n')
|
|
index += 1
|
|
except OSError:
|
|
pass
|
|
except OSError as e:
|
|
print(f"Failed to open the registry key: {e}")
|
|
|
|
|
|
def convert_dword_to_decimal(dword_hex):
|
|
"""Converts a dword key value to decimal numbers."""
|
|
return int(dword_hex, 16)
|
|
|
|
|
|
def process_registry_file(hive, subkey, input_file, config_file):
|
|
"""Processes the exported registry file and converts it to readable text."""
|
|
export_registry_key(hive, subkey, input_file)
|
|
with open(input_file, "r", encoding="utf-8") as file:
|
|
lines = file.readlines()
|
|
|
|
modified_lines = []
|
|
|
|
dword_pattern = re.compile(r"^(\"[^\"]+\")=dword:([0-9a-fA-F]+)", re.IGNORECASE)
|
|
|
|
for line in lines:
|
|
match = dword_pattern.search(line)
|
|
if match:
|
|
key = match.group(1)
|
|
hex_value = match.group(2)
|
|
decimal_value = convert_dword_to_decimal(hex_value)
|
|
modified_line = f"{key}={decimal_value}\n"
|
|
modified_lines.append(modified_line)
|
|
else:
|
|
modified_lines.append(line)
|
|
with open(config_file, "w", encoding="utf-8") as file:
|
|
file.writelines(modified_lines)
|
|
|
|
|
|
def get_resolution(config_file: str) -> tuple[int]:
|
|
"""Retrieve the resolution from local configuration files."""
|
|
width_pattern = re.compile(r"\"ResolutionWidth\"=(\d+)")
|
|
height_pattern = re.compile(r"\"ResolutionHeight\"=(\d+)")
|
|
width = 0
|
|
height = 0
|
|
|
|
with open(config_file, encoding="utf-8") as file:
|
|
lines = file.readlines()
|
|
for line in lines:
|
|
width_match = width_pattern.match(line)
|
|
height_match = height_pattern.match(line)
|
|
|
|
if width_match:
|
|
width = width_match.group(1)
|
|
if height_match:
|
|
height = height_match.group(1)
|
|
|
|
return (height, width)
|
|
|
|
|
|
def get_benchmark_name(config_file: str) -> tuple[int]:
|
|
"""Retrieve the resolution from local configuration files."""
|
|
benchmark_pattern = re.compile(r"\"BenchmarkScene\"=(\d+)")
|
|
benchmark = 0
|
|
|
|
with open(config_file, encoding="utf-8") as file:
|
|
lines = file.readlines()
|
|
for line in lines:
|
|
benchmark_match = benchmark_pattern.match(line)
|
|
if benchmark_match:
|
|
benchmark = int(benchmark_match.group(1))
|
|
|
|
return benchmark
|