slidegen/slides/engine/song_slides.py

"""
Copyright © 2022 Noah Vogt <noah@noahvogt.com>

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

from os import path

from wand.exceptions import BlobError

from utils import (
    log,
    error_msg,
)

import config as const


def generate_song_slides(
    slidegen,
    slide_count,
    template_img,
    zfill_length,
) -> None:
    log("generating song slides...")
    # unique_structures: list = list(set(chosen_structure))

    current_slide_index: int = 0

    for index, structure in enumerate(slidegen.chosen_structure):
        structure_element_splitted: list = slidegen.songtext[
            structure
        ].splitlines()
        line_count = len(structure_element_splitted)
        use_line_ranges_per_index = []
        use_lines_per_index = []
        if line_count <= const.STRUCTURE_ELEMENT_MAX_LINES:
            inner_slide_count = 1
        else:
            inner_slide_count: int = (
                line_count // const.STRUCTURE_ELEMENT_MAX_LINES + 1
            )
            use_lines_per_index = [
                line_count // inner_slide_count
            ] * inner_slide_count

            for inner_slide in range(inner_slide_count):
                if sum(use_lines_per_index) == line_count:
                    break
                use_lines_per_index[inner_slide] = (
                    use_lines_per_index[inner_slide] + 1
                )
            for inner_slide in range(inner_slide_count):
                use_line_ranges_per_index.append(
                    sum(use_lines_per_index[:inner_slide])
                )

        for inner_slide in range(inner_slide_count):
            current_slide_index += 1

            log(
                "generating song slide [{} / {}]...".format(
                    current_slide_index, slide_count
                )
            )

            if inner_slide_count == 1:
                structure_element_value: str = slidegen.songtext[structure]
            else:
                splitted_wanted_range: list = structure_element_splitted[
                    use_line_ranges_per_index[
                        inner_slide
                    ] : use_line_ranges_per_index[inner_slide]
                    + use_lines_per_index[inner_slide]
                ]
                structure_element_value: str = ""
                for element in splitted_wanted_range:
                    structure_element_value += element + "\n"

                structure_element_value = structure_element_value[:-1]

            song_slide = slidegen.song_slide_form()
            song_slide_img = song_slide.get_slide(
                template_img,
                structure_element_value,
                slidegen.chosen_structure,
                index,
                bool(
                    inner_slide_count != 1
                    and inner_slide != inner_slide_count - 1
                ),
            )
            song_slide_img.format = const.IMAGE_FORMAT
            try:
                song_slide_img.save(
                    filename=path.join(
                        slidegen.output_dir,
                        const.FILE_NAMEING
                        + str(current_slide_index + 1).zfill(zfill_length)
                        + "."
                        + const.FILE_EXTENSION,
                    )
                )
            except BlobError:
                error_msg("could not write slide to target directory")