Anchor-annotator 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

{Anchor_annotator-0.3.1.dist-info → Anchor_annotator-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: Anchor_annotator
-Version: 0.3.1
+Version: 0.3.3
 Summary: Anchor annotator is a program for inspecting corpora for the Montreal Forced Aligner and correcting transcriptions and pronunciations.
 Home-page: https://github.com/MontrealCorpusTools/Anchor-annotator
 Author: Montreal Corpus Tools

{Anchor_annotator-0.3.1.dist-info → Anchor_annotator-0.3.3.dist-info}/RECORD

@@ -1,22 +1,22 @@
 anchor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 anchor/__main__.py,sha256=5ufG8lcx2x1am-04xI991AG7saJd24dxPw5JzjmB878,45
-anchor/_version.py,sha256=HzPz9rq3s1AiZXregKlqKaJJ2wGMtvH_a3V9la9CnpM,411
+anchor/_version.py,sha256=FKnJIExgNrZG2xJ0y_dGNBpxGbGBYylvfat-jHhLUuM,411
 anchor/command_line.py,sha256=EucG805HyWk_zkMO9RXv9Yj0I0JVdDLZb1_DX2_ISjM,503
 anchor/db.py,sha256=ef4lO6HtCKoxC9CorIc0ZbPxKpjHa576a0ZIBOWNU9E,4956
-anchor/main.py,sha256=eazWSZC94_nyBx_6rJ85hxC9wkCJYVP5FN_t9vnRm44,120658
-anchor/models.py,sha256=QBVSvmoqee0aIuMUMEbc7D5vDqsZ2sC7njwwI_NOuEg,96005
-anchor/plot.py,sha256=7Z_UeHyrrNk5oEFyFG_RdgiV4ray75juuuq-Nxb9FZo,104379
+anchor/main.py,sha256=wX6LLudTX5-HOxqBB-BGLXTdYJuDGQbV11gA2XBvQ8w,120605
+anchor/models.py,sha256=sCZf5wF6g1KorMgb0AifmqhaxZp5fiYmlmaJoow-tFI,95230
+anchor/plot.py,sha256=eNVG9sDdRA9_KKrHKSb2TF66bIluJsJzVnTGN3q-Brk,104878
 anchor/resources_rc.py,sha256=94wgxDTpP4Oy55Br7CZ_YnmvaqzHr4n-AydBPhZc-es,8427242
 anchor/settings.py,sha256=QdfBtJowHpkBLzJ_3bZRKxF1rJDBW9Z5kp83sJVz0pA,46965
 anchor/ui_corpus_manager.py,sha256=e3ybOd4UdYarrLBATxI8vIFnioa4R_BHrbsEz5mJ5eA,8564
 anchor/ui_error_dialog.py,sha256=c_QS0s1VaJEV9AhcrQZQyWHHpUPudWjJY1NI7Ytipio,3832
 anchor/ui_main_window.py,sha256=MYb4PtV1sHYgnc3QwPphKjU3LepzBJpxllhN4nyDook,63525
 anchor/ui_preferences.py,sha256=MOC2dY4qkViW9cUbC0DVSO7FLg-dGSbmR630WFQ6V9c,41843
-anchor/undo.py,sha256=PLtiAxC7U27vbega_p21vDRy4m9QBiHatDELlcNx08Y,32523
-anchor/widgets.py,sha256=th0zg-bgmHfwEbdfLKkt-OZUsmjy01PSAGGwcnlwwNM,158375
-anchor/workers.py,sha256=WceIGLKC8s9PqQw4VXPppBerHV1JUtJJxawHtQT1MXk,171183
-Anchor_annotator-0.3.1.dist-info/LICENSE,sha256=C0oIsblENEgWQ7XMNdYoXyXsIA5wa3YF0I9lK3H7A1s,1076
-Anchor_annotator-0.3.1.dist-info/METADATA,sha256=jW9m88-K6vEKIRvkqKddvCE9fB7PNMDoqXQamawFwNA,1500
-Anchor_annotator-0.3.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-Anchor_annotator-0.3.1.dist-info/top_level.txt,sha256=wX6ZKxImGRZKFQjs3f6XYw_TfbAp6Xs3SmbLfLbFAJ0,7
-Anchor_annotator-0.3.1.dist-info/RECORD,,
+anchor/undo.py,sha256=HXhrzV-T2JKwPd28KCaR9S6GWmi2Wr2Xk7IBEonRRCs,32564
+anchor/widgets.py,sha256=arL006v_cOGWudEF4Adbh_wRjiyFRGHP9BXOtfCr0h0,158301
+anchor/workers.py,sha256=iWLBGCg6jJr_OmVFJCRGMgF5Rw5G9IhpdNiqDv8ZBxU,171083
+Anchor_annotator-0.3.3.dist-info/LICENSE,sha256=C0oIsblENEgWQ7XMNdYoXyXsIA5wa3YF0I9lK3H7A1s,1076
+Anchor_annotator-0.3.3.dist-info/METADATA,sha256=KPemKhTKt7Rh3N4-k-f8V4tHoAkvoUKrhUlonplJrLo,1500
+Anchor_annotator-0.3.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+Anchor_annotator-0.3.3.dist-info/top_level.txt,sha256=wX6ZKxImGRZKFQjs3f6XYw_TfbAp6Xs3SmbLfLbFAJ0,7
+Anchor_annotator-0.3.3.dist-info/RECORD,,

anchor/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.1'
-__version_tuple__ = version_tuple = (0, 3, 1)
+__version__ = version = '0.3.3'
+__version_tuple__ = version_tuple = (0, 3, 3)

anchor/main.py

@@ -1087,8 +1087,8 @@ class MainWindow(QtWidgets.QMainWindow):
             )
             for m in (
                 session.query(anchor.db.AcousticModel)
-                .order_by(anchor.db.AcousticModel.last_used.desc())
                 .filter_by(available_locally=True)
+                .order_by(anchor.db.AcousticModel.last_used.desc())
             ):
                 a = QtGui.QAction(f"{m.path} [{m.name}]", parent=self)
                 a.setData(m.id)
