Allow functions for Field.coordinate_transform
Compare changes
+ 28
− 11
@@ -15,7 +15,8 @@ import pystencils
@@ -15,7 +15,8 @@ import pystencils
from pystencils.stencil import direction_string_to_offset, offset_to_direction_string, inverse_direction
@@ -328,10 +329,10 @@ class Field(AbstractField):
@@ -328,10 +329,10 @@ class Field(AbstractField):
@@ -432,7 +433,7 @@ class Field(AbstractField):
@@ -432,7 +433,7 @@ class Field(AbstractField):
@@ -453,7 +454,7 @@ class Field(AbstractField):
@@ -453,7 +454,7 @@ class Field(AbstractField):
@@ -528,7 +529,7 @@ class Field(AbstractField):
@@ -528,7 +529,7 @@ class Field(AbstractField):
@@ -562,7 +563,7 @@ class Field(AbstractField):
@@ -562,7 +563,7 @@ class Field(AbstractField):
raise NotImplementedError("staggered_vector_access is not implemented for more than 3 index dimensions")
@@ -613,7 +614,10 @@ class Field(AbstractField):
@@ -613,7 +614,10 @@ class Field(AbstractField):
return self.coordinate_transform @ (self.coordinate_origin + pystencils.x_vector(self.spatial_dimensions))
@@ -623,10 +627,23 @@ class Field(AbstractField):
@@ -623,10 +627,23 @@ class Field(AbstractField):