lingualabpy 0.0.3__tar.gz → 0.0.4__tar.gz

{lingualabpy-0.0.3 → lingualabpy-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lingualabpy
-Version: 0.0.3
+Version: 0.0.4
 Summary: Tools and utilities from the LINGUA laboratory
 Author-email: Christophe Bedetti <christophe.bedetti@umontreal.ca>
 Requires-Python: >=3.8.1

{lingualabpy-0.0.3 → lingualabpy-0.0.4}/pyproject.toml

@@ -38,6 +38,7 @@ feature = []
 dev = ['lingualabpy[test, doc, lint, feature]']
 
 [project.scripts]
+lingualabpy_audio_metrics = 'lingualabpy.cli.audio_metrics:main'
 lingualabpy_audio_triming = 'lingualabpy.cli.audio_triming:main'
 lingualabpy_docx2json = 'lingualabpy.cli.docx2json:main'
 lingualabpy_jsons2csv = 'lingualabpy.cli.jsons2csv:main'

{lingualabpy-0.0.3 → lingualabpy-0.0.4}/src/lingualabpy/__init__.py

@@ -4,12 +4,17 @@
 """lingualabpy"""
 from __future__ import annotations
 
-__version__ = "0.0.3"
+__version__ = "0.0.4"
 
 default_config = {
     "participant_col": "participant_id",
     "participant_label": "IE",
     "clinician_label": "IV",
+    "f0_bounds": {
+        "female": [100.0, 600.0],
+        "male": [75.0, 300.0],
+    },
+    "unit_frequency": "Hertz",
 }
 
 from lingualabpy.io import read_audio, read_docx, read_json, write_json, read_textgrid

lingualabpy-0.0.4/src/lingualabpy/audio/metrics.py

@@ -0,0 +1,85 @@
+from collections import defaultdict
+import numpy as np
+from parselmouth import Sound
+from parselmouth.praat import call
+
+from lingualabpy.tools.data import UnchangeableDict
+
+
+def measure_pitch(sound: Sound, f0min: str, f0max: str, unit: str) -> UnchangeableDict:
+    """
+    This function measures duration, pitch, HNR, jitter, and shimmer
+    This is the function to measure source acoustics using default male parameters.
+    """
+    # compute usefull praat object
+    pitch = call(sound, "To Pitch", 0.0, f0min, f0max)
+    harmonicity = call(sound, "To Harmonicity (cc)", 0.01, f0min, 0.1, 1.0)
+    point_process = call(sound, "To PointProcess (periodic, cc)", f0min, f0max)
+
+    # metrics container
+    metrics = UnchangeableDict()
+
+    # Metrics computation
+    metrics["duration"] = call(sound, "Get total duration")
+    metrics["f0_mean"] = call(pitch, "Get mean", 0, 0, unit)
+    metrics["F0_std"] = call(pitch, "Get standard deviation", 0, 0, unit)
+    metrics["hnr"] = call(harmonicity, "Get mean", 0, 0)
+
+    # jitter
+    jitter_types = ["local", ["local", "absolute"], "rap", "ppq5", "ddp"]
+    for jitter_type in jitter_types:
+        if isinstance(jitter_type, list):
+            metric_name = f"jitter_{'_'.join(jitter_type)}"
+            praat_function = f"Get jitter ({', '.join(jitter_type)})"
+        else:
+            metric_name = f"jitter_{jitter_type}"
+            praat_function = f"Get jitter ({jitter_type})"
+        metrics[metric_name] = call(
+            point_process, praat_function, 0, 0, 0.0001, 0.02, 1.3
+        )
+
+    # shimmer
+    shimmer_types = ["local", "local_dB", "apq3", "apq5", "apq11", "dda"]
+    for shimmer_type in shimmer_types:
+        metric_name = f"shimmer_{shimmer_type}"
+        praat_function = f"Get shimmer ({shimmer_type})"
+        metrics[metric_name] = call(
+            [sound, point_process], praat_function, 0, 0, 0.0001, 0.02, 1.3, 1.6
+        )
+
+    return metrics
+
+
+def measure_formants(
+    sound: Sound, f0min: str, f0max: str, unit: str
+) -> UnchangeableDict:
+    """
+    This function measures formants at each glottal pulse
+
+    Puts, D. A., Apicella, C. L., & Cárdenas, R. A. (2012). Masculine voices signal men's threat potential in forager and industrial societies. Proceedings of the Royal Society of London B: Biological Sciences, 279(1728), 601-609.
+
+    Adapted from: DOI 10.17605/OSF.IO/K2BHS
+    """
+    # compute usefull praat object
+    point_process = call(sound, "To PointProcess (periodic, cc)", f0min, f0max)
+    formants = call(sound, "To Formant (burg)", 0.0025, 5, 5000, 0.025, 50)
+    number_of_points = call(point_process, "Get number of points")
+
+    # metrics container
+    metrics = UnchangeableDict()
+
+    # Measure formants only at glottal pulses
+    formants_list = defaultdict(list)
+    for index in range(1, number_of_points + 1):
+        time = call(point_process, "Get time from index", index)
+        for pulse in [1, 2, 3, 4]:
+            value = call(formants, "Get value at time", pulse, time, unit, "Linear")
+            if str(value) != "nan":
+                formants_list[pulse].append(value)
+
+    # calculate mean and median formants across pulses, median is what is used in all subsequent calculations
+    for pulse in [1, 2, 3, 4]:
+        metrics[f"formants_{pulse}_mean"] = np.mean(formants_list[pulse])
+        metrics[f"formants_{pulse}_median"] = np.median(formants_list[pulse])
+
+    return metrics

lingualabpy-0.0.4/src/lingualabpy/cli/audio_metrics.py

@@ -0,0 +1,57 @@
+import click
+from parselmouth import Sound
+from pathlib import Path
+
+from lingualabpy import default_config, write_json
+from lingualabpy.audio.metrics import measure_pitch, measure_formants
+
+
+@click.command()
+@click.option(
+    "--sex",
+    type=click.Choice(["female", "male"]),
+    help=f"Set f0min and f0max for praat analysis. {default_config['f0_bounds']}",
+)
+@click.option(
+    "--f0min",
+    type=float,
+    help="Define f0min for praat analysis. Not required if sex is specify",
+)
+@click.option(
+    "--f0max",
+    type=float,
+    help="Define f0max for praat analysis. Not required if sex is specify",
+)
+@click.option(
+    "--unit_frequency",
+    default=default_config["unit_frequency"],
+    show_default=True,
+)
+@click.option("--participant_id", "-p", default=None, help="")
+@click.option("--output_json", default=None, help="")
+@click.argument("audiofile", nargs=1, type=click.Path(exists=True))
+def main(sex, f0min, f0max, unit_frequency, participant_id, output_json, audiofile):
+    """Doc"""
+    if sex:
+        f0min, f0max = default_config["f0_bounds"][sex]
+    else:
+        if not f0min or not f0max:
+            raise click.UsageError(
+                "'--f0min' and '--f0max' are required if '--sex' is not specified"
+            )
+
+    sound = Sound(audiofile)
+    metrics = measure_pitch(sound, f0min, f0max, unit_frequency)
+    metrics.update(measure_formants(sound, f0min, f0max, unit_frequency))
+
+    audiofile_stem = Path(audiofile).stem
+
+    if participant_id:
+        metrics["participant_id"] = participant_id
+    else:
+        metrics["participant_id"] = audiofile_stem.split("_")[0]
+
+    if not output_json:
+        output_json = audiofile_stem + "_metric-audio.json"
+
+    write_json(dict(metrics), output_json)

{lingualabpy-0.0.3 → lingualabpy-0.0.4}/src/lingualabpy/cli/audio_triming.py

@@ -23,9 +23,14 @@ from lingualabpy.tools.interval import intervals_masking
 def main(participant_label, clinician_label, textgrid, audiofile, output):
     """Doc"""
     grid = read_textgrid(textgrid)
-    participant_intervals, clinician_intervals = extract_intervals(
-        grid, [participant_label, clinician_label]
-    )
+
+    try:
+        participant_intervals, clinician_intervals = extract_intervals(
+            grid, [participant_label, clinician_label]
+        )
+    except Exception as e:
+        raise Exception(f"Failed to extract intervals for {textgrid}", repr(e))
+
     participant_intervals_clean = intervals_masking(
         participant_intervals, clinician_intervals
     )

{lingualabpy-0.0.3 → lingualabpy-0.0.4}/src/lingualabpy/io.py

@@ -33,7 +33,7 @@ def read_json(json_path: str) -> Union[list, dict]:
 def write_json(data: Union[list, dict], json_path: str) -> None:
     """"""
     with open(json_path, "w") as file:
-        json.dump(data, file)
+        json.dump(data, file, indent=4)
 
 
 # .TextGrid files

{lingualabpy-0.0.3 → lingualabpy-0.0.4}/src/lingualabpy/tools/data.py

@@ -1,8 +1,20 @@
+from collections import UserDict
 from pandas import DataFrame
 
 from typing import Any, Dict, List
 
 
+class UnchangeableDict(UserDict):
+    """A dictionary in which you can add new keys but not modify them in the future."""
+
+    def __setitem__(self, key: Any, item: Any) -> None:
+        try:
+            self.__getitem__(key)
+            raise ValueError("duplicate key '{}' found".format(key))
+        except KeyError:
+            return super().__setitem__(key, item)
+
+
 def merge_participants_to_df(
     data_participants: List[Dict[Any, Any]],
     participant_col: str,