python/lib/gen_{external,code}.py: prepare for double precision

Paul Brossier committed 10 years ago

Commit fbcee4f1 parent c6388f42 files · +7 −5Browse files

Showing the changes to python (2 of 2 files). Show all

Unified · Split

2 files

         }
 pyargparse_chars = {
-        'smpl_t': 'f',
+        'smpl_t': 'f', # if not usedouble else 'd',
         'uint_t': 'I',
         'sint_t': 'I',
         'char_t*': 's',
 class MappedObject(object):
-    def __init__(self, prototypes):
+    def __init__(self, prototypes, usedouble = False):
+        if usedouble:
+            pyargparse_chars['smpl_t'] = 'd'
         self.prototypes = prototypes
         self.shortname = prototypes['shortname']

@@ -66,8 +66,8 @@ def get_cpp_objects(): · 65 unmodified lines
66		66
67	return cpp_output, cpp_objects	67	return cpp_output, cpp_objects
68		68
69	def generate_external(output_path):	69	def generate_external(output_path, usedouble = False):
70	os.mkdir(output_path)	70	if not os.path.isdir(output_path): os.mkdir(output_path)
71	sources_list = []	71	sources_list = []
72	cpp_output, cpp_objects = get_cpp_objects()	72	cpp_output, cpp_objects = get_cpp_objects()
73	lib = {}	73	lib = {}
@@ -123,7 +123,7 @@ def generate_external(output_path): · 49 unmodified lines
123	from .gen_code import MappedObject	123	from .gen_code import MappedObject
124	for o in lib:	124	for o in lib:
125	out = header	125	out = header
126	mapped = MappedObject(lib[o])	126	mapped = MappedObject(lib[o], usedouble = usedouble)
127	out += mapped.gen_code()	127	out += mapped.gen_code()
128	output_file = os.path.join(output_path, 'gen-%s.c' % o)	128	output_file = os.path.join(output_path, 'gen-%s.c' % o)
129	with open(output_file, 'w') as f:	129	with open(output_file, 'w') as f:

Keyboard shortcuts