Skip to content
Permalink
Browse files

Add new element script

  • Loading branch information...
LBPHacker committed May 6, 2019
1 parent 532508d commit 74b0d49131a173fc87b9265dc3008d7c0bda932a
Showing with 64 additions and 0 deletions.
  1. +64 −0 newelement.py
  2. 0 vsproject.py
@@ -0,0 +1,64 @@
import sys
import re
import os

if len(sys.argv) != 2:
sys.exit('needs an element name as an argument')

name = sys.argv[1].upper()
path = 'src/simulation/elements/' + name + '.cpp'

if os.path.isfile(path):
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

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}()
{{
Identifier = "DEFAULT_PT_{0}";
Name = "{0}";
Colour = PIXPACK(0xFFFFFF);
MenuVisible = 1;
MenuSection = SC_SPECIAL;
Enabled = 1;
// element properties here
Update = &Element_{0}::update;
Graphics = &Element_{0}::graphics;
}}
//#TPT-Directive ElementHeader Element_{0} static int update(UPDATE_FUNC_ARGS)
int Element_{0}::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)
{{
// graphics code here
// return 1 if nothing dymanic happens here
return 0;
}}
Element_{0}::~Element_{0}() {{}}
""".format(name, str(max_id + 1)))
elem.close()

with open("generator.py") as f:
exec(compile(f.read(), "generator.py", 'exec'))
0 vsproject.py 100644 → 100755
No changes.

0 comments on commit 74b0d49

Please sign in to comment.
You can’t perform that action at this time.