celldetective 1.0.2.post1__py3-none-any.whl → 1.1.0__py3-none-any.whl

celldetective/__main__.py

@@ -15,10 +15,10 @@ class AppInitWindow(QMainWindow):
 	Initial window to set the experiment folder or create a new one.
 	"""
 
-	def __init__(self, parent=None):
+	def __init__(self, parent_window=None):
 		super().__init__()
 
-		self.parent = parent
+		self.parent_window = parent_window
 		self.Styles = Styles()
 		self.init_styles()
 		self.setWindowTitle("celldetective")

celldetective/events.py

@@ -1,49 +1,7 @@
 import numpy as np
 
-def switch_to_events(classes, times, max_times, first_detections=None, left_censored=False, FrameToMin=None):
-	
-	events = []
-	survival_times = []
-	if first_detections is None:
-		first_detections = np.zeros_like(max_times)
-		
-	for c,t,mt,ft in zip(classes, times, max_times, first_detections):
-
-		if left_censored:
-			#print('left censored is True: exclude cells that exist in first frame')
-			if ft>0.:
-				if c==0:
-					if t>0:
-						dt = t - ft
-						#print('event: dt = ',dt, t, ft)
-						if dt>0:
-							events.append(1)
-							survival_times.append(dt)
-				elif c==1:
-					dt = mt - ft
-					if dt>0:
-						events.append(0)
-						survival_times.append(dt)
-				else:
-					pass
-		else:
-			if c==0:
-				if t>0:
-					events.append(1)
-					survival_times.append(t - ft)
-			elif c==1:
-				events.append(0)
-				survival_times.append(mt - ft)
-			else:
-				pass
 
-	if FrameToMin is not None:
-		print('convert to minutes!', FrameToMin)
-		survival_times = [s*FrameToMin for s in survival_times]
-	return events, survival_times
-	  
-	
-def switch_to_events_v2(classes, event_times, max_times, origin_times=None, left_censored=True, FrameToMin=None):
+def switch_to_events(classes, event_times, max_times, origin_times=None, left_censored=True, FrameToMin=None):
 	
 
 	"""
