mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
a48f3c35f4
argparse will automatically read sys.argv, so we don't need to pass it in manually. Furthermore, none of our scripts customize argv.
94 lines
3.4 KiB
Python
Executable File
94 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Copyright (c) FIRST and other WPILib contributors.
|
|
# Open Source Software; you can modify and/or share it under the terms of
|
|
# the WPILib BSD license file in the root directory of this project.
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
|
|
def generate_usage_reporting(output_directory: Path, template_directory: Path):
|
|
# Gets the folder this script is in (the hal/ directory)
|
|
java_package = "edu/wpi/first/hal"
|
|
# fmt: off
|
|
(output_directory / "main/native/include/hal").mkdir(parents=True, exist_ok=True)
|
|
(output_directory / f"main/java/{java_package}").mkdir(parents=True, exist_ok=True)
|
|
# fmt: on
|
|
usage_reporting_types_cpp = []
|
|
usage_reporting_instances_cpp = []
|
|
usage_reporting_types = []
|
|
usage_reporting_instances = []
|
|
with (template_directory / "Instances.txt").open(encoding="utf-8") as instances:
|
|
for instance in instances:
|
|
usage_reporting_instances_cpp.append(f" {instance.strip()},")
|
|
usage_reporting_instances.append(
|
|
f" /** {instance.strip()}. */\n"
|
|
f" public static final int {instance.strip()};"
|
|
)
|
|
|
|
with (template_directory / "ResourceType.txt").open(
|
|
encoding="utf-8"
|
|
) as resource_types:
|
|
for resource_type in resource_types:
|
|
usage_reporting_types_cpp.append(f" {resource_type.strip()},")
|
|
usage_reporting_types.append(
|
|
f" /** {resource_type.strip()}. */\n"
|
|
f" public static final int {resource_type.strip()};"
|
|
)
|
|
|
|
with (template_directory / "FRCNetComm.java.in").open(
|
|
encoding="utf-8"
|
|
) as java_usage_reporting:
|
|
contents = (
|
|
# fmt: off
|
|
java_usage_reporting.read()
|
|
.replace(r"${usage_reporting_types}", "\n".join(usage_reporting_types))
|
|
.replace(r"${usage_reporting_instances}", "\n".join(usage_reporting_instances))
|
|
# fmt: on
|
|
)
|
|
|
|
frc_net_comm = output_directory / f"main/java/{java_package}/FRCNetComm.java"
|
|
frc_net_comm.write_text(contents, encoding="utf-8", newline="\n")
|
|
|
|
with (template_directory / "FRCUsageReporting.h.in").open(
|
|
encoding="utf-8"
|
|
) as cpp_usage_reporting:
|
|
contents = (
|
|
# fmt: off
|
|
cpp_usage_reporting.read()
|
|
.replace(r"${usage_reporting_types_cpp}", "\n".join(usage_reporting_types_cpp))
|
|
.replace(r"${usage_reporting_instances_cpp}", "\n".join(usage_reporting_instances_cpp))
|
|
# fmt: on
|
|
)
|
|
|
|
usage_reporting_hdr = (
|
|
output_directory / "main/native/include/hal/FRCUsageReporting.h"
|
|
)
|
|
usage_reporting_hdr.write_text(contents, encoding="utf-8", newline="\n")
|
|
|
|
|
|
def main():
|
|
dirname = Path(__file__).parent
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--output_directory",
|
|
help="Optional. If set, will output the generated files to this directory, otherwise it will use a path relative to the script",
|
|
default=dirname / "src/generated",
|
|
type=Path,
|
|
)
|
|
parser.add_argument(
|
|
"--template_root",
|
|
help="Optional. If set, will use this directory as the root for the jinja templates",
|
|
default=dirname / "src/generate",
|
|
type=Path,
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
generate_usage_reporting(args.output_directory, args.template_root)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|