Class Diagnostics

python/tvm/script/parser/core/diagnostics.py:261–305

Diagnostics class for error reporting in parser. Formats parse errors with source location context and raises directly, without going through DiagnosticContext. Parameters ---------- source : Source The source code.

class Diagnostics:
    """Diagnostics class for error reporting in parser.

    Formats parse errors with source location context and raises directly,
    without going through DiagnosticContext.

    Parameters
    ----------
    source : Source
        The source code.
    """

    source: Source

    def __init__(self, source: Source):
        self.source = source

    def error(self, node: doc.AST, message: str) -> None:
        """Emit a diagnostic error by raising with source location context.

        Parameters
        ----------
        node : doc.AST
            The node with diagnostic error.

        message : str
            The diagnostic message.
        """
        lineno = getattr(node, "lineno", 1)
        col_offset = getattr(node, "col_offset", self.source.start_column)
        end_lineno = getattr(node, "end_lineno", lineno)
        end_col_offset = getattr(node, "end_col_offset", col_offset)
        lineno += self.source.start_line - 1
        end_lineno += self.source.start_line - 1
        col_offset += self.source.start_column + 1
        end_col_offset += self.source.start_column + 1

        source_lines = self.source.full_source.splitlines(keepends=True)
        snippet = _format_source_snippet(
            source_lines, lineno, col_offset, end_lineno, end_col_offset
        )

        location = f"{self.source.source_name}:{lineno}:{col_offset}"
        formatted = f"error: {message}\n --> {location}\n{snippet}"
        raise DiagnosticError(formatted)