handleFunctions.py

import settings as s
from constants import *

def handleInput(typeOfInput):
    # Receives a string
    if typeOfInput == typeChar:
        inp = input('>>')
        for i in inp:
            try:
                s.stack.append(ord(i))
            except:
                print("Not and ordinal number :", i, "str :",inp)

    # Receives a int
    elif typeOfInput == typeInt:
        inp = input('>>')
        try:
            s.stack.append(int(inp))
        except:
            print("Invalid integer value", inp)
            exit(1)
    s.point += 1


def handleOutput(stackPointer, typeOfOutput):
    item = s.stack[int(stackPointer)]
    if typeOfOutput == typeChar:
        print(chr(item),end='')
    elif typeOfOutput == typeInt:
        print(item,end='')
    s.point += 1


def handleStackPop(val):
    try:
        s.stack.pop(int(val))
        s.point += 1
    except IndexError:
        print('Pop from empty stack at', val)
        exit(1)


def handleStackPush(val):
    # Maybe enable any type of push?
    s.stack.append(int(val))
    s.point += 1


def handleIncrease(line): 
    add = 0
    try:
        if line[2][0] == sPointer:
            add = s.stack[int(line[2][1:])]
        else:
            add = int(line[2])
        
        s.stack[int(line[1])] = s.stack[int(line[1])] + add
        s.point += 1
    except IndexError:
        print("Invalid decrease at line", s.point)
        exit(0)


def handleDecrease(line): 
    add = 0
    try:
        if line[2][0] == sPointer:
            add = s.stack[int(line[2][1:])]
        else:
            add = int(line[2])
        
        s.stack[int(line[1])] = s.stack[int(line[1])] - add
        s.point += 1
    except IndexError:
        print("Invalid decrease at line", s.point)
        exit(0)


def handleSet(line):
    val = 0
    try:
        if line[2][0] == sPointer:
            val = s.stack[int(line[2][1:])]
        else:
            val = int(line[2])
        s.stack[int(line[1])] = val
        s.point += 1
    except IndexError:
        print("Invalid set at line", s.point)
        exit(0)


def handleIf(line, typeOfCondition):
    try:
        if line[2][0] == sPointer:
            check = s.stack[int(line[2][1:])]
        else:
            check = int(line[2])

        if typeOfCondition == cEqual:
            if s.stack[int(line[1])] == check:
                handleGoto(line[4])
            else:
                s.point += 1

        if typeOfCondition == cNotEqual:
            if s.stack[int(line[1])] != check:
                handleGoto(line[4])
            else:
                s.point += 1

        if typeOfCondition == cBigger:
            if s.stack[int(line[1])] > check:
                handleGoto(line[4])
            else:
                s.point += 1

        if typeOfCondition == cSmaller:
            if s.stack[int(line[1])] < check:
                handleGoto(line[4])
            else:
                s.point += 1
    except IndexError:
        print("Invalid if/nif/bif/sif:", line, typeOfCondition, "at line", s.point)


def handleGoto(goto):
    try:
        indexToJump = int(goto)
        s.point = indexToJump
        return
    except ValueError:
        pass
    try:
        indexToJump = s.gotoMap[goto]
        s.point = indexToJump
        return
    except KeyError:
        print("Invalid goto index or label at line", s.point)
        exit(1)


def handlePrintStack(typeOfOutput):
    for x in s.stack:
        if typeOfOutput:
            if typeOfOutput == typeInt:
                print(x, end='')
            else:
                print(chr(x), end='')
    print()
    s.point += 1



def handleStackSwitch(stack):
    try:
        s.stack = s.dictOfStacks[stack]
    except KeyError:
        s.dictOfStacks[stack] = []
        s.stack = s.dictOfStacks[stack]
    s.point += 1