space-thumbnails/test_speed_aggressive.py at rebuild5 · MicaApps/space-thumbnails · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# -*- coding: utf-8 -*-
import os
import shutil
import subprocess
import time
import re

# Source path with Chinese characters
source_path = r"assets\LGQGJ-00-00 杞瘋鎶涘厜鏈烘€昏.STEP"
temp_input = os.path.join(os.environ["TEMP"], "speed_test_input.step")
temp_output = os.path.join(os.environ["TEMP"], "test_speed.obj")
bat_file = os.path.abspath(r"tools\step2obj.bat")

def run_test(deflection, ang_deflection, relative, format_val, label):
    print(f"\nRunning Test: {label}")
    print(f"Params: Deflection={deflection}, Ang={ang_deflection}, Relative={relative}, Format={format_val}")

    env = os.environ.copy()
    env["STEP2OBJ_INPUT"] = os.path.abspath(temp_input)
    env["STEP2OBJ_OUTPUT"] = os.path.abspath(temp_output)
    env["STEP2OBJ_DEFLECTION"] = str(deflection)
    env["STEP2OBJ_ANG_DEFLECTION"] = str(ang_deflection)
    env["STEP2OBJ_RELATIVE"] = str(relative)
    env["STEP2OBJ_FORMAT"] = format_val

    start_time = time.time()
    # Use shell=True to run the bat file
    result = subprocess.run([bat_file], env=env, shell=True, capture_output=True)
    end_time = time.time()

    print(f"Total Time: {end_time - start_time:.2f}s")

    if result.returncode != 0:
        print(f"Error: Exit code {result.returncode}")
        print("STDERR:", result.stderr.decode('mbcs', errors='replace'))

    log_file = temp_output + ".log"
    if os.path.exists(log_file):
        with open(log_file, "r", encoding="utf-8", errors="replace") as f:
            log_content = f.read()

        read_time = re.search(r"ReadFile took ([\d\.]+)s", log_content)
        if read_time:
            print(f"Internal ReadFile Time: {read_time.group(1)}s")

        transfer_time = re.search(r"Transfer successful \(took ([\d\.]+)s\)", log_content)
        if transfer_time:
            print(f"Internal Transfer Time: {transfer_time.group(1)}s")

        meshing_time = re.search(r"Meshing took ([\d\.]+)s", log_content)
        if meshing_time:
            print(f"Internal Meshing Time: {meshing_time.group(1)}s")

        tri_count = re.search(r"Total Triangles: (\d+)", log_content)
        if tri_count:
            print(f"Total Triangles: {tri_count.group(1)}")

        bbox_info = re.search(r"BBox: .* Diag: ([\d\.]+)", log_content)
        if bbox_info:
            print(f"BBox Diag: {bbox_info.group(1)}")

        color_time = re.search(r"Mapped colors for \d+ faces in ([\d\.]+)s", log_content)
        if color_time:
             print(f"Internal Color Mapping Time: {color_time.group(1)}s")

        exporting_time = re.search(r"Exporting took ([\d\.]+)s", log_content)
        if exporting_time:
            print(f"Internal Exporting Time: {exporting_time.group(1)}s")

        if "Error" in log_content:
            print("--- Log Errors ---")
            print(log_content)
            print("------------------")

        os.remove(log_file)
    else:
        print("Log file not found.")

    if os.path.exists(temp_output):
        os.remove(temp_output)
    mtl_file = temp_output.replace(".obj", ".mtl")
    if os.path.exists(mtl_file):
        os.remove(mtl_file)

def main():
    if not os.path.exists(source_path):
        print(f"Source file not found: {source_path}")
        return

    print(f"Copying {source_path} to {temp_input}...")
    shutil.copy2(source_path, temp_input)

    try:
        # Aggressive 1 (200.0, 1.0) OBJ
        run_test(200.0, 1.0, False, "OBJ", "Aggressive 1 (200.0, 1.0)")

        # Aggressive 2 (500.0, 1.0) OBJ
        run_test(500.0, 1.0, False, "OBJ", "Aggressive 2 (500.0, 1.0)")

        # Aggressive 3 (1000.0, 1.0) OBJ
        run_test(1000.0, 1.0, False, "OBJ", "Aggressive 3 (1000.0, 1.0)")

    finally:
        if os.path.exists(temp_input):
            os.remove(temp_input)

if __name__ == "__main__":
    main()