boris-behav-obs 8.12__py3-none-any.whl → 9.7.6__py3-none-any.whl

boris/select_subj_behav.py

@@ -1,7 +1,7 @@
 """
 BORIS
 Behavioral Observation Research Interactive Software
-Copyright 2012-2023 Olivier Friard
+Copyright 2012-2025 Olivier Friard
 
 This file is part of BORIS.
 
@@ -21,15 +21,16 @@ This file is part of BORIS.
 """
 
 import logging
-
-from PyQt5.QtCore import Qt
-from PyQt5.QtGui import QFont
-from PyQt5.QtWidgets import QCheckBox, QListWidgetItem, QMessageBox
 from decimal import Decimal as dec
+from typing import Optional
+
+from PySide6.QtCore import Qt
+from PySide6.QtGui import QFont
+from PySide6.QtWidgets import QCheckBox, QListWidgetItem, QMessageBox
+
 from . import config as cfg
 from . import gui_utilities, param_panel, project_functions
 from . import utilities as util
-from typing import Optional
 
 
 def choose_obs_subj_behav_category(
@@ -37,33 +38,37 @@ def choose_obs_subj_behav_category(
     selected_observations: list,
     start_coding: Optional[dec] = dec("NaN"),  #  Union[..., None]
     end_coding: Optional[dec] = dec("NaN"),
+    start_interval: Optional[dec] = dec("NaN"),
+    end_interval: Optional[dec] = dec("NaN"),
     maxTime: Optional[dec] = None,
-    flagShowIncludeModifiers: bool = True,
-    flagShowExcludeBehaviorsWoEvents: bool = True,
+    show_include_modifiers: bool = True,
+    show_exclude_non_coded_behaviors: bool = True,
     by_category: bool = False,
     n_observations: int = 1,
     show_time_bin_size: bool = False,
     window_title: str = "Select subjects and behaviors",
-):
+    show_exclude_non_coded_modifiers: bool = False,
+) -> dict:
     """
     show window for:
     - selection of subjects
     - selection of behaviors (based on selected subjects)
     - selection of time interval
     - inclusion/exclusion of modifiers
-    - inclusion/exclusion of behaviors without events (flagShowExcludeBehaviorsWoEvents == True)
+    - inclusion/exclusion of behaviors without events (show_exclude_non_coded_behaviors == True)
     - selection of time bin size (show_time_bin_size == True)
 
     Args:
         selected_observations (list): List of selected observations
         ...
+        show_exclude_non_coded_modifiers (bool): display the Exclude non coded modifiers checkbox
 
     Returns:
         dict: {"selected subjects": selectedSubjects,
             "selected behaviors": selectedBehaviors,
             "include modifiers": True/False,
             "exclude behaviors": True/False,
-            "time": TIME_FULL_OBS / TIME_EVENTS / TIME_ARBITRARY_INTERVAL
+            "time": TIME_FULL_OBS / TIME_EVENTS / TIME_ARBITRARY_INTERVAL / TIME_OBS_INTERVAL
             "start time": startTime,
             "end time": endTime
             }
@@ -77,34 +82,51 @@ def choose_obs_subj_behav_category(
     paramPanelWindow.pj = self.pj
     paramPanelWindow.extract_observed_behaviors = self.extract_observed_behaviors
 
-    paramPanelWindow.cbIncludeModifiers.setVisible(flagShowIncludeModifiers)
-    paramPanelWindow.cbExcludeBehaviors.setVisible(flagShowExcludeBehaviorsWoEvents)
+    paramPanelWindow.cbIncludeModifiers.setVisible(show_include_modifiers)
+    paramPanelWindow.cb_exclude_non_coded_modifiers.setVisible(show_exclude_non_coded_modifiers)
+    paramPanelWindow.cbExcludeBehaviors.setVisible(show_exclude_non_coded_behaviors)
     # show_time_bin_size:
     paramPanelWindow.frm_time_bin_size.setVisible(show_time_bin_size)
 
     if by_category:
         paramPanelWindow.cbIncludeModifiers.setVisible(False)
         paramPanelWindow.cbExcludeBehaviors.setVisible(False)
+        paramPanelWindow.cb_exclude_non_coded_modifiers.setVisible(False)
+
+    # set state of cb_exclude_non_coded_modifiers
+    paramPanelWindow.cb_exclude_non_coded_modifiers.setEnabled(paramPanelWindow.cbIncludeModifiers.isChecked())
 
     paramPanelWindow.media_duration = maxTime
     paramPanelWindow.start_coding = start_coding
     paramPanelWindow.end_coding = end_coding
+    paramPanelWindow.start_interval = start_interval
+    paramPanelWindow.end_interval = end_interval
 
-    paramPanelWindow.start_time.set_format(self.timeFormat)
-    paramPanelWindow.end_time.set_format(self.timeFormat)
+    if self.timeFormat == cfg.S:
+        paramPanelWindow.start_time.rb_seconds.setChecked(True)
+        paramPanelWindow.end_time.rb_seconds.setChecked(True)
+    if self.timeFormat == cfg.HHMMSS:
+        paramPanelWindow.start_time.rb_time.setChecked(True)
+        paramPanelWindow.end_time.rb_time.setChecked(True)
 
     if n_observations > 1:
         paramPanelWindow.frm_time_interval.setVisible(False)
     else:
         if (start_coding is None) or (start_coding.is_nan()):
+            paramPanelWindow.rb_observed_events.setEnabled(False)
             paramPanelWindow.frm_time_interval.setVisible(False)
             paramPanelWindow.rb_user_defined.setVisible(False)
+            paramPanelWindow.rb_obs_interval.setVisible(False)
             paramPanelWindow.rb_media_duration.setVisible(False)
         else:
             paramPanelWindow.frm_time_interval.setEnabled(False)
             paramPanelWindow.start_time.set_time(start_coding)
             paramPanelWindow.end_time.set_time(end_coding)
 
+    # check observation time interval
+    if start_interval is None or start_interval.is_nan() or end_interval is None or end_interval.is_nan():
+        paramPanelWindow.rb_obs_interval.setEnabled(False)
+
     if selected_observations:
         observedSubjects = project_functions.extract_observed_subjects(self.pj, selected_observations)
     else:
@@ -162,7 +184,6 @@ def choose_obs_subj_behav_category(
         categories = ["###no category###"]
 
     for category in categories:
-
         if category != "###no category###":
             if category == "":
                 paramPanelWindow.item = QListWidgetItem("No category")
@@ -180,17 +201,14 @@ def choose_obs_subj_behav_category(
             paramPanelWindow.lwBehaviors.addItem(paramPanelWindow.item)
 
         for behavior in [self.pj[cfg.ETHOGRAM][x][cfg.BEHAVIOR_CODE] for x in util.sorted_keys(self.pj[cfg.ETHOGRAM])]:
-
             if (categories == ["###no category###"]) or (
                 behavior
                 in [
                     self.pj[cfg.ETHOGRAM][x][cfg.BEHAVIOR_CODE]
                     for x in self.pj[cfg.ETHOGRAM]
-                    if cfg.BEHAVIOR_CATEGORY in self.pj[cfg.ETHOGRAM][x]
-                    and self.pj[cfg.ETHOGRAM][x][cfg.BEHAVIOR_CATEGORY] == category
+                    if cfg.BEHAVIOR_CATEGORY in self.pj[cfg.ETHOGRAM][x] and self.pj[cfg.ETHOGRAM][x][cfg.BEHAVIOR_CATEGORY] == category
                 ]
             ):
-
                 paramPanelWindow.item = QListWidgetItem(behavior)
                 if behavior in observedBehaviors:
                     paramPanelWindow.item.setCheckState(Qt.Checked)
@@ -219,30 +237,39 @@ def choose_obs_subj_behav_category(
     logging.debug(f"selected subjects: {selectedSubjects}")
     logging.debug(f"selected behaviors: {selectedBehaviors}")
 
-    startTime = paramPanelWindow.start_time.get_time()
-    endTime = paramPanelWindow.end_time.get_time()
-    """
-    if self.timeFormat == HHMMSS:
-        startTime = time2seconds(paramPanelWindow.teStartTime.time().toString(HHMMSSZZZ))
-        endTime = time2seconds(paramPanelWindow.teEndTime.time().toString(HHMMSSZZZ))
-    if self.timeFormat == S:
-        startTime = Decimal(paramPanelWindow.dsbStartTime.value())
-        endTime = Decimal(paramPanelWindow.dsbEndTime.value())
-    """
-    if startTime > endTime:
-        QMessageBox.warning(
-            None,
-            cfg.programName,
-            "The start time is after the end time",
-            QMessageBox.Ok | QMessageBox.Default,
-            QMessageBox.NoButton,
-        )
-        return {cfg.SELECTED_SUBJECTS: [], cfg.SELECTED_BEHAVIORS: []}
+    if paramPanelWindow.rb_user_defined.isChecked():
+        startTime = paramPanelWindow.start_time.get_time()
+        endTime = paramPanelWindow.end_time.get_time()
+
+        if startTime > endTime:
+            QMessageBox.warning(
+                None,
+                cfg.programName,
+                "The start time is after the end time",
+                QMessageBox.Ok | QMessageBox.Default,
+                QMessageBox.NoButton,
+            )
+            return {cfg.SELECTED_SUBJECTS: [], cfg.SELECTED_BEHAVIORS: []}
+
+    elif paramPanelWindow.rb_obs_interval.isChecked() and not ((start_interval is None) or start_interval.is_nan()):
+        startTime = paramPanelWindow.start_time.get_time()
+        endTime = paramPanelWindow.end_time.get_time()
+
+    else:
+        startTime = None
+        endTime = None
+
+    # if startTime is None:
+    #    startTime = dec("NaN")
+    # if endTime is None:
+    #    endTime = dec("NaN")
 
     if paramPanelWindow.rb_media_duration.isChecked():
         time_param = cfg.TIME_FULL_OBS
     if paramPanelWindow.rb_observed_events.isChecked():
         time_param = cfg.TIME_EVENTS
+    if paramPanelWindow.rb_obs_interval.isChecked():
+        time_param = cfg.TIME_OBS_INTERVAL
     if paramPanelWindow.rb_user_defined.isChecked():
         time_param = cfg.TIME_ARBITRARY_INTERVAL
 
@@ -251,6 +278,7 @@ def choose_obs_subj_behav_category(
         cfg.SELECTED_BEHAVIORS: selectedBehaviors,
         cfg.INCLUDE_MODIFIERS: paramPanelWindow.cbIncludeModifiers.isChecked(),
         cfg.EXCLUDE_BEHAVIORS: paramPanelWindow.cbExcludeBehaviors.isChecked(),
+        cfg.EXCLUDE_NON_CODED_MODIFIERS: paramPanelWindow.cb_exclude_non_coded_modifiers.isChecked(),
         "time": time_param,
         cfg.START_TIME: startTime,
         cfg.END_TIME: endTime,

boris/state_events.py

@@ -1,7 +1,7 @@
 """
 BORIS
 Behavioral Observation Research Interactive Software
