|
|
@@ -57,6 +57,7 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
|
|
|
Core logic of the module.
|
|
|
"""
|
|
|
def run(self, selectedMethod):
|
|
|
+ print("Calculating...")
|
|
|
def group_points(points, threshold):
|
|
|
# Function to group points that are close to each other
|
|
|
grouped_points = []
|
|
|
@@ -116,7 +117,7 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
|
|
|
dimensions = image_data.GetDimensions()
|
|
|
#detected_regions = []
|
|
|
|
|
|
- if yesCbct: #je cbct?
|
|
|
+ if yesCbct: #je cbct ali ct?
|
|
|
valid_x_min, valid_x_max = 0, dimensions[0] - 1
|
|
|
valid_y_min, valid_y_max = 0, dimensions[1] - 1
|
|
|
valid_z_min, valid_z_max = 0, dimensions[2] - 1
|
|
|
@@ -361,8 +362,9 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
|
|
|
for i in range(4):
|
|
|
for j in range(4):
|
|
|
vtk_matrix.SetElement(i, j, transform_matrix[i, j])
|
|
|
- print("Transform matrix: ")
|
|
|
- print(vtk_matrix)
|
|
|
+ #print("Transform matrix:")
|
|
|
+ #for i in range(4):
|
|
|
+ # print(" ".join(f"{vtk_matrix.GetElement(i, j):.6f}" for j in range(4)))
|
|
|
# Create vtkTransform and set the matrix
|
|
|
transform = vtk.vtkTransform()
|
|
|
transform.SetMatrix(vtk_matrix)
|
|
|
@@ -425,7 +427,7 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
|
|
|
if len(ct_points) < 3:
|
|
|
print("CT volumen nima dovolj točk za registracijo.")
|
|
|
else:
|
|
|
- print("CT points: ", np.array(ct_points))
|
|
|
+ #print("CT points: ", np.array(ct_points))
|
|
|
|
|
|
for cbct_volume_name in cbct_list:
|
|
|
# Izvleci točke za trenutni CBCT volumen
|
|
|
@@ -457,21 +459,21 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
|
|
|
if selectedMethod == "Kabsch":
|
|
|
chosen_rotation_matrix = compute_Kabsch_rotation(cbct_points, ct_points)
|
|
|
chosen_translation_vector = compute_translation(cbct_points, ct_points, chosen_rotation_matrix)
|
|
|
- #print("\nKabsch Method:")
|
|
|
- #print("Rotation Matrix:\n", chosen_rotation_matrix)
|
|
|
- #print("Translation Vector:\n", chosen_translation_vector)
|
|
|
+ print("\nKabsch Method:")
|
|
|
+ print("Rotation Matrix:\n", chosen_rotation_matrix)
|
|
|
+ print("Translation Vector:\n", chosen_translation_vector)
|
|
|
elif selectedMethod == "Horn":
|
|
|
chosen_rotation_matrix = compute_Horn_rotation(cbct_points, ct_points)
|
|
|
chosen_translation_vector = compute_translation(cbct_points, ct_points, chosen_rotation_matrix)
|
|
|
- #print("\nHorn Method:")
|
|
|
- #print("Rotation Matrix:\n", chosen_rotation_matrix)
|
|
|
- #print("Translation Vector:\n", chosen_translation_vector)
|
|
|
+ print("\nHorn Method:")
|
|
|
+ print("Rotation Matrix:\n", chosen_rotation_matrix)
|
|
|
+ print("Translation Vector:\n", chosen_translation_vector)
|
|
|
elif selectedMethod == "Iterative Closest Point (Kabsch)":
|
|
|
new_points, chosen_rotation_matrix, chosen_translation_vector = icp_algorithm(cbct_points, ct_points)
|
|
|
#chosen_translation_vector = compute_translation(cbct_points, ct_points, chosen_rotation_matrix)
|
|
|
- #print("\Iterative Closest Point Method:")
|
|
|
- #print("Rotation Matrix:\n", chosen_rotation_matrix)
|
|
|
- #print("Translation Vector:\n", chosen_translation_vector)
|
|
|
+ print("\Iterative Closest Point Method:")
|
|
|
+ print("Rotation Matrix:\n", chosen_rotation_matrix)
|
|
|
+ print("Translation Vector:\n", chosen_translation_vector)
|
|
|
|
|
|
imeTransformNoda = cbct_volume_name + " Transform"
|
|
|
# Ustvari vtkTransformNode in ga poveži z CBCT volumenom
|
Luka