from gbso.program.mutate import create_random_program
from time import time

from gbso.cpu.state import CPUState
from gbso.program.test_case import TestCase
from gbso.cpu.insn import *
from gbso.cpu.regs import R8

from gbso.optimize import OptimizationParameters, cost, optimize
from gbso.program.program import Program

prgm = Program(
    insns=[
        INC_R(R8.A),
        INC_R(R8.A),
        INC_R(R8.A),
        INC_R(R8.A),
    ],
)

outputs = [R8.A]
test_cases = [TestCase(), TestCase(state=CPUState(carry=1))]

params = OptimizationParameters(max_size=4)

print("Program to optimize:")
prgm.display()

print(f"Cost: 0.0")
print(f"Cycles: {prgm.perf()}")

start_time = time()

optimized_prgm = optimize(
    prgm,
    test_cases,
    outputs,
    params,
)

end_time = time()

print("Optimized result:")
optimized_prgm.display()

optimized_cost = cost(prgm, test_cases, outputs, optimized_prgm)
optimized_cycles = optimized_prgm.perf()

print(f"Cost: {optimized_cost[0]}")
print(f"Cycles: {optimized_cycles}")

print(f"Took {round(end_time - start_time, 3)} seconds")