vizi handle mix path

modules/dataset_loader.py

@@ -285,7 +285,7 @@ def resize_and_pad(image, target, new_size=(1333, 800)):
     pad_bottom = new_size[1] - new_scaled_size[1] - pad_top
 
     # Pad the resized image to make it exactly the desired size
-    image = F.pad(image, (pad_left, pad_top, pad_right, pad_bottom), fill=0, padding_mode='constant')
+    image = F.pad(image, (pad_left, pad_top, pad_right, pad_bottom), fill=250, padding_mode='constant')
 
     # Adjust bounding boxes
     target['boxes'] = resize_boxes(target['boxes'], original_size, new_scaled_size)

modules/toWizard.py

@@ -46,12 +46,13 @@ def create_BPMN_id(data):
 
     return data
 
-def check_end(val):
-    if val[1] is None:
+def check_end(link):
+    if link[1] is None:
         return True
     return False
 
 def connect(data, text_mapping, i):
+    next_text = []
     target_idx = data['links'][i][1]  
     # Check if the target index is valid
     if target_idx==None or target_idx >= len(data['links']):
@@ -61,7 +62,12 @@ def connect(data, text_mapping, i):
     current_id = data['BPMN_id'][i]
     next_idx = data['links'][target_idx][1]
     next_id = data['BPMN_id'][next_idx]
-    next_text = text_mapping[next_id]
+    if next_id.split('_')[0] == 'parallelGateway':
+        for idx, link in enumerate(data['links']):
+            if link[0] == next_idx and link[1] is not None:
+                next_text.append(text_mapping[data['BPMN_id'][link[1]]])
+    else:
+        next_text.append(text_mapping[next_id])
     current_text = text_mapping[current_id]
 
     return current_text, next_text
@@ -71,7 +77,44 @@ def check_start(val):
         return True
     return False
 
+def find_merge(bpmn_id, links):
+    merge = []
+    for idx, link in enumerate(links):
+        next_element = link[1]
+        if next_element is None:
+            merge.append(None)
+            continue
+        next_object = links[next_element][1]
+        if next_object is None:
+            merge.append(None)
+            continue
+        if bpmn_id[next_object].split('_')[0] == 'parallelGateway':
+            merge.append(bpmn_id[next_object])
+        else:
+            merge.append(None)
+    merge_elements = merge.copy()
+    for idx, element in enumerate(merge):
+        if element is None:
+            merge_elements[idx] = False
+            continue
+        #count how many time the element is in the list
+        count = merge.count(element)
+        if count > 1:
+            merge_elements[idx] = True
+        else:
+            merge_elements[idx] = False
+    
+    return merge_elements
 
+def find_end_states(data):
+    end_states = {}
+    bpmn_id = data['BPMN_id']
+    for idx, link in enumerate(data['links']):
+        if link[0] is None:
+            continue
+        if bpmn_id[link[0]].split('_')[0] == 'exclusiveGateway':
+            end_states[bpmn_id[idx]] = 'state_'+str(idx)
+    return end_states
 
 def create_wizard_file(data, text_mapping):
     root = ET.Element('methodAndStyleWizard')
@@ -94,24 +137,28 @@ def create_wizard_file(data, text_mapping):
     for idx, Bpmn_id in enumerate(data['BPMN_id']):
         # Start Event
         element_type = Bpmn_id.split('_')[0]
-        if element_type == 'message':
+        if element_type == 'messageEvent':
             eventType = 'Message'
         elif element_type == 'event':
             eventType = 'None'
         if idx >= len(data['links']):
             continue
-        if check_start(data['links'][idx]) and (element_type=='event' or element_type=='message'):
+        if check_start(data['links'][idx]) and (element_type=='event' or element_type=='messageEvent'):
             startEvent = ET.SubElement(root, 'startEvent', attrib={'name': text_mapping[Bpmn_id], 'eventType': eventType}) 
     
     requestMessage = ET.SubElement(root, 'requestMessage')
     requester = ET.SubElement(root, 'requester')
     
     endEvents = ET.SubElement(root, 'endStates')
+
+    print('mapping: ', text_mapping)
+    
+    # Add end states event to the collaboration element
     for idx, Bpmn_id in enumerate(data['BPMN_id']):
         # End States
         if idx >= len(data['links']):
             continue
