mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
51a3876330
This hooks up the bazel build to the robotpyExamples. It can use the (formly pyfrc or whatever) automatic unit tests for an example, as well as exposing the ability to run the example in simulation, with or without `halsim_gui` with a command such as `bazel run //robotpyExamples:AddressableLED-sim` This required building and using wheels instead of just a normal `py_library`, so that things like `ENTRY_POINTS` can be used. I took a bare bones approach to building and naming the wheels (for example the native ones don't have the OS info or python version in them, so they wouldn't be suitable publish to pypi, but that can always be updated later.
119 lines
4.3 KiB
Python
119 lines
4.3 KiB
Python
import argparse
|
|
import json
|
|
|
|
import tomli
|
|
from jinja2 import BaseLoader, Environment
|
|
|
|
from shared.bazel.rules.robotpy.generation_utils import (
|
|
fixup_python_dep_name,
|
|
fixup_root_package_name,
|
|
fixup_shared_lib_name,
|
|
)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--project_cfg")
|
|
parser.add_argument("--output_file")
|
|
parser.add_argument("--third_party_dirs", nargs="+")
|
|
args = parser.parse_args()
|
|
|
|
with open(args.project_cfg, "rb") as fp:
|
|
raw_config = tomli.load(fp)
|
|
|
|
def double_quotes(data):
|
|
if data:
|
|
return json.dumps(data)
|
|
return None
|
|
|
|
def get_pc_dep(library):
|
|
base_project = library.replace("robotpy-native-", "")
|
|
wpilib_project = fixup_root_package_name(base_project)
|
|
return f"//{wpilib_project}:native/{base_project}/{library}.pc"
|
|
|
|
def get_python_dep(library):
|
|
base_project = library.replace("robotpy-native-", "")
|
|
wpilib_project = fixup_root_package_name(base_project)
|
|
return f"//{fixup_root_package_name(wpilib_project)}:{fixup_python_dep_name(library)}"
|
|
|
|
env = Environment(loader=BaseLoader)
|
|
env.filters["double_quotes"] = double_quotes
|
|
env.filters["get_pc_dep"] = get_pc_dep
|
|
env.filters["get_python_dep"] = get_python_dep
|
|
template = env.from_string(BUILD_FILE_TEMPLATE)
|
|
|
|
nativelib_config = raw_config["tool"]["hatch"]["build"]["hooks"]["nativelib"]
|
|
project_name = nativelib_config["pcfile"][0]["name"]
|
|
root_package = fixup_root_package_name(project_name)
|
|
shared_library_name = fixup_shared_lib_name(project_name)
|
|
with open(args.output_file, "w") as f:
|
|
f.write(
|
|
template.render(
|
|
raw_project_config=raw_config["project"],
|
|
nativelib_config=nativelib_config,
|
|
root_package=root_package,
|
|
shared_library_name=shared_library_name,
|
|
third_party_dirs=args.third_party_dirs or [],
|
|
)
|
|
)
|
|
|
|
|
|
BUILD_FILE_TEMPLATE = """# THIS FILE IS AUTO GENERATED
|
|
|
|
load("@aspect_bazel_lib//lib:copy_to_directory.bzl", "copy_to_directory")
|
|
load("//shared/bazel/rules/robotpy:pybind_rules.bzl", "native_wrappery_library")
|
|
|
|
def define_native_wrapper(name, pyproject_toml = None):
|
|
copy_to_directory(
|
|
name = "{}.copy_headers".format(name),
|
|
srcs = native.glob(["src/main/native/include/**"]) + native.glob(["src/generated/main/native/include/**"], allow_empty = True){% if third_party_dirs %} + native.glob([
|
|
{%- for dir in third_party_dirs %}
|
|
"src/main/native/thirdparty/{{dir}}/include/**",
|
|
{%- endfor %}
|
|
]){%- endif %},
|
|
out = "native/{{nativelib_config.pcfile[0].name}}/include",
|
|
root_paths = ["src/main/native/include/"],
|
|
replace_prefixes = {
|
|
"{{root_package}}/src/generated/main/native/include": "",
|
|
"{{root_package}}/src/main/native/include": "",
|
|
{%- for dir in third_party_dirs %}
|
|
"{{root_package}}/src/main/native/thirdparty/{{dir}}/include": "",
|
|
{%- endfor %}
|
|
},
|
|
verbose = False,
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
native_wrappery_library(
|
|
name = name,
|
|
pyproject_toml = pyproject_toml or "src/main/python/native-pyproject.toml",
|
|
libinit_file = "native/{{nativelib_config.pcfile[0].name}}/_init_{{raw_project_config.name.replace("-", "_")}}.py",
|
|
pc_file = "native/{{nativelib_config.pcfile[0].name}}/{{raw_project_config.name}}.pc",
|
|
pc_deps = [
|
|
{%- for dep in nativelib_config.pcfile[0].requires | sort %}
|
|
"{{dep | get_pc_dep}}",
|
|
{%- endfor %}
|
|
],
|
|
deps = [
|
|
{%- for dep in nativelib_config.pcfile[0].requires | sort %}
|
|
"{{dep | get_python_dep}}",
|
|
{%- endfor %}
|
|
],
|
|
headers = "{}.copy_headers".format(name),
|
|
native_shared_library = "shared/{{shared_library_name}}",
|
|
install_path = "native/{{nativelib_config.pcfile[0].name}}/",
|
|
strip_path_prefixes = ["{{root_package}}"],
|
|
requires = {{raw_project_config.dependencies | double_quotes}},
|
|
summary = "{{raw_project_config.description}}",
|
|
entry_points = {
|
|
"pkg_config": [
|
|
"{{nativelib_config.pcfile[0].name}} = native.{{nativelib_config.pcfile[0].name}}",
|
|
],
|
|
},
|
|
)
|
|
|
|
"""
|
|
|
|
if __name__ == "__main__":
|
|
main()
|