Update 'sampeg.sh'

This commit is contained in:
nodemixaholic 2024-02-19 01:57:36 +00:00
parent 15815c5db6
commit 3fbd5067c3

118
sampeg.sh
View file

@ -36,7 +36,7 @@ generate_concat_file() {
combine_using_concat_file() {
concat_file="$1"
output="$2"
resolution="${3:-3840x2160}" # Default resolution is 4k (16:9)
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
@ -104,7 +104,7 @@ scale_clip() {
remove_silence() {
input="$1"
output="$2"
ffmpeg -i "$input" -af silenceremove=1:0:-50dB -c:a copy "$output"
ffmpeg -i "$input" -af silenceremove=1:0:-50dB "$output"
echo "Silence removed successfully"
}
@ -139,29 +139,66 @@ image_stabilization() {
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"
mic="$4"
include_audio="$5"
audio_speakers="$6"
include_audio="$4"
# Set default webcam and microphone devices
webcam_device="/dev/video0" # Default webcam device
microphone_device="$mic" # Default microphone 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 -thread_queue_size 64 -i $webcam_device"
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 pulse -i $mic -ac 2"
microphone_input="-f alsa -i default -ac 1 -f mp3 -acodec libmp3lame"
fi
fi
@ -169,29 +206,41 @@ record_screen() {
audio_output=""
if [[ "$include_audio" == "true" ]]; then
if [[ ! "$audio_speakers" == "" ]]; then
audio_output="pulse -i $audio_speakers 128k -ac 1"
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 24 -i "$DISPLAY" $webcam_input $microphone_input \
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 24 -preset ultrafast \
-r 30 -preset ultrafast -qp 0 \
"$output"
else
# X11 screen capture
ffmpeg -f x11grab -thread_queue_size 450 -video_size "$screen_resolution" -framerate 24 -i :0.0 $webcam_input $microphone_input \
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 24 -preset ultrafast \
-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"
@ -230,18 +279,21 @@ display_commands() {
echo " $0 image-stabilization <input> <output>"
echo " Stabilizes the video to reduce shaking."
echo " $0 record-screen <output> [include_webcam] [include_microphone] [mic] [include_audio] [audioout]"
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, [mic]: microphone device, [include_audio]: true/false, [audioout]: audio output device."
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 0 to 1."
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-contrast <input> <output> <contrast_value>"
echo " Adjusts the contrast 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 help"
echo " Displays this help message."
@ -334,7 +386,7 @@ while [[ $# -gt 0 ]]; do
;;
record-screen)
if [[ $# -lt 2 ]]; then
echo "Usage: $0 record-screen <output> [include_webcam] [include_microphone] [mic] [include_audio] [audioout]"
echo "Usage: $0 record-screen <output> [include_webcam] [include_microphone] [include_audio]"
exit 1
fi
record_screen "$2" "$3" "$4"
@ -343,6 +395,34 @@ while [[ $# -gt 0 ]]; do
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"
;;
*)
exit 1
;;