#!/bin/bash
# Озвучивание русского текста из файла pdf и сохранение в видео
# Параметры
emotion='neutral' #'Default is neutral. Also supported are good (friendly) and evil (angry)'
speaker='ermil' # (Optional): Speaker voice. Default is zahar. Supported female voices are jane, oksana, alyss, omazh and male voices are zahar and ermil.
#["oksana","jane","omazh","zahar","ermil","silaerkan","erkanyavas","alyss", "nick","alena","filipp"]
speed='1.0'
use_half="yes" # yes or no
ffmpeg_pre_options="-loop 1 -r 2" # "-loop 1"
ffmpeg_options="-c:v libx264 -tune stillimage -preset ultrafast -crf 20  -c:a copy -shortest -pix_fmt yuv420p -s 1920*1080"

#----------------------------------------------------

input_pdf_file_name="$1"

page_count=$(pdfinfo "$input_pdf_file_name" | awk '/^Pages:/ {print $2}')

video_file_names_array=()

echo "Всего страниц $page_count"

for ((page=1;page<=${page_count};page++)); do
    
    echo "------------------------------------------------"
    echo "Обрабатываем страницу №$page"
    
    page_text_file="${input_pdf_file_name}_${page}.txt"
    page_image_file="${input_pdf_file_name}_${page}"
    pdftotext -f $page -l $page "$input_pdf_file_name" "$page_text_file"
    pdftoppm -f $page -l $page -png -singlefile "$input_pdf_file_name" "$page_image_file"

    page_image_file="${page_image_file}.png"

    if [ "$use_half"="yes" ]; then
    
        source_text=$(cat "${page_text_file}")
        
        space_char=" "
        split_size=$(( ${#source_text} / 2 + 2)) # Половина с небольшим запасом
        file_index=0
        for ((i=1;i<=${#source_text};i++)); do
            cur_char=${source_text:$i-1:1}
            cur_text="${cur_text}${cur_char}"
            if [ "$cur_char" = "$space_char" ] && [ ${#cur_text} -ge $split_size ] || [ $i = ${#source_text} ]; then
                let file_index+=1
                
                echo "$cur_text" > "${page_text_file}_half${file_index}"
                
                cur_text=""
            fi
        done

        file_mp3_half1="${page_text_file}_half1"
        file_mp3_half2="${page_text_file}_half2"
        
        ~/txt2mp3_by_yatts "${file_mp3_half1}"
        ~/txt2mp3_by_yatts "${file_mp3_half2}"
    
        page_mp3_file_half1="${file_mp3_half1}_yatts.mp3"
        page_mp3_file_half2="${file_mp3_half2}_yatts.mp3"

        width=$(identify -format "%w" "$page_image_file")> /dev/null
        height=$(identify -format "%h" "$page_image_file")> /dev/null

        height_half=$(($height/2))
        
        page_image_file_half1="${page_image_file}_half1.png"
        page_image_file_half2="${page_image_file}_half2.png"

        # format (widthxheight+left+top / wxh+l+t)
        convert "$page_image_file" -crop ${width}x${height_half}+0+0 "$page_image_file_half1"
        convert "$page_image_file" -crop ${width}x${height_half}+0+$height_half "$page_image_file_half2"

        page_mp4_file_half1="${input_pdf_file_name}_${page}_half1.mp4"
        
        ffmpeg $ffmpeg_pre_options -i "$page_image_file_half1" -i "$page_mp3_file_half1" $ffmpeg_options "$page_mp4_file_half1"
        
        SAVE_IFS=$IFS
        IFS=""
        video_file_names_array+=($page_mp4_file_half1)
        IFS=$SAVE_IFS

        page_mp4_file_half2="${input_pdf_file_name}_${page}_half2.mp4"
        
        ffmpeg $ffmpeg_pre_options -i "$page_image_file_half2" -i "$page_mp3_file_half2" $ffmpeg_options "$page_mp4_file_half2"
        
        SAVE_IFS=$IFS
        IFS=""
        video_file_names_array+=($page_mp4_file_half2)
        IFS=$SAVE_IFS
        
        rm "$page_image_file_half1"
        rm "$page_image_file_half2"
        
        rm "$file_mp3_half1"
        rm "$file_mp3_half2"
        rm "$page_mp3_file_half1"
        rm "$page_mp3_file_half2"

    else

        ~/txt2mp3_by_yatts "$page_text_file"
    
        page_mp3_file="${page_text_file}_yatts.mp3"

        page_mp4_file="${input_pdf_file_name}_${page}.mp4"
        
        ffmpeg $ffmpeg_pre_options -i "$page_image_file" -i "$page_mp3_file" $ffmpeg_options "$page_mp4_file"
        
        SAVE_IFS=$IFS
        IFS=""
        video_file_names_array+=($page_mp4_file)
        IFS=$SAVE_IFS
   
        rm "$page_mp3_file"

    fi
    
    rm "$page_image_file"
    rm "$page_text_file"
  
    
done

cmd=""
for ((i = 0; i < ${#video_file_names_array[@]}; i++)) do
    cmd="$cmd|${video_file_names_array[$i]}"
done
cmd=${cmd:1}

out_file="${input_pdf_file_name}.mp4"

echo "Объединяем файлы $cmd в $out_file"
ffmpeg -i "concat:$cmd" -acodec copy -vcodec copy "$out_file"

IFS=""
for ((i = 0; i < ${#video_file_names_array[@]}; i++)) do
    f="${video_file_names_array[$i]}"
    echo "Удаляем файл '$f'"
    rm "$f"
done

echo "Конечный файл создан '$out_file'!"