447 lines
15 KiB
Bash
447 lines
15 KiB
Bash
#!/bin/bash
|
|
|
|
# SaMpeg version
|
|
SAMPEG_VER="2.2024d"
|
|
|
|
# Function to check if NVENC is supported
|
|
check_nvenc_support() {
|
|
if ffmpeg -hide_banner -encoders 2>/dev/null | grep -qE "(nvenc|cuda)"; then
|
|
if [ -e "/dev/nvidia0" ]; then
|
|
echo "NVENC encoding supported"
|
|
return 0
|
|
else
|
|
echo "No NVIDIA GPU detected. NVENC encoding not supported."
|
|
return 1
|
|
fi
|
|
else
|
|
echo "NVENC encoding not supported"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
|
|
# Function to import extension script and execute function from it
|
|
import_extension() {
|
|
extension_script="$1"
|
|
extension_arguments="${@:2}"
|
|
extension_function="main"
|
|
source "$extension_script"
|
|
"$extension_function $extension_arguments"
|
|
echo "Extension function executed successfully"
|
|
}
|
|
|
|
|
|
# Function to generate concat.txt from a folder
|
|
generate_concat_file() {
|
|
folder="$1"
|
|
output="$2"
|
|
cd "$folder" || exit
|
|
files=$(ls *.mp4 | sort -n)
|
|
for file in $files; do
|
|
echo "file '$folder/$file'" >>"$output"
|
|
done
|
|
echo "Generated concat.txt"
|
|
}
|
|
|
|
# Function to combine clips using concat file and scale
|
|
combine_using_concat_file() {
|
|
concat_file="$1"
|
|
output="$2"
|
|
resolution="${3:-3840:2160}" # Default resolution is 4k (16:9)
|
|
if check_nvenc_support; then
|
|
ffmpeg -f concat -safe 0 -i "$concat_file" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -f concat -safe 0 -i "$concat_file" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Clips combined and scaled successfully"
|
|
}
|
|
|
|
# Function to crop a clip
|
|
crop_clip() {
|
|
input="$1"
|
|
output="$2"
|
|
width="$3"
|
|
height="$4"
|
|
x_position="$5"
|
|
y_position="$6"
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -vf "crop=$width:$height:$x_position:$y_position" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -vf "crop=$width:$height:$x_position:$y_position" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Clip cropped successfully"
|
|
}
|
|
|
|
# Function to trim a clip from the beginning
|
|
trim_beginning() {
|
|
input="$1"
|
|
output="$2"
|
|
start_time="$3"
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -ss "$start_time" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -ss "$start_time" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Clip trimmed from beginning successfully"
|
|
}
|
|
|
|
# Function to trim a clip from the end
|
|
trim_end() {
|
|
input="$1"
|
|
output="$2"
|
|
end_time="$3"
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -to "$end_time" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -to "$end_time" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Clip trimmed from end successfully"
|
|
}
|
|
|
|
# Function to scale a clip
|
|
scale_clip() {
|
|
input="$1"
|
|
output="$2"
|
|
resolution="${3:-3840x2160}" # Default resolution is 4k (16:9)
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Clip scaled successfully"
|
|
}
|
|
|
|
# Function to remove parts without audio from a clip
|
|
remove_silence() {
|
|
input="$1"
|
|
output="$2"
|
|
ffmpeg -i "$input" -af silenceremove=1:0:-50dB "$output"
|
|
echo "Silence removed successfully"
|
|
}
|
|
|
|
# Function for picture-in-picture
|
|
picture_in_picture() {
|
|
input_main="$1"
|
|
input_pip="$2"
|
|
output="$3"
|
|
x_scale="$4"
|
|
y_scale="$5"
|
|
x_position="$6"
|
|
y_position="$7"
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input_main" -i "$input_pip" -filter_complex "[1:v]scale=$x_scale:$y_scale [pip]; [0:v][pip]overlay=$x_position:$y_position" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input_main" -i "$input_pip" -filter_complex "[1:v]scale=$x_scale:$y_scale [pip]; [0:v][pip]overlay=$x_position:$y_position" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Picture-in-picture applied successfully"
|
|
}
|
|
|
|
# Function for image stabilization
|
|
image_stabilization() {
|
|
input="$1"
|
|
output="$2"
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -vf vidstabdetect=shakiness=10:accuracy=15:result=transform.trf -f null -
|
|
ffmpeg -i "$input" -vf vidstabtransform=input="transform.trf" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -vf vidstabdetect=shakiness=10:accuracy=15:result=transform.trf -f null -
|
|
ffmpeg -i "$input" -vf vidstabtransform=input="transform.trf" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Image stabilization applied successfully"
|
|
}
|
|
|
|
# Function to set brightness of a clip
|
|
set_brightness() {
|
|
input="$1"
|
|
output="$2"
|
|
brightness="$3"
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -vf "eq=brightness=$brightness" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -vf "eq=brightness=$brightness" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Brightness set successfully"
|
|
}
|
|
|
|
# Function to set hue of a clip
|
|
set_hue() {
|
|
input="$1"
|
|
output="$2"
|
|
hue=$(echo "$3 * 359" | bc)
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -vf "hue=h=$hue" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -vf "hue=h=$hue" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "Hue set successfully"
|
|
}
|
|
|
|
# Function to set saturation of a clip
|
|
set_saturation() {
|
|
input="$1"
|
|
output="$2"
|
|
saturation="$3"
|
|
if check_nvenc_support; then
|
|
ffmpeg -i "$input" -vf "eq=saturation=$saturation" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
else
|
|
ffmpeg -i "$input" -vf "eq=saturation=$saturation" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
|
|
fi
|
|
echo "saturation set successfully"
|
|
}
|
|
|
|
# Function to record screen with optional webcam and microphone as picture-in-picture
|
|
record_screen() {
|
|
screen_resolution=$(xrandr | awk '/ primary/{getline; print $1}')
|
|
output="$1"
|
|
include_webcam="$2"
|
|
include_microphone="$3"
|
|
include_audio="$4"
|
|
|
|
# Set default webcam and microphone devices
|
|
webcam_device="/dev/video0" # Default webcam device
|
|
microphone_device="default" # Default microphone device
|
|
|
|
# Set up inputs for webcam and microphone if included
|
|
webcam_input=""
|
|
microphone_input=""
|
|
if [[ "$include_webcam" == "true" ]]; then
|
|
webcam_input="-f v4l2 -video_size 320x180 -framerate 30 -thread_queue_size 225 -i /dev/video0"
|
|
fi
|
|
if [[ "$include_microphone" == "true" ]]; then
|
|
if [[ ! "$mic" == "" ]]; then
|
|
microphone_input="-f alsa -i default -ac 1 -f mp3 -acodec libmp3lame"
|
|
fi
|
|
fi
|
|
|
|
# Set up audio output options
|
|
audio_output=""
|
|
if [[ "$include_audio" == "true" ]]; then
|
|
if [[ ! "$audio_speakers" == "" ]]; then
|
|
audio_output="-f alsa -i -i hw:Loopback,1,0 -ac 2 -f mp3 -acodec libmp3lame"
|
|
fi
|
|
fi
|
|
|
|
# Determine screen capture method based on display server
|
|
if [[ "$XDG_SESSION_TYPE" == "wayland" ]]; then
|
|
# Wayland screen capture
|
|
ffmpeg -f x11grab -thread_queue_size 450 -video_size "$screen_resolution" -framerate 45 -i "$DISPLAY" $webcam_input $microphone_input \
|
|
-filter_complex "[0:v][1:v]overlay=10:10" \
|
|
$audio_output \
|
|
-r 30 -preset ultrafast -qp 0 \
|
|
"$output"
|
|
else
|
|
# X11 screen capture
|
|
ffmpeg -f x11grab -thread_queue_size 450 -video_size "$screen_resolution" -framerate 45 -i :0.0 $webcam_input $microphone_input \
|
|
-filter_complex "[0:v][1:v]overlay=10:10" \
|
|
$audio_output \
|
|
-r 30 -preset ultrafast -qp 0 \
|
|
"$output"
|
|
fi
|
|
|
|
|
|
echo "Screen recorded successfully"
|
|
}
|
|
|
|
|
|
# Function to add a layer of audio from a video and audio file
|
|
add_audio_layer() {
|
|
input_video="$1"
|
|
input_audio="$2"
|
|
output="$3"
|
|
ffmpeg -i "$input_video" -i "$input_audio" -c:v copy -c:a aac -strict experimental -shortest "$output"
|
|
echo "Audio layer added successfully"
|
|
}
|
|
|
|
# Function to display the SaMpeg header
|
|
display_header() {
|
|
toilet -f pagga -F gay "SaMpeg"
|
|
echo "The video fusing interface."
|
|
echo "Powered by FFmpeg."
|
|
echo "------------"
|
|
}
|
|
|
|
# Function to display the command list
|
|
display_commands() {
|
|
echo "Usage:"
|
|
echo " $0 generate-concat-file <folder> <output>"
|
|
echo " Generates a concat.txt file from a folder containing video files."
|
|
|
|
echo " $0 combine-using-concat-file <concat_file> <output> [resolution]"
|
|
echo " Combines video clips using the concat file and scales them to the specified resolution."
|
|
|
|
echo " $0 crop <input> <output> <width> <height> <x_position> <y_position>"
|
|
echo " Crops a video clip to the specified dimensions and position."
|
|
|
|
echo " $0 trim-beginning <input> <output> <start_time>"
|
|
echo " Trims the beginning of a video clip starting from the specified time."
|
|
|
|
echo " $0 trim-end <input> <output> <end_time>"
|
|
echo " Trims the end of a video clip ending at the specified time."
|
|
|
|
echo " $0 scale <input> <output> [resolution]"
|
|
echo " Scales a video clip to the specified resolution."
|
|
|
|
echo " $0 remove-silence <input> <output>"
|
|
echo " Removes parts of the video clip without audio."
|
|
|
|
echo " $0 picture-in-picture <input_main> <input_pip> <output> <x_scale> <y_scale> <x_position> <y_position>"
|
|
echo " Places a smaller video (picture-in-picture) onto a larger video."
|
|
|
|
echo " $0 image-stabilization <input> <output>"
|
|
echo " Stabilizes the video to reduce shaking."
|
|
|
|
echo " $0 record-screen <output> [include_webcam] [include_microphone] [include_audio]"
|
|
echo " Records the screen with optional webcam, microphone, and desktop audio."
|
|
echo " [include_webcam]: true/false, [include_microphone]: true/false, [include_audio]: true/false."
|
|
|
|
echo " $0 set-brightness <input> <output> <brightness_value>"
|
|
echo " Adjusts the brightness of the video. The value ranges from -1 to 1."
|
|
|
|
echo " $0 set-hue <input> <output> <hue_value>"
|
|
echo " Adjusts the hue of the video. The value ranges from 0 to 1."
|
|
|
|
echo " $0 set-saturation <input> <output> <saturation_value>"
|
|
echo " Adjusts the saturation of the video. The value ranges from -1 to infinity."
|
|
|
|
echo " $0 add-audio-layer <input_video> <input_audio> <output>"
|
|
echo " Adds a layer of audio and outputs to a video."
|
|
|
|
echo " $0 import-extension <extension_script> [extension_arguments]"
|
|
echo " Imports extension, executes main function with arguments specified."
|
|
|
|
echo " $0 help"
|
|
echo " Displays this help message."
|
|
}
|
|
|
|
|
|
# Function to display help message
|
|
display_help() {
|
|
display_header
|
|
display_commands
|
|
}
|
|
|
|
# Main script
|
|
|
|
if [[ $# -lt 1 ]]; then
|
|
display_help
|
|
exit 1
|
|
fi
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
option="$1"
|
|
case $option in
|
|
generate-concat-file)
|
|
if [[ $# -lt 3 ]]; then
|
|
echo "Usage: $0 generate-concat-file <folder> <output>"
|
|
exit 1
|
|
fi
|
|
generate_concat_file "$2" "$3"
|
|
;;
|
|
combine-using-concat-file)
|
|
if [[ $# -lt 3 ]]; then
|
|
echo "Usage: $0 combine-using-concat-file <concat_file> <output> [resolution]"
|
|
exit 1
|
|
fi
|
|
combine_using_concat_file "$2" "$3" "$4"
|
|
;;
|
|
crop)
|
|
if [[ $# -ne 7 ]]; then
|
|
echo "Usage: $0 crop <input> <output> <width> <height> <x_position> <y_position>"
|
|
exit 1
|
|
fi
|
|
crop_clip "$2" "$3" "$4" "$5" "$6" "$7"
|
|
;;
|
|
trim-beginning)
|
|
if [[ $# -ne 4 ]]; then
|
|
echo "Usage: $0 trim-beginning <input> <output> <start_time>"
|
|
exit 1
|
|
fi
|
|
trim_beginning "$2" "$3" "$4"
|
|
;;
|
|
trim-end)
|
|
if [[ $# -ne 4 ]]; then
|
|
echo "Usage: $0 trim-end <input> <output> <end_time>"
|
|
exit 1
|
|
fi
|
|
trim_end "$2" "$3" "$4"
|
|
;;
|
|
scale)
|
|
if [[ $# -lt 3 ]]; then
|
|
echo "Usage: $0 scale <input> <output> [resolution]"
|
|
exit 1
|
|
fi
|
|
scale_clip "$2" "$3" "$4"
|
|
;;
|
|
remove-silence)
|
|
if [[ $# -ne 3 ]]; then
|
|
echo "Usage: $0 remove-silence <input> <output>"
|
|
exit 1
|
|
fi
|
|
remove_silence "$2" "$3"
|
|
;;
|
|
picture-in-picture)
|
|
if [[ $# -ne 8 ]]; then
|
|
echo "Usage: $0 picture-in-picture <input_main> <input_pip> <output> <x_scale> <y_scale> <x_position> <y_position>"
|
|
exit 1
|
|
fi
|
|
picture_in_picture "$2" "$3" "$4" "$5" "$6" "$7" "$8"
|
|
;;
|
|
image-stabilization)
|
|
if [[ $# -ne 3 ]]; then
|
|
echo "Usage: $0 image-stabilization <input> <output>"
|
|
exit 1
|
|
fi
|
|
image_stabilization "$2" "$3"
|
|
;;
|
|
record-screen)
|
|
if [[ $# -lt 2 ]]; then
|
|
echo "Usage: $0 record-screen <output> [include_webcam] [include_microphone] [include_audio]"
|
|
exit 1
|
|
fi
|
|
record_screen "$2" "$3" "$4"
|
|
;;
|
|
help)
|
|
display_help
|
|
exit 0
|
|
;;
|
|
add-audio-layer)
|
|
if [[ $# -lt 3 ]]; then
|
|
echo "Usage: $0 add-audio-layer <input_video> <input_audio> <output>"
|
|
exit 1
|
|
fi
|
|
add_audio_layer "$2" "$3" "$4"
|
|
;;
|
|
set-brightness)
|
|
if [[ $# -lt 3 ]]; then
|
|
echo "Usage: $0 set-brightness <input_video> <input_audio> <output>"
|
|
exit 1
|
|
fi
|
|
set_brightness "$2" "$3" "$4"
|
|
;;
|
|
set-hue)
|
|
if [[ $# -lt 3 ]]; then
|
|
echo "Usage: $0 set-hue <input_video> <input_audio> <output>"
|
|
exit 1
|
|
fi
|
|
set_hue "$2" "$3" "$4"
|
|
;;
|
|
set-saturation)
|
|
if [[ $# -lt 3 ]]; then
|
|
echo "Usage: $0 set-saturation <input_video> <input_audio> <output>"
|
|
exit 1
|
|
fi
|
|
set_saturation "$2" "$3" "$4"
|
|
;;
|
|
import-extension)
|
|
if [[ $# -lt 2 ]]; then
|
|
echo "Usage: $0 import-extension <extension_script> [extension_arguments]"
|
|
exit 1
|
|
fi
|
|
import_extension "$2" "${@:3}"
|
|
;;
|
|
*)
|
|
exit 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|