@@ -144,6 +102,6 @@ def switch_to_events_v2(classes, event_times, max_times, origin_times=None, left
 				pass
 
 	if FrameToMin is not None:
-		print('convert to minutes!', FrameToMin)
+		#print('convert to minutes!', FrameToMin)
 		survival_times = [s*FrameToMin for s in survival_times]
 	return events, survival_times

celldetective/filters.py

@@ -35,8 +35,8 @@ def variance_filter(img, size):
 
 	size = int(size)
 	img = img.astype(float)
-	win_mean = snd.uniform_filter(img, (size,size))
-	win_sqr_mean = snd.uniform_filter(img**2, (size,size))
+	win_mean = snd.uniform_filter(img, (size,size), mode='wrap')
+	win_sqr_mean = snd.uniform_filter(img**2, (size,size), mode='wrap')
 	img = win_sqr_mean - win_mean**2
 
 	return img
@@ -45,8 +45,8 @@ def std_filter(img, size):
 
 	size = int(size)
 	img = img.astype(float)
-	win_mean = snd.uniform_filter(img, (size,size))
-	win_sqr_mean = snd.uniform_filter(img**2, (size, size))
+	win_mean = snd.uniform_filter(img, (size,size), mode='wrap')
+	win_sqr_mean = snd.uniform_filter(img**2, (size, size), mode='wrap')
 	win_sqr_mean[win_sqr_mean!=win_sqr_mean] = 0.
 	win_sqr_mean[win_sqr_mean<=0.] = 0. # add this to prevent sqrt from breaking
 	img = np.sqrt(win_sqr_mean - win_mean**2)
@@ -65,7 +65,6 @@ def otsu_filter(img, *kwargs):
 	return binary.astype(float)
 
 def local_filter(img, *kwargs):
-	print(*kwargs)
 	thresh = threshold_local(img.astype(float), *kwargs)
 	binary = img >= thresh
 	return binary.astype(float)

celldetective/gui/__init__.py

@@ -13,7 +13,7 @@ from .retrain_signal_model_options import ConfigSignalModelTraining
 from .retrain_segmentation_model_options import ConfigSegmentationModelTraining
 from .thresholds_gui import ThresholdConfigWizard
 from .seg_model_loader import SegmentationModelLoader
-from .process_block import ProcessPanel, NeighPanel
+from .process_block import ProcessPanel, NeighPanel, PreprocessingPanel
 from .analyze_block import AnalysisPanel
 from .control_panel import ControlPanel
 from .configure_new_exp import ConfigNewExperiment

celldetective/gui/analyze_block.py

@@ -1,10 +1,12 @@
-from PyQt5.QtWidgets import QFrame, QGridLayout, QComboBox, QLabel, QPushButton, QVBoxLayout, QHBoxLayout, QCheckBox, QMessageBox
+from PyQt5.QtWidgets import QFrame, QGridLayout, QComboBox, QSizePolicy, QSpacerItem, QLabel, QPushButton, QVBoxLayout, QHBoxLayout, QCheckBox, QMessageBox
 from PyQt5.QtCore import Qt, QSize
 from PyQt5.QtGui import QIcon
 
 from superqt.fonticon import icon
 from fonticon_mdi6 import MDI6
 import gc
+
+from celldetective.gui.plot_measurements import ConfigMeasurementsPlot
 from celldetective.io import get_segmentation_models_list, control_segmentation_napari, get_signal_models_list, control_tracking_btrack
 from celldetective.gui import ConfigSurvival, ConfigSignalPlot
 from celldetective.gui.gui_utils import QHSeperationLine
@@ -17,21 +19,23 @@ from glob import glob
 from natsort import natsorted
 import os
 import pandas as pd
+from celldetective.gui import Styles
 
-class AnalysisPanel(QFrame):
-	def __init__(self, parent, title=None):
+class AnalysisPanel(QFrame, Styles):
+	def __init__(self, parent_window, title=None):
 
 		super().__init__()		
-		self.parent = parent
+		self.parent_window = parent_window
 		self.title = title
 		if self.title is None:
 			self.title=''
-		self.exp_channels = self.parent.exp_channels
-		self.exp_dir = self.parent.exp_dir
-		self.soft_path = self.parent.parent.soft_path
+		self.exp_channels = self.parent_window.exp_channels
+		self.exp_dir = self.parent_window.exp_dir
+		self.soft_path = self.parent_window.parent_window.soft_path
 
 		self.setFrameStyle(QFrame.StyledPanel | QFrame.Raised)
 		self.grid = QVBoxLayout(self)
+		self.grid.setSpacing(20)
 		self.generate_header()
 	
 	def generate_header(self):
@@ -41,7 +45,7 @@ class AnalysisPanel(QFrame):
 
 		"""
 
-		panel_title = QLabel(self.title)
+		panel_title = QLabel("Survival")
 		panel_title.setStyleSheet("""
 			font-weight: bold;
 			padding: 0px;
@@ -51,19 +55,37 @@ class AnalysisPanel(QFrame):
 
 		self.survival_btn = QPushButton("plot survival")
 		self.survival_btn.setIcon(QIcon(QIcon(os.sep.join([self.soft_path,'celldetective','icons','survival2.png']))))
-		self.survival_btn.setStyleSheet(self.parent.parent.button_style_sheet_2)
+		self.survival_btn.setStyleSheet(self.button_style_sheet_2)
 		self.survival_btn.setIconSize(QSize(35, 35))
 		self.survival_btn.clicked.connect(self.configure_survival)
 		self.grid.addWidget(self.survival_btn)
 
+		signal_lbl = QLabel("Single-cell signals")
+		signal_lbl.setStyleSheet("""
+			font-weight: bold;
+			padding: 0px;
+			""")
+
+		self.grid.addWidget(signal_lbl, alignment=Qt.AlignCenter)
 
 		self.plot_signal_btn = QPushButton("plot signals")
 		self.plot_signal_btn.setIcon(QIcon(QIcon(os.sep.join([self.soft_path,'celldetective','icons','signals_icon.png']))))
-		self.plot_signal_btn.setStyleSheet(self.parent.parent.button_style_sheet_2)
+		self.plot_signal_btn.setStyleSheet(self.button_style_sheet_2)
 		self.plot_signal_btn.setIconSize(QSize(35, 35))
 		self.plot_signal_btn.clicked.connect(self.configure_plot_signals)
 		self.grid.addWidget(self.plot_signal_btn)
 
+		# self.plot_measurements_btn = QPushButton("plot measurements")
+		# self.plot_measurements_btn.setIcon(icon(MDI6.chart_waterfall, color='black'))
+		# #self.plot_measurements_btn.setIcon(QIcon(QIcon(os.sep.join([self.soft_path,'celldetective','icons','signals_icon.png']))))
+		# self.plot_measurements_btn.setStyleSheet(self.button_style_sheet_2)
+		# self.plot_measurements_btn.setIconSize(QSize(35, 35))
+		# self.plot_measurements_btn.clicked.connect(self.configure_plot_measurements)
+		# self.grid.addWidget(self.plot_measurements_btn)
+
+
+		verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Minimum, QSizePolicy.Expanding)
+		self.grid.addItem(verticalSpacer)
 
 	def configure_survival(self):
 		print('survival analysis starting!!!')
