mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-20 00:51:42 +00:00
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import os
|
|
import shutil
|
|
import sys
|
|
|
|
MAX_NUM = 20
|
|
|
|
dirname, _ = os.path.split(os.path.abspath(__file__))
|
|
cmake_binary_dir = sys.argv[1]
|
|
|
|
with open(f"{dirname}/src/generate/GenericNumber.java.in", "r") as templateFile:
|
|
template = templateFile.read()
|
|
rootPath = f"{cmake_binary_dir}/generated/main/java/edu/wpi/first/wpiutil/math/numbers"
|
|
|
|
if os.path.exists(rootPath):
|
|
shutil.rmtree(rootPath)
|
|
os.makedirs(rootPath)
|
|
|
|
for i in range(MAX_NUM + 1):
|
|
with open(f"{rootPath}/N{i}.java", "w") as f:
|
|
f.write(template.replace("${num}", str(i)))
|
|
|
|
with open(f"{dirname}/src/generate/Nat.java.in", "r") as templateFile:
|
|
template = templateFile.read()
|
|
outputPath = f"{cmake_binary_dir}/generated/main/java/edu/wpi/first/wpiutil/math/Nat.java"
|
|
with open(f"{dirname}/src/generate/NatGetter.java.in", "r") as getterFile:
|
|
getter = getterFile.read()
|
|
|
|
if os.path.exists(outputPath):
|
|
os.remove(outputPath)
|
|
|
|
importsString = ""
|
|
|
|
for i in range(MAX_NUM + 1):
|
|
importsString += f"import edu.wpi.first.wpiutil.math.numbers.N{i};\n"
|
|
template += getter.replace("${num}", str(i))
|
|
|
|
template += "}\n"
|
|
|
|
template = template.replace('{{REPLACEWITHIMPORTS}}', importsString)
|
|
|
|
with open(outputPath, "w") as f:
|
|
f.write(template)
|