mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-21 01:01: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>
84 lines
2.0 KiB
Python
Executable File
84 lines
2.0 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 remove_tag(f: str):
|
|
if "apriltag" in f:
|
|
return False
|
|
if "36h11" in f:
|
|
return False
|
|
if "16h5" in f:
|
|
return False
|
|
if "tag" in f:
|
|
return True
|
|
return False
|
|
|
|
|
|
def copy_upstream_src(wpilib_root: Path):
|
|
apriltag = wpilib_root / "apriltag"
|
|
|
|
# Delete old install
|
|
shutil.rmtree(
|
|
apriltag / "src/main/native/thirdparty/apriltag",
|
|
ignore_errors=True,
|
|
)
|
|
shutil.rmtree(
|
|
apriltag / "src/main/include/thirdparty/apriltag",
|
|
ignore_errors=True,
|
|
)
|
|
|
|
# Copy apriltag source files into allwpilib
|
|
src_files = walk_cwd_and_copy_if(
|
|
lambda dp, f: (f.endswith(".c") or f.endswith(".cpp"))
|
|
and not has_prefix(dp, Path("example"))
|
|
and not has_prefix(dp, Path("test"))
|
|
and not f == "getopt.c"
|
|
and not "py" in f
|
|
and not remove_tag(f),
|
|
apriltag / "src/main/native/thirdparty/apriltag/src",
|
|
)
|
|
|
|
# Copy apriltag header files into allwpilib
|
|
walk_cwd_and_copy_if(
|
|
lambda dp, f: f.endswith(".h")
|
|
and not f == "getopt.h"
|
|
and not f == "postscript_utils.h"
|
|
and not remove_tag(f),
|
|
apriltag / "src/main/native/thirdparty/apriltag/include",
|
|
)
|
|
|
|
for f in src_files:
|
|
comment_out_invalid_includes(
|
|
f,
|
|
[
|
|
apriltag / "src/main/native/thirdparty/apriltag/include",
|
|
apriltag / "src/main/native/thirdparty/apriltag/include/common",
|
|
],
|
|
)
|
|
|
|
|
|
def main():
|
|
name = "apriltag"
|
|
url = "https://github.com/AprilRobotics/apriltag.git"
|
|
tag = "3806edf38ac4400153677e510c9f9dcb81f472c8"
|
|
|
|
patch_options = {
|
|
"ignore_whitespace": True,
|
|
}
|
|
|
|
apriltag = Lib(name, url, tag, copy_upstream_src, patch_options)
|
|
apriltag.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|