From e2b00f4ec963060be98939c7b0d12d6c00e50a02 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Thu, 31 Oct 2019 13:13:00 +0100
Subject: Call symbolic generator inside code templates

This paves the way for dropping in other LBM collision models.
As a side benefit the default momenta calulcation is now fully inlined where possible.
---
 boltzgen/utility/optimizations.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'boltzgen/utility')

diff --git a/boltzgen/utility/optimizations.py b/boltzgen/utility/optimizations.py
index 93dad09..6dc23e9 100644
--- a/boltzgen/utility/optimizations.py
+++ b/boltzgen/utility/optimizations.py
@@ -2,9 +2,17 @@ from sympy import *
 
 from sympy.codegen.rewriting import ReplaceOptim
 
-expand_square = ReplaceOptim(
+expand_pos_square = ReplaceOptim(
     lambda e: e.is_Pow and e.exp.is_integer and e.exp == 2,
     lambda p: UnevaluatedExpr(Mul(p.base, p.base, evaluate = False))
 )
 
-custom = [ (expand_square, expand_square) ] + cse_main.basic_optimizations
+expand_neg_square = ReplaceOptim(
+    lambda e: e.is_Pow and e.exp.is_integer and e.exp == -2,
+    lambda p: UnevaluatedExpr(Mul(p.base, p.base, evaluate = False))
+)
+
+custom = [
+    (expand_pos_square, expand_pos_square),
+    (expand_neg_square, expand_neg_square)
+] + cse_main.basic_optimizations
-- 
cgit v1.2.3