Testing the custom grafts

endograft_selection.py

@@ -99,14 +99,32 @@ def get_endograft_parameters(vessel_diameter, component_type):
 
 def choose_endograft(proximal_diameter, distal_diameter, length, fenestrations_exist, distance_to_last_fenestration):
     strut_amplitude = 15
-    is_tapered = (proximal_diameter - distal_diameter) >= 4
+    custom_oversizing_ratio = 1.15
+    default_custom_overlap = 61
+    is_tapered = (proximal_diameter - distal_diameter) == 4
     if is_tapered:
         proximal_lengths, proximal_graft_diameter, proximal_gap = get_endograft_parameters(proximal_diameter, "proximal_tapered")
-        proximal_prefix = "ZTA_PT"
+        if proximal_graft_diameter is not None:
+            proximal_prefix = "ZTA_PT"
+        else:
+            proximal_lengths, proximal_graft_diameter, proximal_gap = [length], round(proximal_diameter * custom_oversizing_ratio), 8
+            proximal_prefix = "ZTACustom_PT"
     else:
         proximal_lengths, proximal_graft_diameter, proximal_gap = get_endograft_parameters(proximal_diameter, "proximal")
-        proximal_prefix = "ZTA_P"
-    distal_lengths, distal_graft_diameter, distal_gap = get_endograft_parameters(24, "distal")
+        if proximal_graft_diameter is not None:
+            proximal_prefix = "ZTA_P"
+        else:
+            proximal_lengths, proximal_graft_diameter, proximal_gap = [length], round(proximal_diameter * custom_oversizing_ratio), 8
+            proximal_prefix = "ZTACustom_P"
+    if proximal_prefix == "ZTA_PT" or proximal_prefix == "ZTA_P":
+        distal_lengths, distal_graft_diameter, distal_gap = get_endograft_parameters(distal_diameter, "distal")
+        distal_prefix = "ZTA_D"
+        if distal_graft_diameter is None:
+            distal_prefix = "ZTACustom_D"
+            distal_lengths, distal_graft_diameter, distal_gap = [(length + default_custom_overlap) - length_element for length_element in proximal_lengths], round(distal_diameter * custom_oversizing_ratio), 8
+    else:
+        distal_lengths, distal_graft_diameter, distal_gap = [0], 0, 0
+        strut_amplitude = 0
 
     if (proximal_lengths is None) or (distal_lengths is None):
         return None
@@ -136,7 +154,7 @@ def choose_endograft(proximal_diameter, distal_diameter, length, fenestrations_e
 
             total_graft_length = proximal_length + distal_length - distal_overlap
             proximal_code = f"{proximal_prefix}_{proximal_code_diameter}_{proximal_length}_{proximal_gap}"
-            distal_code = f"ZTA_D_{distal_graft_diameter}_{distal_length}_{distal_gap}"
+            distal_code = f"{distal_prefix}_{distal_graft_diameter}_{distal_length}_{distal_gap}"
             extension_code = f"ZTA_E_{0}_{0}_{0}"
             extension_overlap = 0
 
@@ -208,7 +226,6 @@ def choose_endograft(proximal_diameter, distal_diameter, length, fenestrations_e
 
 def start_endograft_pipeline(proximal_diameter, distal_diameter, proximal_start_point, distal_end_point, length, fenestrations_exist, distance_to_last_fenestration):
     selected_graft = choose_endograft(round(proximal_diameter), round(distal_diameter), round(length), fenestrations_exist, round(distance_to_last_fenestration))
-
     if selected_graft is None:
         print("No supported graft; try another manufacturer")
     else: