def od_dct(self, ini_idx, com_idx, match_idx, lst):
tem = []
for i, item in enumerate(lst):
level = 'OUT1'
if i == ini_idx:
level = 'IN2'
elif i == com_idx:
level = 'IN1'
elif i == match_idx:
level = 'IN3'
tem.append((item, COLOR_MAPPING.get(level)))
return OrderedDict(tem)
def od_dct(self, left, right, res, lst):
tem = []
for idx, item in enumerate(lst):
level = 'OUT1'
if item in left:
level = 'IN1'
elif item in right:
level = 'IN3'
elif item in res:
level = 'IN2'
tem.append((item, COLOR_MAPPING.get(level)))
return OrderedDict(tem)
def od_dct(self, left, pivot, right, lst):
pivot = lst.index(pivot)
tem = []
for idx, item in enumerate(lst):
level = 'OUT1'
if idx == pivot:
level = 'IN1'
elif item in left:
level = 'IN2'
elif item in right:
level = 'IN4'
tem.append((item, COLOR_MAPPING.get(level)))
return OrderedDict(tem)
def last_item(self, lst):
tem = []
for i in lst:
tem.append((i, COLOR_MAPPING.get('OUT1')))
return OrderedDict(tem)