Skip to content

Commit

Permalink
use interpolation propagation
Browse files Browse the repository at this point in the history
  • Loading branch information
lachlangrose committed Oct 2, 2024
1 parent 72b574b commit 9ded6e5
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -74,11 +74,11 @@ def propagate_property(self, property, to_level):
yi = np.arange(0, self.levels[to_level]['nsteps'][1], 2 ** (to_level - from_level))
zi = np.arange(0, self.levels[to_level]['nsteps'][2], 2 ** (to_level - from_level))
ii, jj, kk = np.meshgrid(xi, yi, zi, indexing='ij')
new_nodes_idx = np.vstack(
[ii.flatten(order='f'), jj.flatten(order='f'), kk.flatten(order='f')]
).T
# new_nodes_idx = np.vstack(
# [ii.flatten(order='f'), jj.flatten(order='f'), kk.flatten(order='f')]
# ).T
self.nsteps = self.levels[to_level]['nsteps']
gi = self.global_node_indices(new_nodes_idx).astype(int)
# gi = self.global_node_indices(new_nodes_idx).astype(int)
self.set_level(from_level) # reset to the original level
# get property value on all nodes
print(property.shape, self.n_nodes)
Expand Down

0 comments on commit 9ded6e5

Please sign in to comment.