mirror of
https://github.com/bellingcat/open-questions.git
synced 2026-06-08 03:18:29 +03:00
19 lines
688 B
Python
19 lines
688 B
Python
import sys
|
|
from io import StringIO
|
|
from pyenf_extraction.pyenf import pyENF
|
|
|
|
def extract_enf_signal(wav_file_path):
|
|
mysignal = pyENF(filename=wav_file_path, nominal=50, harmonic_multiples=8, duration=0.1, strip_index=0)
|
|
x, fs = mysignal.read_initial_data()
|
|
spectro_strip, frequency_support = mysignal.compute_spectrogam_strips()
|
|
|
|
sys.stdout = StringIO()
|
|
weights = mysignal.compute_combining_weights_from_harmonics()
|
|
sys.stdout = sys.__stdout__
|
|
|
|
OurStripCell, initial_frequency = mysignal.compute_combined_spectrum(
|
|
spectro_strip, weights, frequency_support
|
|
)
|
|
|
|
return mysignal.compute_ENF_from_combined_strip(OurStripCell,initial_frequency)
|