kernpy 0.0.2__py3-none-any.whl → 1.0.1__py3-none-any.whl

kernpy/core/importer.py

@@ -0,0 +1,285 @@
+from __future__ import annotations
+
+import csv
+import io
+from copy import copy
+from pathlib import Path
+from typing import List, Optional
+
+from kernpy.core.tokens import TokenCategory, SignatureToken, MetacommentToken, HeaderToken, SpineOperationToken, \
+    FieldCommentToken, ErrorToken, \
+    BoundingBoxToken, SPINE_OPERATIONS, HEADERS, Token
+from kernpy.core.document import Document, MultistageTree, BoundingBoxMeasures
+from kernpy.core.importer_factory import createImporter
+
+
+class Importer:
+    """
+    Importer class.
+
+    Use this class to import the content from a file or a string to a `Document` object.
+    """
+    def __init__(self):
+        """
+        Create an instance of the importer.
+
+        Raises:
+            Exception: If the importer content is not a valid **kern file.
+
+        Examples:
+            # Create the importer
+            >>> importer = Importer()
+
+            # Import the content from a file
+            >>> document = importer.import_file('file.krn')
+
+            # Import the content from a string
+            >>> document = importer.import_string("**kern\n*clefF4\nc4\n4d\n4e\n4f\n*-")
+        """
+        self.last_measure_number = None
+        self.last_bounding_box = None
+        self.errors = []
+
+        self._tree = MultistageTree()
+        self._document = Document(self._tree)
+        self._importers = {}
+        self._header_row_number = None
+        self._row_number = 1
+        self._tree_stage = 0
+        self._next_stage_parents = None
+        self._prev_stage_parents = None
+        self._last_node_previous_to_header = self._tree.root
+
+    @staticmethod
+    def get_last_spine_operator(parent):
+        if parent is None:
+            return None
+        elif isinstance(parent.token, SpineOperationToken):
+            return parent
+        else:
+            return parent.last_spine_operator_node
+
+    #TODO Documentar cómo propagamos los header_node y last_spine_operator_node...
+    def run(self, reader) -> Document:
+        for row in reader:
+            if len(row) <= 0:
+                # Found an empty row, usually the last one. Ignore it.
+                continue
+
+            self._tree_stage = self._tree_stage + 1
+            is_barline = False
+            if self._next_stage_parents:
+                self._prev_stage_parents = copy(self._next_stage_parents)
+            self._next_stage_parents = []
+
+            if row[0].startswith("!!"):
+                self._compute_metacomment_token(row[0].strip())
+            else:
+                for icolumn, column in enumerate(row):
+                    if column.startswith("**"):
+                        self._compute_header_token(icolumn, column)
+                        # go to next row
+                        continue
+
+                    if column in SPINE_OPERATIONS:
+                        self._compute_spine_operator_token(icolumn, column, row)
+                    else:  # column is not a spine operation
+                        if column.startswith("!"):
+                            token = FieldCommentToken(column)
+                        else:
+                            if self._prev_stage_parents is None:
+                                raise ValueError(f'Any spine header found in the column #{icolumn}. '
+                                                 f'Expected a previous line with valid content. '
+                                                 f'The token in column #{icolumn} and row #{self._row_number - 1}'
+                                                 f' was not created correctly. Error detected in '
+                                                 f'column #{icolumn} in row #{self._row_number}. '
+                                                 f'Found {column}. ')
+                            if icolumn >= len(self._prev_stage_parents):
+                                # TODO: Try to fix the kern in runtime. Add options to public API
+                                # continue  # ignore the column
+                                raise ValueError(f'Wrong columns number in row {self._row_number}. '
+                                                 f'The token in column #{icolumn} and row #{self._row_number}'
+                                                 f' has more columns than expected in its row. '
+                                                 f'Expected {len(self._prev_stage_parents)} columns '
+                                                 f'but found {len(row)}.')
+                            parent = self._prev_stage_parents[icolumn]
+                            if not parent:
+                                raise Exception(f'Cannot find a parent node for column #{icolumn} in row {self._row_number}')
+                            if not parent.header_node:
+                                raise Exception(f'Cannot find a header node for column #{icolumn} in row {self._row_number}')
+                            importer = self._importers.get(parent.header_node.token.encoding)
+                            if not importer:
+                                raise Exception(f'Cannot find an importer for header {parent.header_node.token.encoding}')
+                            try:
+                                token = importer.import_token(column)
+                            except Exception as error:
+                                token = ErrorToken(column, self._row_number, str(error))
+                                self.errors.append(token)
+                        if not token:
+                            raise Exception(
+                                f'No token generated for input {column} in row number #{self._row_number} using importer {importer}')
+
+                        parent = self._prev_stage_parents[icolumn]
+                        node = self._tree.add_node(self._tree_stage, parent, token, self.get_last_spine_operator(parent), parent.last_signature_nodes, parent.header_node)
+                        self._next_stage_parents.append(node)
+
+                        if (token.category == TokenCategory.BARLINES
+                                or TokenCategory.is_child(child=token.category, parent=TokenCategory.CORE)
+                                    and len(self._document.measure_start_tree_stages) == 0):
+                            is_barline = True
+                        elif isinstance(token, BoundingBoxToken):
+                            self.handle_bounding_box(self._document, token)
+                        elif isinstance(token, SignatureToken):
+                            node.last_signature_nodes.update(node)
+
+                if is_barline:
+                    self._document.measure_start_tree_stages.append(self._tree_stage)
+                    self.last_measure_number = len(self._document.measure_start_tree_stages)
+                    if self.last_bounding_box:
+                        self.last_bounding_box.to_measure = self.last_measure_number
+            self._row_number = self._row_number + 1
+        return self._document
+
+    def handle_bounding_box(self, document: Document, token: BoundingBoxToken):
+        page_number = token.page_number
+        last_page_bb = document.page_bounding_boxes.get(page_number)
+        if last_page_bb is None:
+            if self.last_measure_number is None:
+                self.last_measure_number = 0
+            self.last_bounding_box = BoundingBoxMeasures(token.bounding_box, self.last_measure_number,
+                                                         self.last_measure_number)
+            document.page_bounding_boxes[page_number] = self.last_bounding_box
+        else:
+            last_page_bb.bounding_box.extend(token.bounding_box)
+            last_page_bb.to_measure = self.last_measure_number
+
+    def import_file(self, file_path: Path) -> Document:
+        """
+        Import the content from the importer to the file.
+        Args:
+            file_path: The path to the file.
+
+        Returns:
+            Document - The document with the imported content.
+
+        Examples:
+            # Create the importer and read the file
+            >>> importer = Importer()
+            >>> importer.import_file('file.krn')
+        """
+        with open(file_path, 'r', newline='', encoding='utf-8', errors='ignore') as file:
+            reader = csv.reader(file, delimiter='\t')
+            return self.run(reader)
+
+    def import_string(self, text: str) -> Document:
+        """
+        Import the content from the content of the score in string format.
+
+        Args:
+            text: The content of the score in string format.
+
+        Returns:
+            Document - The document with the imported content.
+
+        Examples:
+            # Create the importer and read the file
+            >>> importer = Importer()
+            >>> importer.import_string("**kern\n*clefF4\nc4\n4d\n4e\n4f\n*-")
+            # Read the content from a file
+            >>> with open('file.krn',  'r', newline='', encoding='utf-8', errors='ignore') as f: # We encourage you to use these open file options
+            >>>     content = f.read()
+            >>> importer.import_string(content)
+            >>> document = importer.import_string(content)
+        """
+        lines = text.splitlines()
+        reader = csv.reader(lines, delimiter='\t')
+        return self.run(reader)
+
+    def get_error_messages(self) -> str:
+        """
+        Get the error messages of the importer.
+
+        Returns: str - The error messages split by a new line character.
+
+        Examples:
+            # Create the importer and read the file
+            >>> importer = Importer()
+            >>> importer.import_file(Path('file.krn'))
+            >>> print(importer.get_error_messages())
+            'Error: Invalid token in row 1'
+        """
+        result = ''
+        for err in self.errors:
+            result += str(err)
+            result += '\n'
+        return result
+
+    def has_errors(self) -> bool:
+        """
+        Check if the importer has any errors.
+
+        Returns: bool - True if the importer has errors, False otherwise.
+
+        Examples:
+            # Create the importer and read the file
+            >>> importer = Importer()
+            >>> importer.import_file(Path('file.krn'))    # file.krn has an error
+            >>> print(importer.has_errors())
+            True
+            >>> importer.import_file(Path('file2.krn'))   # file2.krn has no errors
+            >>> print(importer.has_errors())
+            False
+        """
+        return len(self.errors) > 0
+
+    def _compute_metacomment_token(self, raw_token: str):
+        token = MetacommentToken(raw_token)
+        if self._header_row_number is None:
+            node = self._tree.add_node(self._tree_stage, self._last_node_previous_to_header, token, None, None, None)
+            self._last_node_previous_to_header = node
+        else:
+            for parent in self._prev_stage_parents:
+                node = self._tree.add_node(self._tree_stage, parent, token, self.get_last_spine_operator(parent), parent.last_signature_nodes, parent.header_node) # the same reference for all spines - TODO Recordar documentarlo
+                self._next_stage_parents.append(node)
+
+    def _compute_header_token(self, column_index: int, column_content: str):
+        if self._header_row_number is not None and self._header_row_number != self._row_number:
+            raise Exception(
+                f"Several header rows not supported, there is a header row in #{self._header_row_number} and another in #{self._row_number} ")
+
+            # it's a spine header
+        self._document.header_stage = self._tree_stage
+        importer = self._importers.get(column_content)
+        if not importer:
+            importer = createImporter(column_content)
+            self._importers[column_content] = importer
+
+        token = HeaderToken(column_content, spine_id=column_index)
+        node = self._tree.add_node(self._tree_stage, self._last_node_previous_to_header, token, None, None)
+        node.header_node = node # this value will be propagated
+        self._next_stage_parents.append(node)
+
+    def _compute_spine_operator_token(self, column_index: int, column_content: str, row: List[str]):
+        token = SpineOperationToken(column_content)
+
+        if column_index >= len(self._prev_stage_parents):
+            raise Exception(f'Expected at least {column_index+1} parents in row {self._row_number}, but found {len(self._prev_stage_parents)}: {row}')
+
+        parent = self._prev_stage_parents[column_index]
+        node = self._tree.add_node(self._tree_stage, parent, token, self.get_last_spine_operator(parent), parent.last_signature_nodes, parent.header_node)
+
+        if column_content == '*-':
+            if node.last_spine_operator_node is not None:
+                node.last_spine_operator_node.token.cancelled_at_stage = self._tree_stage
+            pass # it's terminated, no continuation
+        elif column_content == "*+" or column_content == "*^":
+            self._next_stage_parents.append(node)
+            self._next_stage_parents.append(node) # twice, the next stage two children will have this one as parent
+        elif column_content == "*v":
+            if node.last_spine_operator_node is not None:
+                node.last_spine_operator_node.token.cancelled_at_stage = self._tree_stage
+
+            if column_index == 0 or row[column_index-1] != '*v' or self._prev_stage_parents[column_index-1].header_node != self._prev_stage_parents[column_index].header_node: # don't collapse two different spines
+                self._next_stage_parents.append(node) # just one spine each two
+        else:
+            raise Exception(f'Unknown spine operation in column #{column_content} and row #{self._row_number}')

