Mercurial > hg > orthanc-stone
annotate Resources/CodeGeneration/stonegentool.py @ 494:fc17251477d6 bgo-commands-codegen
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
author | bgo-osimis |
---|---|
date | Sat, 23 Feb 2019 10:18:13 +0100 |
parents | 6fbf2eae7c88 |
children | 8b6ceae45ba0 |
rev | line source |
---|---|
471 | 1 import json |
491 | 2 import yaml |
471 | 3 import re |
494
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
4 import os |
473 | 5 import sys |
491 | 6 from jinja2 import Template |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
7 from typing import ( |
489 | 8 Any, |
9 Dict, | |
10 Generator, | |
11 Iterable, | |
12 Iterator, | |
13 List, | |
14 Match, | |
15 Optional, | |
16 Tuple, | |
17 Union, | |
18 cast, | |
19 ) | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
20 from io import StringIO |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
21 import time |
469 | 22 |
472 | 23 """ |
24 1 2 3 4 5 6 7 | |
25 12345678901234567890123456789012345678901234567890123456789012345678901234567890 | |
26 """ | |
27 | |
491 | 28 # see https://stackoverflow.com/a/2504457/2927708 |
29 def trim(docstring): | |
30 if not docstring: | |
31 return '' | |
32 # Convert tabs to spaces (following the normal Python rules) | |
33 # and split into a list of lines: | |
34 lines = docstring.expandtabs().splitlines() | |
35 # Determine minimum indentation (first line doesn't count): | |
36 indent = sys.maxsize | |
37 for line in lines[1:]: | |
38 stripped = line.lstrip() | |
39 if stripped: | |
40 indent = min(indent, len(line) - len(stripped)) | |
41 # Remove indentation (first line is special): | |
42 trimmed = [lines[0].strip()] | |
43 if indent < sys.maxsize: | |
44 for line in lines[1:]: | |
45 trimmed.append(line[indent:].rstrip()) | |
46 # Strip off trailing and leading blank lines: | |
47 while trimmed and not trimmed[-1]: | |
48 trimmed.pop() | |
49 while trimmed and not trimmed[0]: | |
50 trimmed.pop(0) | |
51 # Return a single string: | |
52 return '\n'.join(trimmed) | |
53 | |
54 | |
55 class GenCode: | |
489 | 56 def __init__(self): |
57 | |
58 # file-wide preamble (#include directives, comment...) | |
59 self.cppPreamble = StringIO() | |
60 | |
61 self.cppEnums = StringIO() | |
62 self.cppStructs = StringIO() | |
63 self.cppDispatcher = StringIO() | |
64 self.cppHandler = StringIO() | |
473 | 65 |
489 | 66 # file-wide preamble (module directives, comment...) |
67 self.tsPreamble = StringIO() | |
68 | |
69 self.tsEnums = StringIO() | |
70 self.tsStructs = StringIO() | |
71 self.tsDispatcher = StringIO() | |
72 self.tsHandler = StringIO() | |
486
8e40355a172b
Unit tests OK for preambles, enums and structs in both TS and C++
bgo-osimis
parents:
485
diff
changeset
|
73 |
489 | 74 def FlattenToFiles(self, outputDir: str): |
75 raise NotImplementedError() | |
76 | |
77 | |
473 | 78 class JsonHelpers: |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
79 """A set of utilities to perform JSON operations""" |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
80 |
473 | 81 @staticmethod |
82 def removeCommentsFromJsonContent(string): | |
83 """ | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
84 Remove comments from a JSON file |
473 | 85 |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
86 Comments are not allowed in JSON but, i.e., Orthanc configuration files |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
87 contains C++ like comments that we need to remove before python can |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
88 parse the file |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
89 """ |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
90 # remove all occurrence streamed comments (/*COMMENT */) from string |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
91 string = re.sub(re.compile("/\*.*?\*/", re.DOTALL), "", string) |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
92 |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
93 # remove all occurrence singleline comments (//COMMENT\n ) from string |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
94 string = re.sub(re.compile("//.*?\n"), "", string) |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
95 |
473 | 96 return string |
97 | |
98 @staticmethod | |
99 def loadJsonWithComments(path): | |
100 """ | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
101 Reads a JSON file that may contain C++ like comments |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
102 """ |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
103 with open(path, "r") as fp: |
473 | 104 fileContent = fp.read() |
105 fileContent = JsonHelpers.removeCommentsFromJsonContent(fileContent) | |
106 return json.loads(fileContent) | |
107 | |
108 | |
490 | 109 def LoadSchemaFromJson(filePath: str): |
473 | 110 return JsonHelpers.loadJsonWithComments(filePath) |
469 | 111 |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
112 def CanonToCpp(canonicalTypename: str) -> str: |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
113 # C++: prefix map vector and string with std::map, std::vector and |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
114 # std::string |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
115 # replace int32 by int32_t |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
116 # replace float32 by float |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
117 # replace float64 by double |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
118 retVal: str = canonicalTypename |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
119 retVal = retVal.replace("map", "std::map") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
120 retVal = retVal.replace("vector", "std::vector") |
494
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
121 retVal = retVal.replace("string", "std::string") |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
122 retVal = retVal.replace("int32", "int32_t") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
123 retVal = retVal.replace("float32", "float") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
124 retVal = retVal.replace("float64", "double") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
125 return retVal |
470 | 126 |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
127 def CanonToTs(canonicalTypename: str) -> str: |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
128 # TS: replace vector with Array and map with Map |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
129 # string remains string |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
130 # replace int32 by number |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
131 # replace float32 by number |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
132 # replace float64 by number |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
133 retVal: str = canonicalTypename |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
134 retVal = retVal.replace("map", "Map") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
135 retVal = retVal.replace("vector", "Array") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
136 retVal = retVal.replace("int32", "number") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
137 retVal = retVal.replace("float32", "number") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
138 retVal = retVal.replace("float64", "number") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
139 retVal = retVal.replace("bool", "boolean") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
140 return retVal |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
141 |
494
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
142 def NeedsTsConstruction(enums: Dict, tsType: str): |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
143 if tsType == 'boolean': |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
144 return False |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
145 elif tsType == 'number': |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
146 return False |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
147 elif tsType == 'string': |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
148 return False |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
149 else: |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
150 enumNames = [] |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
151 for enum in enums: |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
152 enumNames.append(enum['name']) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
153 if tsType in enumNames: |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
154 return False |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
155 return True |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
156 |
494
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
157 def NeedsCppConstruction(canonTypename): |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
158 return False |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
159 |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
160 def RegisterTemplateFunction(template,func): |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
161 """Makes a function callable by a jinja2 template""" |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
162 template.globals[func.__name__] = func |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
163 return func |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
164 |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
165 def MakeTemplate(templateStr): |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
166 template = Template(templateStr) |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
167 RegisterTemplateFunction(template,CanonToCpp) |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
168 RegisterTemplateFunction(template,CanonToTs) |
494
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
169 RegisterTemplateFunction(template,NeedsTsConstruction) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
170 RegisterTemplateFunction(template,NeedsCppConstruction) |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
171 return template |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
172 |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
173 def MakeTemplateFromFile(templateFileName): |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
174 templateFile = open(templateFileName, "r") |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
175 templateFileContents = templateFile.read() |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
176 return MakeTemplate(templateFileContents) |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
177 templateFile.close() |
469 | 178 |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
179 def EatToken(sentence: str) -> Tuple[str, str]: |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
180 """splits "A,B,C" into "A" and "B,C" where A, B and C are type names |
471 | 181 (including templates) like "int32", "TotoTutu", or |
182 "map<map<int32,vector<string>>,map<string,int32>>" """ | |
472 | 183 |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
184 if sentence.count("<") != sentence.count(">"): |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
185 raise Exception( |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
186 f"Error in the partial template type list {sentence}." |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
187 + " The number of < and > do not match!" |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
188 ) |
471 | 189 |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
190 # the template level we're currently in |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
191 templateLevel = 0 |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
192 for i in range(len(sentence)): |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
193 if (sentence[i] == ",") and (templateLevel == 0): |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
194 return (sentence[0:i], sentence[i + 1 :]) |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
195 elif sentence[i] == "<": |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
196 templateLevel += 1 |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
197 elif sentence[i] == ">": |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
198 templateLevel -= 1 |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
199 return (sentence, "") |
471 | 200 |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
201 |
491 | 202 def SplitListOfTypes(typename: str) -> List[str]: |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
203 """Splits something like |
471 | 204 vector<string>,int32,map<string,map<string,int32>> |
205 in: | |
206 - vector<string> | |
207 - int32 | |
208 map<string,map<string,int32>> | |
209 | |
210 This is not possible with a regex so | |
211 """ | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
212 stillStuffToEat: bool = True |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
213 tokenList = [] |
491 | 214 restOfString = typename |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
215 while stillStuffToEat: |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
216 firstToken, restOfString = EatToken(restOfString) |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
217 tokenList.append(firstToken) |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
218 if restOfString == "": |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
219 stillStuffToEat = False |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
220 return tokenList |
471 | 221 |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
222 |
490 | 223 templateRegex = \ |
224 re.compile(r"([a-zA-Z0-9_]*[a-zA-Z0-9_]*)<([a-zA-Z0-9_,:<>]+)>") | |
472 | 225 |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
226 |
491 | 227 def ParseTemplateType(typename) -> Tuple[bool, str, List[str]]: |
228 """ If the type is a template like "SOMETHING<SOME<THING,EL<SE>>>", | |
229 then it returns (true,"SOMETHING","SOME<THING,EL<SE>>") | |
470 | 230 otherwise it returns (false,"","")""" |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
231 |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
232 # let's remove all whitespace from the type |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
233 # split without argument uses any whitespace string as separator |
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
234 # (space, tab, newline, return or formfeed) |
491 | 235 typename = "".join(typename.split()) |
236 matches = templateRegex.match(typename) | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
237 if matches == None: |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
238 return (False, "", []) |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
239 else: |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
240 m = cast(Match[str], matches) |
489 | 241 assert len(m.groups()) == 2 |
242 # we need to split with the commas that are outside of the | |
243 # defined types. Simply splitting at commas won't work | |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
244 listOfDependentTypes = SplitListOfTypes(m.group(2)) |
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
245 return (True, m.group(1), listOfDependentTypes) |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
246 |
491 | 247 |
248 def ComputeOrderFromTypeTree( | |
249 ancestors: List[str], | |
250 genOrder: List[str], | |
251 shortTypename: str, schema: Dict[str, Dict]) -> None: | |
252 | |
253 if shortTypename in ancestors: | |
489 | 254 raise Exception( |
255 f"Cyclic dependency chain found: the last of {ancestors} " | |
491 | 256 + f"depends on {shortTypename} that is already in the list." |
489 | 257 ) |
258 | |
491 | 259 if not (shortTypename in genOrder): |
260 (isTemplate, _, dependentTypenames) = ParseTemplateType(shortTypename) | |
489 | 261 if isTemplate: |
491 | 262 # if it is a template, it HAS dependent types... They can be |
263 # anything (primitive, collection, enum, structs..). | |
264 # Let's process them! | |
265 for dependentTypename in dependentTypenames: | |
489 | 266 # childAncestors = ancestors.copy() NO TEMPLATE ANCESTOR!!! |
491 | 267 # childAncestors.append(typename) |
268 ComputeOrderFromTypeTree( | |
269 ancestors, genOrder, dependentTypename, schema | |
489 | 270 ) |
271 else: | |
491 | 272 # If it is not template, we are only interested if it is a |
273 # dependency that we must take into account in the dep graph, | |
274 # i.e., a struct. | |
275 if IsShortStructType(shortTypename, schema): | |
276 struct:Dict = schema[GetLongTypename(shortTypename, schema)] | |
277 # The keys in the struct dict are the member names | |
278 # The values in the struct dict are the member types | |
279 for field in struct.keys(): | |
280 # we fill the chain of dependent types (starting here) | |
281 ancestors.append(shortTypename) | |
282 ComputeOrderFromTypeTree( | |
283 ancestors, genOrder, struct[field], schema) | |
284 # don't forget to restore it! | |
285 ancestors.pop() | |
286 | |
287 # now we're pretty sure our dependencies have been processed, | |
288 # we can start marking our code for generation (it might | |
289 # already have been done if someone referenced us earlier) | |
290 if not shortTypename in genOrder: | |
291 genOrder.append(shortTypename) | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
292 |
491 | 293 # +-----------------------+ |
294 # | Utility functions | | |
295 # +-----------------------+ | |
296 | |
297 def IsShortStructType(typename: str, schema: Dict[str, Dict]) -> bool: | |
298 fullStructName = "struct " + typename | |
299 return (fullStructName in schema) | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
300 |
491 | 301 def GetLongTypename(shortTypename: str, schema: Dict): |
302 if shortTypename.startswith("enum "): | |
303 raise RuntimeError('shortTypename.startswith("enum "):') | |
304 enumName: str = "enum " + shortTypename | |
305 isEnum = enumName in schema | |
490 | 306 |
491 | 307 if shortTypename.startswith("struct "): |
308 raise RuntimeError('shortTypename.startswith("struct "):') | |
309 structName: str = "struct " + shortTypename | |
310 isStruct = ("struct " + shortTypename) in schema | |
474 | 311 |
491 | 312 if isEnum and isStruct: |
313 raise RuntimeError('Enums and structs cannot have the same name') | |
474 | 314 |
491 | 315 if isEnum: |
316 return enumName | |
317 if isStruct: | |
318 return structName | |
319 | |
320 def IsTypename(fullName: str) -> bool: | |
321 return (fullName.startswith("enum ") or fullName.startswith("struct ")) | |
322 | |
323 def IsEnumType(fullName: str) -> bool: | |
324 return fullName.startswith("enum ") | |
490 | 325 |
491 | 326 def IsStructType(fullName: str) -> bool: |
327 return fullName.startswith("struct ") | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
328 |
491 | 329 def GetShortTypename(fullTypename: str) -> str: |
330 if fullTypename.startswith("struct "): | |
331 return fullTypename[7:] | |
332 elif fullTypename.startswith("enum"): | |
333 return fullTypename[5:] | |
334 else: | |
335 raise RuntimeError \ | |
336 ('fullTypename should start with either "struct " or "enum "') | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
337 |
491 | 338 def CheckSchemaSchema(schema: Dict) -> None: |
339 if not "rootName" in schema: | |
340 raise Exception("schema lacks the 'rootName' key") | |
341 for name in schema.keys(): | |
342 if (not IsEnumType(name)) and (not IsStructType(name)) and \ | |
343 (name != 'rootName'): | |
344 raise RuntimeError \ | |
345 (f'Type "{name}" should start with "enum " or "struct "') | |
346 | |
347 # TODO: check enum fields are unique (in whole namespace) | |
348 # TODO: check struct fields are unique (in each struct) | |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
349 # TODO: check that in the source schema, there are spaces after each colon |
491 | 350 |
351 # +-----------------------+ | |
352 # | Main processing logic | | |
353 # +-----------------------+ | |
354 | |
355 def ComputeRequiredDeclarationOrder(schema: dict) -> List[str]: | |
356 # sanity check | |
357 CheckSchemaSchema(schema) | |
474 | 358 |
491 | 359 # we traverse the type dependency graph and we fill a queue with |
360 # the required struct types, in a bottom-up fashion, to compute | |
361 # the declaration order | |
362 # The genOrder list contains the struct full names in the order | |
363 # where they must be defined. | |
364 # We do not care about the enums here... They do not depend upon | |
365 # anything and we'll handle them, in their original declaration | |
366 # order, at the start | |
367 genOrder: List = [] | |
368 for fullName in schema.keys(): | |
369 if IsStructType(fullName): | |
370 realName: str = GetShortTypename(fullName) | |
371 ancestors: List[str] = [] | |
372 ComputeOrderFromTypeTree(ancestors, genOrder, realName, schema) | |
373 return genOrder | |
489 | 374 |
491 | 375 def ProcessSchema(schema: dict, genOrder: List[str]) -> Dict: |
376 # sanity check | |
377 CheckSchemaSchema(schema) | |
472 | 378 |
491 | 379 # let's doctor the schema to clean it up a bit |
380 # order DOES NOT matter for enums, even though it's a list | |
381 enums: List[Dict] = [] | |
382 for fullName in schema.keys(): | |
383 if IsEnumType(fullName): | |
384 # convert "enum Toto" to "Toto" | |
385 typename:str = GetShortTypename(fullName) | |
386 enum = {} | |
387 enum['name'] = typename | |
388 assert(type(schema[fullName]) == list) | |
389 enum['fields'] = schema[fullName] # must be a list | |
390 enums.append(enum) | |
489 | 391 |
491 | 392 # now that the order has been established, we actually store\ |
393 # the structs in the correct order | |
394 # the structs are like: | |
395 # example = [ | |
396 # { | |
397 # "name": "Message1", | |
398 # "fields": { | |
399 # "someMember":"int32", | |
400 # "someOtherMember":"vector<string>" | |
401 # } | |
402 # }, | |
403 # { | |
404 # "name": "Message2", | |
405 # "fields": { | |
406 # "someMember":"int32", | |
407 # "someOtherMember22":"vector<Message1>" | |
408 # } | |
409 # } | |
410 # ] | |
489 | 411 |
491 | 412 structs: List[Dict] = [] |
413 for i in range(len(genOrder)): | |
414 # this is already the short name | |
415 typename = genOrder[i] | |
416 fieldDict = schema["struct " + typename] | |
417 struct = {} | |
418 struct['name'] = typename | |
419 struct['fields'] = fieldDict | |
420 structs.append(struct) | |
489 | 421 |
491 | 422 templatingDict = {} |
423 templatingDict['enums'] = enums | |
424 templatingDict['structs'] = structs | |
425 templatingDict['rootName'] = schema['rootName'] | |
489 | 426 |
491 | 427 return templatingDict |
489 | 428 |
491 | 429 # +-----------------------+ |
430 # | Write to files | | |
431 # +-----------------------+ | |
474 | 432 |
491 | 433 # def WriteStreamsToFiles(rootName: str, genc: Dict[str, StringIO]) \ |
434 # -> None: | |
435 # pass | |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
436 |
491 | 437 def LoadSchema(fn): |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
438 # latin-1 is a trick, when we do NOT care about NON-ascii chars but |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
439 # we wish to avoid using a decoding error handler |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
440 # (see http://python-notes.curiousefficiency.org/en/latest/python3/text_file_processing.html#files-in-an-ascii-compatible-encoding-best-effort-is-acceptable) |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
441 # TL;DR: all 256 values are mapped to characters in latin-1 so the file |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
442 # contents never cause an error. |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
443 with open(fn, 'r', encoding='latin-1') as f: |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
444 schemaText = f.read() |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
445 assert(type(schemaText) == str) |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
446 # ensure there is a space after each colon. Otherwise, dicts could be |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
447 # erroneously recognized as an array of strings containing ':' |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
448 for i in range(len(schemaText)-1): |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
449 ch = schemaText[i] |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
450 nextCh = schemaText[i+1] |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
451 if ch == ':': |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
452 if not (nextCh == ' ' or nextCh == '\n'): |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
453 assert(False) |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
454 schema = yaml.load(schemaText) |
491 | 455 return schema |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
456 |
491 | 457 def GetTemplatingDictFromSchemaFilename(fn): |
458 obj = LoadSchema(fn) | |
459 genOrder: str = ComputeRequiredDeclarationOrder(obj) | |
460 templatingDict = ProcessSchema(obj, genOrder) | |
461 return templatingDict | |
470 | 462 |
491 | 463 # +-----------------------+ |
464 # | ENTRY POINT | | |
465 # +-----------------------+ | |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
466 |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
467 if __name__ == "__main__": |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
468 import argparse |
468 | 469 |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
470 parser = argparse.ArgumentParser( |
491 | 471 usage="""stonegentool.py [-h] [-o OUT_DIR] [-v] input_schema |
472 EXAMPLE: python stonegentool.py -o "generated_files/" """ | |
473 + """ "mainSchema.yaml,App Specific Commands.json" """ | |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
474 ) |
491 | 475 parser.add_argument("input_schema", type=str, \ |
476 help="path to the schema file") | |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
477 parser.add_argument( |
489 | 478 "-o", |
479 "--out_dir", | |
480 type=str, | |
481 default=".", | |
482 help="""path of the directory where the files | |
483 will be generated. Default is current | |
484 working folder""", | |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
485 ) |
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
486 parser.add_argument( |
489 | 487 "-v", |
488 "--verbosity", | |
489 action="count", | |
490 default=0, | |
491 help="""increase output verbosity (0 == errors | |
492 only, 1 == some verbosity, 2 == nerd | |
493 mode""", | |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
494 ) |
468 | 495 |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
496 args = parser.parse_args() |
494
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
497 schemaFile = args.input_schema |
485
772516adcbf6
Ongoing work on code generation. Enums and structs OK in ts and cpp
bgo-osimis
parents:
482
diff
changeset
|
498 outDir = args.out_dir |
468 | 499 |
494
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
500 tdico: Dict = GetTemplatingDictFromSchemaFilename(schemaFile) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
501 |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
502 tsTemplateFile = \ |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
503 os.path.join(os.path.dirname(__file__), 'template.in.ts') |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
504 template = MakeTemplateFromFile(tsTemplateFile) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
505 renderedTsCode: str = template.render(**tdico) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
506 outputTsFile = os.path.join( \ |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
507 outDir,f"{tdico['rootName']}_generated.ts") |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
508 with open(outputTsFile,"wt",encoding='utf8') as outFile: |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
509 outFile.write(renderedTsCode) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
510 |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
511 cppTemplateFile = \ |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
512 os.path.join(os.path.dirname(__file__), 'template.in.h') |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
513 template = MakeTemplateFromFile(cppTemplateFile) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
514 renderedCppCode: str = template.render(**tdico) |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
515 outputCppFile = os.path.join( \ |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
516 outDir,f"{tdico['rootName']}_generated.hpp") |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
517 with open(outputCppFile,"wt",encoding='utf8') as outFile: |
fc17251477d6
TS and CPP tests OK. Ongoing code for C++ program that reads list of serialized messages in N files. Requires conan
bgo-osimis
parents:
493
diff
changeset
|
518 outFile.write(renderedCppCode) |
491 | 519 |
520 | |
521 # def GenEnumDecl(genc: GenCode, fullName: str, schema: Dict) -> None: | |
522 # """Writes the enumerations in genc""" | |
523 # enumDict:Dict=schema[fullName] | |
524 # # jinja2 template | |
525 # j2cppEnum = Template(trim( | |
526 # """ {{fullName}} | |
527 # { | |
528 # {% for key in enumDict.keys()%} | |
529 # {{key}}, | |
530 # {%endfor%} | |
531 # }; | |
532 # """)) | |
533 # j2cppEnumR = j2cppEnum.render(locals()) | |
534 # genc.cppEnums.write(j2cppEnumR) | |
535 | |
536 # j2tsEnum = Template(trim( | |
537 # """ export {{fullName}} | |
538 # { | |
539 # {% for key in enumDict.keys()%} | |
540 # {{key}}, | |
541 # {%endfor%} | |
542 # }; | |
543 # """)) | |
544 # j2cppEnumR = j2cppEnum.render(locals()) | |
545 # genc.tsEnums.write(j2cppEnumR) | |
546 | |
547 | |
548 | |
549 # def GetSerializationCode(typename: str,valueName: str, tempName: str) | |
550 # if IsPrimitiveType(typename) or IsTemplateCollection(typename): | |
551 # # no need to write code for the primitive types or collections. | |
552 # # It is handled in C++ by the template functions and in TS by | |
553 # # the JSON.stringify code. | |
554 # elif IsStructType(typename): | |
555 # pass | |
556 | |
557 # def GenStructTypeDeclAndSerialize(genc: GenCode, type, schema) -> None: | |
558 # ###### | |
559 # # CPP | |
560 # ###### | |
561 # sampleCpp = """ struct Message1 | |
562 # { | |
563 # int32_t a; | |
564 # std::string b; | |
565 # EnumMonth0 c; | |
566 # bool d; | |
567 # }; | |
568 | |
569 # Json::Value StoneSerialize(const Message1& value) | |
570 # { | |
571 # Json::Value result(Json::objectValue); | |
572 # result["a"] = StoneSerialize(value.a); | |
573 # result["b"] = StoneSerialize(value.b); | |
574 # result["c"] = StoneSerialize(value.c); | |
575 # result["d"] = StoneSerialize(value.d); | |
576 # return result; | |
577 # } | |
578 # """ | |
579 | |
580 | |
581 # ###### | |
582 # # TS | |
583 # ###### | |
584 # sampleTs = """ | |
585 # { | |
586 # export class Message1 { | |
587 # a: number; | |
588 # b: string; | |
589 # c: EnumMonth0; | |
590 # d: boolean; | |
591 # public StoneSerialize(): string { | |
592 # let container: object = {}; | |
593 # container['type'] = 'Message1'; | |
594 # container['value'] = this; | |
595 # return JSON.stringify(container); | |
596 # } | |
597 # }; | |
598 # } | |
599 # """ | |
600 | |
601 | |
602 | |
603 | |
604 # tsText: StringIO = StringIO() | |
605 # cppText: StringIO = StringIO() | |
606 | |
607 # tsText.write("class %s\n" % typeDict["name"]) | |
608 # tsText.write("{\n") | |
609 | |
610 # cppText.write("struct %s\n" % typeDict["name"]) | |
611 # cppText.write("{\n") | |
612 | |
613 # """ | |
614 | |
615 # GenerateSerializationCode(typename,valueName) | |
482
f58fe38c8c04
Ongoing work on codegen: ts and cpp enum and struct writing seem to be OK. No file write yet
bgo-osimis
parents:
474
diff
changeset
|
616 |
491 | 617 # primitives: |
618 # ----------- | |
619 # int | |
620 # jsonValue val(objectInt); | |
621 # val.setValue("$name") | |
622 # parent.add(("$name",$name) | |
623 # double | |
624 # ... | |
625 # string | |
626 # ... | |
627 | |
628 # collections: | |
629 # ----------- | |
630 # dict { } | |
631 | |
632 # serializeValue() | |
633 # """ | |
634 | |
635 # for i in range(len(typeDict["fields"])): | |
636 # field = typeDict["fields"][i] | |
637 # name = field["name"] | |
638 # tsType = GetTypeScriptTypenameFromCanonical(field["type"]) | |
639 # tsText.write(" public %s %s;\n" % (tsType, name)) | |
640 # cppType = GetCppTypenameFromCanonical(field["type"]) | |
641 # cppText.write(" %s %s;\n" % (cppType, name)) | |
642 | |
643 # tsText.write("};\n\n") | |
644 # cppText.write("};\n\n") | |
645 | |
646 # genc.tsStructs.write(tsText.getvalue()) | |
647 # genc.cppStructs.write(cppText.getvalue()) | |
648 | |
649 | |
650 # def GenerateCodeFromTsTemplate(genc) | |
651 | |
652 | |
653 # +-----------------------+ | |
654 # | CODE GENERATION | | |
655 # +-----------------------+ | |
656 | |
657 # def GenPreambles(rootName: str, genc: GenCode) -> None: | |
658 # cppPreambleT = Template(trim( | |
659 # """// autogenerated by stonegentool on {{time.ctime()}} | |
660 # // for module {{rootName}} | |
661 # #include <cstdint> | |
662 # #include <string> | |
663 # #include <vector> | |
664 # #include <map> | |
665 # namespace {{rootName}} | |
666 # { | |
667 # Json::Value StoneSerialize(int32_t value) | |
668 # { | |
669 # Json::Value result(value); | |
670 # return result; | |
671 # } | |
672 # Json::Value StoneSerialize(double value) | |
673 # { | |
674 # Json::Value result(value); | |
675 # return result; | |
676 # } | |
677 # Json::Value StoneSerialize(bool value) | |
678 # { | |
679 # Json::Value result(value); | |
680 # return result; | |
681 # } | |
682 # Json::Value StoneSerialize(const std::string& value) | |
683 # { | |
684 # // the following is better than | |
685 # Json::Value result(value.data(),value.data()+value.size()); | |
686 # return result; | |
687 # } | |
688 # template<typename T> | |
689 # Json::Value StoneSerialize(const std::map<std::string,T>& value) | |
690 # { | |
691 # Json::Value result(Json::objectValue); | |
692 | |
693 # for (std::map<std::string, T>::const_iterator it = value.cbegin(); | |
694 # it != value.cend(); ++it) | |
695 # { | |
696 # // it->first it->second | |
697 # result[it->first] = StoneSerialize(it->second); | |
698 # } | |
699 # return result; | |
700 # } | |
701 # template<typename T> | |
702 # Json::Value StoneSerialize(const std::vector<T>& value) | |
703 # { | |
704 # Json::Value result(Json::arrayValue); | |
705 # for (size_t i = 0; i < value.size(); ++i) | |
706 # { | |
707 # result.append(StoneSerialize(value[i])); | |
708 # } | |
709 # return result; | |
710 # } | |
711 # """ | |
712 # cppPreambleR = cppPreambleT.render(locals()) | |
713 # genc.cppPreamble.write(cppPreambleR) | |
714 | |
715 # tsPreambleT = Template(trim( | |
716 # """// autogenerated by stonegentool on {{time.ctime()}} | |
717 # // for module {{rootName}} | |
718 | |
719 # namespace {{rootName}} | |
720 # { | |
721 # """ | |
722 # tsPreambleR = tsPreambleT.render(locals()) | |
723 # genc.tsPreamble.write(tsPreambleR) | |
724 | |
725 # def ComputeOrder_ProcessStruct( \ | |
726 # genOrder: List[str], name:str, schema: Dict[str, str]) -> None: | |
727 # # let's generate the code according to the | |
728 # struct = schema[name] | |
729 | |
730 # if not IsStructType(name): | |
731 # raise Exception(f'{typename} should start with "struct "') | |
493
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
732 |
6fbf2eae7c88
All unit tests pass for generation, including handler and dispatcher
bgo-osimis
parents:
491
diff
changeset
|
733 |