-Copyright 2012-2023 Olivier Friard
+Copyright 2012-2025 Olivier Friard
 
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
@@ -17,15 +17,15 @@ Copyright 2012-2023 Olivier Friard
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
   MA 02110-1301, USA.
-"""
-"""
+
 Module containing functions for state events
 
 """
 
+import time
 from decimal import Decimal as dec
 
-from PyQt5.QtWidgets import QMessageBox
+from PySide6.QtWidgets import QMessageBox, QAbstractItemView
 
 from . import config as cfg
 from . import dialog, project_functions, select_observations
@@ -82,17 +82,16 @@ def check_state_events(self, mode: str = "all") -> None:
     results.exec_()
 
 
-def fix_unpaired_events(self):
+def fix_unpaired_events(self, silent_mode: bool = False):
     """
     fix unpaired state events
     """
 
     if self.observationId:
-
         r, msg = project_functions.check_state_events_obs(
             self.observationId, self.pj[cfg.ETHOGRAM], self.pj[cfg.OBSERVATIONS][self.observationId]
         )
-        if "not PAIRED" not in msg:
+        if not silent_mode and "not PAIRED" not in msg:
             QMessageBox.information(
                 None,
                 cfg.programName,
@@ -102,33 +101,53 @@ def fix_unpaired_events(self):
             )
             return
 
-        w = dialog.Ask_time(self.timeFormat)
+        w = dialog.Ask_time(0)
         w.setWindowTitle("Fix UNPAIRED state events")
-        w.label.setText("Fix UNPAIRED events at time")
+        w.label.setText("Fix UNPAIRED events at time:")
 
-        if w.exec_():
-            fix_at_time = w.time_widget.get_time()
+        if not w.exec_():
+            return
 
-            events_to_add = project_functions.fix_unpaired_state_events(
-                self.pj[cfg.ETHOGRAM],
-                self.pj[cfg.OBSERVATIONS][self.observationId],
-                fix_at_time - dec("0.001"),
+        fix_at_time = w.time_widget.get_time()
+        if fix_at_time.is_nan():
+            QMessageBox.warning(
+                self,
+                cfg.programName,
+                ("Select a time format"),
             )
+            return
 
-            if events_to_add:
-                self.pj[cfg.OBSERVATIONS][self.observationId][cfg.EVENTS].extend(events_to_add)
-                self.project_changed()
-                self.pj[cfg.OBSERVATIONS][self.observationId][cfg.EVENTS].sort()
-                self.load_tw_events(self.observationId)
-                item = self.twEvents.item(
-                    [
-                        event_idx
-                        for event_idx, event in enumerate(self.pj[cfg.OBSERVATIONS][self.observationId][cfg.EVENTS])
-                        if event[cfg.TIME] == fix_at_time
-                    ][0],
-                    0,
-                )
-                self.twEvents.scrollToItem(item)
+        events_to_add = project_functions.fix_unpaired_state_events(
+            self.pj[cfg.ETHOGRAM],
+            self.pj[cfg.OBSERVATIONS][self.observationId],
+            fix_at_time - dec("0.001"),
+        )
+
+        if events_to_add:
+            # determine the new frame index
+            if (self.pj[cfg.OBSERVATIONS][self.observationId][cfg.TYPE] == cfg.MEDIA) and self.playerType == cfg.MEDIA:
+                mem_time = self.getLaps()
+                for event in events_to_add:
+                    if not self.seek_mediaplayer(event[0]):
+                        time.sleep(0.1)
+                        frame_idx = self.get_frame_index()
+                        event[cfg.PJ_OBS_FIELDS[cfg.MEDIA][cfg.FRAME_INDEX]] = frame_idx
+                self.seek_mediaplayer(mem_time)
+
+            self.pj[cfg.OBSERVATIONS][self.observationId][cfg.EVENTS].extend(events_to_add)
+            self.project_changed()
+            self.pj[cfg.OBSERVATIONS][self.observationId][cfg.EVENTS].sort(key=lambda x: x[:3])
+            self.load_tw_events(self.observationId)
+
+            index = self.tv_events.model().index(
+                [
+                    event_idx
+                    for event_idx, event in enumerate(self.pj[cfg.OBSERVATIONS][self.observationId][cfg.EVENTS])
+                    if event[cfg.PJ_OBS_FIELDS[self.playerType][cfg.TIME]] == fix_at_time
+                ][0],
+                0,
+            )
+            self.tv_events.scrollTo(index, QAbstractItemView.EnsureVisible)
 
     # selected observations
     else:
@@ -139,11 +158,11 @@ def fix_unpaired_events(self):
         # check if state events are paired
         out: str = ""
         for obs_id in selected_observations:
-            r, msg = project_functions.check_state_events_obs(
-                obs_id, self.pj[cfg.ETHOGRAM], self.pj[cfg.OBSERVATIONS][obs_id]
-            )
+            r, msg = project_functions.check_state_events_obs(obs_id, self.pj[cfg.ETHOGRAM], self.pj[cfg.OBSERVATIONS][obs_id])
             if "NOT PAIRED" in msg.upper():
+                # determine max time of events
                 fix_at_time = max(x[0] for x in self.pj[cfg.OBSERVATIONS][obs_id][cfg.EVENTS])
+                # list of events to add to fix unpaired events
                 events_to_add = project_functions.fix_unpaired_state_events(
                     self.pj[cfg.ETHOGRAM], self.pj[cfg.OBSERVATIONS][obs_id], fix_at_time
                 )
@@ -152,9 +171,7 @@ def fix_unpaired_events(self):
                     self.pj[cfg.OBSERVATIONS][obs_id][cfg.EVENTS].extend(events_to_add)
 
                     # check if modified obs if fixed
-                    r, msg = project_functions.check_state_events_obs(
-                        obs_id, self.pj[cfg.ETHOGRAM], self.pj[cfg.OBSERVATIONS][obs_id]
-                    )
+                    r, msg = project_functions.check_state_events_obs(obs_id, self.pj[cfg.ETHOGRAM], self.pj[cfg.OBSERVATIONS][obs_id])
                     if "NOT PAIRED" in msg.upper():
                         out += f"The observation <b>{obs_id}</b> can not be automatically fixed.<br><br>"
                         self.pj[cfg.OBSERVATIONS][obs_id][cfg.EVENTS] = events_backup

boris/subjects_pad.py

@@ -1,7 +1,7 @@
 """
 BORIS
 Behavioral Observation Research Interactive Software
