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

niwrap_mrtrix/mrtrix/dwi2mask.py

@@ -6,7 +6,7 @@ import pathlib
 from styxdefs import *
 
 DWI2MASK_METADATA = Metadata(
-    id="9d585c1eb8931fbf6a7b9b4efcfcbe79a0ec151f.boutiques",
+    id="5345247ad41518ec6a872de4d754e7c58b01dff1.boutiques",
     name="dwi2mask",
     package="mrtrix",
     container_image_tag="mrtrix3/mrtrix3:3.0.4",
@@ -14,21 +14,47 @@ DWI2MASK_METADATA = Metadata(
 
 
 Dwi2maskFslgradParameters = typing.TypedDict('Dwi2maskFslgradParameters', {
-    "__STYXTYPE__": typing.Literal["fslgrad"],
+    "@type": typing.NotRequired[typing.Literal["fslgrad"]],
+    "bvecs": InputPathType,
+    "bvals": InputPathType,
+})
+Dwi2maskFslgradParametersTagged = typing.TypedDict('Dwi2maskFslgradParametersTagged', {
+    "@type": typing.Literal["fslgrad"],
     "bvecs": InputPathType,
     "bvals": InputPathType,
 })
 
 
 Dwi2maskConfigParameters = typing.TypedDict('Dwi2maskConfigParameters', {
-    "__STYXTYPE__": typing.Literal["config"],
+    "@type": typing.NotRequired[typing.Literal["config"]],
+    "key": str,
+    "value": str,
+})
+Dwi2maskConfigParametersTagged = typing.TypedDict('Dwi2maskConfigParametersTagged', {
+    "@type": typing.Literal["config"],
     "key": str,
     "value": str,
 })
 
 
 Dwi2maskParameters = typing.TypedDict('Dwi2maskParameters', {
-    "__STYXTYPE__": typing.Literal["dwi2mask"],
+    "@type": typing.NotRequired[typing.Literal["mrtrix/dwi2mask"]],
+    "clean_scale": typing.NotRequired[int | None],
+    "grad": typing.NotRequired[InputPathType | None],
+    "fslgrad": typing.NotRequired[Dwi2maskFslgradParameters | None],
+    "info": bool,
+    "quiet": bool,
+    "debug": bool,
+    "force": bool,
+    "nthreads": typing.NotRequired[int | None],
+    "config": typing.NotRequired[list[Dwi2maskConfigParameters] | None],
+    "help": bool,
+    "version": bool,
+    "input": InputPathType,
+    "output": str,
+})
+Dwi2maskParametersTagged = typing.TypedDict('Dwi2maskParametersTagged', {
+    "@type": typing.Literal["mrtrix/dwi2mask"],
     "clean_scale": typing.NotRequired[int | None],
     "grad": typing.NotRequired[InputPathType | None],
     "fslgrad": typing.NotRequired[Dwi2maskFslgradParameters | None],
@@ -43,46 +69,12 @@ Dwi2maskParameters = typing.TypedDict('Dwi2maskParameters', {
     "input": InputPathType,
     "output": str,
 })
-
-
-def dyn_cargs(
-    t: str,
-) -> typing.Any:
-    """
-    Get build cargs function by command type.
-    
-    Args:
-        t: Command type.
-    Returns:
-        Build cargs function.
-    """
-    return {
-        "dwi2mask": dwi2mask_cargs,
-        "fslgrad": dwi2mask_fslgrad_cargs,
-        "config": dwi2mask_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 {
-        "dwi2mask": dwi2mask_outputs,
-    }.get(t)
 
 
 def dwi2mask_fslgrad_params(
     bvecs: InputPathType,
     bvals: InputPathType,
-) -> Dwi2maskFslgradParameters:
+) -> Dwi2maskFslgradParametersTagged:
     """
     Build parameters.
     
@@ -99,7 +91,7 @@ def dwi2mask_fslgrad_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "fslgrad",
+        "@type": "fslgrad",
         "bvecs": bvecs,
         "bvals": bvals,
     }
@@ -121,15 +113,15 @@ def dwi2mask_fslgrad_cargs(
     """
     cargs = []
     cargs.append("-fslgrad")
-    cargs.append(execution.input_file(params.get("bvecs")))
-    cargs.append(execution.input_file(params.get("bvals")))
+    cargs.append(execution.input_file(params.get("bvecs", None)))
+    cargs.append(execution.input_file(params.get("bvals", None)))
     return cargs
 
 
 def dwi2mask_config_params(
     key: str,
     value: str,
-) -> Dwi2maskConfigParameters:
+) -> Dwi2maskConfigParametersTagged:
     """
     Build parameters.
     
@@ -140,7 +132,7 @@ def dwi2mask_config_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "config",
+        "@type": "config",
         "key": key,
         "value": value,
     }
@@ -162,14 +154,14 @@ def dwi2mask_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 Dwi2maskOutputs(typing.NamedTuple):
     """
-    Output object returned when calling `dwi2mask(...)`.
+    Output object returned when calling `Dwi2maskParameters(...)`.
     """
     root: OutputPathType
     """Output root folder. This is the root folder for all outputs."""
@@ -191,7 +183,7 @@ def dwi2mask_params(
     config: list[Dwi2maskConfigParameters] | None = None,
     help_: bool = False,
     version: bool = False,
-) -> Dwi2maskParameters:
+) -> Dwi2maskParametersTagged:
     """
     Build parameters.
     
@@ -228,7 +220,7 @@ def dwi2mask_params(
         Parameter dictionary
     """
     params = {
-        "__STYXTYPE__": "dwi2mask",
+        "@type": "mrtrix/dwi2mask",
         "info": info,
         "quiet": quiet,
         "debug": debug,
@@ -266,39 +258,39 @@ def dwi2mask_cargs(
     """
     cargs = []
     cargs.append("dwi2mask")
-    if params.get("clean_scale") is not None:
+    if params.get("clean_scale", None) is not None:
         cargs.extend([
             "-clean_scale",
-            str(params.get("clean_scale"))
+            str(params.get("clean_scale", None))
         ])
-    if params.get("grad") is not None:
+    if params.get("grad", None) is not None:
         cargs.extend([
             "-grad",
-            execution.input_file(params.get("grad"))
+            execution.input_file(params.get("grad", None))
         ])
-    if params.get("fslgrad") is not None:
-        cargs.extend(dyn_cargs(params.get("fslgrad")["__STYXTYPE__"])(params.get("fslgrad"), execution))
-    if params.get("info"):
+    if params.get("fslgrad", None) is not None:
+        cargs.extend(dwi2mask_fslgrad_cargs(params.get("fslgrad", None), execution))
+    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 [dwi2mask_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")))
-    cargs.append(params.get("output"))
+    cargs.append(execution.input_file(params.get("input", None)))
+    cargs.append(params.get("output", None))
     return cargs
 
 
@@ -317,16 +309,18 @@ def dwi2mask_outputs(
     """
     ret = Dwi2maskOutputs(
         root=execution.output_file("."),
-        output=execution.output_file(params.get("output")),
+        output=execution.output_file(params.get("output", None)),
     )
     return ret
 
 
 def dwi2mask_execute(
     params: Dwi2maskParameters,
-    execution: Execution,
+    runner: Runner | None = None,
 ) -> Dwi2maskOutputs:
     """
+    dwi2mask
+    
     Generates a whole brain mask from a DWI image.
     
     All diffusion weighted and b=0 volumes are used to obtain a mask that
@@ -349,10 +343,12 @@ def dwi2mask_execute(
     
     Args:
         params: The parameters.
-        execution: The execution object.
+        runner: Command runner.
     Returns:
         NamedTuple of outputs (described in `Dwi2maskOutputs`).
     """
+    runner = runner or get_global_runner()
+    execution = runner.start_execution(DWI2MASK_METADATA)
     params = execution.params(params)
     cargs = dwi2mask_cargs(params, execution)
     ret = dwi2mask_outputs(params, execution)
@@ -377,6 +373,8 @@ def dwi2mask(
     runner: Runner | None = None,
 ) -> Dwi2maskOutputs:
     """
+    dwi2mask
+    
     Generates a whole brain mask from a DWI image.
     
     All diffusion weighted and b=0 volumes are used to obtain a mask that
@@ -430,8 +428,6 @@ def dwi2mask(
     Returns:
         NamedTuple of outputs (described in `Dwi2maskOutputs`).
     """
-    runner = runner or get_global_runner()
-    execution = runner.start_execution(DWI2MASK_METADATA)
     params = dwi2mask_params(
         clean_scale=clean_scale,
         grad=grad,
@@ -447,17 +443,15 @@ def dwi2mask(
         input_=input_,
         output=output,
     )
-    return dwi2mask_execute(params, execution)
+    return dwi2mask_execute(params, runner)
 
 
 __all__ = [
     "DWI2MASK_METADATA",
-    "Dwi2maskConfigParameters",
-    "Dwi2maskFslgradParameters",
     "Dwi2maskOutputs",
-    "Dwi2maskParameters",
     "dwi2mask",
     "dwi2mask_config_params",
+    "dwi2mask_execute",
     "dwi2mask_fslgrad_params",
     "dwi2mask_params",
 ]