pandoc_beamer_arrow._main.tikz()

Complexity

Conditions

17

Size

Total Lines	96
Code Lines	52

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

#!/usr/bin/env python
def tikz(elem: Element, doc: Doc) -> RawInline | None:
    """
    Add admonition to elem.

    Arguments
    ---------
    elem
        The current element
    doc
        The pandoc document

    Returns
    -------
    RawInline | None
        The modified element or None
    """
    # Is it in the right format and is it Div or a CodeBlock?
    if doc.format in ("beamer") and elem.tag in ("Span"):
        # Is there a latex-admonition-color attribute?
        if "beamer-arrow-node" in elem.classes:
            text = convert_text(
                Plain(*elem.content), input_format="panflute", output_format="latex"
            )

            options = ["anchor=base"]

            color = get_color(elem, doc)
            if color:
                options.append(f"fill={color}")

            (from_value, to_value) = get_range(elem, doc)
            if from_value or to_value:
                display = f"\\only<{from_value}-{to_value}>"
            else:
                display = ""

            return RawInline(
                f"\\tikz[baseline]{{"
                f"{display}{{\\node[{','.join(options)}] "
                f"({elem.identifier}) "
                f"{{{text}}}"
                f";}}}}",
                format="tex",
            )

        if "beamer-arrow-edge" in elem.classes:
            angles = []
            if "angle_src" in elem.attributes:
                try:
                    angle = elem.attributes["angle_src"]
                    angle = int(angle)
                    angles.append(f"out={angle}")
                except ValueError:
                    debug(f"pandoc-beamer-arrow: angle_src '{angle}' is not correct")

            if "angle_dest" in elem.attributes:
                try:
                    angle = elem.attributes["angle_dest"]
                    angle = int(angle)
                    angles.append(f"in={angle}")
                except ValueError:
                    debug(f"pandoc-beamer-arrow: angle_dest '{angle}' is not correct")

            options = ["->"]

            color = get_color(elem, doc)
            if color:
                options.append(color)

            if "linewidth" in elem.attributes:
                try:
                    linewidth = elem.attributes["linewidth"]
                    linewidth = int(linewidth)
                    options.append(f"line width={linewidth}pt")
                except ValueError:
                    debug(
                        f"pandoc-beamer-arrow: linewidth '{linewidth}' is not correct"
                    )

            (from_value, to_value) = get_range(elem, doc)
            if from_value or to_value:
                display = f"<{from_value}-{to_value}>"
            else:
                display = ""

            return RawInline(
                f"\\begin{{tikzpicture}}[overlay]"
                f"\\path[{','.join(options)}]{display} "
                f"({elem.attributes['src']}) "
                f"edge "
                f"[{','.join(angles)}] "
                f"({elem.attributes['dest']});"
                f"\\end{{tikzpicture}}",
                format="tex",
            )
    return None