-        if check_end(data['links'][idx]) and Bpmn_id.split('_')[0] == 'event':
+        if check_end(data['links'][idx]) and (Bpmn_id.split('_')[0] == 'event' or Bpmn_id.split('_')[0] == 'messageEvent'):
             if text_mapping[Bpmn_id] == '':
                 text_mapping[Bpmn_id] = '(unnamed)'
             ET.SubElement(endEvents, 'endState', attrib={'name': text_mapping[Bpmn_id], 'eventType': 'None', 'isRegular': 'False'})
@@ -125,27 +172,56 @@ def create_wizard_file(data, text_mapping):
             activity = ET.SubElement(activities, 'activity', attrib={'name': text_mapping.get(activity_name, activity_name), 'performer': ''})
             endStates = ET.SubElement(activity, 'endStates')
             current_text, next_text = connect(data, text_mapping, idx)
-            if next_text is not None:
-                ET.SubElement(endStates, 'endState', attrib={'name': next_text, 'isRegular': 'True'})
+            if next_text is not None and len(next_text) == 1:
+                ET.SubElement(endStates, 'endState', attrib={'name': next_text[0], 'isRegular': 'True'})
             ET.SubElement(activity, 'subActivities')
             ET.SubElement(activity, 'subActivityFlows')
             ET.SubElement(activity, 'messageFlows')
+
+    endStates_text = find_end_states(data)
+    #print(endStates_text)
+
+    merge_object = find_merge(data['BPMN_id'], data['links'])
     
-    """activityFlows = ET.SubElement(root, 'activityFlows')
+    activityFlows = ET.SubElement(root, 'activityFlows')
     i=0
     for i, link in enumerate(data['links']):
-        if link[0] is None and link[1] is not None and (data['BPMN_id'][i].split('_')[0] == 'event' or data['BPMN_id'][i].split('_')[0] == 'message'):
+        # create flow with start event
+        if link[0] is None and link[1] is not None and (data['BPMN_id'][i].split('_')[0] == 'event' or data['BPMN_id'][i].split('_')[0] == 'messageEvent'):
             current_text, next_text = connect(data, text_mapping, i)
+            print('current_text: ', current_text)
+            print('next_text: ', next_text)
             if current_text is None or next_text is None:
                 continue
-            ET.SubElement(activityFlows, 'activityFlow', attrib={'startEvent': current_text, 'endState': '---', 'target': next_text, 'isMerging': 'False', 'isPredefined': 'True'})
+            ET.SubElement(activityFlows, 'activityFlow', attrib={'startEvent': current_text, 'endState': '---', 'target': next_text[0], 'isMerging': 'False', 'isPredefined': 'True'})
             i+=1
+        # create flow with tasks
         if link[0] is not None and link[1] is not None and data['BPMN_id'][i].split('_')[0] == 'task':
             current_text, next_text = connect(data, text_mapping, i)
             if current_text is None or next_text is None:
                 continue
-            ET.SubElement(activityFlows, 'activityFlow', attrib={'activity': current_text, 'endState': '---', 'target': next_text, 'isMerging': 'False', 'isPredefined': 'True'})
-            i+=1"""
+
+            if merge_object[i] == True:
+                merge = 'True'
+            else:
+                merge = 'False'
+  
+            if len(next_text) > 1:
+                for next in next_text:
+                    ET.SubElement(activityFlows, 'activityFlow', attrib={'activity': current_text, 'endState': '---', 'target': next, 'isMerging': merge, 'isPredefined': 'True'})
+            elif len(next_text) == 1:
+                ET.SubElement(activityFlows, 'activityFlow', attrib={'activity': current_text, 'endState': '---', 'target': next_text[0], 'isMerging': merge, 'isPredefined': 'True'})
+            else:
+                ET.SubElement(activityFlows, 'activityFlow', attrib={'activity': current_text, 'endState': '---', 'target': next_text, 'isMerging': merge, 'isPredefined': 'True'})
+            
+                
+
+            """if data['BPMN_id'][i] in endStates_text:
+                endState = endStates_text[data['BPMN_id'][i]]
+            else:
+                endState = '---'"""
+
+            i+=1
     
     ET.SubElement(root, 'participants')