Files
allwpilib/wpilibj/generate_wpilibj.py

33 lines
1004 B
Python

import argparse
import os
from pathlib import Path
from wpilibj.generate_hids import generate_hids
from wpilibj.generate_pwm_motor_controllers import generate_pwm_motor_controllers
def main():
dirname, _ = os.path.split(os.path.abspath(__file__))
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=os.path.join(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=os.path.join(dirname, "src/generate"),
type=Path,
)
args = parser.parse_args()
generate_hids(args.output_directory, args.template_root)
generate_pwm_motor_controllers(args.output_directory, args.template_root)
if __name__ == "__main__":
main()