kernpy/core/importer_factory.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+import sys
+
+from .dyn_importer import DynSpineImporter
+from .dynam_spine_importer import DynamSpineImporter
+from .fing_spine_importer import FingSpineImporter
+from .harm_spine_importer import HarmSpineImporter
+from .kern_spine_importer import KernSpineImporter
+from .mens_spine_importer import MensSpineImporter
+from .mhxm_spine_importer import MxhmSpineImporter
+from .root_spine_importer import RootSpineImporter
+from .spine_importer import SpineImporter
+from .text_spine_importer import TextSpineImporter
+from .basic_spine_importer import BasicSpineImporter
+
+
+def createImporter(spine_type: str) -> SpineImporter:
+    if spine_type == '**mens':
+        return MensSpineImporter()
+    elif spine_type == '**kern':
+        return KernSpineImporter()
+    elif spine_type == '**text':
+        return TextSpineImporter()
+    elif spine_type == '**harm':
+        return HarmSpineImporter()
+    elif spine_type == '**mxhm':
+        return MxhmSpineImporter()
+    elif spine_type == '**root':
+        return RootSpineImporter()
+    elif spine_type == '**dyn':
+        return DynSpineImporter()
+    elif spine_type == '**dynam':
+        return DynamSpineImporter()
+    elif spine_type == '**fing':
+        return FingSpineImporter()
+    else:
+        #print(f'Invalid spine header type found: {spine_type} '
+        #      f'using a basic spine importer instead',
+        #      file=sys.stderr)
+        # TODO: Should we use a logger? Global variable for verbosity?
+
+        return BasicSpineImporter()  # Only parse basic token categories
+

