Anchor-annotator 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

{Anchor_annotator-0.3.0.dist-info → Anchor_annotator-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: Anchor_annotator
-Version: 0.3.0
+Version: 0.3.2
 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.0.dist-info → Anchor_annotator-0.3.2.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=Jk2iAU7m-7Vx9XV1TtdD9ZoJraIncDq_4_Wd-qtUotg,411
+anchor/_version.py,sha256=9jP8Fo8egXoMs_T3DFqSuJYg4n9o9mnwYubl_hnut4k,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=jpSisbtASWnD_In1jgcvH7xA8-9HJFylY2AVGw1tqMU,95858
-anchor/plot.py,sha256=R0SCcK-QmkTkQRsGWRRq1M9vnmve7pyQRaYMvcKlC3Q,104473
+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=nfcSBJ0_30V_uWNDFLZ48T9p9hJcu0mpbW31VMfxMZg,63518
+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=Uqre4W6kh9TEm72oocBitj-KT6lSR3WX7eB3wVZZYQM,171238
-Anchor_annotator-0.3.0.dist-info/LICENSE,sha256=C0oIsblENEgWQ7XMNdYoXyXsIA5wa3YF0I9lK3H7A1s,1076
-Anchor_annotator-0.3.0.dist-info/METADATA,sha256=CRpxOymwZoBYlSWMAnvByNOMEI73Nnv4vn0LyYZl36Y,1500
-Anchor_annotator-0.3.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-Anchor_annotator-0.3.0.dist-info/top_level.txt,sha256=wX6ZKxImGRZKFQjs3f6XYw_TfbAp6Xs3SmbLfLbFAJ0,7
-Anchor_annotator-0.3.0.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.2.dist-info/LICENSE,sha256=C0oIsblENEgWQ7XMNdYoXyXsIA5wa3YF0I9lK3H7A1s,1076
+Anchor_annotator-0.3.2.dist-info/METADATA,sha256=YcbnBTVx6O59V8W-6t9wu5GVrQBvGcXWDK3fCWHhkvE,1500
+Anchor_annotator-0.3.2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+Anchor_annotator-0.3.2.dist-info/top_level.txt,sha256=wX6ZKxImGRZKFQjs3f6XYw_TfbAp6Xs3SmbLfLbFAJ0,7
+Anchor_annotator-0.3.2.dist-info/RECORD,,

anchor/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.0'
-__version_tuple__ = version_tuple = (0, 3, 0)
+__version__ = version = '0.3.2'
+__version_tuple__ = version_tuple = (0, 3, 2)

anchor/models.py

@@ -68,18 +68,20 @@ class TextFilterQuery:
         if not self.regex:
             text = re.escape(text)
         word_break_set = r"\b"
-        if posix:
-            word_break_set = r"\y"
-            text = text.replace(r"\b", word_break_set)
         if self.word:
             if not text.startswith(word_break_set):
                 text = word_break_set + text
             if not text.endswith(word_break_set):
                 text += word_break_set
+        if posix:
+            text = text.replace(r"\b", word_break_set)
+            if text.startswith(r"\b"):
+                text = r"((?<=\s)|(?<=^))" + text[2:]
+            if text.endswith(r"\b"):
+                text = text[:-2] + r"((?=\s)|(?=$))"
         if self.regex or self.word:
             if not self.case_sensitive:
                 text = "(?i)" + text
-
         return text
 
 
@@ -164,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)
@@ -323,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(
@@ -846,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)
@@ -941,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):
@@ -2273,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()
 
@@ -2584,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

@@ -11,7 +11,6 @@ import numpy as np
 import pyqtgraph as pg
 import sqlalchemy
 from Bio import pairwise2
-from line_profiler_pycharm import profile
 from montreal_forced_aligner.data import CtmInterval
 from montreal_forced_aligner.db import Speaker, Utterance
 from PySide6 import QtCore, QtGui, QtWidgets
@@ -1457,7 +1456,6 @@ class MfaRegion(pg.LinearRegionItem):
 
     settings = AnchorSettings()
 
-    @profile
     def __init__(
         self,
         item: CtmInterval,
@@ -1566,7 +1564,6 @@ class MfaRegion(pg.LinearRegionItem):
 
 
 class AlignmentRegion(MfaRegion):
-    @profile
     def __init__(
         self,
         phone_interval: CtmInterval,
@@ -1681,7 +1678,6 @@ class WordRegion(AlignmentRegion):
 
 
 class UtteranceRegion(MfaRegion):
-    @profile
     def __init__(
         self,
         utterance: workers.UtteranceData,
@@ -2514,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)
@@ -2573,7 +2579,6 @@ class SpeakerTier(pg.GraphicsObject):
                 reg.scene().removeItem(reg)
         self.visible_utterances = {}
 
-    @profile
     def refresh(self, *args):
         self.hide()
         if self.selection_model.plot_min is None:

anchor/ui_main_window.py

@@ -8,7 +8,6 @@
 ## WARNING! All changes made in this file will be lost when recompiling UI file!
 ################################################################################
 
-import resources_rc
 from PySide6.QtCore import (
     QCoreApplication,
     QDate,
@@ -55,6 +54,7 @@ from PySide6.QtWidgets import (
     QWidget,
 )
 
+import anchor.resources_rc
 from anchor.widgets import (
     AcousticModelWidget,
     AlignmentWidget,

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

@@ -29,7 +29,6 @@ from _kalpy.feat import compute_pitch
 from _kalpy.ivector import Plda, ivector_normalize_length
 from _kalpy.matrix import DoubleVector, FloatVector
 from kalpy.feat.pitch import PitchComputer
-from line_profiler_pycharm import profile
 from montreal_forced_aligner import config
 from montreal_forced_aligner.alignment import PretrainedAligner
 from montreal_forced_aligner.config import IVECTOR_DIMENSION, XVECTOR_DIMENSION
@@ -3151,7 +3150,6 @@ class SpectrogramWorker(FunctionWorker):  # pragma: no cover
             self.end = end
             self.channel = channel
 
-    @profile
     def run(self):
         self.stopped.clear()
         dynamic_range = self.settings.value(self.settings.SPEC_DYNAMIC_RANGE)
@@ -3173,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():