project init

src/speccpu/x264.py

@@ -0,0 +1,84 @@
+import os
+import shutil
+from pathlib import Path
+from subprocess import run
+
+
+def get_mc_path(build):
+    return build / "src" / "common" / "mc.c"
+
+
+def get_ref_add_pragma(mc_lines):
+    # Initialize variables
+    modified_lines = []
+    pragma_added = False
+
+    pragma_str = "#pragma clang loop vectorize(disable)"
+
+    # Iterate through each line to find the target for loop
+    for i, line in enumerate(mc_lines):
+        # If there are already #pragma string, skip it.
+        if i > 1 and pragma_str in mc_lines[i - 1]:
+            continue
+        if "for( int x = 0; x < i_width; x++ )" in line:
+            modified_lines.append("#pragma clang loop vectorize(disable)\n")
+            pragma_added = True
+
+        modified_lines.append(line)
+
+    return modified_lines, pragma_added
+
+
+def disable_mc_loop_vectorize(mc_path: Path) -> bool:
+    """
+    Disable x264 mc.c loop vectorizing, mainly for "get_ref" performance.
+    If needed, update the file located at mc_path, returns true if that file is updated.
+    """
+    # Read the content of mc.c
+    with open(mc_path, "r") as mc_file:
+        mc_lines = mc_file.readlines()
+
+    mc_pragma_lines, mc_modified = get_ref_add_pragma(mc_lines)
+
+    if mc_modified:
+        with open(mc_path, "w") as mc:
+            mc.writelines(mc_pragma_lines)
+
+    return mc_modified
+
+
+def update_exe(build_exe, exe_dir):
+    exe_file = exe_dir / "x264_s_base.mytest-m64"
+
+    shutil.copy2(build_exe, exe_file)
+    print(f"Copied to {exe_file}")
+
+
+def make(build, spec_env):
+    build_exe = build / "x264_s"
+
+    run(
+        [
+            "make",
+            "TARGET=s",
+        ],
+        check=True,
+        env=spec_env,
+        cwd=build,
+    )
+
+    # Copy "s" into run_dir
+    if not build_exe.exists():
+        raise RuntimeError(f"x264 exe not generated! Please verify make process")
+    return build_exe
+
+
+def recompile_mc(mc_path: Path, build: Path, spec_env: os._Environ):
+    # Recompile mc.o
+    mc_o_path = mc_path.with_suffix(".o")
+    if mc_o_path.exists():
+        print(f"Found exist mc.o file at {mc_o_path}")
+        mc_o_path.unlink()
+
+    # Run "make"
+    return make(build, spec_env)