|
|
@@ -33,7 +33,7 @@ class SeekTransformModuleWidget(ScriptedLoadableModuleWidget):
|
|
|
|
|
|
# Dropdown menu za izbiro metode
|
|
|
self.rotationMethodComboBox = qt.QComboBox()
|
|
|
- self.rotationMethodComboBox.addItems(["SVD", "Horn", "Quaternion"])
|
|
|
+ self.rotationMethodComboBox.addItems(["Kabsch", "Horn", "Quaternion"])
|
|
|
self.layout.addWidget(self.rotationMethodComboBox)
|
|
|
|
|
|
# Load button
|
|
|
@@ -365,7 +365,7 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
|
|
|
shNode = slicer.vtkMRMLSubjectHierarchyNode.GetSubjectHierarchyNode(slicer.mrmlScene)
|
|
|
imageItem = shNode.GetItemByDataNode(volumeNode)
|
|
|
|
|
|
- modality = shNode.GetItemAttribute(imageItem, 'DICOM.Modality')
|
|
|
+ #modality = shNode.GetItemAttribute(imageItem, 'DICOM.Modality')
|
|
|
#print(modality)
|
|
|
|
|
|
# Check if the volume is loaded into the scene
|
|
|
@@ -431,7 +431,7 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
|
|
|
if selectedMethod == "SVD":
|
|
|
chosen_rotation_matrix = compute_Kabsch_rotation(cbct_points, ct_points)
|
|
|
chosen_translation_vector = compute_translation(cbct_points, ct_points, chosen_rotation_matrix)
|
|
|
- print("\nSVD Method:")
|
|
|
+ print("\nKabsch Method:")
|
|
|
print("Rotation Matrix:\n", chosen_rotation_matrix)
|
|
|
print("Translation Vector:\n", chosen_translation_vector)
|
|
|
elif selectedMethod == "Horn":
|
Luka