kernpy/core/kern_spine_importer.py

@@ -0,0 +1,73 @@
+from __future__ import annotations
+
+from typing import List
+
+from antlr4 import InputStream, CommonTokenStream, ParseTreeWalker, BailErrorStrategy, \
+    PredictionMode
+from typing import Optional
+
+from .base_antlr_importer import BaseANTLRListenerImporter
+from .base_antlr_spine_parser_listener import BaseANTLRSpineParserListener
+from .error_listener import ErrorListener
+from .generated.kernSpineLexer import kernSpineLexer
+from .generated.kernSpineParser import kernSpineParser
+from .spine_importer import SpineImporter
+from .tokens import SimpleToken, TokenCategory, Subtoken, ChordToken, BoundingBox, \
+    BoundingBoxToken, ClefToken, KeySignatureToken, TimeSignatureToken, MeterSymbolToken, BarToken, NoteRestToken, \
+    KeyToken, InstrumentToken
+
+
+class KernSpineListener(BaseANTLRSpineParserListener):
+
+    def __init__(self):
+        super().__init__()
+
+class KernListenerImporter(BaseANTLRListenerImporter):
+
+    def createListener(self):
+        return KernSpineListener()
+
+    def createLexer(self, charStream):
+        return kernSpineLexer(charStream)
+
+    def createParser(self, tokenStream):
+        return kernSpineParser(tokenStream)
+
+    def startRule(self):
+        return self.parser.start()
+
+
+class KernSpineImporter(SpineImporter):
+    def __init__(self, verbose: Optional[bool] = False):
+        """
+        KernSpineImporter constructor.
+
+        Args:
+            verbose (Optional[bool]): Level of verbosity for error messages.
+        """
+        super().__init__(verbose=verbose)
+
+    def import_listener(self) -> BaseANTLRSpineParserListener:
+        return KernSpineListener()
+
+    def import_token(self, encoding: str):
+        self._raise_error_if_wrong_input(encoding)
+
+        # self.listenerImporter = KernListenerImporter(token) # TODO ¿Por qué no va esto?
+        # self.listenerImporter.start()
+        lexer = kernSpineLexer(InputStream(encoding))
+        lexer.removeErrorListeners()
+        lexer.addErrorListener(self.error_listener)
+        stream = CommonTokenStream(lexer)
+        parser = kernSpineParser(stream)
+        parser._interp.predictionMode = PredictionMode.SLL  # it improves a lot the parsing
+        parser.removeErrorListeners()
+        parser.addErrorListener(self.error_listener)
+        parser.errHandler = BailErrorStrategy()
+        tree = parser.start()
+        walker = ParseTreeWalker()
+        listener = KernSpineListener()
+        walker.walk(listener, tree)
+        if self.error_listener.getNumberErrorsFound() > 0:
+            raise Exception(self.error_listener.errors)
+        return listener.token

