update channel_mapping.py

channel_mapping.py

@@ -55,8 +55,8 @@ def mapping(input_file, loc_file, fill_mode):
 	#template_montage.plot()
 	#input_montage.plot()
 	
-	template = {}
-	input = {}
+	template_dict = {}
+	input_dict = {}
 
 	# convert all channel names to uppercase
 	for i in range(30):
@@ -64,14 +64,14 @@ def mapping(input_file, loc_file, fill_mode):
 		template_montage.rename_channels({channel: str.upper(channel)})
 
 		channel = str.upper(channel)
-		template[channel] = Channel(index=i, name=channel)
+		template_dict[channel] = Channel(index=i, name=channel)
 
 	for i in range(len(input_montage.ch_names)):
 		channel = input_montage.ch_names[i]
 		input_montage.rename_channels({channel: str.upper(channel)})
 
 		channel = str.upper(channel)
-		input[channel] = Channel(index=i, coord=input_montage.get_positions()['ch_pos'][channel])
+		input_dict[channel] = Channel(index=i, coord=input_montage.get_positions()['ch_pos'][channel])
 
 
 	new_idx = [-1]*30
@@ -95,23 +95,23 @@ def mapping(input_file, loc_file, fill_mode):
 	
 	for i in range(30):
 		channel = template_montage.ch_names[i]
-		if channel not in input.keys() | alias.keys():
+		if channel not in input_dict.keys() | alias.keys():
 			finish_flag = 0
 			continue
 
-		if channel not in input and channel in alias:
-			if alias[channel] in input:
+		if channel not in input_dict and channel in alias:
+			if alias[channel] in input_dict:
 				template_montage.rename_channels({channel: alias[channel]})
-				template[alias[channel]] = template.pop(channel)
-				template[alias[channel]].name = alias[channel]
+				template_dict[alias[channel]] = template_dict.pop(channel)
+				template_dict[alias[channel]].name = alias[channel]
 				channel = alias[channel]
 			else:
 				finish_flag = 0
 				continue
 
-		new_idx[i] = input[channel].index
+		new_idx[i] = input_dict[channel].index
 		new_idx_name[i] = channel  # tmp
-		input[channel].used = True
+		input_dict[channel].used = True
 
 	if finish_flag == 1:
 		second2 = time.time()
@@ -119,7 +119,7 @@ def mapping(input_file, loc_file, fill_mode):
 		#print('new idx order:', new_idx)
 		#print('new_idx_name:', new_idx_name) # tmp
 
-		reorder_data(input_file, new_idx, fill_mode) # & save data to mapped.csv
+		reorder_data(input_file, new_idx, fill_mode)
 		return
 	
 	
@@ -143,8 +143,8 @@ def mapping(input_file, loc_file, fill_mode):
 
 				ver = 'front' if i<3 else 'center' if i==3 else 'back'
 				hor = 'left' if j<2 else 'center' if j==2 else 'right'
-				template[channel].topo_index = [i, j]
-				template[channel].topo_location = [ver, hor]
+				template_dict[channel].topo_index = [i, j]
+				template_dict[channel].topo_location = [ver, hor]
 
 				if i > 1 and j in [0, 4]:
 					temporal_channels.append(channel)
@@ -153,20 +153,18 @@ def mapping(input_file, loc_file, fill_mode):
 		
 
 	# ensure that CZ is found or imputed by another channel
-	if 'CZ' not in input and fill_mode=='adjacent':
+	if 'CZ' not in input_dict and fill_mode=='adjacent':
 		min_dist = 1e5
 		for channel in input_montage.ch_names:
-			curr_x, curr_y, curr_z = input[channel].coord.round(6)
-
+			curr_x, curr_y, curr_z = input_dict[channel].coord.round(6)
 			if curr_x**2 + curr_y**2 < min_dist:
 				nearest_channel = channel
 				min_dist = curr_x**2 + curr_y**2
 
-		if input[nearest_channel].used == True:
+		if input_dict[nearest_channel].used == True:
 			missing_channels.append('CZ')
-		input[nearest_channel].used = True
-		input['CZ'] = input[nearest_channel]
-
+		input_dict[nearest_channel].used = True
+		input_dict['CZ'] = input_dict[nearest_channel]
 		print("CZ's nearest neighbor:", nearest_channel)
 
 
@@ -176,13 +174,13 @@ def mapping(input_file, loc_file, fill_mode):
 
 		channel = template_montage.ch_names[i]
 
-		curr_prefix = template[channel].prefix()
-		curr_suffix = template[channel].suffix()
+		curr_prefix = template_dict[channel].prefix()
+		curr_suffix = template_dict[channel].suffix()
 
-		curr_row = template[channel].topo_index[0]
-		curr_col = template[channel].topo_index[1]
-		curr_ver = template[channel].topo_location[0]
-		curr_hor = template[channel].topo_location[1]
+		curr_row = template_dict[channel].topo_index[0]
+		curr_col = template_dict[channel].topo_index[1]
+		curr_ver = template_dict[channel].topo_location[0]
+		curr_hor = template_dict[channel].topo_location[1]
 
 		impute_channel = ''
 
