mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
ca05ffa1b9
A noteworthy change is the replacement of the `dp.startswith(os.path.join(".", "subdir"))` pattern. pathlib doesn't offer something with similar semantics besides `match` and `full_match`, so there's now a helper function that replicates the behavior.
Other notable changes include the addition of type annotations to ensure code correctness, using == to check file names instead of `endswith` for clarity (`endswith` is still used to check extensions), manual walking and copying being refactored in googletest, json, memory, nanopb, protobuf, and sleipnir to use `walk_cwd_and_copy_if`, and matching functions being shortened to the point where they can just be inlined into the lambda.
Co-authored-by: Tyler Veness <calcmogul@gmail.com>
Co-authored-by: David Vo <auscompgeek@users.noreply.github.com>
143 lines
3.4 KiB
Python
Executable File
143 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
from upstream_utils import (
|
|
Lib,
|
|
comment_out_invalid_includes,
|
|
has_prefix,
|
|
walk_cwd_and_copy_if,
|
|
)
|
|
|
|
|
|
def eigen_inclusions(dp: Path, f: str):
|
|
"""Returns true if the given file in the "Eigen" include directory of the
|
|
Eigen git repo should be copied into allwpilib
|
|
|
|
Keyword arguments:
|
|
dp -- directory path
|
|
f -- filename
|
|
"""
|
|
if not has_prefix(dp, Path("Eigen")):
|
|
return False
|
|
|
|
# Exclude NonMPL2.h since all non-MPL2 code will be excluded anyway
|
|
if f == "NonMPL2.h":
|
|
return False
|
|
|
|
# Exclude BLAS support
|
|
if f.endswith("_BLAS.h") or "blas" in f:
|
|
return False
|
|
|
|
# Exclude LAPACK support
|
|
if f.endswith("_LAPACKE.h") or "lapack" in f:
|
|
return False
|
|
|
|
# Exclude MKL support
|
|
if "MKL" in f:
|
|
return False
|
|
|
|
# Include architectures we care about by filtering for Core/arch
|
|
if "Core" in dp.parts and "arch" in dp.parts:
|
|
return (
|
|
"AVX" in dp.parts
|
|
or "Default" in dp.parts
|
|
or "NEON" in dp.parts
|
|
or "SSE" in dp.parts
|
|
)
|
|
|
|
# Include the following modules
|
|
modules = [
|
|
"Cholesky",
|
|
"Core",
|
|
"Eigenvalues",
|
|
"Geometry",
|
|
"Householder",
|
|
"IterativeLinearSolvers",
|
|
"Jacobi",
|
|
"LU",
|
|
"OrderingMethods",
|
|
"QR",
|
|
"SVD",
|
|
"SparseCholesky",
|
|
"SparseCore",
|
|
"SparseLU",
|
|
"SparseQR",
|
|
"misc",
|
|
"plugins",
|
|
]
|
|
for m in modules:
|
|
if m in dp.parts or f == m:
|
|
return True
|
|
return False
|
|
|
|
|
|
def unsupported_inclusions(dp: Path, f: str):
|
|
"""Returns true if the given file in the "unsupported" include directory of
|
|
the Eigen git repo should be copied into allwpilib
|
|
|
|
Keyword arguments:
|
|
dp -- directory path
|
|
f -- filename
|
|
"""
|
|
if not has_prefix(dp, Path("unsupported")):
|
|
return False
|
|
|
|
abspath = dp / f
|
|
|
|
# Exclude build system and READMEs
|
|
if f == "CMakeLists.txt" or "README" in f:
|
|
return False
|
|
|
|
# Include the MatrixFunctions module
|
|
return "MatrixFunctions" in abspath.parts
|
|
|
|
|
|
def copy_upstream_src(wpilib_root: Path):
|
|
wpimath = wpilib_root / "wpimath"
|
|
|
|
# Delete old install
|
|
for d in ["src/main/native/thirdparty/eigen/include"]:
|
|
shutil.rmtree(wpimath / d, ignore_errors=True)
|
|
|
|
# Copy Eigen headers into allwpilib
|
|
eigen_files = walk_cwd_and_copy_if(
|
|
eigen_inclusions,
|
|
wpimath / "src/main/native/thirdparty/eigen/include",
|
|
)
|
|
|
|
# Copy unsupported headers into allwpilib
|
|
unsupported_files = walk_cwd_and_copy_if(
|
|
unsupported_inclusions,
|
|
wpimath / "src/main/native/thirdparty/eigen/include",
|
|
)
|
|
|
|
for f in eigen_files:
|
|
comment_out_invalid_includes(
|
|
f, [wpimath / "src/main/native/thirdparty/eigen/include"]
|
|
)
|
|
for f in unsupported_files:
|
|
comment_out_invalid_includes(
|
|
f, [wpimath / "src/main/native/thirdparty/eigen/include"]
|
|
)
|
|
|
|
shutil.copyfile(
|
|
".clang-format",
|
|
wpimath / "src/main/native/thirdparty/eigen/include/.clang-format",
|
|
)
|
|
|
|
|
|
def main():
|
|
name = "eigen"
|
|
url = "https://gitlab.com/libeigen/eigen.git"
|
|
# master on 2025-05-18
|
|
tag = "d81aa18f4dc56264b2cd7e2f230807d776a2d385"
|
|
|
|
eigen = Lib(name, url, tag, copy_upstream_src)
|
|
eigen.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|