@@ -75,6 +97,11 @@ class AnalysisPanel(QFrame):
 		self.ConfigSignalPlot = ConfigSignalPlot(self)
 		self.ConfigSignalPlot.show()
 
+	def configure_plot_measurements(self):
+		print('plot measurements analysis starting!!!')
+		self.ConfigMeasurementsPLot = ConfigMeasurementsPlot(self)
+		self.ConfigMeasurementsPLot.show()
+
 
 	# 	self.select_all_btn = QPushButton()
 	# 	self.select_all_btn.setIcon(icon(MDI6.checkbox_blank_outline,color="black"))

celldetective/gui/btrack_options.py

@@ -14,27 +14,28 @@ import os
 import matplotlib.pyplot as plt
 from mpl_toolkits.axes_grid1 import make_axes_locatable
 from glob import glob
+from celldetective.gui import Styles
 
-class ConfigTracking(QMainWindow):
+class ConfigTracking(QMainWindow, Styles):
 	
 	"""
 	UI to set tracking parameters for bTrack.
 
 	"""
 
-	def __init__(self, parent=None):
+	def __init__(self, parent_window=None):
 		
 		super().__init__()
-		self.parent = parent
+		self.parent_window = parent_window
 		self.setWindowTitle("Configure tracking")
-		self.mode = self.parent.mode
-		self.exp_dir = self.parent.exp_dir
+		self.mode = self.parent_window.mode
+		self.exp_dir = self.parent_window.exp_dir
 		if self.mode=="targets":
 			self.config_name = os.sep.join(["configs", "btrack_config_targets.json"])
-			self.track_instructions_write_path = self.parent.exp_dir + os.sep.join(["configs","tracking_instructions_targets.json"])
+			self.track_instructions_write_path = self.parent_window.exp_dir + os.sep.join(["configs","tracking_instructions_targets.json"])
 		elif self.mode=="effectors":
 			self.config_name = os.sep.join(["configs","btrack_config_effectors.json"])
-			self.track_instructions_write_path = self.parent.exp_dir + os.sep.join(["configs", "tracking_instructions_effectors.json"])
+			self.track_instructions_write_path = self.parent_window.exp_dir + os.sep.join(["configs", "tracking_instructions_effectors.json"])
 		self.soft_path = get_software_location()
 		
 		exp_config = self.exp_dir +"config.ini"
