niwrap-mrtrix 0.6.2__py3-none-any.whl → 0.7.0__py3-none-any.whl

niwrap_mrtrix/mrtrix/mredit.py

@@ -6,7 +6,7 @@ import pathlib
 from styxdefs import *
 
 MREDIT_METADATA = Metadata(
-    id="9ffef6548a8d55005ba7462a213b391f4d2da89f.boutiques",
+    id="678a737a8aca3b1d9299ee0e4e4b44fe21b937dd.boutiques",
     name="mredit",
     package="mrtrix",
     container_image_tag="mrtrix3/mrtrix3:3.0.4",
@@ -14,7 +14,13 @@ MREDIT_METADATA = Metadata(
 
 
 MreditPlaneParameters = typing.TypedDict('MreditPlaneParameters', {
-    "__STYXTYPE__": typing.Literal["plane"],
+    "@type": typing.NotRequired[typing.Literal["plane"]],
+    "axis": int,
+    "coord": list[int],
+    "value": float,
+})
+MreditPlaneParametersTagged = typing.TypedDict('MreditPlaneParametersTagged', {
+    "@type": typing.Literal["plane"],
     "axis": int,
     "coord": list[int],
     "value": float,
@@ -22,7 +28,13 @@ MreditPlaneParameters = typing.TypedDict('MreditPlaneParameters', {
 
 
 MreditSphereParameters = typing.TypedDict('MreditSphereParameters', {
-    "__STYXTYPE__": typing.Literal["sphere"],
+    "@type": typing.NotRequired[typing.Literal["sphere"]],
+    "position": list[float],
+    "radius": float,
+    "value": float,
+})
+MreditSphereParametersTagged = typing.TypedDict('MreditSphereParametersTagged', {
+    "@type": typing.Literal["sphere"],
     "position": list[float],
     "radius": float,
     "value": float,
@@ -30,21 +42,48 @@ MreditSphereParameters = typing.TypedDict('MreditSphereParameters', {
 
 
 MreditVoxelParameters = typing.TypedDict('MreditVoxelParameters', {
-    "__STYXTYPE__": typing.Literal["voxel"],
+    "@type": typing.NotRequired[typing.Literal["voxel"]],
+    "position": list[float],
+    "value": float,
+})
+MreditVoxelParametersTagged = typing.TypedDict('MreditVoxelParametersTagged', {
+    "@type": typing.Literal["voxel"],
     "position": list[float],
     "value": float,
 })
 
 
 MreditConfigParameters = typing.TypedDict('MreditConfigParameters', {
-    "__STYXTYPE__": typing.Literal["config"],
+    "@type": typing.NotRequired[typing.Literal["config"]],
+    "key": str,
+    "value": str,
+})
+MreditConfigParametersTagged = typing.TypedDict('MreditConfigParametersTagged', {
+    "@type": typing.Literal["config"],
     "key": str,
     "value": str,
 })
 
 
 MreditParameters = typing.TypedDict('MreditParameters', {
-    "__STYXTYPE__": typing.Literal["mredit"],
+    "@type": typing.NotRequired[typing.Literal["mrtrix/mredit"]],
+    "plane": typing.NotRequired[list[MreditPlaneParameters] | None],
+    "sphere": typing.NotRequired[list[MreditSphereParameters] | None],
+    "voxel": typing.NotRequired[list[MreditVoxelParameters] | None],
+    "scanner": bool,
+    "info": bool,
+    "quiet": bool,
+    "debug": bool,
+    "force": bool,
+    "nthreads": typing.NotRequired[int | None],
+    "config": typing.NotRequired[list[MreditConfigParameters] | None],
+    "help": bool,
+    "version": bool,
+    "input": InputPathType,
+    "output": typing.NotRequired[str | None],
+})
+MreditParametersTagged = typing.TypedDict('MreditParametersTagged', {
+    "@type": typing.Literal["mrtrix/mredit"],
     "plane": typing.NotRequired[list[MreditPlaneParameters] | None],
     "sphere": typing.NotRequired[list[MreditSphereParameters] | None],
     "voxel": typing.NotRequired[list[MreditVoxelParameters] | None],
@@ -60,49 +99,13 @@ MreditParameters = typing.TypedDict('MreditParameters', {
     "input": InputPathType,
     "output": typing.NotRequired[str | None],
 })
-
-
-def dyn_cargs(
-    t: str,
-) -> typing.Any:
-    """
-    Get build cargs function by command type.
-    
-    Args:
-        t: Command type.
-    Returns:
-        Build cargs function.
-    """
-    return {
-        "mredit": mredit_cargs,
-        "plane": mredit_plane_cargs,
-        "sphere": mredit_sphere_cargs,
-        "voxel": mredit_voxel_cargs,
-        "config": mredit_config_cargs,
-    }.get(t)
-
-
-def dyn_outputs(
-    t: str,
-) -> typing.Any:
-    """
-    Get build outputs function by command type.
-    
-    Args:
-        t: Command type.
-    Returns:
-        Build outputs function.
-    """
-    return {
-        "mredit": mredit_outputs,
-    }.get(t)
 
 
 def mredit_plane_params(
     axis: int,
     coord: list[int],
     value: float,
-) -> MreditPlaneParameters:
+) -> MreditPlaneParametersTagged:
     """
     Build parameters.
     
@@ -114,7 +117,7 @@ def mredit_plane_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "plane",
+        "@type": "plane",
         "axis": axis,
         "coord": coord,
         "value": value,
@@ -137,9 +140,9 @@ def mredit_plane_cargs(
     """
     cargs = []
     cargs.append("-plane")
-    cargs.append(str(params.get("axis")))
-    cargs.extend(map(str, params.get("coord")))
-    cargs.append(str(params.get("value")))
+    cargs.append(str(params.get("axis", None)))
+    cargs.extend(map(str, params.get("coord", None)))
+    cargs.append(str(params.get("value", None)))
     return cargs
 
 
@@ -147,7 +150,7 @@ def mredit_sphere_params(
     position: list[float],
     radius: float,
     value: float,
-) -> MreditSphereParameters:
+) -> MreditSphereParametersTagged:
     """
     Build parameters.
     
@@ -159,7 +162,7 @@ def mredit_sphere_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "sphere",
+        "@type": "sphere",
         "position": position,
         "radius": radius,
         "value": value,
@@ -182,16 +185,16 @@ def mredit_sphere_cargs(
     """
     cargs = []
     cargs.append("-sphere")
-    cargs.extend(map(str, params.get("position")))
-    cargs.append(str(params.get("radius")))
-    cargs.append(str(params.get("value")))
+    cargs.extend(map(str, params.get("position", None)))
+    cargs.append(str(params.get("radius", None)))
+    cargs.append(str(params.get("value", None)))
     return cargs
 
 
 def mredit_voxel_params(
     position: list[float],
     value: float,
-) -> MreditVoxelParameters:
+) -> MreditVoxelParametersTagged:
     """
     Build parameters.
     
@@ -202,7 +205,7 @@ def mredit_voxel_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "voxel",
+        "@type": "voxel",
         "position": position,
         "value": value,
     }
@@ -224,15 +227,15 @@ def mredit_voxel_cargs(
     """
     cargs = []
     cargs.append("-voxel")
-    cargs.extend(map(str, params.get("position")))
-    cargs.append(str(params.get("value")))
+    cargs.extend(map(str, params.get("position", None)))
+    cargs.append(str(params.get("value", None)))
     return cargs
 
 
 def mredit_config_params(
     key: str,
     value: str,
-) -> MreditConfigParameters:
+) -> MreditConfigParametersTagged:
     """
     Build parameters.
     
@@ -243,7 +246,7 @@ def mredit_config_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "config",
+        "@type": "config",
         "key": key,
         "value": value,
     }
@@ -265,14 +268,14 @@ def mredit_config_cargs(
     """
     cargs = []
     cargs.append("-config")
-    cargs.append(params.get("key"))
-    cargs.append(params.get("value"))
+    cargs.append(params.get("key", None))
+    cargs.append(params.get("value", None))
     return cargs
 
 
 class MreditOutputs(typing.NamedTuple):
     """
-    Output object returned when calling `mredit(...)`.
+    Output object returned when calling `MreditParameters(...)`.
     """
     root: OutputPathType
     """Output root folder. This is the root folder for all outputs."""
@@ -295,7 +298,7 @@ def mredit_params(
     help_: bool = False,
     version: bool = False,
     output: str | None = None,
-) -> MreditParameters:
+) -> MreditParametersTagged:
     """
     Build parameters.
     
@@ -323,7 +326,7 @@ def mredit_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "mredit",
+        "@type": "mrtrix/mredit",
         "scanner": scanner,
         "info": info,
         "quiet": quiet,
@@ -363,36 +366,36 @@ def mredit_cargs(
     """
     cargs = []
     cargs.append("mredit")
-    if params.get("plane") is not None:
-        cargs.extend([a for c in [dyn_cargs(s["__STYXTYPE__"])(s, execution) for s in params.get("plane")] for a in c])
-    if params.get("sphere") is not None:
-        cargs.extend([a for c in [dyn_cargs(s["__STYXTYPE__"])(s, execution) for s in params.get("sphere")] for a in c])
-    if params.get("voxel") is not None:
-        cargs.extend([a for c in [dyn_cargs(s["__STYXTYPE__"])(s, execution) for s in params.get("voxel")] for a in c])
-    if params.get("scanner"):
+    if params.get("plane", None) is not None:
+        cargs.extend([a for c in [mredit_plane_cargs(s, execution) for s in params.get("plane", None)] for a in c])
+    if params.get("sphere", None) is not None:
+        cargs.extend([a for c in [mredit_sphere_cargs(s, execution) for s in params.get("sphere", None)] for a in c])
+    if params.get("voxel", None) is not None:
+        cargs.extend([a for c in [mredit_voxel_cargs(s, execution) for s in params.get("voxel", None)] for a in c])
+    if params.get("scanner", False):
         cargs.append("-scanner")
-    if params.get("info"):
+    if params.get("info", False):
         cargs.append("-info")
-    if params.get("quiet"):
+    if params.get("quiet", False):
         cargs.append("-quiet")
-    if params.get("debug"):
+    if params.get("debug", False):
         cargs.append("-debug")
-    if params.get("force"):
+    if params.get("force", False):
         cargs.append("-force")
-    if params.get("nthreads") is not None:
+    if params.get("nthreads", None) is not None:
         cargs.extend([
             "-nthreads",
-            str(params.get("nthreads"))
+            str(params.get("nthreads", None))
         ])
-    if params.get("config") is not None:
-        cargs.extend([a for c in [dyn_cargs(s["__STYXTYPE__"])(s, execution) for s in params.get("config")] for a in c])
-    if params.get("help"):
+    if params.get("config", None) is not None:
+        cargs.extend([a for c in [mredit_config_cargs(s, execution) for s in params.get("config", None)] for a in c])
+    if params.get("help", False):
         cargs.append("-help")
-    if params.get("version"):
+    if params.get("version", False):
         cargs.append("-version")
-    cargs.append(execution.input_file(params.get("input")))
-    if params.get("output") is not None:
-        cargs.append(params.get("output"))
+    cargs.append(execution.input_file(params.get("input", None)))
+    if params.get("output", None) is not None:
+        cargs.append(params.get("output", None))
     return cargs
 
 
@@ -411,16 +414,18 @@ def mredit_outputs(
     """
     ret = MreditOutputs(
         root=execution.output_file("."),
-        output=execution.output_file(params.get("output")) if (params.get("output") is not None) else None,
+        output=execution.output_file(params.get("output", None)) if (params.get("output") is not None) else None,
     )
     return ret
 
 
 def mredit_execute(
     params: MreditParameters,
-    execution: Execution,
+    runner: Runner | None = None,
 ) -> MreditOutputs:
     """
+    mredit
+    
     Directly edit the intensities within an image from the command-line.
     
     A range of options are provided to enable direct editing of voxel
@@ -439,10 +444,12 @@ def mredit_execute(
     
     Args:
         params: The parameters.
-        execution: The execution object.
+        runner: Command runner.
     Returns:
         NamedTuple of outputs (described in `MreditOutputs`).
     """
+    runner = runner or get_global_runner()
+    execution = runner.start_execution(MREDIT_METADATA)
     params = execution.params(params)
     cargs = mredit_cargs(params, execution)
     ret = mredit_outputs(params, execution)
@@ -468,6 +475,8 @@ def mredit(
     runner: Runner | None = None,
 ) -> MreditOutputs:
     """
+    mredit
+    
     Directly edit the intensities within an image from the command-line.
     
     A range of options are provided to enable direct editing of voxel
@@ -508,8 +517,6 @@ def mredit(
     Returns:
         NamedTuple of outputs (described in `MreditOutputs`).
     """
-    runner = runner or get_global_runner()
-    execution = runner.start_execution(MREDIT_METADATA)
     params = mredit_params(
         plane=plane,
         sphere=sphere,
@@ -526,19 +533,15 @@ def mredit(
         input_=input_,
         output=output,
     )
-    return mredit_execute(params, execution)
+    return mredit_execute(params, runner)
 
 
 __all__ = [
     "MREDIT_METADATA",
-    "MreditConfigParameters",
     "MreditOutputs",
-    "MreditParameters",
-    "MreditPlaneParameters",
-    "MreditSphereParameters",
-    "MreditVoxelParameters",
     "mredit",
     "mredit_config_params",
+    "mredit_execute",
     "mredit_params",
     "mredit_plane_params",
     "mredit_sphere_params",