LABA4

class Stack(object):
    def __init__(self):
        self.items = []

    def isEmpty(self):
        return self.items == []

    def push(self, item):
        self.items.append(item)

    def pop(self):
        return self.items.pop()

    def peek(self):
        return self.items[len(self.items)-1]

    def size(self):
        return len(self.items)

    def __str__(self):
        out = self.items[:]
        out.reverse()
        return str(out)


class MyComplex(object):
    def __init__(self, k1, k2):
        self.complex1 = complex(*[int(i) for i in k1.split(" ")])
        self.complex2 = complex(*[int(i) for i in k2.split(" ")])
        print(self.complex1)
        print(self.complex2)

    def add(self):
        return self.complex1 + self.complex2

    def mul(self):
        return self.complex1 * self.complex2

    def sub(self):
        return self.complex1 - self.complex2


print("Лабораторна робота 4")
print("Завдання 15")
initStack = input("Ініціалізація стеку (введіть за бажанням кілька значень): ")
s = Stack()
for i in initStack.split(" "):
    s.push(i)
print(s)
newItem = input("Введіть елемент який треба додати в стек: ")
s.push(newItem)
print(s)
print("Стек після операції pop")
s.pop()
print(s)

print("\nЗавдання 22")
ans=True
while ans:
    print("""
    1.Додати комплексні числа
    2.Відняти комплексні числа
    3.Множити комплексні числа
    4.Вихід з програми
    """)
    ans = input("Що ви хочете зробити? (введіть число яке вказує пункт меню) ")
    if ans == "1":
        comp1 = input(" Введіть 1 комплексне число (дійсна умовна частина): ")
        comp2 = input(" Введіть 2 комплексне число (дійсна умовна частина): ")
        c = MyComplex(comp1, comp2)
        result = c.add()
        print("Результат додавання : {}".format(result))

    elif ans == "2":
        comp1 = input(" Введіть 1 комплексне число (дійсна умовна частина): ")
        comp2 = input(" Введіть 2 комплексне число (дійсна умовна частина): ")
        c = MyComplex(comp1, comp2)
        result = c.sub()
        print("Результат віднімання : {}".format(result))

    elif ans == "3":
        comp1 = input(" Введіть 1 комплексне число (дійсна умовна частина): ")
        comp2 = input(" Введіть 2 комплексне число (дійсна умовна частина): ")
        c = MyComplex(comp1, comp2)
        result = c.mul()
        print("Результат множення : {}".format(result))

    elif ans == "4":
        print("\n До побачення")
        ans = None
    else:
        print("\n Введене число не належить до списку визначених в програмі, спробуйте ще раз ")