@@ -42,7 +43,7 @@ class ConfigTracking(QMainWindow):
 		self.channel_names, self.channels = extract_experiment_channels(exp_config)
 		self.channel_names = np.array(self.channel_names)
 		self.channels = np.array(self.channels)
-		self.screen_height = self.parent.parent.parent.screen_height
+		self.screen_height = self.parent_window.parent_window.parent_window.screen_height
 
 		center_window(self)
 		self.setMinimumWidth(540)
@@ -84,7 +85,7 @@ class ConfigTracking(QMainWindow):
 		main_layout.addWidget(self.post_proc_frame)
 
 		self.submit_btn = QPushButton('Save')
-		self.submit_btn.setStyleSheet(self.parent.parent.parent.button_style_sheet)
+		self.submit_btn.setStyleSheet(self.parent_window.parent_window.parent_window.button_style_sheet)
 		self.submit_btn.clicked.connect(self.write_instructions)
 		main_layout.addWidget(self.submit_btn)
 
@@ -114,7 +115,7 @@ class ConfigTracking(QMainWindow):
 
 		self.select_post_proc_btn = QPushButton()
 		self.select_post_proc_btn.clicked.connect(self.activate_post_proc_options)
-		self.select_post_proc_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.select_post_proc_btn.setStyleSheet(self.button_select_all)
 		grid.addWidget(self.select_post_proc_btn, 0,0,1,4,alignment=Qt.AlignLeft)
 
 		self.post_proc_lbl = QLabel("POST-PROCESSING")
@@ -127,7 +128,7 @@ class ConfigTracking(QMainWindow):
 		self.collapse_post_proc_btn = QPushButton()
 		self.collapse_post_proc_btn.setIcon(icon(MDI6.chevron_down,color="black"))
 		self.collapse_post_proc_btn.setIconSize(QSize(20, 20))
-		self.collapse_post_proc_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.collapse_post_proc_btn.setStyleSheet(self.button_select_all)
 		grid.addWidget(self.collapse_post_proc_btn, 0, 0, 1, 4, alignment=Qt.AlignRight)
 
 		self.generate_post_proc_panel_contents()
@@ -148,7 +149,7 @@ class ConfigTracking(QMainWindow):
 
 		self.select_features_btn = QPushButton()
 		self.select_features_btn.clicked.connect(self.activate_feature_options)
-		self.select_features_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.select_features_btn.setStyleSheet(self.button_select_all)
 		grid.addWidget(self.select_features_btn, 0,0,1,4,alignment=Qt.AlignLeft)
 
 
@@ -162,7 +163,7 @@ class ConfigTracking(QMainWindow):
 		self.collapse_features_btn = QPushButton()
 		self.collapse_features_btn.setIcon(icon(MDI6.chevron_down,color="black"))
 		self.collapse_features_btn.setIconSize(QSize(20, 20))
-		self.collapse_features_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.collapse_features_btn.setStyleSheet(self.button_select_all)
 		grid.addWidget(self.collapse_features_btn, 0, 0, 1, 4, alignment=Qt.AlignRight)
 
 		self.generate_feature_panel_contents()
@@ -209,7 +210,7 @@ class ConfigTracking(QMainWindow):
 		self.min_tracklength_slider.setTickInterval(1)
 		self.min_tracklength_slider.setSingleStep(1)
 		self.min_tracklength_slider.setOrientation(1)
-		self.min_tracklength_slider.setRange(0,self.parent.parent.len_movie)
+		self.min_tracklength_slider.setRange(0,self.parent_window.parent_window.len_movie)
 		self.min_tracklength_slider.setValue(0)
 		tracklength_layout.addWidget(QLabel('Min. tracklength: '),40)
 		tracklength_layout.addWidget(self.min_tracklength_slider, 60)