-Copyright 2012-2023 Olivier Friard
+Copyright 2012-2025 Olivier Friard
 
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
@@ -19,19 +19,17 @@ Copyright 2012-2023 Olivier Friard
   MA 02110-1301, USA.
 """
 
-from PyQt5.QtCore import *
-from PyQt5.QtGui import *
-from PyQt5.QtWidgets import *
+from PySide6.QtCore import Signal, QRect, QEvent, Qt
+from PySide6.QtWidgets import QGridLayout, QPushButton, QHBoxLayout, QWidget
 
 from . import config as cfg
 from . import utilities as util
 
 
 class SubjectsPad(QWidget):
-
-    clickSignal = pyqtSignal(str)
-    sendEventSignal = pyqtSignal(QEvent)
-    close_signal = pyqtSignal(QRect)
+    clickSignal = Signal(str)
+    sendEventSignal = Signal(QEvent)
+    close_signal = Signal(QRect)
 
     def __init__(self, pj, filtered_subjects, parent=None):
         super(SubjectsPad, self).__init__(parent)
@@ -63,7 +61,6 @@ class SubjectsPad(QWidget):
                 c += 1
 
     def addWidget(self, subject, i, j):
-
         self.grid.addWidget(Button(), i, j)
         index = self.grid.count() - 1
         widget = self.grid.itemAt(index).widget()

boris/synthetic_time_budget.py

@@ -1,7 +1,7 @@
 """
 BORIS
 Behavioral Observation Research Interactive Software
