Changeset fbcee4f

Timestamp:

Apr 25, 2016, 12:35:53 AM (8 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

c4b2183

Parents:

c6388f4

Message:

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

Location:

python/lib

Files:

• gen_code.py (modified) (2 diffs)
• gen_external.py (modified) (2 diffs)

python/lib/gen_code.py

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

python/lib/gen_external.py

@@ -67 +67 @@
     return cpp_output, cpp_objects
 
-def generate_external(output_path):
-    os.mkdir(output_path)
+def generate_external(output_path, usedouble = False):
+    if not os.path.isdir(output_path): os.mkdir(output_path)
     sources_list = []
     cpp_output, cpp_objects = get_cpp_objects()
@@ -124 +124 @@
     for o in lib:
         out = header
-        mapped = MappedObject(lib[o])
+        mapped = MappedObject(lib[o], usedouble = usedouble)
         out += mapped.gen_code()
         output_file = os.path.join(output_path, 'gen-%s.c' % o)