Source code for

from pathlib import Path
import subprocess
import logging

_logger = logging.getLogger('ibllib')

[docs]def compress(file_in, file_out, command, remove_original=True): """ runs a command of the form 'ffmpeg -i {file_in} -c:a flac -nostats {file_out}' using a supbprocess audio compression for ephys: `"ffmpeg -i {file_in} -c:a flac -nostats {file_out}"` video compression for ephys: `"('ffmpeg -i {file_in} -codec:v libx264 -preset slow -crf 17 ' '-nostats -loglevel 0 -codec:a copy {file_out}')"` :param file_in: full file path of input :param file_out: full file path of output :param command: string ready to be formatted with `file_in` and `file_out` return: file_out """ file_in = Path(file_in) file_out = Path(file_out) # if the output file already exists, overwrite it if file_out.exists(): file_out.unlink() command2run = command.format(file_in=str(file_in), file_out=str(file_out)) process = subprocess.Popen(command2run, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) info, error = process.communicate() if process.returncode != 0: _logger.error(f'compression failed for {file_in}: {error}') _logger.error(command2run) return process.returncode, None else: # if the command was successful delete the original file if remove_original: file_in.unlink() return process.returncode, file_out
[docs]def iblrig_video_compression(session_path, command): """ Compresses avi files from the raw_videao_data folder into mp4. Once the compression completes the avi file is removed. If mp4 already exist, the funciton returns their file path :param session_path: :param command: for ephys: >>> command = ('ffmpeg -i {file_in} -y -codec:v libx264 -preset slow -crf 17 ' >>> '-nostats -loglevel 0 -codec:a copy {file_out}') for training: >>> command = ('ffmpeg -i {file_in} -y -codec:v libx264 -preset slow -crf 29 ' >>> '-nostats -loglevel 0 -codec:a copy {file_out}') :return: list of compressed files """ output_files = list(session_path.joinpath("raw_video_data").rglob('_iblrig_*.mp4')) rig_avi_files = list(session_path.joinpath("raw_video_data").rglob('_iblrig_*.avi')) # first compress everything (the rationale is not to delete anything if there is a crash) for file_in in rig_avi_files:" compressing {file_in}") file_out = file_in.with_suffix('.mp4') status, fout = compress(file_in=file_in, file_out=file_out, command=command, remove_original=False) output_files.append(fout) # then remove everything for file_in in rig_avi_files: file_in.unlink() return output_files