generate.py
2.8 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env python3
# Copyright (c) 2023 Xiaomi Corporation
import argparse
import re
from pathlib import Path
import jinja2
SHERPA_ONNX_DIR = Path(__file__).resolve().parent.parent.parent
def get_version():
cmake_file = SHERPA_ONNX_DIR / "CMakeLists.txt"
with open(cmake_file) as f:
content = f.read()
version = re.search(r"set\(SHERPA_ONNX_VERSION (.*)\)", content).group(1)
return version.strip('"')
def read_proj_file(filename):
with open(filename) as f:
return f.read()
def get_dict():
version = get_version()
return {
"version": get_version(),
}
def process_linux(s):
libs = [
"libkaldi-native-fbank-core.so",
"libonnxruntime.so.1.14.0",
"libsherpa-onnx-c-api.so",
"libsherpa-onnx-core.so",
]
prefix = f"{SHERPA_ONNX_DIR}/linux/sherpa_onnx/lib/"
libs = [prefix + lib for lib in libs]
libs = "\n ;".join(libs)
d = get_dict()
d["dotnet_rid"] = "linux-x64"
d["libs"] = libs
environment = jinja2.Environment()
template = environment.from_string(s)
s = template.render(**d)
with open("./linux/sherpa-onnx.runtime.csproj", "w") as f:
f.write(s)
def process_macos(s):
libs = [
"libkaldi-native-fbank-core.dylib",
"libonnxruntime.1.14.0.dylib",
"libsherpa-onnx-c-api.dylib",
"libsherpa-onnx-core.dylib",
]
prefix = f"{SHERPA_ONNX_DIR}/macos/sherpa_onnx/lib/"
libs = [prefix + lib for lib in libs]
libs = "\n ;".join(libs)
d = get_dict()
d["dotnet_rid"] = "osx-x64"
d["libs"] = libs
environment = jinja2.Environment()
template = environment.from_string(s)
s = template.render(**d)
with open("./macos/sherpa-onnx.runtime.csproj", "w") as f:
f.write(s)
def process_windows(s):
libs = [
"kaldi-native-fbank-core.dll",
"onnxruntime.dll",
"sherpa-onnx-c-api.dll",
"sherpa-onnx-core.dll",
]
prefix = f"{SHERPA_ONNX_DIR}/windows/sherpa_onnx/lib/"
libs = [prefix + lib for lib in libs]
libs = "\n ;".join(libs)
d = get_dict()
d["dotnet_rid"] = "win-x64"
d["libs"] = libs
environment = jinja2.Environment()
template = environment.from_string(s)
s = template.render(**d)
with open("./windows/sherpa-onnx.runtime.csproj", "w") as f:
f.write(s)
def main():
s = read_proj_file("./sherpa-onnx.csproj.runtime.in")
process_macos(s)
process_linux(s)
process_windows(s)
s = read_proj_file("./sherpa-onnx.csproj.in")
d = get_dict()
d["packages_dir"] = str(SHERPA_ONNX_DIR / "scripts/dotnet/packages")
environment = jinja2.Environment()
template = environment.from_string(s)
s = template.render(**d)
with open("./all/sherpa-onnx.csproj", "w") as f:
f.write(s)
if __name__ == "__main__":
main()