e0c9d018aca7bf05486c385484d5ed1c37bac423,pynets/diffconnectometry.py,,coreg_WM_mask_to_diff,#Any#Any#,151

Before Change

    flirt.inputs.out_matrix_file = "/tmp/out_flirt.mat"
    flirt.inputs.in_matrix_file = "%s%s" % (dwi_dir, "/xfms/MNI2diff.mat")
    flirt.inputs.apply_xfm = True
    flirt.run()
    args = "-bin"
    maths = fsl.ImageMaths(in_file=out_file, op_string=args, out_file=out_file)
    print("\nBinarizing WM mask...")
    os.system(maths.cmdline)

After Change

        dwi_infile = "%s%s" % (dwi_dir, "/dwi.nii.gz")

    out_file = "%s%s" % (dwi_dir, "/wm_mask_diff.nii.gz")
    wm_transforms_wf = pe.Workflow(name="wm_transforms")
    inputnode = pe.Node(niu.IdentityInterface(fields=["dwi_infile", "wm_mask_loc", "out_file", "mat_file", "dwi_dir"]),
                        name="inputnode")
    inputnode.inputs.dwi_infile = dwi_infile
    inputnode.inputs.wm_mask_loc = wm_mask_loc
    inputnode.inputs.out_file = out_file
    inputnode.inputs.mat_file = "%s%s" % (dwi_dir, "/xfms/MNI2diff.mat")
    inputnode.inputs.dwi_dir = dwi_dir

    flirt = pe.Node(interface=fsl.FLIRT(cost_func="mutualinfo"), name="coregister_wm")
    flirt.inputs.apply_xfm = True
    flirt.inputs.out_matrix_file = "/tmp/out_flirt.mat"

    def bin_wm(out_file):
        import nipype.interfaces.fsl as fsl
        import os
        args = "-bin"
        maths = fsl.ImageMaths(in_file=out_file, op_string=args, out_file=out_file)
        print("\nBinarizing WM mask...")
        os.system(maths.cmdline)
        return out_file

    bin_wm = pe.Node(niu.Function(input_names=["out_file"], output_names=["out_file"],
                                  function=bin_wm), name="bin_wm_node")

    wm_transforms_wf.connect([(inputnode, flirt, [("dwi_infile", "reference"), ("wm_mask_loc", "in_file"),
                                                  ("out_file", "out_file"), ("mat_file", "in_matrix_file")]),
                              (flirt, bin_wm, [("out_file", "out_file")])
                              ])

    return wm_transforms_wf

Instances