@@ -1105,8 +1105,8 @@ class MainWindow(QtWidgets.QMainWindow):
 
             for m in (
                 session.query(anchor.db.Dictionary)
-                .order_by(anchor.db.Dictionary.last_used.desc())
                 .filter_by(available_locally=True)
+                .order_by(anchor.db.Dictionary.last_used.desc())
             ):
                 a = QtGui.QAction(text=f"{m.path} [{m.name}]", parent=self)
                 a.setData(m.id)
@@ -1122,8 +1122,8 @@ class MainWindow(QtWidgets.QMainWindow):
 
             for m in (
                 session.query(anchor.db.LanguageModel)
-                .order_by(anchor.db.LanguageModel.last_used.desc())
                 .filter_by(available_locally=True)
+                .order_by(anchor.db.LanguageModel.last_used.desc())
             ):
                 a = QtGui.QAction(text=f"{m.path} [{m.name}]", parent=self)
                 a.setData(m.id)
@@ -1139,8 +1139,8 @@ class MainWindow(QtWidgets.QMainWindow):
 
             for m in (
                 session.query(anchor.db.G2PModel)
-                .order_by(anchor.db.G2PModel.last_used.desc())
                 .filter_by(available_locally=True)
+                .order_by(anchor.db.G2PModel.last_used.desc())
             ):
                 a = QtGui.QAction(text=f"{m.path} [{m.name}]", parent=self)
                 a.setData(m.id)
@@ -1932,7 +1932,7 @@ class MainWindow(QtWidgets.QMainWindow):
                 session.query(anchor.db.AnchorCorpus).filter_by(current=True).update(
                     {anchor.db.AnchorCorpus.acoustic_model_id: m_id}
                 )