kernpy/core/mens_spine_importer.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+from typing import Optional
+
+from .base_antlr_spine_parser_listener import BaseANTLRSpineParserListener
+from .spine_importer import SpineImporter
+from .tokens import Token
+
+
+class MensSpineImporter(SpineImporter):
+    def __init__(self, verbose: Optional[bool] = False):
+        """
+        MensSpineImporter constructor.
+
+        Args:
+            verbose (Optional[bool]): Level of verbosity for error messages.
+        """
+        super().__init__(verbose=verbose)
+
+    def import_listener(self) -> BaseANTLRSpineParserListener:
+        raise NotImplementedError()
+
+    def import_token(self, encoding: str) -> Token:
+        raise NotImplementedError()

kernpy/core/mhxm_spine_importer.py

@@ -0,0 +1,44 @@
+from __future__ import annotations
+from typing import Optional
+
+from .kern_spine_importer import KernSpineListener, KernSpineImporter
+from .base_antlr_spine_parser_listener import BaseANTLRSpineParserListener
+from .spine_importer import SpineImporter
+from .tokens import MHXMToken, Token, TokenCategory
+
+
+class MxhmSpineImporter(SpineImporter):
+    def __init__(self, verbose: Optional[bool] = False):
+        """
+        KernSpineImporter constructor.
+
+        Args:
+            verbose (Optional[bool]): Level of verbosity for error messages.
+        """
+        super().__init__(verbose=verbose)
+
+    def import_listener(self) -> BaseANTLRSpineParserListener:
+        return KernSpineListener()
+
+    def import_token(self, encoding: str) -> Token:
+        self._raise_error_if_wrong_input(encoding)
+
+        kern_spine_importer = KernSpineImporter()
+        token = kern_spine_importer.import_token(encoding)
+
+        ACCEPTED_CATEGORIES = {
+            TokenCategory.STRUCTURAL,
+            TokenCategory.SIGNATURES,
+            TokenCategory.EMPTY,
+            TokenCategory.IMAGE_ANNOTATIONS,
+            TokenCategory.BARLINES,
+            TokenCategory.COMMENTS,
+        }
+
+        if any(TokenCategory.is_child(child=token.category, parent=cat) for cat in ACCEPTED_CATEGORIES):
+            return SimpleToken(encoding, TokenCategory.HARMONY)
+
+        return token
+
+        return MHXMToken(encoding)
+