From bb7afdcc0bf6d74bbba7ea52ec116c6e46f98964 Mon Sep 17 00:00:00 2001 From: Wei Ji Date: Thu, 11 Jun 2020 00:31:03 +1200 Subject: [PATCH] Allow virtualfile_from_grid to accept a registration argument Default registration is still "GMT_GRID_NODE_REG" as per upstream, but we might want to override that in some cases. --- pygmt/clib/session.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/pygmt/clib/session.py b/pygmt/clib/session.py index 961965102a4..0289212059f 100644 --- a/pygmt/clib/session.py +++ b/pygmt/clib/session.py @@ -1171,7 +1171,7 @@ def virtualfile_from_matrix(self, matrix): yield vfile @contextmanager - def virtualfile_from_grid(self, grid): + def virtualfile_from_grid(self, grid, registration="GMT_GRID_NODE_REG"): """ Store a grid in a virtual file. @@ -1240,7 +1240,12 @@ def virtualfile_from_grid(self, grid): family = "GMT_IS_GRID|GMT_VIA_MATRIX" geometry = "GMT_IS_SURFACE" gmt_grid = self.create_data( - family, geometry, mode="GMT_CONTAINER_ONLY", ranges=region, inc=inc + family, + geometry, + mode="GMT_CONTAINER_ONLY", + ranges=region, + inc=inc, + registration=registration, ) self.put_matrix(gmt_grid, matrix) args = (family, geometry, "GMT_IN|GMT_IS_REFERENCE", gmt_grid)