@@ -261,13 +262,13 @@ class ConfigTracking(QMainWindow):
 
 		self.feature_lbl = QLabel("Add features:")
 		self.del_feature_btn = QPushButton("")
-		self.del_feature_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.del_feature_btn.setStyleSheet(self.button_select_all)
 		self.del_feature_btn.setIcon(icon(MDI6.trash_can,color="black"))
 		self.del_feature_btn.setToolTip("Remove feature")
 		self.del_feature_btn.setIconSize(QSize(20, 20))
 
 		self.add_feature_btn = QPushButton("")
-		self.add_feature_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.add_feature_btn.setStyleSheet(self.button_select_all)
 		self.add_feature_btn.setIcon(icon(MDI6.filter_plus,color="black"))
 		self.add_feature_btn.setToolTip("Add feature")
 		self.add_feature_btn.setIconSize(QSize(20, 20))		
@@ -324,7 +325,7 @@ class ConfigTracking(QMainWindow):
 		self.haralick_percentile_min_le = QLineEdit('0.01')
 		self.haralick_percentile_max_le = QLineEdit('99.9')
 		self.haralick_normalization_mode_btn = QPushButton()
-		self.haralick_normalization_mode_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.haralick_normalization_mode_btn.setStyleSheet(self.button_select_all)
 		self.haralick_normalization_mode_btn.setIcon(icon(MDI6.percent_circle,color="black"))
 		self.haralick_normalization_mode_btn.setIconSize(QSize(20, 20))		
 		self.haralick_normalization_mode_btn.setToolTip("Switch to absolute normalization values.")
@@ -336,12 +337,12 @@ class ConfigTracking(QMainWindow):
 		self.haralick_hist_btn = QPushButton()
 		self.haralick_hist_btn.clicked.connect(self.control_haralick_intensity_histogram)
 		self.haralick_hist_btn.setIcon(icon(MDI6.poll,color="k"))
-		self.haralick_hist_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.haralick_hist_btn.setStyleSheet(self.button_select_all)
 
 		self.haralick_digit_btn = QPushButton()
 		self.haralick_digit_btn.clicked.connect(self.control_haralick_digitalization)
 		self.haralick_digit_btn.setIcon(icon(MDI6.image_check,color="k"))
-		self.haralick_digit_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.haralick_digit_btn.setStyleSheet(self.button_select_all)
 
 		self.haralick_layout = QVBoxLayout()
 		self.haralick_layout.setContentsMargins(20,20,20,20)
@@ -431,7 +432,7 @@ class ConfigTracking(QMainWindow):
 		self.collapse_config_btn = QPushButton()
 		self.collapse_config_btn.setIcon(icon(MDI6.chevron_down,color="black"))
 		self.collapse_config_btn.setIconSize(QSize(20, 20))
-		self.collapse_config_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.collapse_config_btn.setStyleSheet(self.button_select_all)
 		grid.addWidget(self.collapse_config_btn, 0, 0, 1, 4, alignment=Qt.AlignRight)
 
 		self.generate_config_panel_contents()
@@ -471,7 +472,7 @@ class ConfigTracking(QMainWindow):
 		self.upload_btrack_config_btn.setIcon(icon(MDI6.plus,color="black"))
 		self.upload_btrack_config_btn.setIconSize(QSize(20, 20))
 		self.upload_btrack_config_btn.setToolTip("Upload a new bTrack configuration.")
-		self.upload_btrack_config_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.upload_btrack_config_btn.setStyleSheet(self.button_select_all)
 		self.upload_btrack_config_btn.clicked.connect(self.upload_btrack_config)
 		btrack_config_layout.addWidget(self.upload_btrack_config_btn, 5)  #4,3,1,1, alignment=Qt.AlignLeft
 
