From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Sat, 18 Nov 2017 00:32:09 +0000 (+0000)
Subject: symbolic.py: wip ccode
X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ian/git?a=commitdiff_plain;h=9331fa21a9f5ae85a206478f7f0eaccc522b3390;p=moebius3.git

symbolic.py: wip ccode

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

diff --git a/symbolic.py b/symbolic.py
index d0bcdc2..1c90561 100755
--- 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)