def dispW1(dataD, G, b, org, spa, mat, mu, shpV, nvol): scene = slicer.MRMLScene r1 = slicer.vtkMRMLDiffusionWeightedVolumeNode() r11 = slicer.vtkMRMLDiffusionWeightedVolumeDisplayNode() scene.AddNode(r11) r1.AddAndObserveDisplayNodeID(r11.GetName()) imgD = slicer.vtkImageData() imgD.SetDimensions(shpV[0], shpV[1], shpV[2]) imgD.SetScalarTypeToShort() r1.SetAndObserveImageData(imgD) r1.SetNumberOfGradients(shpV[3]) r1.SetIJKToRASMatrix(mat) r1.SetMeasurementFrameMatrix(mu) r1.SetDiffusionGradients(G) r1.SetBValues(b) r1.SetOrigin(org[0], org[1], org[2]) r1.SetSpacing(spa[0], spa[1], spa[2]) scene.AddNode(r1) tmp = r1.GetImageData().ToArray() tmp[...] = dataD[..., nvol] r1.GetDisplayNode().SetDefaultColorMap() r1.Modified() return
def dispW0(dataD, inputVolume, shpV, nvol): scene = slicer.MRMLScene r1 = slicer.vtkMRMLDiffusionWeightedVolumeNode() r11 = slicer.vtkMRMLDiffusionWeightedVolumeDisplayNode() scene.AddNode(r11) r1.AddAndObserveDisplayNodeID(r11.GetName()) imgD = slicer.vtkImageData() imgD.SetDimensions(shpV[0], shpV[1], shpV[2]) imgD.SetScalarTypeToShort() org = inputVolume.GetOrigin() spa = inputVolume.GetSpacing() mat = slicer.vtkMatrix4x4() inputVolume.GetIJKToRASMatrix(mat) r1.SetAndObserveImageData(imgD) r1.SetIJKToRASMatrix(mat) r1.SetOrigin(org[0], org[1], org[2]) r1.SetSpacing(spa[0], spa[1], spa[2]) scene.AddNode(r1) tmp = r1.GetImageData().ToArray() tmp[...] = dataD[..., nvol] r1.GetDisplayNode().SetDefaultColorMap() r1.Modified() return