Ditch element and tool classes

This commit is contained in:
Tamás Bálint Misius
2020-01-09 10:34:15 +01:00
parent 78203fc219
commit 4d52531889
223 changed files with 1939 additions and 2504 deletions

View File

@@ -1,7 +1,6 @@
#!/usr/bin/env python3
import sys
import re
import os
if len(sys.argv) != 2:
name = input('element name: ')
@@ -13,24 +12,30 @@ if re.search('[^A-Z0-9-]', name):
path = 'src/simulation/elements/' + name + '.cpp'
if os.path.isfile(path):
def get_elements():
elements = dict()
with open('src/simulation/ElementNumbers.h', 'r') as numbers:
for nm, pt in re.findall('ELEMENT_DEFINE\\s*\\(\\s*(\\S+)\\s*,\\s*(\\d+)\\s*\\)', numbers.read()):
elements[nm] = int(pt)
return elements
elements = get_elements()
if name in elements:
sys.exit('element already exists')
with open("generator.py") as f:
exec(compile(f.read(), "generator.py", 'exec'))
max_id = 0
with open('generated/ElementClasses.h', 'r') as classes:
for pt in re.findall('#define PT_\\S+ (\\d+)', classes.read()):
pt_id = int(pt)
if max_id < pt_id:
max_id = pt_id
for nm, pt in elements.items():
pt_id = int(pt)
if max_id < pt_id:
max_id = pt_id
new_id = max_id + 1
with open(path, 'w') as elem:
elem.write(r"""#include "simulation/ElementCommon.h"
//#TPT-Directive ElementClass Element_{0} PT_{0} {1}
Element_{0}::Element_{0}()
static int update(UPDATE_FUNC_ARGS);
static int graphics(GRAPHICS_FUNC_ARGS);
void Element::Element_{0}()
{{
Identifier = "DEFAULT_PT_{0}";
Name = "{0}";
@@ -41,20 +46,18 @@ Element_{0}::Element_{0}()
// element properties here
Update = &Element_{0}::update;
Graphics = &Element_{0}::graphics;
Update = &update;
Graphics = &graphics;
}}
//#TPT-Directive ElementHeader Element_{0} static int update(UPDATE_FUNC_ARGS)
int Element_{0}::update(UPDATE_FUNC_ARGS)
static int update(UPDATE_FUNC_ARGS)
{{
// update code here
return 0;
}}
//#TPT-Directive ElementHeader Element_{0} static int graphics(GRAPHICS_FUNC_ARGS)
int Element_{0}::graphics(GRAPHICS_FUNC_ARGS)
static int graphics(GRAPHICS_FUNC_ARGS)
{{
// graphics code here
// return 1 if nothing dymanic happens here
@@ -62,9 +65,13 @@ int Element_{0}::graphics(GRAPHICS_FUNC_ARGS)
return 0;
}}
Element_{0}::~Element_{0}() {{}}
""".format(name, str(max_id + 1)))
""".format(name))
elem.close()
with open("generator.py") as f:
exec(compile(f.read(), "generator.py", 'exec'))
print('element file \'{0}\' successfully created '.format(path))
input('now add \'ELEMENT_DEFINE({0}, {1});\' to \'src/simulation/ElementNumbers.h\', then press enter'.format(name, str(new_id)))
while True:
elements = get_elements()
if name in elements and elements[name] == new_id:
break
input('nope; try doing that again, then press enter')