wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Adding flag for file writting recording case

This commit is contained in:
CarolinePascal 2025-04-09 15:00:18 +02:00
parent d53035d047
commit 1e5e631743
No known key found for this signature in database
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/common/robot_devices/microphones/microphone.py
View File

@ -158,6 +158,7 @@ class Microphone:
self.logs = {} self.logs = {}
self.is_connected = False self.is_connected = False
self.is_recording = False self.is_recording = False
self.is_writing = False
def connect(self) -> None: def connect(self) -> None:
if self.is_connected: if self.is_connected:
@ -220,7 +221,8 @@ class Microphone:
logging.warning(status) logging.warning(status)
# Slicing makes copy unecessary # Slicing makes copy unecessary
# Two separate queues are necessary because .get() also pops the data from the queue # Two separate queues are necessary because .get() also pops the data from the queue
self.record_queue.put(indata[:,self.channels]) if self.is_writing:
self.record_queue.put(indata[:,self.channels])
self.read_queue.put(indata[:,self.channels]) self.read_queue.put(indata[:,self.channels])
@staticmethod @staticmethod
@ -300,6 +302,8 @@ class Microphone:
self.record_thread = Thread(target=Microphone._record_loop, args=(self.record_queue, self.record_stop_event, self.sample_rate, self.channels, output_file, )) self.record_thread = Thread(target=Microphone._record_loop, args=(self.record_queue, self.record_stop_event, self.sample_rate, self.channels, output_file, ))
self.record_thread.daemon = True self.record_thread.daemon = True
self.record_thread.start() self.record_thread.start()
self.is_writing = True
self.is_recording = True self.is_recording = True
self.stream.start() self.stream.start()