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

boris/transitions.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.
 
@@ -24,9 +24,9 @@ import logging
 import os
 import subprocess
 import tempfile
-from decimal import Decimal as dec
+from pathlib import Path
 
-from PyQt5.QtWidgets import QFileDialog, QMessageBox
+from PySide6.QtWidgets import QFileDialog, QMessageBox
 
 from . import config as cfg
 from . import dialog, export_observation, select_subj_behav
@@ -135,17 +135,15 @@ def create_transitions_gv_from_matrix(matrix, cutoff_all=0, cutoff_behavior=0, e
             else:
                 transitions[row.split("\t")[0]][behaviours[idx]] = int(r)
 
-    transitions_total_number = sum([sum(transitions[x].values()) for x in transitions])
+    """transitions_total_number = sum([sum(transitions[x].values()) for x in transitions])"""
 
     out = "digraph G { \n"
 
     for behaviour1 in behaviours:
         for behaviour2 in behaviours:
-
             if behaviour1 not in transitions or behaviour2 not in transitions:
                 return True, "Error: the file does not seem a transition matrix"
             if transitions[behaviour1][behaviour2]:
-
                 if edge_label == "percent_node":
                     if transitions[behaviour1][behaviour2] > cutoff_all:
                         out += '"{behaviour1}" -> "{behaviour2}" [label="{label:0.3f}"];\n'.format(
@@ -173,6 +171,8 @@ def transitions_matrix(self, mode):
     * number
     * frequencies_after_behaviors
     """
+    logging.debug("flag transitions_matrix function")
+
     # ask user observations to analyze
     _, selected_observations = select_observations.select_observations2(
         self, cfg.MULTIPLE, windows_title="Select observations for transitions matrix"
@@ -184,8 +184,8 @@ def transitions_matrix(self, mode):
     parameters = select_subj_behav.choose_obs_subj_behav_category(
         self,
         selected_observations,
-        flagShowIncludeModifiers=True,
-        flagShowExcludeBehaviorsWoEvents=False,
+        show_include_modifiers=True,
+        show_exclude_non_coded_behaviors=False,
         n_observations=len(selected_observations),
     )
 
@@ -197,21 +197,20 @@ def transitions_matrix(self, mode):
 
     flagMulti = False
     if len(parameters[cfg.SELECTED_SUBJECTS]) == 1:
-
-        fn = QFileDialog().getSaveFileName(
+        file_name, _ = QFileDialog().getSaveFileName(
             None,
             "Create matrix of transitions " + mode,
             "",
             "Transitions matrix files (*.txt *.tsv);;All files (*)",
         )
-        fileName = fn[0] if type(fn) is tuple else fn  # PyQt4/5
-
+        if not file_name:
+            return
     else:
-        exportDir = QFileDialog(self).getExistingDirectory(
+        exportDir = QFileDialog.getExistingDirectory(
             self,
             "Choose a directory to save the transitions matrices",
-            os.path.expanduser("~"),
-            options=QFileDialog(self).ShowDirsOnly,
+            str(Path.home()),
+            options=QFileDialog.ShowDirsOnly,
         )
         if not exportDir:
             return
@@ -219,15 +218,12 @@ def transitions_matrix(self, mode):
 
     flag_overwrite_all = False
     for subject in parameters[cfg.SELECTED_SUBJECTS]:
-
         logging.debug(f"subjects: {subject}")
 
         strings_list = []
         for obs_id in selected_observations:
             strings_list.append(
-                export_observation.events_to_behavioral_sequences(
-                    self.pj, obs_id, subject, parameters, self.behav_seq_separator
-                )
+                export_observation.events_to_behavioral_sequences(self.pj, obs_id, subject, parameters, self.behav_seq_separator)
             )
 
         sequences, observed_behaviors = behavioral_strings_analysis(strings_list, self.behav_seq_separator)
@@ -244,7 +240,6 @@ def transitions_matrix(self, mode):
 
         if flagMulti:
             try:
-
                 nf = f"{exportDir}{os.sep}{subject}_transitions_{mode}_matrix.tsv"
 
                 if os.path.isfile(nf) and not flag_overwrite_all:
@@ -264,11 +259,11 @@ def transitions_matrix(self, mode):
                 QMessageBox.critical(self, cfg.programName, f"The file {nf} can not be saved")
         else:
             try:
-                with open(fileName, "w") as outfile:
+                with open(file_name, "w") as outfile:
                     outfile.write(observed_matrix)
 
             except Exception:
-                QMessageBox.critical(self, cfg.programName, f"The file {fileName} can not be saved")
+                QMessageBox.critical(self, cfg.programName, f"The file {file_name} can not be saved")
 
 
 def transitions_dot_script():
@@ -276,21 +271,18 @@ def transitions_dot_script():
     create dot script (graphviz language) from transitions frequencies matrix
     """
 
-    fn = QFileDialog().getOpenFileNames(
+    file_names, _ = QFileDialog().getOpenFileNames(
         None,
         "Select one or more transitions matrix files",
         "",
         "Transitions matrix files (*.txt *.tsv);;All files (*)",
     )
-    fileNames = fn[0] if type(fn) is tuple else fn
 
     out = ""
 
-    for fileName in fileNames:
-        with open(fileName, "r") as infile:
-            result, gv = create_transitions_gv_from_matrix(
-                infile.read(), cutoff_all=0, cutoff_behavior=0, edge_label="percent_node"
-            )
+    for file_name in file_names:
+        with open(file_name, "r") as infile:
+            result, gv = create_transitions_gv_from_matrix(infile.read(), cutoff_all=0, cutoff_behavior=0, edge_label="percent_node")
             if result:
                 QMessageBox.critical(
                     None,
@@ -299,16 +291,24 @@ def transitions_dot_script():
                 )
                 return
 
-            with open(fileName + ".gv", "w") as f:
-                f.write(gv)
+            try:
+                with open(file_name + ".gv", "w") as file_out:
+                    file_out.write(gv)
+            except Exception:
+                QMessageBox.critical(
+                    None,
+                    cfg.programName,
+                    ("Error saving the file"),
+                )
+                return
 
-            out += f"<b>{fileName}.gv</b> created<br>"
+            out += f"<b>{file_name}.gv</b> created<br>"
 
     if out:
         QMessageBox.information(
             None,
             cfg.programName,
-            (f"{out}<br><br>The DOT scripts can be used with Graphviz or WebGraphviz " "to generate diagram"),
+            (f"{out}<br><br>The DOT scripts can be used with the Graphviz package or WebGraphviz to generate diagram"),
         )
 
 
@@ -332,20 +332,17 @@ def transitions_flow_diagram():
         )
         return
 
-    fn = QFileDialog().getOpenFileNames(
+    file_names, _ = QFileDialog().getOpenFileNames(
         None,
         "Select one or more transitions matrix files",
         "",
         "Transitions matrix files (*.txt *.tsv);;All files (*)",
     )
-    fileNames = fn[0] if type(fn) is tuple else fn
 
     out = ""
-    for fileName in fileNames:
-        with open(fileName, "r") as infile:
-            result, gv = create_transitions_gv_from_matrix(
-                infile.read(), cutoff_all=0, cutoff_behavior=0, edge_label="percent_node"
-            )
+    for file_name in file_names:
+        with open(file_name, "r") as infile:
+            result, gv = create_transitions_gv_from_matrix(infile.read(), cutoff_all=0, cutoff_behavior=0, edge_label="percent_node")
             if result:
                 QMessageBox.critical(
                     None,
@@ -354,18 +351,15 @@ def transitions_flow_diagram():
                 )
                 return
 
-            with open(tempfile.gettempdir() + os.sep + os.path.basename(fileName) + ".tmp.gv", "w") as f:
+            with open(tempfile.gettempdir() + os.sep + os.path.basename(file_name) + ".tmp.gv", "w") as f:
                 f.write(gv)
             result = subprocess.getoutput(
-                (
-                    f'dot -Tpng -o "{fileName}.png" '
-                    f'"{tempfile.gettempdir() + os.sep + os.path.basename(fileName)}.tmp.gv"'
-                )
+                (f'dot -Tpng -o "{file_name}.png" "{tempfile.gettempdir() + os.sep + os.path.basename(file_name)}.tmp.gv"')
             )
             if not result:
-                out += f"<b>{fileName}.png</b> created<br>"
+                out += f"<b>{file_name}.png</b> created<br>"
             else:
-                out += f"Problem with <b>{fileName}</b><br>"
+                out += f"Problem with <b>{file_name}</b><br>"
 
     if out:
         QMessageBox.information(None, cfg.programName, out)