From f1cc4051abec79c26f6a6ea06afcf6453e9f5e05 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Fri, 30 Jun 2017 16:19:21 +0200
Subject: [PATCH] Steady state Chapman Enskog
---
sympyextensions.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/sympyextensions.py b/sympyextensions.py
index c580680be..79fe02448 100644
--- a/sympyextensions.py
+++ b/sympyextensions.py
@@ -25,6 +25,21 @@ def equationsToMatrix(equations, degreesOfFreedom):
lambda row, col: equations[row].coeff(degreesOfFreedom[col]))
+def kroneckerDelta(*args):
+ """Kronecker delta for variable number of arguments,
+ 1 if all args are equal, otherwise 0"""
+ for a in args:
+ if a != args[0]:
+ return 0
+ return 1
+
+
+def multidimensionalSummation(i, dim):
+ """Multidimensional summation"""
+ prodArgs = [range(dim)] * i
+ return itertools.product(*prodArgs)
+
+
def normalizeProduct(product):
"""
Expects a sympy expression that can be interpreted as a product and
--
GitLab