-Copyright 2012-2023 Olivier Friard
+Copyright 2012-2025 Olivier Friard
 
 This file is part of BORIS.
 
@@ -22,10 +22,9 @@ This file is part of BORIS.
 
 import logging
 import pathlib as pl
-from decimal import Decimal as dec
 
-from PyQt5.QtWidgets import QFileDialog, QMessageBox
-from PyQt5.QtGui import QFont, QTextOption, QTextCursor
+from PySide6.QtWidgets import QFileDialog, QMessageBox
+from PySide6.QtGui import QFont, QTextOption, QTextCursor
 
 from . import config as cfg
 from . import (
@@ -43,6 +42,8 @@ def synthetic_time_budget(self) -> None:
     Synthetic time budget
     """
 
+    logging.debug("synthetic time budget function")
+
     _, selected_observations = select_observations.select_observations2(
         self, mode=cfg.MULTIPLE, windows_title="Select observations for synthetic time budget"
     )
@@ -59,21 +60,26 @@ def synthetic_time_budget(self) -> None:
     if not_ok or not selected_observations:
         return
 
-    max_media_duration_all_obs, _ = observation_operations.media_duration(
-        self.pj[cfg.OBSERVATIONS], selected_observations
-    )
+    max_media_duration_all_obs, _ = observation_operations.media_duration(self.pj[cfg.OBSERVATIONS], selected_observations)
 
     start_coding, end_coding, _ = observation_operations.coding_time(self.pj[cfg.OBSERVATIONS], selected_observations)
 
+    start_interval, end_interval = observation_operations.time_intervals_range(self.pj[cfg.OBSERVATIONS], selected_observations)
+
     synth_tb_param = select_subj_behav.choose_obs_subj_behav_category(
         self,
         selected_observations,
         start_coding=start_coding,
         end_coding=end_coding,
+        # start_interval=start_interval,
+        # end_interval=end_interval,
+        start_interval=None,
+        end_interval=None,
         maxTime=max_media_duration_all_obs,
-        flagShowExcludeBehaviorsWoEvents=False,
+        show_exclude_non_coded_behaviors=False,
         by_category=False,
         n_observations=len(selected_observations),
+        show_exclude_non_coded_modifiers=True,
     )
 
     if synth_tb_param == {}:
@@ -87,9 +93,9 @@ def synthetic_time_budget(self) -> None:
     if not start_coding.is_nan():
         cancel_pressed, synth_tb_param[cfg.EXCLUDED_BEHAVIORS] = self.filter_behaviors(
             title="Select behaviors to exclude from the total time",
-            text=("The duration of the selected behaviors will " "be subtracted from the total time"),
+            text="The duration of the selected behaviors will be subtracted from the total time",
             table="",
-            behavior_type=[cfg.STATE_EVENT],
+            behavior_type=cfg.STATE_EVENT_TYPES,
         )
         if cancel_pressed:
             return
@@ -133,21 +139,22 @@ def synthetic_time_budget(self) -> None:
 
         output_format = cfg.FILE_NAME_SUFFIX[filter_]
 
-        if filter_ != cfg.TEXT_FILE and pl.Path(file_name).suffix != "." + output_format:
-            file_name = str(pl.Path(file_name)) + "." + output_format
+        if pl.Path(file_name).suffix != "." + output_format:
+            if filter_ != cfg.TEXT_FILE:
+                file_name = str(pl.Path(file_name)) + "." + output_format
+            else:
+                file_name = str(pl.Path(file_name))
             if pl.Path(file_name).is_file():
                 if (
-                    dialog.MessageDialog(
-                        cfg.programName, f"The file {file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE]
-                    )
+                    dialog.MessageDialog(cfg.programName, f"The file {file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE])
                     == cfg.CANCEL
                 ):
                     return
 
         with open(file_name, "wb") as f:
-            if filter_ in [cfg.TSV, cfg.CSV, cfg.HTML, cfg.TEXT_FILE]:
+            if filter_ in (cfg.TSV, cfg.CSV, cfg.HTML, cfg.TEXT_FILE):
                 f.write(str.encode(data_report.export(output_format)))
-            if output_format in [cfg.ODS, cfg.XLSX, cfg.XLS]:
+            if filter_ in (cfg.ODS, cfg.XLSX, cfg.XLS):
                 f.write(data_report.export(output_format))
 
 
@@ -178,6 +185,8 @@ def synthetic_binned_time_budget(self) -> None:
 
     start_coding, end_coding, _ = observation_operations.coding_time(self.pj[cfg.OBSERVATIONS], selected_observations)
 
+    start_interval, end_interval = observation_operations.time_intervals_range(self.pj[cfg.OBSERVATIONS], selected_observations)
+
     # exit with message if events do not have timestamp
     if start_coding.is_nan():
         QMessageBox.critical(
@@ -194,11 +203,16 @@ def synthetic_binned_time_budget(self) -> None:
         selected_observations,
         start_coding=start_coding,
         end_coding=end_coding,
+        # start_interval=start_interval,
+        # end_interval=end_interval,
+        start_interval=None,
+        end_interval=None,
         maxTime=max_media_duration_all_obs,
-        flagShowExcludeBehaviorsWoEvents=False,
+        show_exclude_non_coded_behaviors=False,
         by_category=False,
         n_observations=len(selected_observations),
         show_time_bin_size=True,
+        show_exclude_non_coded_modifiers=True,
     )
 
     if synth_tb_param == {}:
@@ -211,9 +225,9 @@ def synthetic_binned_time_budget(self) -> None:
     # ask for excluding behaviors durations from total time
     cancel_pressed, synth_tb_param[cfg.EXCLUDED_BEHAVIORS] = self.filter_behaviors(
         title="Select behaviors to exclude",
-        text=("The duration of the selected behaviors will " "be subtracted from the total time"),
+        text=("The duration of the selected behaviors will be subtracted from the total time"),
         table="",
-        behavior_type=[cfg.STATE_EVENT],
+        behavior_type=cfg.STATE_EVENT_TYPES,
     )
     if cancel_pressed:
         return
@@ -250,24 +264,27 @@ def synthetic_binned_time_budget(self) -> None:
         ]
 
         file_name, filter_ = QFileDialog().getSaveFileName(
-            self, "Synthetic time budget with time bin", "", ";;".join(file_formats)
+            self, "Save the Synthetic time budget with time bin", "", ";;".join(file_formats)
         )
         if not file_name:
             return
 
-        if filter_ != cfg.TEXT_FILE and pl.Path(file_name).suffix != "." + cfg.FILE_NAME_SUFFIX[filter_]:
-            file_name = str(pl.Path(file_name)) + "." + cfg.FILE_NAME_SUFFIX[filter_]
+        output_format = cfg.FILE_NAME_SUFFIX[filter_]
+
+        if pl.Path(file_name).suffix != "." + output_format:
+            if filter_ != cfg.TEXT_FILE:
+                file_name = str(pl.Path(file_name)) + "." + output_format
+            else:
+                file_name = str(pl.Path(file_name))
             if pl.Path(file_name).is_file():
                 if (
-                    dialog.MessageDialog(
-                        cfg.programName, f"The file {file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE]
-                    )
+                    dialog.MessageDialog(cfg.programName, f"The file {file_name} already exists.", (cfg.CANCEL, cfg.OVERWRITE))
                     == cfg.CANCEL
                 ):
                     return
 
         with open(file_name, "wb") as f:
             if filter_ in (cfg.TSV, cfg.CSV, cfg.HTML, cfg.TEXT_FILE):
-                f.write(str.encode(data_report.export(cfg.FILE_NAME_SUFFIX[filter_])))
+                f.write(str.encode(data_report.export(output_format)))
             if filter_ in (cfg.ODS, cfg.XLSX, cfg.XLS):
-                f.write(data_report.export(cfg.FILE_NAME_SUFFIX[filter_]))
+                f.write(data_report.export(output_format))