glyphs.py

from typing import Tuple

from reportlab.platypus import Flowable

from lyric import Lyric
from neume_chunk import NeumeChunk
from neume_dict import stand_alone


class Glyph(Flowable):
    def __init__(self,
                 neume_chunk: NeumeChunk = None,
                 neume_chunk_pos: Tuple[float, float] = None,
                 lyric: Lyric = None,
                 lyric_pos: Tuple[float, float] = None,
                 # fthora: Fthora = None,
                 # fthora_pos: Tuple = [float, float]
                 ):
        super().__init__()
        self.neume_chunk = neume_chunk  # A list of neumes, one base neume and some zero width supporting characters
        self.neume_chunk_pos = neume_chunk_pos if neume_chunk_pos is not None else [0, 0]
        self.lyric: Lyric = lyric
        self.lyric_pos = lyric_pos if lyric_pos is not None else [0, 0]
        # self.fthora = fthora
        # self.fthora_pos = fthora_pos if fthora_pos is not None else [0, 0]
        self.set_size()

    def set_size(self):
        self.width = max(getattr(self.neume_chunk, 'width', 0), getattr(self.lyric, 'width', 0))
        self.height = getattr(self.neume_chunk, 'height', 0)\
            + getattr(self.lyric, 'height', 0)

    def draw(self, canvas):
        canvas.saveState()
        canvas.translate(self.neume_chunk_pos[0], self.neume_chunk_pos[1])
        pos_x: float = 0
        for i, neume in enumerate(self.neume_chunk):
            canvas.setFillColor(neume.color)
            canvas.setFont(neume.font_family, neume.font_size)
            if i > 0:
                pos_x += self.neume_chunk[i - 1].width
            canvas.drawString(pos_x, self.neume_chunk_pos[1], neume.char)
        canvas.restoreState()

        if self.lyric and self.lyric.text != '_':
            canvas.setFillColor(self.lyric.color)
            canvas.setFont(self.lyric.font_family, self.lyric.font_size)
            canvas.drawString(self.lyric_pos[0], self.lyric_pos[1], self.lyric.text)

    def has_lyric_text(self, text: str):
        if self.lyric and self.lyric.text == text:
            return True
        else:
            return False

    def get_standalone_neume(self):
        for neume in self.neume_chunk:
            if stand_alone(neume):
                return neume