Updated tutorial 02

apps/tutorials/codegen/02_LBMLatticeModelGeneration.cpp

@@ -29,7 +29,6 @@
 #include "geometry/all.h"
 
 #include "lbm/boundary/factories/DefaultBoundaryHandling.h"
-#include "lbm/communication/PdfFieldPackInfo.h"
 #include "lbm/field/AddToStorage.h"
 #include "lbm/field/PdfField.h"
 #include "lbm/field/initializer/all.h"

apps/tutorials/codegen/02_LBMLatticeModelGeneration.prm

 
 Parameters 
 {
-   omega           1.8;
+    omega           1.8;
 	timesteps       10000;
 
 	remainingTimeLoggerFrequency 3; // in seconds
@@ -35,7 +35,7 @@ StabilityChecker
 
 Boundaries 
 {   
-	Border { direction S,N; walldistance -1; NoSlip {} }		
+	Border { direction S, N; walldistance -1; NoSlip {} }
 }
 
 

apps/tutorials/codegen/02_LBMLatticeModelGeneration.py

@@ -9,28 +9,30 @@ from lbmpy_walberla import generate_lattice_model
 #      General Parameters
 #   ========================
 
-STENCIL = 'D2Q9'
-OMEGA = sp.Symbol('omega')
-LAYOUT = 'fzyx'
+stencil = 'D2Q9'
+omega = sp.Symbol('omega')
+layout = 'fzyx'
 
-#   Optimization
-OPT = {'target': 'cpu', 'cse_global': True, 'field_layout': LAYOUT}
+#   Optimizations to be used by the code generator
+optimizations = {'target': 'cpu', 'cse_global': True, 'field_layout': layout}
 
 #   ===========================
 #      SRT Method Definition
 #   ===========================
 
-srt_params = {'stencil': STENCIL,
+srt_params = {'stencil': stencil,
               'method': 'srt',
-              'relaxation_rate': OMEGA}
+              'relaxation_rate': omega}
 
-srt_collision_rule = create_lb_collision_rule(optimization=OPT, **srt_params)
-srt_update_rule = create_lb_update_rule(collision_rule=srt_collision_rule, optimization=OPT)
+srt_collision_rule = create_lb_collision_rule(optimization=optimizations, **srt_params)
+srt_update_rule = create_lb_update_rule(collision_rule=srt_collision_rule, optimization=optimizations)
 
 #   =====================
 #      Code Generation
 #   =====================
 
 with CodeGeneration() as ctx:
-    generate_lattice_model(ctx, "SRTLatticeModel", srt_collision_rule, field_layout=LAYOUT)
+    # generation of the lattice model ...
+    generate_lattice_model(ctx, "SRTLatticeModel", srt_collision_rule, field_layout=layout)
+    # ... and generation of the pack information to be used for the MPI communication
     generate_pack_info_from_kernel(ctx, "SRTPackInfo", srt_update_rule)