mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
109 lines
2.9 KiB
Python
Executable File
109 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import shutil
|
|
|
|
from upstream_utils import (
|
|
walk_if,
|
|
copy_to,
|
|
Lib,
|
|
)
|
|
|
|
|
|
def run_source_replacements(memory_files):
|
|
for wpi_file in memory_files:
|
|
with open(wpi_file) as f:
|
|
content = f.read()
|
|
|
|
# Fix #includes
|
|
content = content.replace('include "', 'include "wpi/memory/')
|
|
content = content.replace(
|
|
"wpi/memory/free_list_utils.hpp", "free_list_utils.hpp"
|
|
)
|
|
|
|
with open(wpi_file, "w") as f:
|
|
f.write(content)
|
|
|
|
|
|
def run_header_replacements(memory_files):
|
|
for wpi_file in memory_files:
|
|
if "detail" not in wpi_file:
|
|
continue
|
|
with open(wpi_file) as f:
|
|
content = f.read()
|
|
|
|
# Fix #includes
|
|
content = content.replace('include "config.hpp', 'include "../config.hpp')
|
|
|
|
with open(wpi_file, "w") as f:
|
|
f.write(content)
|
|
|
|
|
|
def run_global_replacements(memory_files):
|
|
for wpi_file in memory_files:
|
|
with open(wpi_file) as f:
|
|
content = f.read()
|
|
|
|
# Rename namespace from foonathan to wpi
|
|
content = content.replace("namespace foonathan", "namespace wpi")
|
|
content = content.replace("foonathan::", "wpi::")
|
|
content = content.replace("FOONATHAN_", "WPI_")
|
|
|
|
# Fix #includes
|
|
content = content.replace('include "foonathan', 'include "wpi')
|
|
|
|
with open(wpi_file, "w") as f:
|
|
f.write(content)
|
|
|
|
|
|
def copy_upstream_src(wpilib_root):
|
|
wpiutil = os.path.join(wpilib_root, "wpiutil")
|
|
|
|
# Delete old install
|
|
for d in [
|
|
"src/main/native/thirdparty/memory/src",
|
|
"src/main/native/thirdparty/memory/include",
|
|
]:
|
|
shutil.rmtree(os.path.join(wpiutil, d), ignore_errors=True)
|
|
|
|
# Copy sources
|
|
src_files = walk_if("src", lambda dp, f: f.endswith(".cpp") or f.endswith(".hpp"))
|
|
src_files = copy_to(
|
|
src_files, os.path.join(wpiutil, "src/main/native/thirdparty/memory")
|
|
)
|
|
run_global_replacements(src_files)
|
|
run_source_replacements(src_files)
|
|
|
|
# Copy headers
|
|
os.chdir(os.path.join("include", "foonathan"))
|
|
include_files = walk_if(".", lambda dp, f: f.endswith(".hpp"))
|
|
include_files = copy_to(
|
|
include_files,
|
|
os.path.join(wpiutil, "src/main/native/thirdparty/memory/include/wpi"),
|
|
)
|
|
os.chdir(os.path.join("..", ".."))
|
|
run_global_replacements(include_files)
|
|
run_header_replacements(include_files)
|
|
|
|
# Copy config_impl.hpp
|
|
shutil.copyfile(
|
|
os.path.join(wpilib_root, "upstream_utils/memory_files/config_impl.hpp"),
|
|
os.path.join(
|
|
wpiutil,
|
|
"src/main/native/thirdparty/memory/include/wpi/memory/config_impl.hpp",
|
|
),
|
|
)
|
|
|
|
|
|
def main():
|
|
name = "memory"
|
|
url = "https://github.com/foonathan/memory"
|
|
tag = "v0.7-3"
|
|
|
|
memory = Lib(name, url, tag, [], copy_upstream_src)
|
|
memory.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|