Changeset 0bd0877 in sasmodels for sasmodels/py2c.py

Timestamp:

Jan 8, 2018 11:07:32 AM (7 years ago)

Author:

Omer Eisenberg <omereis@…>

Children:

ff431ca

Parents:

4339764

Message:

working main in translated python

File:

• sasmodels/py2c.py (modified) (3 diffs)

sasmodels/py2c.py

@@ -123 +123 @@
     generator = SourceGenerator(constants=constants, fname=fname, lineno=lineno)
     generator.visit(tree)
-    c_code = "\n".join(generator.c_proc)
+#    c_code = "\n".join(generator.c_proc)
     c_code = "".join(generator.c_proc)
     return (c_code, generator.warnings)
@@ -1280 +1280 @@
     return valid_params, fname_in,fname_out
 
+C_HEADER = """
+#include <stdio.h>
+#include <stdbool.h>
+#include <math.h>
+#define constant const
+double square(double x) { return x*x; }
+double cube(double x) { return x*x*x; }
+double polyval(constant double *coef, double x, int N)
+{
+    int i = 0;
+    double ans = coef[0];
+
+    while (i < N) {
+        ans = ans * x + coef[i++];
+    }
+
+    return ans;
+}
+"""
 def main():
     print("Parsing...using Python" + sys.version)
@@ -1293 +1312 @@
             .replace(name+'.n', 'GAUSS_N')
             .replace(name+'.z', 'GAUSS_Z')
-            .replace(name+'.w', 'GAUSS_W'))
-
+            .replace(name+'.w', 'GAUSS_W')
+            .replace('if __name__ == "__main__"', "def main()")
+           )
     translation,warnings = translate([(code, fname_in, 1)])
+    translation = translation.replace("double main()", "int main(int argc, char *argv[])")
 
     with open(fname_out, "w") as file_out:
+        file_out.write(C_HEADER)
         file_out.write(str(translation))
     if warnings: