lingualabpy 0.0.5__py3-none-any.whl → 0.0.6__py3-none-any.whl

lingualabpy/__init__.py

@@ -4,7 +4,7 @@
 """lingualabpy"""
 from __future__ import annotations
 
-__version__ = "0.0.5"
+__version__ = "0.0.6"
 
 default_config = {
     "participant_col": "participant_id",

lingualabpy/cli/plot_sound.py

@@ -0,0 +1,55 @@
+import cv2
+import os
+import click
+from parselmouth import Sound
+import matplotlib.pyplot as plt
+from pathlib import Path
+
+from lingualabpy.plot import draw_pitch, draw_spectrogram
+
+
+@click.command()
+@click.option("--output", default=None, help="")
+@click.argument("audiofile", nargs=1, type=click.Path(exists=True))
+def main(audiofile, output):
+    if not output:
+        output = Path(audiofile).stem + ".png"
+
+    sound = Sound(audiofile)
+
+    pitch = sound.to_pitch()
+
+    # If desired, pre-emphasize the sound fragment before calculating the spectrogram
+    pre_emphasized_snd = sound.copy()
+    pre_emphasized_snd.pre_emphasize()
+    spectrogram = pre_emphasized_snd.to_spectrogram(
+        window_length=0.03, maximum_frequency=8000
+    )
+
+    # amplitude figure
+    tmp_amplitude_png = "tmp_amplitude.png"
+    amplitude = plt.figure()
+    plt.plot(sound.xs(), sound.values.T)
+    plt.xlim([sound.xmin, sound.xmax])
+    plt.xlabel("time [s]")
+    plt.ylabel("amplitude")
+    amplitude.set_figwidth(sound.xmax / 4)
+    plt.savefig(tmp_amplitude_png)
+
+    # spectro pitch figure
+    tmp_spectro_pitch_png = "tmp_spectro_pitch.png"
+    spectro_pitch = plt.figure()
+    draw_spectrogram(spectrogram)
+    plt.twinx()
+    draw_pitch(pitch)
+    plt.xlim([sound.xmin, sound.xmax])
+    spectro_pitch.set_figwidth(sound.xmax / 4)
+    plt.savefig(tmp_spectro_pitch_png)
+
+    # concatenation
+    fig_concat = cv2.vconcat(
+        [cv2.imread(tmp_amplitude_png), cv2.imread(tmp_spectro_pitch_png)]
+    )
+    cv2.imwrite(output, fig_concat)
+    os.remove(tmp_amplitude_png)
+    os.remove(tmp_spectro_pitch_png)

lingualabpy/plot.py

@@ -0,0 +1,23 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+def draw_spectrogram(spectrogram, dynamic_range=70):
+    X, Y = spectrogram.x_grid(), spectrogram.y_grid()
+    sg_db = 10 * np.log10(spectrogram.values)
+    plt.pcolormesh(X, Y, sg_db, vmin=sg_db.max() - dynamic_range, cmap="afmhot")
+    plt.ylim([spectrogram.ymin, spectrogram.ymax])
+    plt.xlabel("time [s]")
+    plt.ylabel("frequency [Hz]")
+
+
+def draw_pitch(pitch):
+    # Extract selected pitch contour, and
+    # replace unvoiced samples by NaN to not plot
+    pitch_values = pitch.selected_array["frequency"]
+    pitch_values[pitch_values == 0] = np.nan
+    plt.plot(pitch.xs(), pitch_values, "o", markersize=5, color="w")
+    plt.plot(pitch.xs(), pitch_values, "o", markersize=2)
+    plt.grid(False)
+    plt.ylim(0, pitch.ceiling)
+    plt.ylabel("fundamental frequency [Hz]")

{lingualabpy-0.0.5.dist-info → lingualabpy-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lingualabpy
-Version: 0.0.5
+Version: 0.0.6
 Summary: Tools and utilities from the LINGUA laboratory
 Author-email: Christophe Bedetti <christophe.bedetti@umontreal.ca>
 Requires-Python: >=3.8.1
@@ -19,6 +19,8 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: click
+Requires-Dist: matplotlib
+Requires-Dist: opencv-python
 Requires-Dist: pandas
 Requires-Dist: praat-parselmouth
 Requires-Dist: praat-textgrids

{lingualabpy-0.0.5.dist-info → lingualabpy-0.0.6.dist-info}/RECORD

@@ -1,5 +1,6 @@
-lingualabpy/__init__.py,sha256=_oiqa1RV-lhkXP60vELtzeIEbRJWn9DagQRqNSDzl-s,642
+lingualabpy/__init__.py,sha256=Ej3OuJSypbV3r_Dv_S1oIf-6Cl6nbzJC2tfOs8HavlQ,642
 lingualabpy/io.py,sha256=r3Y781XKAea_T3V2L8aJlxoOEQcYCsdd7c-f-8NcEeI,1155
+lingualabpy/plot.py,sha256=KvXXjzUULWy5ibjlgLu4GOHN62YWY2q3IwvBxW_-gfg,859
 lingualabpy/audio/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lingualabpy/audio/metrics.py,sha256=u5FlADmqeYQOSpqhY2l1l8CSC4tfBV6cWp3g6Hri6bE,3502
 lingualabpy/audio/triming.py,sha256=6CY9pH43KFGAPj8Nw34y1YnlOb8gxGLU1btcuRy-Hgc,288
@@ -8,6 +9,7 @@ lingualabpy/cli/audio_metrics.py,sha256=N_kIKmro5bjFEKjHPNfmuY4G5TXQB9qegbc55pUc
 lingualabpy/cli/audio_triming.py,sha256=H62FTRmwhha-qAXoZ450TNNf_3sHg69GATmbMHQBoCM,1566
 lingualabpy/cli/docx2json.py,sha256=Bj5f89B76NtA7Xx71xXGnSucrDEyaH9mUFifQo0wfn4,590
 lingualabpy/cli/jsons2csv.py,sha256=_AcIXiQUCF5SsKqMg6WjTr8fhbuflaJNFrCP91ccSYs,596
+lingualabpy/cli/plot_sound.py,sha256=Vvddh2PWxXLt_uXrHFvHfchEyOCrcMsAhEldgA1LzqE,1658
 lingualabpy/resources/FilledPauses.praat,sha256=8tY0tAcG71KwnNhawgqyZ7vT__Udf5mSaVutH1eetd8,19623
 lingualabpy/resources/syllablenucleiv3.praat,sha256=aZYRGB2iLGpYxL3ma-UMutQEyP_NvbbbF1yKhGvVcsA,35818
 lingualabpy/text/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -16,8 +18,8 @@ lingualabpy/text/textgrid.py,sha256=nTVj3UOCJRsdybxAW0M2jlDq0eMMRYSSo6GdXYkH2wU,
 lingualabpy/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lingualabpy/tools/data.py,sha256=FU0_3TaeAZNCu1WpNIOkVBV3bYiEhI1rPw_l8q8z0gk,1523
 lingualabpy/tools/interval.py,sha256=50lzbMTNHF26mPRG50mykCUQE3pdyRjPWMwsskwy0tg,2060
-lingualabpy-0.0.5.dist-info/entry_points.txt,sha256=IXEsa7Cgqjph5bkKSBMXZIBVP4ocrRaSh13dFPBwBmE,247
-lingualabpy-0.0.5.dist-info/LICENSE,sha256=s3hbMsmwGq2XFcxpMD3oHc8GSUeXAmPVXJbn7SYXdos,1095
-lingualabpy-0.0.5.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-lingualabpy-0.0.5.dist-info/METADATA,sha256=ocw3j6BfTe0NSSKORBs-JjU5KacofzOi81fYRbZ47F4,1703
-lingualabpy-0.0.5.dist-info/RECORD,,
+lingualabpy-0.0.6.dist-info/entry_points.txt,sha256=RJc47XeEp4rf6QvUe-o0b9cxPQs3NmreaY_w4hTmIto,302
+lingualabpy-0.0.6.dist-info/LICENSE,sha256=s3hbMsmwGq2XFcxpMD3oHc8GSUeXAmPVXJbn7SYXdos,1095
+lingualabpy-0.0.6.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+lingualabpy-0.0.6.dist-info/METADATA,sha256=RKuikNjTkwYwXiuwXefV-xpYDmGulMOs6EBgvudyftk,1758
+lingualabpy-0.0.6.dist-info/RECORD,,

{lingualabpy-0.0.5.dist-info → lingualabpy-0.0.6.dist-info}/entry_points.txt

@@ -3,4 +3,5 @@ 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_plot_sound=lingualabpy.cli.plot_sound:main