celldetective 1.1.1.post4__tar.gz → 1.2.1__tar.gz

{celldetective-1.1.1.post4 → celldetective-1.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: celldetective
-Version: 1.1.1.post4
+Version: 1.2.1
 Summary: description
 Home-page: http://github.com/remyeltorro/celldetective
 Author: Rémy Torro

celldetective-1.2.1/celldetective/__init__.py

@@ -0,0 +1,3 @@
+#from .tracking import track, clean_trajectories
+#from .measure import contour_of_instance_segmentation
+

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/extra_properties.py

@@ -1,15 +1,31 @@
 """
-Add extra properties to directly to regionprops
-Functions must take regionmask as first argument and optionally intensity_image as second argument
-If intensity is in function name, it will be replaced by the name of the channel. These measurements are applied automatically to all channels
+Copyright © 2022 Laboratoire Adhesion et Inflammation, Authored by Remy Torro, 
+Ksenija Dervanova.
+
+Add extra properties to measure using regionprops (measure_features). 
+The functions must take "regionmask" as the first argument, corresponding
+to the binary mask of each cell of interest generated by regionprops 
+and an optional "intensity_image" second argument, i.e. the associated 
+multichannel image crop. The measurement will be named after the function
+name. If a function function() has several outputs (e.g. due to a multichannel image),
+they will be labelled with an index as function-0, function-1... A routine 
+automatically replacements keywords "intensity-0", "intensity-1" and so on,
+with the actual name of the intensity channel (e.g. "intensity-0" -> 
+"brightfield_channel"). Due to this indexing behavior, avoid using digits 
+in the function names and prefer text instead: "intensity-99()" --> 
+"intensity-ninety-nine()".
+
+New functions appear automatically in the ConfigMeasurements widget of the
+GUI.
 
 """
+
 import warnings
 
 import numpy as np
 from scipy.ndimage import distance_transform_edt, center_of_mass
 from scipy.spatial.distance import euclidean
-
+from celldetective.utils import interpolate_nan, contour_of_instance_segmentation
 
 # Percentiles
 
@@ -44,6 +60,9 @@ def intensity_nanmean(regionmask, intensity_image):
 	return np.nanmean(intensity_image[regionmask])
 
 def intensity_centre_of_mass_displacement(regionmask, intensity_image):
+
+	intensity_image = interpolate_nan(intensity_image.copy())
+
 	y, x = np.mgrid[:regionmask.shape[0], :regionmask.shape[1]]
 	xtemp = x.copy()
 	ytemp = y.copy()
@@ -51,60 +70,69 @@ def intensity_centre_of_mass_displacement(regionmask, intensity_image):
 	centroid_x = intensity_weighted_center[1]
 	centroid_y = intensity_weighted_center[0]
 
+	#centroid_x = np.sum(xtemp * intensity_image) / np.sum(intensity_image)
 	geometric_centroid_x = np.sum(xtemp * regionmask) / np.sum(regionmask)
 	geometric_centroid_y = np.sum(ytemp * regionmask) / np.sum(regionmask)
-	distance = euclidean(np.array((geometric_centroid_y, geometric_centroid_x)), np.array((centroid_y, centroid_x)))
+	try:
+		distance = euclidean(np.array((geometric_centroid_y, geometric_centroid_x)), np.array((centroid_y, centroid_x)))
+	except:
+		distance = np.nan
+
 	delta_x = geometric_centroid_x - centroid_x
 	delta_y = geometric_centroid_y - centroid_y
 	direction_arctan = np.arctan2(delta_y, delta_x) * 180 / np.pi
 	if direction_arctan < 0:
 		direction_arctan += 360
-	return distance, direction_arctan
+
+	return distance, direction_arctan, centroid_x - geometric_centroid_x, centroid_y - geometric_centroid_y
 
 def intensity_radial_gradient(regionmask, intensity_image):
-	warnings.filterwarnings('ignore', message="Polyfit may be poorly conditioned")
-	cell_mask = regionmask.copy()
-	intensity = intensity_image.copy()
-	y = intensity[cell_mask].flatten()
-	x = distance_transform_edt(cell_mask)
-	x = x[cell_mask].flatten()
-	params = np.polyfit(x, y, 1)
-	line = np.poly1d(params)
 
-	return line.coefficients[0], line.coefficients[1]
+	try:
+		warnings.filterwarnings('ignore', message="Polyfit may be poorly conditioned")
+		cell_mask = regionmask.copy()
+		intensity = intensity_image.copy()
+		y = intensity[cell_mask].flatten()
+		x = distance_transform_edt(cell_mask)
+		x = x[cell_mask].flatten()
+		params = np.polyfit(x, y, 1)
+		line = np.poly1d(params)
+
+		return line.coefficients[0], line.coefficients[1]
+	except Exception as e:
+		print(e)
+		return np.nan, np.nan
 
 
 def intensity_centre_of_mass_displacement_edge(regionmask, intensity_image):
-	edt = distance_transform_edt(regionmask)
-	min_distance = 0
-	max_distance = 0.1*edt.max()
-	thresholded = (edt <= max_distance) * (edt > min_distance)
-	edge_mask = np.copy(regionmask)
-	edge_mask[np.where(thresholded == 0)] = 0
-	y, x = np.mgrid[:edge_mask.shape[0], :edge_mask.shape[1]]
-	xtemp = x.copy()
-	ytemp = y.copy()
-	intensity_edge = intensity_image.copy()
-	intensity_edge[np.where(edge_mask == 0)] = 0.
-	sum_intensity_edge = np.sum(intensity_edge)
-	sum_regionmask = np.sum(regionmask)
+	
+	intensity_image = interpolate_nan(intensity_image.copy())
+	edge_mask = contour_of_instance_segmentation(regionmask, 3)
 
-	if sum_intensity_edge != 0 and sum_regionmask != 0:
-		y, x = np.mgrid[:regionmask.shape[0], :regionmask.shape[1]]
+	if np.sum(edge_mask)>0:
+		
+		y, x = np.mgrid[:edge_mask.shape[0], :edge_mask.shape[1]]
 		xtemp = x.copy()
 		ytemp = y.copy()
-		intensity_weighted_center = center_of_mass(intensity_image, regionmask)
+		intensity_weighted_center = center_of_mass(intensity_image, edge_mask)
 		centroid_x = intensity_weighted_center[1]
 		centroid_y = intensity_weighted_center[0]
 
+		#centroid_x = np.sum(xtemp * intensity_image) / np.sum(intensity_image)
 		geometric_centroid_x = np.sum(xtemp * regionmask) / np.sum(regionmask)
 		geometric_centroid_y = np.sum(ytemp * regionmask) / np.sum(regionmask)
-		distance = euclidean(np.array((geometric_centroid_y, geometric_centroid_x)), np.array((centroid_y, centroid_x)))
+		
+		try:
+			distance = euclidean(np.array((geometric_centroid_y, geometric_centroid_x)), np.array((centroid_y, centroid_x)))
+		except:
+			distance = np.nan
+
 		delta_x = geometric_centroid_x - centroid_x
 		delta_y = geometric_centroid_y - centroid_y
 		direction_arctan = np.arctan2(delta_y, delta_x) * 180 / np.pi
 		if direction_arctan < 0:
 			direction_arctan += 360
-		return distance, direction_arctan
+
+		return distance, direction_arctan, centroid_x - geometric_centroid_x, centroid_y - geometric_centroid_y
 	else:
-		return np.nan, np.nan
+		return np.nan, np.nan, np.nan, np.nan

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/gui/__init__.py

@@ -9,6 +9,7 @@ from .survival_ui import ConfigSurvival
 from .plot_signals_ui import ConfigSignalPlot
 from .signal_annotator_options import ConfigSignalAnnotator
 from .signal_annotator import SignalAnnotator
+from .signal_annotator2 import SignalAnnotator2
 from .retrain_signal_model_options import ConfigSignalModelTraining
 from .retrain_segmentation_model_options import ConfigSegmentationModelTraining
 from .thresholds_gui import ThresholdConfigWizard

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/gui/analyze_block.py

@@ -1,4 +1,5 @@
-from PyQt5.QtWidgets import QFrame, QGridLayout, QComboBox, QSizePolicy, QSpacerItem, QLabel, QPushButton, QVBoxLayout, QHBoxLayout, QCheckBox, QMessageBox
+from PyQt5.QtWidgets import QFrame, QGridLayout, QComboBox, QLabel, QPushButton, QVBoxLayout, QHBoxLayout, QCheckBox, \
+    QMessageBox, QSpacerItem, QSizePolicy
 from PyQt5.QtCore import Qt, QSize
 from PyQt5.QtGui import QIcon
 

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/gui/classifier_widget.py

@@ -279,9 +279,12 @@ class ClassifierWidget(QWidget, Styles):
 			try:
 				if cols_in_df:
 					self.selection = self.df.dropna(subset=cols).query(query).index
+					null_selection = self.df[self.df.loc[:,cols].isna().any(axis=1)].index
+					self.df.loc[null_selection, self.class_name] = np.nan
+					self.df.loc[self.selection, self.class_name] = 0
 				else:
-					self.selection = self.df.query(query).index
-				self.df.loc[self.selection, self.class_name] = 0
+					self.df.loc[:, self.class_name] = np.nan
+
 			except Exception as e:
 				print(e)
 				print(self.df.columns)
@@ -375,14 +378,16 @@ class ClassifierWidget(QWidget, Styles):
 						frames = track['FRAME'].to_numpy()
 						t_first = track['t_firstdetection'].to_numpy()[0]
 						median_status = np.nanmedian(status_values[frames>=t_first])
-						c = ceil(median_status)
-						if c==0:
-							self.df.loc[indices, self.class_name_user] = 1
-							self.df.loc[indices, self.class_name_user.replace('class','t')] = -1
-						elif c==1:
-							self.df.loc[indices, self.class_name_user] = 2
-							self.df.loc[indices, self.class_name_user.replace('class','t')] = -1
+						if median_status==median_status:
+							c = ceil(median_status)
+							if c==0:
+								self.df.loc[indices, self.class_name_user] = 1
+								self.df.loc[indices, self.class_name_user.replace('class','t')] = -1
+							elif c==1:
+								self.df.loc[indices, self.class_name_user] = 2
+								self.df.loc[indices, self.class_name_user.replace('class','t')] = -1
 				if self.irreversible_event_btn.isChecked():
+					self.df.loc[self.df[self.class_name_user]!=2, self.class_name_user.replace('class', 't')] = -1
 					self.estimate_time()
 		else:
 			self.group_name_user = 'group_' + self.name_le.text()
@@ -414,6 +419,7 @@ class ClassifierWidget(QWidget, Styles):
 		# reset
 		#self.init_class()
 		#self.update_props_scatter()
+		self.parent_window.parent_window.update_position_options()
 		self.close()
 
 	def switch_to_log(self, i):

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/gui/control_panel.py

@@ -3,7 +3,7 @@ from PyQt5.QtWidgets import QMainWindow, QComboBox, QPushButton, QHBoxLayout, QL
 from PyQt5.QtCore import Qt, QSize
 from PyQt5.QtGui import QIcon
 from celldetective.gui.gui_utils import center_window, QHSeperationLine
-from celldetective.utils import _extract_labels_from_config, ConfigSectionMap, extract_experiment_channels
+from celldetective.utils import _extract_labels_from_config, ConfigSectionMap, extract_experiment_channels, extract_identity_col
 from celldetective.gui import ConfigEditor, ProcessPanel, PreprocessingPanel, AnalysisPanel, NeighPanel
 from natsort import natsorted
 from glob import glob
@@ -414,26 +414,46 @@ class ControlPanel(QMainWindow, Styles):
 		self.pos = self.position_list.currentText()
 		panels = [self.ProcessEffectors, self.ProcessTargets]
 		if self.position_list.currentText()=="*":
+			
 			for p in panels:
 				p.check_seg_btn.setEnabled(False)
 				p.check_tracking_result_btn.setEnabled(False)
+
 			self.ProcessTargets.view_tab_btn.setEnabled(True)
 			self.ProcessEffectors.view_tab_btn.setEnabled(True)
+			self.NeighPanel.view_tab_btn.setEnabled(True)
+			self.ProcessEffectors.signal_analysis_action.setEnabled(True)
+			self.ProcessTargets.signal_analysis_action.setEnabled(True)
+
 			self.ProcessTargets.check_seg_btn.setEnabled(False)
 			self.ProcessEffectors.check_seg_btn.setEnabled(False)
+
 			self.ProcessTargets.check_tracking_result_btn.setEnabled(False)
 			self.ProcessEffectors.check_tracking_result_btn.setEnabled(False)
+
 			self.ProcessEffectors.check_measurements_btn.setEnabled(False)
 			self.ProcessTargets.check_measurements_btn.setEnabled(False)
 			#self.ProcessTargets.signal_analysis_action.setEnabled(False)
 			#self.ProcessEffectors.signal_analysis_action.setEnabled(False)
+
 			self.ProcessTargets.check_signals_btn.setEnabled(False)
 			self.ProcessEffectors.check_signals_btn.setEnabled(False)
+			self.NeighPanel.check_signals_btn.setEnabled(False)
+			self.ProcessTargets.delete_tracks_btn.hide()
+			self.ProcessEffectors.delete_tracks_btn.hide()
+
 			self.view_stack_btn.setEnabled(False)
 		elif self.well_list.currentText()=='*':
 			self.ProcessTargets.view_tab_btn.setEnabled(True)
 			self.ProcessEffectors.view_tab_btn.setEnabled(True)	
-			self.view_stack_btn.setEnabled(False)	
+			self.NeighPanel.view_tab_btn.setEnabled(True)
+			self.view_stack_btn.setEnabled(False)
+			self.ProcessEffectors.signal_analysis_action.setEnabled(True)
+			self.ProcessTargets.signal_analysis_action.setEnabled(True)
+
+			self.delete_tracks_btn.hide()
+			self.ProcessTargets.delete_tracks_btn.hide()
+			self.ProcessEffectors.delete_tracks_btn.hide()
 		else:
 			if not self.well_list.currentText()=="*":
 				self.locate_selected_position()
@@ -453,10 +473,16 @@ class ControlPanel(QMainWindow, Styles):
 					self.ProcessEffectors.check_tracking_result_btn.setEnabled(False)
 
 				if os.path.exists(os.sep.join([self.pos,'output','tables','trajectories_effectors.csv'])):
-					cols = pd.read_csv(os.sep.join([self.pos,'output','tables','trajectories_effectors.csv']), nrows=1).columns.tolist()
+					df = pd.read_csv(os.sep.join([self.pos,'output','tables','trajectories_effectors.csv']), nrows=1)
+					id_col = extract_identity_col(df)
 					self.ProcessEffectors.check_measurements_btn.setEnabled(True)
-					if 'TRACK_ID' in cols:
+					if id_col=='TRACK_ID':
 						self.ProcessEffectors.check_signals_btn.setEnabled(True)
+						self.ProcessEffectors.delete_tracks_btn.show()
+						self.ProcessEffectors.signal_analysis_action.setEnabled(True)
+					else:
+						self.ProcessEffectors.signal_analysis_action.setEnabled(False)						
+
 					#self.ProcessEffectors.signal_analysis_action.setEnabled(True)
 					self.ProcessEffectors.view_tab_btn.setEnabled(True)
 					self.ProcessEffectors.classify_btn.setEnabled(True)
@@ -466,12 +492,19 @@ class ControlPanel(QMainWindow, Styles):
 					#self.ProcessEffectors.signal_analysis_action.setEnabled(False)
 					self.ProcessEffectors.view_tab_btn.setEnabled(False)
 					self.ProcessEffectors.classify_btn.setEnabled(False)
+					self.ProcessEffectors.delete_tracks_btn.hide()
+					self.ProcessEffectors.signal_analysis_action.setEnabled(False)
 
 				if os.path.exists(os.sep.join([self.pos,'output','tables','trajectories_targets.csv'])):
-					cols = pd.read_csv(os.sep.join([self.pos,'output','tables','trajectories_targets.csv']), nrows=1).columns.tolist()
+					df = pd.read_csv(os.sep.join([self.pos,'output','tables','trajectories_targets.csv']), nrows=1)
+					id_col = extract_identity_col(df)
 					self.ProcessTargets.check_measurements_btn.setEnabled(True)
-					if 'TRACK_ID' in cols:
+					if id_col=='TRACK_ID':
 						self.ProcessTargets.check_signals_btn.setEnabled(True)
+						self.ProcessTargets.signal_analysis_action.setEnabled(True)
+						self.ProcessTargets.delete_tracks_btn.show()
+					else:
+						self.ProcessTargets.signal_analysis_action.setEnabled(False)						
 					#self.ProcessTargets.signal_analysis_action.setEnabled(True)
 					self.ProcessTargets.view_tab_btn.setEnabled(True)
 					self.ProcessTargets.classify_btn.setEnabled(True)
@@ -481,4 +514,15 @@ class ControlPanel(QMainWindow, Styles):
 					#self.ProcessTargets.signal_analysis_action.setEnabled(False)
 					self.ProcessTargets.view_tab_btn.setEnabled(False)
 					self.ProcessTargets.classify_btn.setEnabled(False)
+					self.ProcessTargets.signal_analysis_action.setEnabled(False)				
+					self.ProcessTargets.delete_tracks_btn.hide()
+					self.ProcessTargets.signal_analysis_action.setEnabled(False)
+
+				if os.path.exists(os.sep.join([self.pos,'output','tables','trajectories_pairs.csv'])):
+					self.NeighPanel.view_tab_btn.setEnabled(True)
+					self.NeighPanel.check_signals_btn.setEnabled(True)
+				else:
+					self.NeighPanel.view_tab_btn.setEnabled(False)
+					self.NeighPanel.check_signals_btn.setEnabled(False)
+
 

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/gui/layouts.py

@@ -73,7 +73,7 @@ class ChannelNormGenerator(QVBoxLayout, Styles):
 			for tab in tables:
 				cols = pd.read_csv(tab, nrows=1).columns.tolist()
 				all_measurements.extend(cols)
-			all_measurements = np.unique(all_measurements)		
+			all_measurements = np.unique(all_measurements)
 
 		if self.mode=='signals':
 			generic_measurements = ['brightfield_channel', 'live_nuclei_channel', 'dead_nuclei_channel', 
@@ -245,13 +245,14 @@ class ChannelNormGenerator(QVBoxLayout, Styles):
 
 	def check_valid_channels(self):
 
-		if np.all([cb.currentText()=='--' for cb in self.channel_cbs]):
-			self.parent_window.submit_btn.setEnabled(False)
+		if hasattr(self.parent_window, "submit_btn"):
+			if np.all([cb.currentText()=='--' for cb in self.channel_cbs]):
+				self.parent_window.submit_btn.setEnabled(False)
 
 		if hasattr(self.parent_window, "spatial_calib_le"):
 			if self.parent_window.spatial_calib_le.text()!='--':
 				self.parent_window.submit_btn.setEnabled(True)
-		else:
+		elif hasattr(self.parent_window, "submit_btn"):
 			self.parent_window.submit_btn.setEnabled(True)
 
 

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/gui/neighborhood_options.py

@@ -90,6 +90,8 @@ class ConfigNeighborhoods(QWidget, Styles):
 
 		self.clear_previous_btn = QCheckBox('clear previous neighborhoods')
 		self.clear_previous_btn.setToolTip('Clear all previous neighborhood measurements.')
+		self.clear_previous_btn.setIcon(icon(MDI6.broom, color='black'))
+
 		main_layout.addWidget(self.clear_previous_btn, alignment=Qt.AlignRight)
 
 		main_layout.addWidget(QLabel(''))
@@ -228,11 +230,12 @@ class ConfigNeighborhoods(QWidget, Styles):
 
 		status_layout = QHBoxLayout()
 
-		status_layout.addWidget(QLabel('status: '), 30)
+		#status_layout.addWidget(QLabel('status: '), 30)
 
 		status_sublayout = QHBoxLayout()
 		self.reference_population_status_cb = QComboBox()
 		self.reference_population_status_cb.setToolTip('Status of the reference population.')
+		self.reference_population_status_cb.hide()
 		status_sublayout.addWidget(self.reference_population_status_cb,95)
 
 		self.reference_switch_status_btn = QPushButton("")
@@ -241,6 +244,7 @@ class ConfigNeighborhoods(QWidget, Styles):
 		self.reference_switch_status_btn.setIconSize(QSize(20, 20))
 		self.reference_switch_status_btn.clicked.connect(self.switch_not_reference)
 		self.reference_switch_status_btn.setToolTip('Invert status values.')
+		self.reference_switch_status_btn.hide()
 
 		status_sublayout.addWidget(self.reference_switch_status_btn, 5)
 
@@ -305,11 +309,13 @@ class ConfigNeighborhoods(QWidget, Styles):
 
 		self.cumulated_presence_btn = QCheckBox('cumulated presence')
 		self.cumulated_presence_btn.setToolTip("Compute the cumulated presence time of each neighbor around a reference cell.")
+		self.cumulated_presence_btn.setIcon(icon(MDI6.timer_outline, color='black'))
+
 		layout.addWidget(self.cumulated_presence_btn)
 
-		self.symmetrize_btn = QCheckBox('symmetrize')
-		self.symmetrize_btn.setToolTip("Write the neighborhood of the neighbor cells with respect to the reference cells.")
-		layout.addWidget(self.symmetrize_btn)
+		# self.symmetrize_btn = QCheckBox('symmetrize')
+		# self.symmetrize_btn.setToolTip("Write the neighborhood of the neighbor cells with respect to the reference cells.")
+		# layout.addWidget(self.symmetrize_btn)
 		
 		self.fill_cbs_of_neighbor_population()
 		self.neighbor_population_cb.currentIndexChanged.connect(self.fill_cbs_of_neighbor_population)
@@ -424,7 +430,7 @@ class ConfigNeighborhoods(QWidget, Styles):
 		neighborhood_options.update({'event_time_col': event_time_col})		
 
 		neighborhood_kwargs = {'mode': mode, 'status': status_options, 'not_status_option': [self.not_status_reference, self.not_status_neighbor], 
-							  'compute_cum_sum': self.cumulated_presence_btn.isChecked(), 'attention_weight': True, 'symmetrize': self.symmetrize_btn.isChecked(),
+							  'compute_cum_sum': self.cumulated_presence_btn.isChecked(), 'attention_weight': True, 'symmetrize': False,
 							  'include_dead_weight': True}
 
 		neighborhood_options.update({'neighborhood_kwargs': neighborhood_kwargs})
@@ -493,8 +499,6 @@ class ConfigNeighborhoods(QWidget, Styles):
 						neighborhood_kwargs = neigh_instructions['neighborhood_kwargs']
 						if 'compute_cum_sum' in neighborhood_kwargs:
 							self.cumulated_presence_btn.setChecked(neighborhood_kwargs['compute_cum_sum'])
-						if 'symmetrize' in neighborhood_kwargs:
-							self.symmetrize_btn.setChecked(neighborhood_kwargs['symmetrize'])
 						if 'status' in neighborhood_kwargs:
 							status_options = neighborhood_kwargs['status']
 							status_options = ['--' if s is None else s for s in status_options]
@@ -505,6 +509,6 @@ class ConfigNeighborhoods(QWidget, Styles):
 						if 'not_status_option' in neighborhood_kwargs:
 							not_status_option = neighborhood_kwargs['not_status_option']
 							if not_status_option[0]:
-								self.switch_not_reference.click()
+								self.reference_switch_status_btn.click()
 							if not_status_option[1]:
-								self.switch_not_neigh.click()
+								self.neighbor_switch_status_btn.click()

{celldetective-1.1.1.post4 → celldetective-1.2.1}/celldetective/gui/plot_signals_ui.py

@@ -6,7 +6,7 @@ from PyQt5.QtGui import QIcon, QDoubleValidator
 from sklearn.preprocessing import MinMaxScaler
 
 from celldetective.gui.gui_utils import center_window, FeatureChoice, ListWidget, QHSeperationLine, FigureCanvas, GeometryChoice, OperationChoice
-from superqt import QLabeledSlider
+from superqt import QLabeledSlider, QColormapComboBox
 from superqt.fonticon import icon
 from fonticon_mdi6 import MDI6
 from celldetective.utils import extract_experiment_channels, get_software_location, _extract_labels_from_config
@@ -33,6 +33,7 @@ from matplotlib.cm import viridis, tab10
 import math
 from celldetective.gui import Styles
 from matplotlib import colormaps
+import matplotlib.cm as mcm
 
 
 
@@ -117,9 +118,11 @@ class ConfigSignalPlot(QWidget, Styles):
 			""")
 		main_layout.addWidget(panel_title, alignment=Qt.AlignCenter)
 
-		labels = [QLabel('population: '), QLabel('class: '), QLabel('time of\ninterest: ')]
-		self.cb_options = [['targets','effectors'],['class'], ['t0']]
+		labels = [QLabel('population: '), QLabel('class: '), QLabel('time of\ninterest: '), QLabel('cmap: ')]
+		self.cb_options = [['targets','effectors'],['class'], ['t0'], list(plt.colormaps())]
 		self.cbs = [QComboBox() for i in range(len(labels))]
+		self.cbs[-1] = QColormapComboBox()
+		
 		self.cbs[0].currentIndexChanged.connect(self.set_classes_and_times)
 
 		choice_layout = QVBoxLayout()
@@ -128,9 +131,15 @@ class ConfigSignalPlot(QWidget, Styles):
 			hbox = QHBoxLayout()
 			hbox.addWidget(labels[i], 33)
 			hbox.addWidget(self.cbs[i],66)
-			self.cbs[i].addItems(self.cb_options[i])
+			if i < len(labels)-1:
+				self.cbs[i].addItems(self.cb_options[i])
 			choice_layout.addLayout(hbox)
-
+		
+		for cm in list(colormaps):
+			try:
+				self.cbs[-1].addColormap(cm)
+			except:
+				pass
 
 		self.cbs[0].setCurrentIndex(1)
 		self.cbs[0].setCurrentIndex(0)
@@ -150,6 +159,11 @@ class ConfigSignalPlot(QWidget, Styles):
 
 		self.abs_time_checkbox.stateChanged.connect(self.switch_ref_time_mode)
 
+		select_layout = QHBoxLayout()
+		select_layout.addWidget(QLabel('select cells\nwith query: '), 33)
+		self.query_le = QLineEdit()
+		select_layout.addWidget(self.query_le, 66)
+		main_layout.addLayout(select_layout)
 
 		time_calib_layout = QHBoxLayout()
 		time_calib_layout.setContentsMargins(20,20,20,20)
@@ -266,9 +280,18 @@ class ConfigSignalPlot(QWidget, Styles):
 			self.feature_two_cb.setEnabled(True)
 		else:
 			self.feature_two_cb.setEnabled(False)
+	
 	def compute_signals(self):
 
 		if self.df is not None:
+
+			try:
+				query_text = self.query_le.text()
+				if query_text != '':
+					self.df = self.df.query(query_text)
+			except Exception as e:
+				print(e, ' The query is misunderstood and will not be applied...')
+
 			self.feature_selected = self.feature_cb.currentText()
 			if self.checkBox_feature.isChecked():
 				self.second_feature_selected=self.feature_two_cb.currentText()
@@ -334,6 +357,10 @@ class ConfigSignalPlot(QWidget, Styles):
 				self.initialize_axis()
 			plt.tight_layout()
 
+			cmap_lbl = self.cbs[-1].currentText()
+			self.cmap = getattr(mcm, cmap_lbl)
+			self.ax.set_prop_cycle('color',[self.cmap(i) for i in np.linspace(0, 1, len(self.well_indices))])
+
 
 			self.fig.set_facecolor('none')  # or 'None'
 			self.fig.canvas.setStyleSheet("background-color: transparent;")
@@ -353,7 +380,6 @@ class ConfigSignalPlot(QWidget, Styles):
 				radio_hbox.addWidget(self.plot_options[i], 33, alignment=Qt.AlignCenter)
 			self.plot_btn_group.buttonClicked[int].connect(self.plot_survivals)
 
-			print(self.well_indices, self.position_indices)
 			if self.position_indices is not None:
 				if len(self.well_indices)>1 and len(self.position_indices)==1:
 					self.plot_btn_group.buttons()[0].click()
@@ -725,10 +751,10 @@ class ConfigSignalPlot(QWidget, Styles):
 				second_matrix='second_matrix_no_event'
 
 
-		colors = np.array([tab10(i / len(self.df_pos_info)) for i in range(len(self.df_pos_info))])
+		colors = np.array([self.cmap(i / len(self.df_pos_info)) for i in range(len(self.df_pos_info))])
 		if self.checkBox_feature.isChecked():
-			second_colors = tab10(np.linspace(0.5, 1.5, len(self.df_pos_info)))
-		well_color = [tab10(i / len(self.df_well_info)) for i in range(len(self.df_well_info))]
+			second_colors = self.cmap(np.linspace(0.5, 1.5, len(self.df_pos_info)))
+		well_color = [self.cmap(i / len(self.df_well_info)) for i in range(len(self.df_well_info))]
 
 		if self.plot_mode=='pos':
 			self.initialize_axis()
@@ -801,6 +827,8 @@ class ConfigSignalPlot(QWidget, Styles):
 		self.survival_window.canvas.draw()
 
 	def plot_line(self, line, color, label, mean_signal, ci_option=True, cell_lines_option=False, alpha_ci=0.5, alpha_cell_lines=0.5, std_signal=None, matrix=None):
+		
+
 		try:
 			if 'second' in str(mean_signal):
 				self.ax2.plot(line['timeline'] * self.FrameToMin, line[mean_signal], color=color, label=label)
@@ -1005,10 +1033,10 @@ class ConfigSignalPlot(QWidget, Styles):
 	def switch_ref_time_mode(self):
 		if self.abs_time_checkbox.isChecked():
 			self.frame_slider.setEnabled(True)
-			self.cbs[-1].setEnabled(False)
+			self.cbs[-2].setEnabled(False)
 		else:
 			self.frame_slider.setEnabled(False)
-			self.cbs[-1].setEnabled(True)
+			self.cbs[-2].setEnabled(True)
 
 	def switch_ci(self):