-                session.query(anchor.db.AcousticModel).filter_by(
+                session.query(anchor.db.AcousticModel).filter(
                     anchor.db.AcousticModel.id == m_id
                 ).update(
                     {
@@ -2003,9 +2003,7 @@ class MainWindow(QtWidgets.QMainWindow):
                 session.query(anchor.db.AnchorCorpus).filter_by(current=True).update(
                     {anchor.db.AnchorCorpus.dictionary_id: m_id}
                 )
-                session.query(anchor.db.Dictionary).filter_by(
-                    anchor.db.Dictionary.id == m_id
-                ).update(
+                session.query(anchor.db.Dictionary).filter(anchor.db.Dictionary.id == m_id).update(
                     {
                         anchor.db.Dictionary.last_used: datetime.datetime.now(),
                     }
@@ -2060,7 +2058,7 @@ class MainWindow(QtWidgets.QMainWindow):
                 session.query(anchor.db.AnchorCorpus).filter_by(current=True).update(
                     {anchor.db.AnchorCorpus.language_model_id: m_id}
                 )
-                session.query(anchor.db.LanguageModel).filter_by(
+                session.query(anchor.db.LanguageModel).filter(
                     anchor.db.LanguageModel.id == m_id
                 ).update(
                     {
@@ -2139,7 +2137,7 @@ class MainWindow(QtWidgets.QMainWindow):
                         anchor.db.AnchorCorpus.g2p_model_id: m_id,
                     }
                 )
-                session.query(anchor.db.G2PModel).filter_by(anchor.db.G2PModel.id == m_id).update(
+                session.query(anchor.db.G2PModel).filter(anchor.db.G2PModel.id == m_id).update(
                     {
                         anchor.db.G2PModel.last_used: datetime.datetime.now(),
                     }
@@ -2182,7 +2180,7 @@ class MainWindow(QtWidgets.QMainWindow):
                 session.query(anchor.db.AnchorCorpus).filter_by(current=True).update(
                     {anchor.db.AnchorCorpus.ivector_extractor_id: m_id}
                 )
-                session.query(anchor.db.IvectorExtractor).filter_by(
+                session.query(anchor.db.IvectorExtractor).filter(
                     anchor.db.IvectorExtractor.id == m_id
                 ).update(
                     {

anchor/models.py

@@ -166,6 +166,7 @@ class FileUtterancesModel(QtCore.QAbstractListModel):
 
     waveformReady = QtCore.Signal()
     utterancesReady = QtCore.Signal()
+    speakersChanged = QtCore.Signal()
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -325,6 +326,7 @@ class FileUtterancesModel(QtCore.QAbstractListModel):
             if utterance.speaker_id not in self.speakers:
                 self.speakers.append(utterance.speaker_id)
                 self.speaker_channel_mapping[utterance.speaker_id] = utterance.channel
+                self.speakersChanged.emit()
             self._speaker_indices[index] = utterance.speaker_id
 
     def merge_table_utterances(
@@ -848,6 +850,8 @@ class FileSelectionModel(QtCore.QItemSelectionModel):
         except sqlalchemy.orm.exc.DetachedInstanceError:
             new_file = True
         self.requested_utterance_id = utterance_id
+        self.selected_min_time = None
+        self.selected_max_time = None
         if new_file:
             self.fileAboutToChange.emit()
             self.model().set_file(file_id)
@@ -943,8 +947,7 @@ class CorpusSelectionModel(QtCore.QItemSelectionModel):
         self.currentRowChanged.connect(self.switch_utterance)
         # self.selectionChanged.connect(self.update_selection_audio)
         # self.selectionChanged.connect(self.update_selection_audio)
-        # self.model().changeCommandFired.connect(self.expire_current)
-        self.model().layoutChanged.connect(self.check_selection)
+        self.model().newResults.connect(self.check_selection)
         self.model().unlockCorpus.connect(self.fileChanged.emit)
 
     def set_current_utterance(self, utterance_id):
@@ -2275,32 +2278,6 @@ class CorpusModel(TableModel):
             self._speaker_indices[row],
         )
 
-    def fileAt(self, index) -> int:
-        if not isinstance(index, int):
-            if not index.isValid():
-                return None
-            index = index.row()
-        return self._file_indices[index]
-
-    def indexForUtterance(self, utterance_id: int, column: int = 1):
-        return self.createIndex(self.reversed_indices[utterance_id], column)
-
-    def rollback_changes(self):
-        self.unsaved_files = set()
-        self.session.rollback()
-        # self.query_session.expire_all()
-        self.databaseSynced.emit(False)
-        self.update_data()
-
-    def commit_changes(self):
-        self.session.bulk_update_mappings(
-            File, [{"id": x, "modified": True} for x in self.unsaved_files]
-        )
-
-        self.unsaved_files = set()
-        self.session.commit()
-        self.databaseSynced.emit(True)
-
     def finish_export_files(self):
         self.filesSaved.emit()
 
@@ -2586,7 +2563,6 @@ class CorpusTableModel(QtCore.QAbstractTableModel):
         self.session.commit()
         corpus_name = os.path.basename(corpus_directory)
         c = anchor.db.AnchorCorpus(name=corpus_name, path=corpus_directory, current=True, **kwargs)
-        print(c.current)
         self.session.add(c)
         self.session.commit()
         self.corpora.append(c)

anchor/plot.py

@@ -2510,6 +2510,16 @@ class SpeakerTier(pg.GraphicsObject):
         self.corpus_model.refreshUtteranceText.connect(self.refreshTexts)
         self.selection_model.selectionChanged.connect(self.update_select)
         self.selection_model.model().utterancesReady.connect(self.refresh)
+        self.file_model.speakersChanged.connect(self.update_available_speakers)
+        self.available_speakers = {}
+
+    def update_available_speakers(self):
+        self.available_speakers = {}
+        for speaker_id in self.file_model.speakers:
+            speaker_name = self.corpus_model.get_speaker_name(speaker_id)
+            self.available_speakers[speaker_name] = speaker_id
+        for reg in self.visible_utterances.values():
+            reg.available_speakers = self.available_speakers
 
     def wheelEvent(self, ev):
         self.receivedWheelEvent.emit(ev)

anchor/undo.py

@@ -6,7 +6,7 @@ import typing
 import pynini.lib
 import sqlalchemy
 from montreal_forced_aligner.data import WordType
-from montreal_forced_aligner.db import File, Pronunciation, Speaker, Utterance, Word
+from montreal_forced_aligner.db import File, Pronunciation, Speaker, Utterance, Word, bulk_update
 from PySide6 import QtCore, QtGui
 from sqlalchemy.orm import make_transient
 
@@ -643,18 +643,18 @@ class UpdateUtteranceSpeakerCommand(FileCommand):
                 )
             )
             session.flush()
+        mappings = []
         for u in self.utterances:
             u.speaker_id = self.new_speaker_id
-        session.query(Speaker).filter(
-            Speaker.id.in_(self.old_speaker_ids + [self.new_speaker_id])
-        ).update({Speaker.modified: True})
+            mappings.append({"id": u.id, "speaker_id": self.new_speaker_id})
+        bulk_update(self.corpus_model.session, Utterance, mappings)
 
     def _undo(self, session) -> None:
+        mappings = []
         for i, u in enumerate(self.utterances):
             u.speaker_id = self.old_speaker_ids[i]
-        session.query(Speaker).filter(
-            Speaker.id.in_(self.old_speaker_ids + [self.new_speaker_id])
-        ).update({Speaker.modified: True})
+            mappings.append({"id": u.id, "speaker_id": self.old_speaker_ids[i]})
+        bulk_update(self.corpus_model.session, Utterance, mappings)
 
     def update_data(self):
         super().update_data()

anchor/widgets.py

@@ -125,6 +125,7 @@ class MediaPlayer(QtMultimedia.QMediaPlayer):  # pragma: no cover
             or self.currentTime() >= self.maxTime()
         ):
             self.setCurrentTime(self.startTime())
+        self.timer.start()
         super(MediaPlayer, self).play()
         if fade_in:
             self.fade_in_anim.start()
@@ -155,8 +156,6 @@ class MediaPlayer(QtMultimedia.QMediaPlayer):  # pragma: no cover
             self.timer.stop()
             self.setCurrentTime(self.startTime())
             self.timeChanged.emit(self.currentTime())
-        elif state == QtMultimedia.QMediaPlayer.PlaybackState.PlayingState:
-            self.timer.start()
         elif state == QtMultimedia.QMediaPlayer.PlaybackState.PausedState:
             self.timer.stop()
 
@@ -183,10 +182,10 @@ class MediaPlayer(QtMultimedia.QMediaPlayer):  # pragma: no cover
     def set_volume(self, volume: int):
         if self.audioOutput() is None:
             return
-        linearVolume = QtMultimedia.QAudio.convertVolume(
+        linearVolume = QtMultimedia.QtAudio.convertVolume(
             volume / 100.0,
-            QtMultimedia.QAudio.VolumeScale.LogarithmicVolumeScale,
-            QtMultimedia.QAudio.VolumeScale.LinearVolumeScale,
+            QtMultimedia.QtAudio.VolumeScale.LogarithmicVolumeScale,
+            QtMultimedia.QtAudio.VolumeScale.LinearVolumeScale,
         )
         self.audioOutput().setVolume(linearVolume)
 
@@ -195,10 +194,10 @@ class MediaPlayer(QtMultimedia.QMediaPlayer):  # pragma: no cover
             return 100
         volume = self.audioOutput().volume()
         volume = int(
-            QtMultimedia.QAudio.convertVolume(
+            QtMultimedia.QtAudio.convertVolume(
                 volume,
-                QtMultimedia.QAudio.VolumeScale.LinearVolumeScale,
-                QtMultimedia.QAudio.VolumeScale.LogarithmicVolumeScale,
+                QtMultimedia.QtAudio.VolumeScale.LinearVolumeScale,
+                QtMultimedia.QtAudio.VolumeScale.LogarithmicVolumeScale,
             )
             * 100
         )

anchor/workers.py

@@ -3171,7 +3171,6 @@ class SpectrogramWorker(FunctionWorker):  # pragma: no cover
                 self.y = scipy.signal.resample(
                     self.y, int(self.y.shape[0] * max_sr / self.sample_rate)
                 )
-                # self.y = resampy.resample(self.y, self.sample_rate, max_sr, filter='kaiser_fast')
                 self.sample_rate = max_sr
             self.y = librosa.effects.preemphasis(self.y, coef=pre_emph_coeff)
             if self.stopped.is_set():