"""Wrapper functions for input interface."""
from pathlib import Path
from typing import Any, List, Optional, Union
from mido import MidiFile
from music21.stream import Stream
from numpy import ndarray
from pretty_midi import PrettyMIDI
from pypianoroll import Multitrack
from ..classes import Track
from ..music import Music
from .abc import read_abc
from .event import from_event_representation
from .json import load_json
from .midi import from_mido, from_pretty_midi, read_midi
from .music21 import from_music21
from .musicxml import read_musicxml
from .note import from_note_representation
from .pianoroll import from_pianoroll_representation, from_pypianoroll
from .pitch import from_pitch_representation
from .yaml import load_yaml
[docs]def load(
path: Union[str, Path], kind: Optional[str] = None, **kwargs: Any
) -> Music:
"""Load a JSON or a YAML file into a Music object.
Parameters
----------
path : str or Path
Path to the file to load.
kind : {'json', 'yaml'}, optional
Format to save (case-insensitive). Defaults to infer the format from
the extension.
**kwargs : dict
Keyword arguments to pass to the target function. See
:func:`muspy.load_json` or :func:`muspy.load_yaml` for available
arguments.
Returns
-------
:class:`muspy.Music`
Loaded Music object.
See Also
--------
:func:`muspy.read` :
Read a MIDI/MusicXML/ABC file into a Music object.
"""
# pylint: disable=unused-argument
if kind is None:
if str(path).lower().endswith(".json"):
kind = "json"
elif str(path).lower().endswith((".yaml", ".yml")):
kind = "yaml"
else:
raise ValueError(
"Got unsupported file format (expect JSON or YAML)."
)
if kind == "json":
return load_json(path)
if kind == "yaml":
return load_yaml(path)
raise ValueError("`kind` must be either 'json' or 'yaml'.")
[docs]def read(
path: Union[str, Path], kind: Optional[str] = None, **kwargs: Any
) -> Union[Music, List[Music]]:
"""Read a MIDI/MusicXML/ABC file into a Music object.
Parameters
----------
path : str or Path
Path to the file to read.
kind : {'midi', 'musicxml', 'abc'}, optional
Format to save (case-insensitive). Defaults to infer the format from
the extension.
Returns
-------
:class:`muspy.Music` or list of :class:`muspy.Music`
Converted Music object(s).
See Also
--------
:func:`muspy.load` : Load a JSON or a YAML file into a Music object.
"""
if kind is None:
if str(path).lower().endswith((".mid", ".midi")):
kind = "midi"
elif str(path).lower().endswith((".mxl", ".xml", ".musicxml")):
kind = "musicxml"
elif str(path).lower().endswith(".abc"):
kind = "abc"
else:
raise ValueError(
"Got unsupported file format (expect MIDI, MusicXML or ABC)."
)
if kind == "midi":
return read_midi(path, **kwargs)
if kind == "musicxml":
return read_musicxml(path, **kwargs)
if kind == "abc":
return read_abc(path, **kwargs)
raise ValueError("`kind` must be one of 'midi', 'musicxml' and 'abc'.")
[docs]def from_object(
obj: Union[Stream, MidiFile, PrettyMIDI, Multitrack], **kwargs: Any
) -> Union[Music, List[Music], Track, List[Track]]:
"""Return an outside object as a Music object.
Parameters
----------
obj
Object to convert. Supported objects are `music21.Stream`,
:class:`mido.MidiTrack`, :class:`pretty_midi.PrettyMIDI`, and
:class:`pypianoroll.Multitrack` objects.
Returns
-------
:class:`muspy.Music`
Converted Music object.
"""
if isinstance(obj, Stream):
return from_music21(obj, **kwargs)
if isinstance(obj, MidiFile):
return from_mido(obj, **kwargs)
if isinstance(obj, PrettyMIDI):
return from_pretty_midi(obj)
if isinstance(obj, Multitrack):
return from_pypianoroll(obj, **kwargs)
raise TypeError(
"`obj` must be of type music21.Stream, mido.MidiFile, "
"pretty_midi.PrettyMIDI or pypianoroll.Multitrack."
)
[docs]def from_representation(array: ndarray, kind: str, **kwargs: Any) -> Music:
"""Update with the given representation.
Parameters
----------
array : :class:`numpy.ndarray`
Array in a supported representation.
kind : str, {'pitch', 'pianoroll', 'event', 'note'}
Data representation type (case-insensitive).
Returns
-------
:class:`muspy.Music`
Converted Music object.
"""
if kind.lower() in ("pitch", "pitch-based"):
return from_pitch_representation(array, **kwargs)
if kind.lower() in ("pianoroll", "piano-roll", "piano roll"):
return from_pianoroll_representation(array, **kwargs)
if kind.lower() in ("event", "event-based"):
return from_event_representation(array, **kwargs)
if kind.lower() in ("note", "note-based"):
return from_note_representation(array, **kwargs)
raise ValueError(
"`kind` must be one of 'pitch', 'pianoroll', 'event' and 'note'."
)