update

channel_mapping.py

@@ -11,41 +11,59 @@ def reorder_data(filename, old_idx, fill_mode):
 	filepath = os.path.dirname(str(filename))
 	old_data = utils.read_train_data(filename)
 	new_data = np.zeros((30, old_data.shape[1]))
-	print('old data shape: ', old_data.shape)
+	#print('old data shape: ', old_data.shape)
+	
+	zero_arr = np.zeros((1, old_data.shape[1]))
+	old_data = np.concatenate((old_data, zero_arr), axis=0)
 	
-	if fill_mode == 'zero':
-		zero_arr = np.zeros((1, old_data.shape[1]))
-		old_data = np.concatenate((old_data, zero_arr), axis=0)
-
 	for i in range(30):
-		new_data[i, :] = old_data[old_idx[i]-1, :]
-
-	print('new data shape: ', new_data.shape)
-	utils.save_data(new_data, filepath+'/mapped.csv')
+		curr_idx_set = old_idx[i]
+		#print("channel_{}'s index set: {}".format(i, curr_idx_set))
+		
+		if curr_idx_set == []:
+			new_data[i, :] = zero_arr
+		else:
+			tmp_data = [old_data[j, :] for j in curr_idx_set]
+			new_data[i, :] = np.mean(tmp_data, axis=0)
+
+	#print('new data shape: ', new_data.shape)
+	utils.save_data(new_data, filepath+'/temp_data/mapped.csv')
 	return
 
 
-class Channel:  # (DigMontage):
+class Channel:
 
-    def __init__(self, index, name=None, used=False, coord=None, topo_index=None, topo_location=None):
+	def __init__(self, index, name=None, used=False, coord=None, topo_index=None, topo_location=None):
 
-		# super().__init__()
-		# self._montage = montage
+		self.name = name
+		self.index = index
+		self.used = used
+		self.coord = coord
+		self.topo_index = topo_index
+		self.topo_location = topo_location
 
-        self.name = name
-        self.index = index
-        self.used = used
-        self.coord = coord
-        self.topo_index = topo_index
-        self.topo_location = topo_location
+	def prefix(self):
+		ret = ''.join(filter(str.isalpha, self.name))
+		return ret[:len(ret) - 1] if ret[-1] == 'Z' else ret
 
-    def prefix(self):
-        ret = ''.join(filter(str.isalpha, self.name))
-        return ret[:len(ret) - 1] if ret[-1] == 'Z' else ret
+	def suffix(self):
+		return -1 if self.name[-1] == 'Z' else int(''.join(filter(str.isdigit, self.name)))
 
-    def suffix(self):
-        return -1 if self.name[-1] == 'Z' else int(''.join(filter(str.isdigit, self.name)))
 
+def pack_data(new_idx, missing_channels, tpl_dict, in_dict, tpl_ordered_name, in_ordered_name):
+
+	# tmp...
+	for ch in in_dict:
+		in_dict[ch].coord = None
+	
+	return {
+		"newOrder" : [([i] if i!=-1 else []) for i in new_idx],
+		"missingIndex" : missing_channels,
+		"templateByName" : {k : v.__dict__ for k,v in tpl_dict.items()},  # dict, {name:object}
+		"templateByIndex" : tpl_ordered_name, # list
+		"inputByName" : {k : v.__dict__ for k,v in in_dict.items()},
+		"inputByIndex" : in_ordered_name
+	}
 
 def mapping(input_file, loc_file, fill_mode):
 	second1 = time.time()
@@ -71,12 +89,13 @@ def mapping(input_file, loc_file, fill_mode):
 		input_montage.rename_channels({channel: str.upper(channel)})
 
 		channel = str.upper(channel)
-		input_dict[channel] = Channel(index=i, coord=input_montage.get_positions()['ch_pos'][channel])
+		input_dict[channel] = Channel(index=i, name=channel, coord=input_montage.get_positions()['ch_pos'][channel])
 
 
 	new_idx = [-1]*30
-	new_idx_name = ['']*30  # tmp
+	#new_idx_name = ['']*30  # tmp
 	missing_channels = []
+	exact_missing_channels = []
 	z_row_idx = len(input_montage.ch_names)
 	
 	
@@ -96,6 +115,7 @@ def mapping(input_file, loc_file, fill_mode):
 	for i in range(30):
 		channel = template_montage.ch_names[i]
 		if channel not in input_dict.keys() | alias.keys():
+			exact_missing_channels.append(i)
 			finish_flag = 0
 			continue
 
@@ -106,11 +126,12 @@ def mapping(input_file, loc_file, fill_mode):
 				template_dict[alias[channel]].name = alias[channel]
 				channel = alias[channel]
 			else:
+				exact_missing_channels.append(i)
 				finish_flag = 0
 				continue
 
 		new_idx[i] = input_dict[channel].index
-		new_idx_name[i] = channel  # tmp
+		#new_idx_name[i] = channel  # tmp
 		input_dict[channel].used = True
 
 	if finish_flag == 1:
@@ -119,8 +140,18 @@ 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)
-		return
+		#reorder_data(input_file, new_idx, fill_mode)
+		channels_obj = pack_data(new_idx, [],
+						template_dict, input_dict,
+						template_montage.ch_names, input_montage.ch_names)
+		return channels_obj
+		
+	elif fill_mode == "mean":
+		channels_obj = pack_data(new_idx, exact_missing_channels,
+						template_dict, input_dict,
+						template_montage.ch_names, input_montage.ch_names)
+		return channels_obj
+	
 	
 	
 	
@@ -162,7 +193,7 @@ def mapping(input_file, loc_file, fill_mode):
 				min_dist = curr_x**2 + curr_y**2
 
 		if input_dict[nearest_channel].used == True:
-			missing_channels.append('CZ')
+			missing_channels.append(template_dict['CZ'].index)
 		input_dict[nearest_channel].used = True
 		input_dict['CZ'] = input_dict[nearest_channel]
 		print("CZ's nearest neighbor:", nearest_channel)
@@ -195,7 +226,7 @@ def mapping(input_file, loc_file, fill_mode):
 			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)
+				missing_channels.append(i)
 				continue
 		
 		elif fill_mode == 'adjacent':
@@ -255,17 +286,19 @@ def mapping(input_file, loc_file, fill_mode):
 					impute_channel = 'CZ'
 
 		new_idx[i] = input_dict[impute_channel].index
-		new_idx_name[i] = impute_channel  # tmp
+		#new_idx_name[i] = impute_channel  # tmp
 		if input_dict[impute_channel].used == True:  # this channel is shared with others
-			missing_channels.append(channel)
+			missing_channels.append(i)
 		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)
+	#print('missing_channels:', missing_channels)
+	#reorder_data(input_file, new_idx, fill_mode)
 	
-	return
+	channels_obj = pack_data(new_idx, missing_channels,
+						template_dict, input_dict,
+						template_montage.ch_names, input_montage.ch_names)
+	return channels_obj