Handle context reassignment for Grid/GridSet/File objects

1 files changed, 9 insertions, 0 deletions

diff --git a/src/transformations/hgridshift.cpp b/src/transformations/hgridshift.cpp
index 3b6e366f..24da4dde 100644
--- a/src/transformations/hgridshift.cpp
+++ b/src/transformations/hgridshift.cpp
@@ -111,10 +111,19 @@ static PJ *destructor (PJ *P, int errlev) {
     return pj_default_destructor(P, errlev);
 }
 
+static void reassign_context( PJ* P, PJ_CONTEXT* ctx )
+{
+    auto Q = (struct hgridshiftData *) P->opaque;
+    for( auto& grid: Q->grids ) {
+        grid->reassign_context(ctx);
+    }
+}
+
 PJ *TRANSFORMATION(hgridshift,0) {
     auto Q = new hgridshiftData;
     P->opaque = (void *) Q;
     P->destructor = destructor;
+    P->reassign_context = reassign_context;
 
     P->fwd4d  = forward_4d;
     P->inv4d  = reverse_4d;