set_node_number
This commit is contained in:
Laura Klünder
parent
8c8ef12cf0
commit
732c6f8449
1 changed files with 179 additions and 104 deletions
|
|
@ -1,116 +1,191 @@
|
||||||
from dataclasses import fields
|
|
||||||
|
|
||||||
from django.core.management.base import BaseCommand
|
from django.core.management.base import BaseCommand
|
||||||
|
from django.utils.translation import gettext_lazy as _
|
||||||
|
|
||||||
from c3nav.mesh.baseformats import StructType, normalize_name
|
from c3nav.mesh.models import MeshNode
|
||||||
from c3nav.mesh.messages import MeshMessage
|
|
||||||
from c3nav.mesh.utils import indent_c
|
names = {
|
||||||
|
1: "Alpheratz",
|
||||||
|
2: "Ankaa",
|
||||||
|
3: "Schedar",
|
||||||
|
4: "Diphda",
|
||||||
|
5: "Achernar",
|
||||||
|
6: "Hamal",
|
||||||
|
7: "Acamar",
|
||||||
|
8: "Menkar",
|
||||||
|
9: "Mirfak",
|
||||||
|
10: "Aldebaran",
|
||||||
|
11: "Rigel",
|
||||||
|
12: "Capella",
|
||||||
|
13: "Bellatrix",
|
||||||
|
14: "Elnath",
|
||||||
|
15: "Alnilam",
|
||||||
|
16: "Betelgeuse",
|
||||||
|
17: "Canopus",
|
||||||
|
18: "Sirius",
|
||||||
|
19: "Adhara",
|
||||||
|
20: "Procyon",
|
||||||
|
21: "Pollux",
|
||||||
|
22: "Avior",
|
||||||
|
23: "Suhail",
|
||||||
|
24: "Miaplacidus",
|
||||||
|
25: "Alphard",
|
||||||
|
26: "Regulus",
|
||||||
|
27: "Dubhe",
|
||||||
|
28: "Denebola",
|
||||||
|
29: "Gienah",
|
||||||
|
30: "Acrux",
|
||||||
|
31: "Gacrux",
|
||||||
|
32: "Alioth",
|
||||||
|
33: "Spica",
|
||||||
|
34: "Alkaid",
|
||||||
|
35: "Hadar",
|
||||||
|
36: "Menkent",
|
||||||
|
37: "Arcturus",
|
||||||
|
38: "Rigil Kentaurus",
|
||||||
|
39: "Zubenelgenubi",
|
||||||
|
40: "Kochab",
|
||||||
|
41: "Alphecca",
|
||||||
|
42: "Antares",
|
||||||
|
43: "Atria",
|
||||||
|
44: "Sabik",
|
||||||
|
45: "Shaula",
|
||||||
|
46: "Rasalgethi",
|
||||||
|
47: "Eltanin",
|
||||||
|
48: "Kaus Australis",
|
||||||
|
49: "Vega",
|
||||||
|
50: "Nunki",
|
||||||
|
51: "Altair",
|
||||||
|
52: "Peacock",
|
||||||
|
53: "Deneb",
|
||||||
|
54: "Enif",
|
||||||
|
55: "Alnair",
|
||||||
|
56: "Fomalhaut",
|
||||||
|
57: "Markab",
|
||||||
|
58: "Mimosa",
|
||||||
|
59: "Toliman",
|
||||||
|
60: "Alnitak",
|
||||||
|
61: "Wezen",
|
||||||
|
62: "Sargas",
|
||||||
|
63: "Menkalinan",
|
||||||
|
64: "Alhena",
|
||||||
|
65: "Polaris",
|
||||||
|
66: "Castor",
|
||||||
|
67: "Mirzam",
|
||||||
|
68: "Alsephina",
|
||||||
|
69: "Saiph",
|
||||||
|
70: "Rasalhague",
|
||||||
|
71: "Algol",
|
||||||
|
72: "Almach",
|
||||||
|
73: "Tiaki",
|
||||||
|
74: "Aspidiske",
|
||||||
|
75: "Naos",
|
||||||
|
76: "Mizar",
|
||||||
|
77: "Sadr",
|
||||||
|
78: "Mintaka",
|
||||||
|
79: "Caph",
|
||||||
|
80: "Dschubba",
|
||||||
|
81: "Larawag",
|
||||||
|
82: "Merak",
|
||||||
|
83: "Izar",
|
||||||
|
84: "Phecda",
|
||||||
|
85: "Scheat",
|
||||||
|
86: "Alderamin",
|
||||||
|
87: "Aludra",
|
||||||
|
88: "Acrab",
|
||||||
|
89: "Markeb",
|
||||||
|
90: "Zosma",
|
||||||
|
91: "Arneb",
|
||||||
|
92: "Ascella",
|
||||||
|
93: "Algieba",
|
||||||
|
94: "Zubeneschamali",
|
||||||
|
95: "Unukalhai",
|
||||||
|
96: "Sheratan",
|
||||||
|
97: "Kraz",
|
||||||
|
98: "Mahasim",
|
||||||
|
99: "Phact",
|
||||||
|
100: "Ruchbah",
|
||||||
|
101: "Muphrid",
|
||||||
|
102: "Hassaleh",
|
||||||
|
103: "Lesath",
|
||||||
|
104: "Kaus Media",
|
||||||
|
105: "Tarazed",
|
||||||
|
106: "Athebyne",
|
||||||
|
107: "Yed Prior",
|
||||||
|
108: "Porrima",
|
||||||
|
109: "Imai",
|
||||||
|
110: "Zubenelhakrabi",
|
||||||
|
111: "Cebalrai",
|
||||||
|
112: "Cursa",
|
||||||
|
113: "Kornephoros",
|
||||||
|
114: "Rastaban",
|
||||||
|
115: "Hatysa",
|
||||||
|
116: "Nihal",
|
||||||
|
117: "Paikauhale",
|
||||||
|
118: "Kaus Borealis",
|
||||||
|
119: "Algenib",
|
||||||
|
120: "Tureis",
|
||||||
|
121: "Alcyone",
|
||||||
|
122: "Deneb Algedi",
|
||||||
|
123: "Vindemiatrix",
|
||||||
|
124: "Tejat",
|
||||||
|
125: "Albaldah",
|
||||||
|
126: "Cor Caroli",
|
||||||
|
127: "Fang",
|
||||||
|
128: "Gomeisa",
|
||||||
|
129: "Fawaris",
|
||||||
|
130: "Alniyat",
|
||||||
|
131: "Sadalsuud",
|
||||||
|
132: "Matar",
|
||||||
|
133: "Algorab",
|
||||||
|
134: "Sadalmelik",
|
||||||
|
135: "Tianguan",
|
||||||
|
136: "Zaurak",
|
||||||
|
137: "Alnasl",
|
||||||
|
138: "Okab",
|
||||||
|
139: "Aldhanab",
|
||||||
|
140: "Pherkad",
|
||||||
|
141: "Xamidimura",
|
||||||
|
142: "Furud",
|
||||||
|
143: "Almaaz",
|
||||||
|
144: "Seginus",
|
||||||
|
145: "Albireo",
|
||||||
|
146: "Dabih",
|
||||||
|
147: "Mebsuta",
|
||||||
|
148: "Tania Australis",
|
||||||
|
149: "Altais",
|
||||||
|
150: "Sarin",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
class Command(BaseCommand):
|
class Command(BaseCommand):
|
||||||
help = 'export mesh message structs for c code'
|
help = 'set node number of non-named node'
|
||||||
|
|
||||||
|
def add_arguments(self, parser):
|
||||||
|
parser.add_argument('number', type=int, help=_('number to set'))
|
||||||
|
|
||||||
def handle(self, *args, **options):
|
def handle(self, *args, **options):
|
||||||
done_struct_names = set()
|
number = options["number"]
|
||||||
nodata = set()
|
try:
|
||||||
struct_lines = {}
|
name = names[options["number"]]
|
||||||
struct_sizes = []
|
except KeyError:
|
||||||
struct_max_sizes = []
|
print('number without name')
|
||||||
done_definitions = set()
|
return
|
||||||
|
|
||||||
for include in StructType.c_includes:
|
if MeshNode.objects.filter(name__startswith=f"{number} – ").exists():
|
||||||
print(f'#include {include}')
|
print('number is already taken')
|
||||||
|
return
|
||||||
|
|
||||||
ignore_names = set(field_.name for field_ in fields(MeshMessage))
|
unnamed_nodes = list(MeshNode.objects.filter(name__isnull=True))
|
||||||
for msg_type, msg_class in MeshMessage.get_types().items():
|
|
||||||
if msg_class.c_struct_name:
|
|
||||||
if msg_class.c_struct_name in done_struct_names:
|
|
||||||
continue
|
|
||||||
done_struct_names.add(msg_class.c_struct_name)
|
|
||||||
if MeshMessage.c_structs[msg_class.c_struct_name] != msg_class:
|
|
||||||
# the purpose of MeshMessage.c_structs is unclear, currently this never triggers
|
|
||||||
# todo get rid of the whole c_structs thing if it doesn't turn out to be useful for anything
|
|
||||||
raise ValueError('what happened?')
|
|
||||||
|
|
||||||
base_name = (msg_class.c_struct_name or normalize_name(
|
if not unnamed_nodes:
|
||||||
getattr(msg_type, 'name', msg_class.__name__)
|
print('no unnamed nodes')
|
||||||
))
|
return
|
||||||
name = "mesh_msg_%s_t" % base_name
|
|
||||||
|
|
||||||
for definition_name, definition in msg_class.get_c_definitions().items():
|
if len(unnamed_nodes) > 1:
|
||||||
if definition_name not in done_definitions:
|
print('more than on unnamed node')
|
||||||
done_definitions.add(definition_name)
|
return
|
||||||
print(definition)
|
|
||||||
print()
|
|
||||||
|
|
||||||
code = msg_class.get_c_code(name, ignore_fields=ignore_names, no_empty=True)
|
node = unnamed_nodes[0]
|
||||||
if code:
|
node.name = f"{number} – {name}"
|
||||||
size = msg_class.get_size(no_inherited_fields=True, calculate_max=False)
|
node.save()
|
||||||
max_size = msg_class.get_size(no_inherited_fields=True, calculate_max=True)
|
print('done')
|
||||||
struct_lines[base_name] = "%s %s;" % (name, base_name.replace('_announce', ''))
|
|
||||||
struct_sizes.append(size)
|
|
||||||
struct_max_sizes.append(max_size)
|
|
||||||
print(code)
|
|
||||||
print("static_assert(sizeof(%s) == %d, \"size of generated message structs is calculated wrong\");" %
|
|
||||||
(name, size))
|
|
||||||
print()
|
|
||||||
else:
|
|
||||||
nodata.add(msg_class)
|
|
||||||
|
|
||||||
print("/** union between all message data structs */")
|
|
||||||
print("typedef union __packed {")
|
|
||||||
for line in struct_lines.values():
|
|
||||||
print(indent_c(line))
|
|
||||||
print("} mesh_msg_data_t; ")
|
|
||||||
print(
|
|
||||||
"static_assert(sizeof(mesh_msg_data_t) == %d, \"size of generated message structs is calculated wrong\");"
|
|
||||||
% max(struct_sizes)
|
|
||||||
)
|
|
||||||
|
|
||||||
print()
|
|
||||||
print('#define MESH_MSG_MAX_LENGTH (%d)' % max(struct_max_sizes))
|
|
||||||
|
|
||||||
print()
|
|
||||||
|
|
||||||
max_msg_type = max(MeshMessage.get_types().keys())
|
|
||||||
macro_data = []
|
|
||||||
for i in range(((max_msg_type//16)+1)*16):
|
|
||||||
msg_class = MeshMessage.get_types().get(i, None)
|
|
||||||
if msg_class:
|
|
||||||
name = (msg_class.c_struct_name or normalize_name(
|
|
||||||
getattr(msg_class.msg_type, 'name', msg_class.__name__)
|
|
||||||
))
|
|
||||||
macro_data.append((
|
|
||||||
msg_class.get_c_enum_name(),
|
|
||||||
("nodata" if msg_class in nodata else name),
|
|
||||||
msg_class.get_var_num(),
|
|
||||||
msg_class.get_size(no_inherited_fields=True, calculate_max=True),
|
|
||||||
msg_class.__doc__.strip(),
|
|
||||||
))
|
|
||||||
else:
|
|
||||||
macro_data.append((
|
|
||||||
"RESERVED_%02X" % i,
|
|
||||||
"nodata",
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
"",
|
|
||||||
))
|
|
||||||
|
|
||||||
max0 = max(len(d[0]) for d in macro_data)
|
|
||||||
max1 = max(len(d[1]) for d in macro_data)
|
|
||||||
max2 = max(len(str(d[2])) for d in macro_data)
|
|
||||||
max3 = max(len(str(d[3])) for d in macro_data)
|
|
||||||
lines = []
|
|
||||||
for i, (macro_name, struct_name, num_len, max_len, comment) in enumerate(macro_data):
|
|
||||||
lines.append(indent_c(
|
|
||||||
"FN(%s %s %s %s) /** 0x%02X %s*/" % (
|
|
||||||
f'{macro_name},'.ljust(max0+1),
|
|
||||||
f'{struct_name},'.ljust(max1+1),
|
|
||||||
f'{num_len},'.rjust(max2+1),
|
|
||||||
f'{max_len}'.rjust(max3),
|
|
||||||
i,
|
|
||||||
comment+(" " if comment else ""),
|
|
||||||
)
|
|
||||||
))
|
|
||||||
print("#define FOR_ALL_MESH_MSG_TYPES(FN) \\")
|
|
||||||
print(" \\\n".join(lines))
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue