newelement.py

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'))