ddnet/datasrc/crosscompile.py
def 13a8dfa307 Add pylint
and fix occurences
2020-12-09 10:40:28 +01:00

53 lines
1.2 KiB
Python

import network
import seven.network
def get_msgs():
return ["NETMSG_INVALID"] + [m.enum_name for m in network.Messages]
def get_msgs_7():
return ["NETMSG_INVALID"] + [m.enum_name for m in seven.network.Messages]
def get_objs():
return ["NETOBJ_INVALID"] + [m.enum_name for m in network.Objects if m.ex is None]
def get_objs_7():
return ["NETOBJ_INVALID"] + [m.enum_name for m in seven.network.Objects]
def generate_map(a, b):
result = []
for m in a:
try:
result += [b.index(m)]
except ValueError:
result += [-1]
return result
def output_map(name, m):
print("static const int gs_{}[{}] = {{".format(name, len(m)))
print(*m, sep=',')
print("};")
print("inline int {0}(int a) {{ if(a < 0 || a >= {1}) return -1; return gs_{0}[a]; }}".format(name, len(m)))
def main():
guard = "GAME_GENERATED_PROTOCOLGLUE"
print("#ifndef " + guard)
print("#define " + guard)
msgs = get_msgs()
msgs7 = get_msgs_7()
output_map("Msg_SixToSeven", generate_map(msgs, msgs7))
output_map("Msg_SevenToSix", generate_map(msgs7, msgs))
objs = get_objs()
objs7 = get_objs_7()
output_map("Obj_SixToSeven", generate_map(objs, objs7))
output_map("Obj_SevenToSix", generate_map(objs7, objs))
print("#endif //" + guard)
if __name__ == "__main__":
main()