fix initial condition

Author: jaimesouza

.gitignore

@@ -4,6 +4,7 @@
 *.png
 dist/
 build/
+tmp/*
 /src/wavefield/*
 simwave.egg-info/*
 .ipynb_checkpoints

simwave/kernel/frontend/solver.py

@@ -141,8 +141,8 @@ def forward(self, initial_grid=None):
             halo = self.space_model.halo_size[0]
             nbl = self.space_model.nbl
 
-            print("Initial grid")
-            print(initial_grid)
+            # print("Initial grid")
+            # print(initial_grid)
             print("Initial grid shape:", initial_grid.shape)
 
             if self.space_model.dimension == 2:
@@ -151,7 +151,7 @@ def forward(self, initial_grid=None):
                 z2 = halo + nbl[1]
                 x1 = halo + nbl[2]
                 x2 = halo + nbl[3]
-                u[0, z1:-z2, x1:-x2] = initial_grid
+                u[1, z1:-z2, x1:-x2] = initial_grid
 
             else:
 
@@ -162,7 +162,7 @@ def forward(self, initial_grid=None):
                 y1 = halo + nbl[4]
                 y2 = halo + nbl[5]
 
-                u[0, z1:-z2, x1:-x2, y1:-y2] = initial_grid
+                u[1, z1:-z2, x1:-x2, y1:-y2] = initial_grid
 
         u_full, recv = self._middleware.exec(
             operator='forward',