Session.virtualf_in: Allow passing the data kind if it's already known

Author: seisman

pygmt/clib/session.py

@@ -1758,6 +1758,7 @@ def virtualfile_from_stringio(
     def virtualfile_in(
         self,
         check_kind=None,
+        kind=None,
         data=None,
         x=None,
         y=None,
@@ -1847,7 +1848,9 @@ def virtualfile_in(
             )
             mincols = 3
 
-        kind = data_kind(data, required=required, check_kind=check_kind)
+        # Determine the data kind if not given.
+        if kind is None:
+            kind = data_kind(data, required=required, check_kind=check_kind)
         _validate_data_input(
             data=data,
             x=x,

pygmt/src/grdcut.py

@@ -128,7 +128,7 @@ def grdcut(
 
     with Session() as lib:
         with (
-            lib.virtualfile_in(data=grid) as vingrd,
+            lib.virtualfile_in(data=grid, kind=inkind) as vingrd,
             lib.virtualfile_out(kind=outkind, fname=outgrid) as voutgrd,
         ):
             kwargs["G"] = voutgrd

pygmt/src/legend.py

@@ -95,5 +95,5 @@ def legend(
         raise GMTInvalidInput(msg)
 
     with Session() as lib:
-        with lib.virtualfile_in(data=spec, required=False) as vintbl:
+        with lib.virtualfile_in(data=spec, required=False, kind=kind) as vintbl:
             lib.call_module(module="legend", args=build_arg_list(kwargs, infile=vintbl))

pygmt/src/plot.py

@@ -234,6 +234,7 @@ def plot(  # noqa: PLR0912
 
     kind = data_kind(data, check_kind="vector")
     if kind == "empty":  # Data is given via a series of vectors.
+        kind = "vectors"
         data = {"x": x, "y": y}
         # Parameters for vector styles
         if (
@@ -280,5 +281,5 @@ def plot(  # noqa: PLR0912
         kwargs["S"] = "s0.2c"
 
     with Session() as lib:
-        with lib.virtualfile_in(data=data) as vintbl:
+        with lib.virtualfile_in(data=data, kind=kind) as vintbl:
             lib.call_module(module="plot", args=build_arg_list(kwargs, infile=vintbl))

pygmt/src/plot3d.py

@@ -212,6 +212,7 @@ def plot3d(  # noqa: PLR0912
 
     kind = data_kind(data, check_kind="vector")
     if kind == "empty":  # Data is given via a series of vectors.
+        kind = "vectors"
         data = {"x": x, "y": y, "z": z}
         # Parameters for vector styles
         if (
@@ -259,5 +260,5 @@ def plot3d(  # noqa: PLR0912
         kwargs["S"] = "u0.2c"
 
     with Session() as lib:
-        with lib.virtualfile_in(data=data, mincols=3) as vintbl:
+        with lib.virtualfile_in(data=data, mincols=3, kind=kind) as vintbl:
             lib.call_module(module="plot3d", args=build_arg_list(kwargs, infile=vintbl))

pygmt/src/text.py

@@ -42,7 +42,7 @@
     w="wrap",
 )
 @kwargs_to_strings(R="sequence", c="sequence_comma", p="sequence")
-def text_(  # noqa: PLR0912
+def text_(  # noqa: PLR0912, PLR0915
     self,
     textfiles: PathLike | TableLike | None = None,
     x=None,
@@ -225,6 +225,7 @@ def text_(  # noqa: PLR0912
     confdict = {}
     data = None
     if kind == "empty":
+        kind = "vectors"
         data = {"x": x, "y": y}
 
         for arg, flag, name in array_args:
@@ -261,7 +262,9 @@ def text_(  # noqa: PLR0912
 
     with Session() as lib:
         with lib.virtualfile_in(
-            data=textfiles or data, required=data_is_required
+            data=textfiles or data,
+            required=data_is_required,
+            kind=kind,
         ) as vintbl:
             lib.call_module(
                 module="text",