@@ -194,25 +192,25 @@ def mapping(input_file, loc_file, fill_mode):
 		if fill_mode == 'zero':
 
 			impute_channel = curr_prefix+str(1) if curr_hor=='center' else curr_prefix+str(curr_suffix-2)
-			if impute_channel not in input or input[impute_channel].used==True:
+			if impute_channel not in input_dict or input_dict[impute_channel].used==True:
 				impute_channel = ''
 				new_idx[i] = z_row_idx
 				missing_channels.append(channel)
 				continue
 		
 		elif fill_mode == 'adjacent':
+			
+			ver_dir = 1 if curr_ver == 'front' else -1
+            
+			if curr_hor == 'center':  # FZ, FPZ...
 
-			if curr_hor == 'center':  # FZ, FPZ, CZ...
-
-				if curr_prefix+str(1) in input:  # ex: FZ<-F1
+				if curr_prefix+str(1) in input_dict:  # ex: FZ<-F1
 					impute_channel = curr_prefix + str(1)
-				elif curr_ver=='front' and template_topo_pos[curr_row+1][curr_col] in input: # ex: FZ<-FCZ
-					impute_channel = template_topo_pos[curr_row+1][curr_col]
 					
-				elif curr_ver=='back' and template_topo_pos[curr_row-1][curr_col] in input: # ex: PZ<-CPZ
-					impute_channel = template_topo_pos[curr_row-1][curr_col]
+				elif template_topo_pos[curr_row+ver_dir][curr_col] in input_dict: # ex: front:FZ<-FCZ,
+				    impute_channel = template_topo_pos[curr_row+ver_dir][curr_col]
 					
-				elif curr_prefix+str(3) in input: # ex: FZ<-F3
+				elif curr_prefix+str(3) in input_dict: # ex: FZ<-F3
 					impute_channel = curr_prefix + str(3)
 					
 				else:
@@ -221,23 +219,23 @@ def mapping(input_file, loc_file, fill_mode):
 			elif curr_hor == 'left' or curr_hor == 'right':
 
 				ver_ctrl = 1 if curr_ver=='front' else 2 if curr_ver=='back' else 3 # bit0: row+1, bit1: row-1
-				ver_dir = 1 if curr_ver == 'front' else -1
+				#ver_dir = 1 if curr_ver == 'front' else -1
 
 				# search horizontally
 				cnt = 0
 				tmp_suffix = curr_suffix
 				while tmp_suffix > 0:  # ex: F7<-F5/F3/F1
 					tmp_suffix = curr_suffix - 2*cnt
-					if curr_prefix+str(tmp_suffix) in input:
+					if curr_prefix+str(tmp_suffix) in input_dict:
 						impute_channel = curr_prefix + str(tmp_suffix)
 						break
 
 					if cnt == 2:
 						# check row+1/row-1
-						if ver_ctrl&1 and template_topo_pos[curr_row+1][curr_col] in input:
+						if ver_ctrl&1 and template_topo_pos[curr_row+1][curr_col] in input_dict:
 							impute_channel = template_topo_pos[curr_row+1][curr_col]
 							break
-						if ver_ctrl&2 and template_topo_pos[curr_row-1][curr_col] in input:
+						if ver_ctrl&2 and template_topo_pos[curr_row-1][curr_col] in input_dict:
 							impute_channel = template_topo_pos[curr_row-1][curr_col]
 							break
 					cnt += 1
@@ -247,7 +245,7 @@ def mapping(input_file, loc_file, fill_mode):
 					cnt = 0
 					tmp_row = curr_row + ver_dir
 					while tmp_row-ver_dir != 3:  # terminate if the last channel is a middle one
-						if template_topo_pos[tmp_row][curr_col] in input:
+						if template_topo_pos[tmp_row][curr_col] in input_dict:
 							impute_channel = template_topo_pos[tmp_row][curr_col]
 							break
 						tmp_row += ver_dir
@@ -256,18 +254,18 @@ def mapping(input_file, loc_file, fill_mode):
 				if impute_channel == '':
 					impute_channel = 'CZ'
 
-		new_idx[i] = input[impute_channel].index
+		new_idx[i] = input_dict[impute_channel].index
 		new_idx_name[i] = impute_channel  # tmp
-		if input[impute_channel].used == True:  # this channel is shared with others
+		if input_dict[impute_channel].used == True:  # this channel is shared with others
 			missing_channels.append(channel)
-		input[impute_channel].used = True
+		input_dict[impute_channel].used = True
 	
 	second2 = time.time()
 	print('Finish at stage 2 ! (',second2 - second1,'s)')
 	#print('new_idx:', new_idx)
 	#print('new_idx_name:', new_idx_name) # tmp
 	print('missing_channels:', missing_channels)
-	reorder_data(input_file, new_idx, fill_mode) # & save data to mapped.csv
+	reorder_data(input_file, new_idx, fill_mode)
 	
 	return