aboutsummaryrefslogtreecommitdiff
path: root/ldc_3d.py
diff options
context:
space:
mode:
Diffstat (limited to 'ldc_3d.py')
-rw-r--r--ldc_3d.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/ldc_3d.py b/ldc_3d.py
index 62f820e..c008b06 100644
--- a/ldc_3d.py
+++ b/ldc_3d.py
@@ -5,7 +5,8 @@ import matplotlib
import matplotlib.pyplot as plt
matplotlib.use('AGG')
-from lbm import Lattice, Geometry
+from simulation import Lattice, Geometry
+from symbolic.generator import LBM
import symbolic.D3Q19 as D3Q19
@@ -68,12 +69,14 @@ moments = []
print("Initializing simulation...\n")
+lbm = LBM(D3Q19)
+
lattice = Lattice(
descriptor = D3Q19,
geometry = Geometry(128, 128, 128),
- moments = D3Q19.moments(optimize = False),
- collide = D3Q19.bgk(tau = 0.52),
+ moments = lbm.moments(optimize = False),
+ collide = lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.52),
boundary_src = boundary)