Source code for textmate_grammar.parsers.matlab

from __future__ import annotations

import plistlib
import re
from pathlib import Path

import yaml

from ..base import LanguageParser




class MatlabParser(LanguageParser):
    """
    Represents a grammar for the MATLAB language.
    """

    def __init__(self, remove_line_continuations: bool = False, **kwargs):
        """
        Initializes a new instance of the MatlabGrammar class.

        Args:
            remove_line_continuations (bool, optional): Whether to remove line continuations. Defaults to False.
        """

        self._rlc = remove_line_continuations

        tmLanguageFile = (
            Path(__file__).parents[3]
            / "syntaxes"
            / "matlab"
            / "Matlab.tmbundle"
            / "Syntaxes"
            / "MATLAB.tmLanguage"
        )
        tmLanguageYAML = Path(__file__).parent / "grammar.yaml"

        if tmLanguageFile.exists():
            with open(tmLanguageFile, "rb") as tmFile:
                grammar = plistlib.load(tmFile, fmt=plistlib.FMT_XML)
            with open(tmLanguageYAML, "w") as ymlFile:
                ymlFile.write(yaml.dump(grammar, indent=2))
        else:
            with open(tmLanguageYAML) as ymlFile:
                try:
                    grammar = yaml.load(ymlFile.read(), Loader=yaml.CLoader)
                except ImportError:
                    grammar = yaml.load(ymlFile.read(), Loader=yaml.Loader)

        super().__init__(grammar, **kwargs)



    def pre_process(self, input: str) -> str:
        """
        Pre-processes the input text.
        """
        if self._rlc:
            input = self._remove_line_continuations(input)
        return input


    def _remove_line_continuations(self, input: str) -> str:
        """
        Removes line continuations from the input text.
        """
        if "..." not in input:
            return input
        output = ""
        for split in input.split("..."):
            matching = re.search(r"\n[\t\f\v ]*", split)
            if matching:
                output += split[matching.span()[1] :]
            else:
                output += split
        return output