include("../../TUG/src/TUG.jl")
using .TUG

function main()
    # **** GRID ****
    rows::Int = 20
    cols::Int = 20

    alphaX = fill(1.0, rows, cols)
    alphaY = fill(1.0, rows, cols)
    grid::Grid = Grid{Float64}(rows, cols, alphaX, alphaY)

    concentrations = fill(0.0, rows, cols)
    concentrations[11, 11] = 2000
    setConcentrations!(grid, concentrations)

    # **** BOUNDARY ****
    bc::Boundary = Boundary(grid)
    setBoundarySideClosed!(bc, LEFT)
    setBoundarySideClosed!(bc, RIGHT)
    setBoundarySideClosed!(bc, TOP)
    setBoundarySideClosed!(bc, BOTTOM)

    # **** SIMULATION ****
    simulation::Simulation = Simulation(grid, bc, FTCS)
    simulation = setTimestep(simulation, 0.1)
    simulation = setIterations(simulation, 500)
    simulation = setOutputConsole(simulation, CONSOLE_OUTPUT_OFF)
    simulation = setOutputCSV(simulation, CSV_OUTPUT_ON)

    # **** RUN SIMULATION ****
    print((@elapsed TUG.run(simulation)) * 1e9)
end

main()