mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-26 01:51:41 +00:00
60 lines
1.9 KiB
Python
Executable File
60 lines
1.9 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 json
|
|
import os
|
|
|
|
from jinja2 import Environment, FileSystemLoader
|
|
|
|
|
|
def write_controller_file(outPath, controllerName, contents):
|
|
if not os.path.exists(outPath):
|
|
os.makedirs(outPath)
|
|
|
|
outpathname = f"{outPath}/{controllerName}"
|
|
|
|
if os.path.exists(outpathname):
|
|
with open(outpathname, "r") as f:
|
|
if f.read() == contents:
|
|
return
|
|
|
|
# File either doesn't exist or has different contents
|
|
with open(outpathname, "w", newline="\n") as f:
|
|
f.write(contents)
|
|
|
|
|
|
def main():
|
|
dirname, _ = os.path.split(os.path.abspath(__file__))
|
|
|
|
with open(f"{dirname}/src/generate/hids.json") as f:
|
|
controllers = json.load(f)
|
|
|
|
# Java files
|
|
env = Environment(
|
|
loader=FileSystemLoader(f"{dirname}/src/generate/"),
|
|
autoescape=False,
|
|
keep_trailing_newline=True,
|
|
)
|
|
rootPath = f"{dirname}/src/generated/main/java/edu/wpi/first/wpilibj"
|
|
template = env.get_template("hid.java.jinja")
|
|
for controller in controllers:
|
|
controllerName = os.path.basename(f"{controller['ConsoleName']}Controller.java")
|
|
output = template.render(controller)
|
|
write_controller_file(rootPath, controllerName, output)
|
|
|
|
# Java simulation files
|
|
rootPath = f"{dirname}/src/generated/main/java/edu/wpi/first/wpilibj/simulation"
|
|
template = env.get_template("hidsim.java.jinja")
|
|
for controller in controllers:
|
|
controllerName = os.path.basename(
|
|
f"{controller['ConsoleName']}ControllerSim.java"
|
|
)
|
|
output = template.render(controller)
|
|
write_controller_file(rootPath, controllerName, output)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|