@@ -479,7 +480,7 @@ class ConfigTracking(QMainWindow):
 		self.reset_config_btn.setIcon(icon(MDI6.arrow_u_right_top,color="black"))
 		self.reset_config_btn.setIconSize(QSize(20, 20))
 		self.reset_config_btn.setToolTip("Reset the configuration to the default bTrack config.")
-		self.reset_config_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.reset_config_btn.setStyleSheet(self.button_select_all)
 		self.reset_config_btn.clicked.connect(self.reset_btrack_config)
 		btrack_config_layout.addWidget(self.reset_config_btn, 5)  #4,3,1,1, alignment=Qt.AlignLeft
 
@@ -817,7 +818,7 @@ class ConfigTracking(QMainWindow):
 		Load the first frame of the first movie found in the experiment folder as a sample.
 		"""
 
-		movies = glob(self.parent.parent.exp_dir + os.sep.join(["*","*","movie",self.parent.parent.movie_prefix+"*.tif"]))
+		movies = glob(self.parent_window.parent_window.exp_dir + os.sep.join(["*","*","movie",self.parent_window.parent_window.movie_prefix+"*.tif"]))
 		if len(movies)==0:
 			msgBox = QMessageBox()
 			msgBox.setIcon(QMessageBox.Warning)

celldetective/gui/classifier_widget.py

@@ -1,5 +1,5 @@
 from PyQt5.QtWidgets import QWidget, QLineEdit, QMessageBox, QHBoxLayout, QVBoxLayout, QPushButton, QLabel, QComboBox, \
-	QCheckBox
+	QCheckBox, QRadioButton, QButtonGroup
 from celldetective.gui.gui_utils import FigureCanvas, center_window, color_from_class
 import numpy as np
 import matplotlib.pyplot as plt
@@ -10,25 +10,27 @@ from PyQt5.QtCore import Qt, QSize
 import os
 from sklearn.metrics import r2_score
 from scipy.optimize import curve_fit
+from celldetective.gui import Styles
+from math import ceil
 
 def step_function(t, t_shift, dt):
 	return 1/(1+np.exp(-(t-t_shift)/dt))
 
-class ClassifierWidget(QWidget):
+class ClassifierWidget(QWidget, Styles):
 
-	def __init__(self, parent):
+	def __init__(self, parent_window):
 
 		super().__init__()
 
-		self.parent = parent
-		self.screen_height = self.parent.parent.parent.screen_height
-		self.screen_width = self.parent.parent.parent.screen_width
+		self.parent_window = parent_window
+		self.screen_height = self.parent_window.parent_window.parent_window.screen_height
+		self.screen_width = self.parent_window.parent_window.parent_window.screen_width
 		self.currentAlpha = 1.0
 
 		self.setWindowTitle("Custom classification")
 
-		self.mode = self.parent.mode
-		self.df = self.parent.df
+		self.mode = self.parent_window.mode
+		self.df = self.parent_window.df
 
 		is_number = np.vectorize(lambda x: np.issubdtype(x, np.number))
 		is_number_test = is_number(self.df.dtypes)
@@ -53,7 +55,7 @@ class ClassifierWidget(QWidget):
 		fig_btn_hbox = QHBoxLayout()
 		fig_btn_hbox.addWidget(QLabel(''), 95)
 		self.project_times_btn = QPushButton('')
-		self.project_times_btn.setStyleSheet(self.parent.parent.parent.button_select_all)
+		self.project_times_btn.setStyleSheet(self.parent_window.parent_window.parent_window.button_select_all)
 		self.project_times_btn.setIcon(icon(MDI6.math_integral,color="black"))
 		self.project_times_btn.setToolTip("Project measurements at all times.")
 		self.project_times_btn.setIconSize(QSize(20, 20))
@@ -111,7 +113,7 @@ class ClassifierWidget(QWidget):
 			self.features_cb[i].setCurrentIndex(i)
 
 			self.log_btns[i].setIcon(icon(MDI6.math_log,color="black"))
-			self.log_btns[i].setStyleSheet(self.parent.parent.parent.button_select_all)
+			self.log_btns[i].setStyleSheet(self.button_select_all)
 			self.log_btns[i].clicked.connect(lambda ch, i=i: self.switch_to_log(i))
 
 		hbox_classify = QHBoxLayout()
@@ -124,19 +126,47 @@ class ClassifierWidget(QWidget):
 		hbox_classify.addWidget(self.submit_query_btn, 20)
 		layout.addLayout(hbox_classify)
 
-		self.time_corr = QCheckBox('Time correlated event')
+		self.time_corr = QCheckBox('Time correlated')
+		self.time_corr.toggled.connect(self.activate_time_corr_options)
 		if "TRACK_ID" in self.df.columns:
 			self.time_corr.setEnabled(True)
 		else:
 			self.time_corr.setEnabled(False)
 		layout.addWidget(self.time_corr,alignment=Qt.AlignCenter)
+
+		self.irreversible_event_btn = QRadioButton('irreversible event')
+		self.unique_state_btn = QRadioButton('unique state')
+		self.time_corr_options = [self.irreversible_event_btn, self.unique_state_btn]
+		time_corr_btn_group = QButtonGroup()
+		self.unique_state_btn.click()
+		for btn in self.time_corr_options:
+			time_corr_btn_group.addButton(btn)
+			btn.setEnabled(False)
+
+		time_corr_layout = QHBoxLayout()
+		time_corr_layout.addWidget(self.unique_state_btn, 50, alignment=Qt.AlignCenter)
+		time_corr_layout.addWidget(self.irreversible_event_btn, 50,alignment=Qt.AlignCenter)
+		layout.addLayout(time_corr_layout)
+
+		layout.addWidget(QLabel())
+
 		self.submit_btn = QPushButton('apply')
+		self.submit_btn.setStyleSheet(self.button_style_sheet)
 		self.submit_btn.clicked.connect(self.submit_classification)
 		layout.addWidget(self.submit_btn, 30)
 
 		self.frame_slider.valueChanged.connect(self.set_frame)
 		self.alpha_slider.valueChanged.connect(self.set_transparency)
 
+	def activate_time_corr_options(self):
+
+		if self.time_corr.isChecked():
+			for btn in self.time_corr_options:
+				btn.setEnabled(True)
+		else:
+			for btn in self.time_corr_options:
+				btn.setEnabled(False)
+
 	def init_class(self):
 
 		self.class_name = 'custom'
@@ -277,14 +307,27 @@ class ClassifierWidget(QWidget):
 				for tid,track in self.df.groupby(['position','TRACK_ID']):
 					indices = track[self.class_name_user].index
 					status_values = track[stat_col].to_numpy()
-					if np.all([s==0 for s in status_values]):
-						self.df.loc[indices, self.class_name_user] = 1
-					elif np.all([s==1 for s in status_values]):
-						self.df.loc[indices, self.class_name_user] = 2
-						self.df.loc[indices, self.class_name_user.replace('class','status')] = 2
-					else:
-						self.df.loc[indices, self.class_name_user] = 2
-				self.estimate_time()
+					if self.irreversible_event_btn.isChecked():
+						if np.all([s==0 for s in status_values]):
+							self.df.loc[indices, self.class_name_user] = 1
+						elif np.all([s==1 for s in status_values]):
+							self.df.loc[indices, self.class_name_user] = 2
+							self.df.loc[indices, self.class_name_user.replace('class','status')] = 2
+						else:
+							self.df.loc[indices, self.class_name_user] = 2
+					elif self.unique_state_btn.isChecked():
+						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 self.irreversible_event_btn.isChecked():
+					self.estimate_time()
 		else:
 			self.group_name_user = 'group_' + self.name_le.text()
 			print(f'User defined characteristic group name: {self.group_name_user}.')