import numpy as np
import tabulate as tb
from Simplex import Simplex

X_COUNT = 2 # Задача содержит две переменные: x1 и x2.

table = np.array([[18, 1, 3],
                  [16, 2, 1],
                  [5, 0, 1],
                  [21, 3, 0],
                  [0, -2, -3]])

# Инициализация массива result нулями для хранения решения (значений переменных).
result = np.zeros(X_COUNT)
S = Simplex(table)
# Вызов метода calculate объекта S для решения задачи.
table_result = S.calculate(result)

print("Результирующая симплексная таблица:")
print(tb.tabulate(table_result))

print("\nРезультат:")
print(f"X[max] = {result}")
print(f"F[max] = {table_result[-1][0]:.2f}")