mirror of
https://github.com/F5OEO/rpitx.git
synced 2026-03-23 16:56:54 +01:00
b85ece8df6
Wav files include the length of the data in the file header. libsndfile expects this length to be there and be sane. Unfortunately, when piping data from avconv, this length is dummied out because the full length is not known yet, so libsndfile can't be used to process a pipe data stream. This removes the libsndfile dependency and replaces the used functions with hand written ones.
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
"""Hides imports and other irrelevant things so that ipython works nicely."""
|
|
|
|
import _rpitx
|
|
import ffmpegwrapper
|
|
import libavwrapper
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
import threading
|
|
|
|
PIPE = 'pipe:1'
|
|
DUMMY_FILE = 'dummy-file'
|
|
|
|
|
|
def broadcast_fm(media_file_name, frequency):
|
|
"""Play a media file's audio over FM."""
|
|
|
|
logging.basicConfig()
|
|
logger = logging.getLogger('rpitx')
|
|
|
|
dev_null = open(os.devnull, 'w')
|
|
if subprocess.call(('which', 'ffmpeg'), stdout=dev_null) == 0:
|
|
Input = ffmpegwrapper.Input
|
|
Output = ffmpegwrapper.Output
|
|
VideoCodec = ffmpegwrapper.VideoCodec
|
|
AudioCodec = ffmpegwrapper.AudioCodec
|
|
Stream = ffmpegwrapper.FFmpeg
|
|
command = 'ffmpeg'
|
|
elif subprocess.call(('which', 'avconv'), stdout=dev_null) == 0:
|
|
Input = libavwrapper.Input
|
|
Output = libavwrapper.Output
|
|
VideoCodec = libavwrapper.VideoCodec
|
|
AudioCodec = libavwrapper.AudioCodec
|
|
Stream = libavwrapper.AVConv
|
|
Stream.add_option = Stream.add_parameter
|
|
command = 'avconv'
|
|
else:
|
|
raise NotImplementedError(
|
|
'Broadcasting audio requires either avconv or ffmpeg to be installed\n'
|
|
'sudo apt install libav-tools'
|
|
)
|
|
|
|
logger.debug('Using %s', command)
|
|
|
|
input_media = Input(media_file_name)
|
|
codec = AudioCodec('pcm_s16le').frequence(48000).channels(1)
|
|
output_audio = Output(DUMMY_FILE, codec)
|
|
stream = Stream(command, input_media, output_audio)
|
|
command_line = list(stream)
|
|
# The format needs to be specified manually because we're writing to
|
|
# stderr and not a named file. Normally, it would infer the format from
|
|
# the file name extension. Also, ffmpeg will only write to a pipe if it's
|
|
# the last named argument.
|
|
command_line.remove(DUMMY_FILE)
|
|
command_line += ('-f', 'wav', PIPE)
|
|
logger.debug('Running command "%s"', ' '.join(command_line))
|
|
stream_process = subprocess.Popen(command_line, stdout=subprocess.PIPE)
|
|
|
|
def log_stdout():
|
|
"""Log stdout from the stream process."""
|
|
lines = stream.run()
|
|
for line in lines:
|
|
logger.debug(line)
|
|
|
|
thread = threading.Thread(target=log_stdout).start()
|
|
|
|
logger.debug('Calling broadcast_fm')
|
|
_rpitx.broadcast_fm(stream_process.stdout.fileno(), frequency)
|
|
thread.join()
|