diff --git a/parallel.py b/parallel.py index 3d5ddb8..f33a336 100644 --- a/parallel.py +++ b/parallel.py @@ -280,11 +280,11 @@ class PPP: else: raise ValueError("Invalid shifting state.") self.field[key_out] = self.field[key].derivative(axis,padding=padding) - self.origin[key_out] = tuple(origin) - self.spacing[key_out] = self.spacing[key].copy() - self.symmetries[key_out] = self.symmetries[key].copy() - # TBD: copy everything field specific - # TBD: make sure processor distribution is fine + self.origin[key_out] = tuple(origin) + self.spacing[key_out] = self.spacing[key].copy() + self.symmetries[key_out] = self.symmetries[key].copy() + self.proc_grid[key_out] = self.proc_grid[key].copy() + self.proc_grid_ext[key_out] = self.proc_grid_ext[key].copy() if axis==0: self.symmetries[key_out][0,1,1] = -self.symmetries[key_out][0,1,1] self.symmetries[key_out][2,1,1] = -self.symmetries[key_out][2,1,1]