symbolic.py: wip ccode

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>

diff --git a/symbolic.py b/symbolic.py
index d0bcdc24f41393d66b1168c7062c8e2384728070..1c9056163e5052cb440d9604e90a6470a65dd114 100755 (executable)
--- a/symbolic.py
+++ b/symbolic.py
@@ -1,6 +1,7 @@
 #!/usr/bin/python
 
 from sympy import *
+import itertools
 
 import sys
 
@@ -161,6 +162,14 @@ for sm_sh in ((), (sh*sh*la,)):
       )
     print smalls
     dbg('result_fordiff');
+    for v in 'sh','th','la','mu','gamma','kappa':
+      dbg('diff(result_fordiff,%s)' % v)
+      d = diff(result_fordiff,eval(v))
+      print ccode(d, assign_to='FOO')
+      (defs, v) = cse(d)
+      for vn, val in defs:
+        print ccode(val, assign_to=vn)
+      print ccode(v[0], assign_to='BAR')
 
 #bad = q_orgcoords[0]
 #badd = diff(bad, la)