27 lines
1.0 KiB
Python
27 lines
1.0 KiB
Python
from smnp.function.model import Function, CombinedFunction
|
|
from smnp.function.signature import varargSignature
|
|
from smnp.note.interval import intervalToString
|
|
from smnp.note.model import Note
|
|
from smnp.type.model import Type
|
|
from smnp.type.signature.matcher.list import listOf
|
|
from smnp.type.signature.matcher.type import ofTypes
|
|
|
|
_signature1 = varargSignature(ofTypes(Type.NOTE, Type.INTEGER))
|
|
def _function1(env, vararg):
|
|
withoutPauses = [note.value for note in vararg if note.type == Type.NOTE]
|
|
if len(withoutPauses) < 2:
|
|
return Type.list([])
|
|
semitones = [Note.checkInterval(withoutPauses[i-1], withoutPauses[i]) for i in range(1, len(withoutPauses))]
|
|
return Type.list([Type.string(intervalToString(s)) for s in semitones]).decompose()
|
|
|
|
|
|
_signature2 = varargSignature(listOf(Type.NOTE, Type.INTEGER))
|
|
def _function2(env, vararg):
|
|
return Type.list([_function1(env, arg.value) for arg in vararg]).decompose()
|
|
|
|
|
|
function = CombinedFunction(
|
|
'interval',
|
|
Function(_signature1, _function1),
|
|
Function(_signature2, _function2)
|
|
) |