mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
95 lines
3.4 KiB
Python
Executable File
95 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.
|
|
from pathlib import Path
|
|
import sys
|
|
import argparse
|
|
|
|
|
|
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")
|
|
|
|
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")
|
|
|
|
|
|
def main(argv):
|
|
|
|
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(argv)
|
|
|
|
generate_usage_reporting(args.output_directory, args.template_root)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|