smnp-py/Environment.py

import sys
from Evaluator import RuntimeException, objectString
from Note import *
import random
import Synth
import time

types = {
    int: 'integer',
    str: 'string',
    list: 'list',
    float: 'percent',
    Note: 'note', 
    type(None): 'void'
}

class Environment():
    def __init__(self, scopes, functions):
        self.scopes = scopes
        self.functions = functions
    
    def findVariable(self, name, type=None):
        for scope in reversed(self.scopes):
            if name in scope:
                value = scope[name]
                if type is not None:
                    if isinstance(value, type): 
                        return value         
                else:
                    return value
        raise RuntimeException(f"Variable '{name}' is not declared" + ("" if type is None else f" (expected type: {types[type]})"))
    
    def findVariableScope(self, name, type=None):
        for scope in reversed(self.scopes):
            if name in scope:                
                if type is not None:
                    if isinstance(scope[name], type): 
                        return scope         
                else:
                    return scope        

def sample(args, env):
    if len(args) == 1 and isinstance(args[0], list):
        return _sample(args[0])
    elif len(args) == 0:
        return _sample(Note.range(Note(NotePitch.C), Note(NotePitch.H)))
    elif all(isinstance(x, Note) for x in args):
        return _sample(args)
    else:
        pass # not valid signature
 
def _sample(list):
    return list[int(random.uniform(0, len(list)))]
 
def doPrint(args, env):
    print("".join([objectString(arg) for arg in args]))

def semitonesList(list):
    for x in list:
        if not isinstance(x, Note) and not isistance(x, int):
            pass # invalid arguments
    withoutPauses = tuple(filter(lambda x: isinstance(x, Note), list))    
    r = [Note.checkInterval(withoutPauses[i-1], withoutPauses[i]) for i, _ in enumerate(withoutPauses) if i != 0]
    return returnElementOrList(r)

def returnElementOrList(list):
    return list[0] if len(list) == 1 else list

def semitones(args, env):
    if len(args) > 0 and isinstance(args[0], list):
        return semitonesList(args[0])
    return semitonesList(args)

def intervalList(list):    
    r = [intervalToString(x) for x in list]
    return returnElementOrList(r)

def interval(args, env):    
    if len(args) > 0 and isinstance(args[0], list):        
        return intervalList(args[0])
    return intervalList(args)

def transposeTo(args, env):
    if len(args) > 1 and isinstance(args[0], Note) and all(isinstance(x, list) for i, x in enumerate(args) if i != 0):        
        target = args[0]        
        result = []
        for i, notes in enumerate(args):
            if i == 0:
                continue
            if len(notes) > 0:
                first = notes[0]
                semitones = semitonesList([target, first])
                result.append([note.transpose(semitones) for note in notes if isinstance(note, Note)])
            else:
                result.append([])
        return returnElementOrList(result)    
    else:
        pass # not valid signature
                
 
def transpose(args, env):    
    if len(args) > 1 and isinstance(args[0], int) and all(isinstance(arg, list) for i, arg in enumerate(args) if i != 0):
        value = args[0]
        transposed = []
        for i, arg in enumerate(args):            
            if i == 0:
                continue
            if not isinstance(arg, list):
                return # is not list            
            transposed.append([note.transpose(value) for note in arg if isinstance(note, Note)])
        return returnElementOrList(transposed)
    if len(args) > 1 and all(isinstance(arg, Note) for i, arg in enumerate(args) if i != 0):        
        value = args[0]
        transposed = [note.transpose(value) for i, note in enumerate(args) if i != 0]
        return returnElementOrList(transposed)
    else:
        return # not valid signature

def objectType(args, env):
    if len(args) == 1:
        return types[type(args[0])]
    else:
        pass # not valid signature

def exit(args, env):
    if len(args) == 1 and isinstance(args[0], int):
        sys.exit(args[0])
    else:
        pass # not valid signature

def sleep(args, env):
    if len(args) == 1 and isinstance(args[0], int):
        time.sleep(args[0])
    else:
        pass # not valid signature

def rand(args, env):
    if not all(isinstance(x, list) and len(x) == 2 and isinstance(x[0], float) for x in args):
        return # not valid signature
    if sum([x[0] for x in args]) != 1.0:
        return # not sums to 100%
    choice = random.random()
    acc = 0
    for e in args:
        acc += e[0]
        if choice <= acc:
            return e[1]

def createEnvironment():
    functions = {
        'print': doPrint,
        'synth': Synth.play,
        'pause': Synth.pause,
        'type': objectType,
        'sample': sample,
        'semitones': semitones,
        'interval': interval,
        'transpose': transpose,   
        'transposeTo': transposeTo,
        'sleep': sleep,
        'random': rand,
        'exit': exit
        
    }
    
    variables = {
        "bpm": 120
    }
    
    return Environment([ variables ], functions)