github/turntable
1
0
Fork 0
mirror of https://github.com/correl/turntable.git synced 2024-11-23 19:19:55 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add a delay before capturing fingerprinting audio

Browse source
This commit is contained in:
Correl Roush 2020-09-01 23:09:18 -04:00
parent 9859ca21e9
commit 77b28a54ad
1 changed files with 17 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
turntable/turntable.py
View file

@ -19,6 +19,7 @@ from turntable.models import PCM
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
FINGERPRINT_DELAY = 5
FINGERPRINT_IDENTIFY_SECONDS = 5 FINGERPRINT_IDENTIFY_SECONDS = 5
FINGERPRINT_STORE_SECONDS = 30 FINGERPRINT_STORE_SECONDS = 30
SAMPLE_SECONDS = 30 SAMPLE_SECONDS = 30
@ -104,7 +105,7 @@ class Turntable(Process):
maximum = audioop.max(fragment.raw, 2) maximum = audioop.max(fragment.raw, 2)
self.update_audiolevel(maximum) self.update_audiolevel(maximum)
def update_audiolevel(self, level: int): def update_audiolevel(self, level: int) -> None:
newstate = self.state newstate = self.state
now = time.time() now = time.time()
if self.state == State.idle: if self.state == State.idle:
@ -116,12 +117,15 @@ class Turntable(Process):
if level <= SILENCE_THRESHOLD: if level <= SILENCE_THRESHOLD:
self.transition(State.silent, now) self.transition(State.silent, now)
elif ( elif (
now - self.last_update >= FINGERPRINT_IDENTIFY_SECONDS now - self.last_update
>= FINGERPRINT_DELAY + FINGERPRINT_IDENTIFY_SECONDS
and self.identified == False and self.identified == False
): ):
sample = self.buffer[ startframe = self.buffer.framerate * FINGERPRINT_DELAY
0 : self.buffer.framerate * FINGERPRINT_IDENTIFY_SECONDS endframe = (
] startframe + self.buffer.framerate * FINGERPRINT_IDENTIFY_SECONDS
)
sample = self.buffer[startframe:endframe]
identification = self.recognizer.recognize(sample) identification = self.recognizer.recognize(sample)
logger.debug("Dejavu results: %s", identification) logger.debug("Dejavu results: %s", identification)
if results := identification[dejavu.config.settings.RESULTS]: if results := identification[dejavu.config.settings.RESULTS]:
@ -132,12 +136,14 @@ class Turntable(Process):
) )
self.identified = True self.identified = True
elif ( elif (
now - self.last_update >= FINGERPRINT_STORE_SECONDS now - self.last_update >= FINGERPRINT_DELAY + FINGERPRINT_STORE_SECONDS
and self.captured == False and self.captured == False
): ):
sample = self.buffer[ startframe = self.buffer.framerate * FINGERPRINT_DELAY
0 : self.buffer.framerate * FINGERPRINT_STORE_SECONDS endframe = (
] startframe + self.buffer.framerate * FINGERPRINT_STORE_SECONDS
)
sample = self.buffer[startframe:endframe]
with wave.open("/tmp/fingerprint.wav", "wb") as wavfile: with wave.open("/tmp/fingerprint.wav", "wb") as wavfile:
wavfile.setsampwidth(2) wavfile.setsampwidth(2)
wavfile.setnchannels(sample.channels) wavfile.setnchannels(sample.channels)
@ -154,10 +160,10 @@ class Turntable(Process):
elif now - self.last_update >= STOP_DELAY: elif now - self.last_update >= STOP_DELAY:
self.transition(State.idle, now) self.transition(State.idle, now)
def transition(self, to_state: State, updated_at: float): def transition(self, to_state: State, updated_at: float) -> None:
logger.debug("Transition: %s => %s", self.state, to_state)
self.state = to_state self.state = to_state
self.last_update = updated_at self.last_update = updated_at
logger.debug("State: %s", self.state)
if to_state == State.idle: if to_state == State.idle:
self.events_out.put(StoppedPlaying()) self.events_out.put(StoppedPlaying())