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

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

    alphaX = fill(1.0, rows, cols)
    alphaY = fill(1.0, rows, cols)
    alphaX[1:100, :] .= 0.5
    alphaX[101:200, :] .= 0.8
    alphaY[:, 1:200] .= 0.6
    alphaY[:, 201:400] .= 0.9
    grid::Grid = Grid{Float64}(rows, cols, alphaX, alphaY)

    concentrations = fill(0.0, rows, cols)
    concentrations[11, 11] = 1500
    concentrations[441, 661] = 750
    concentrations[441, 11] = 750
    concentrations[11, 661] = 750
    concentrations[221, 336] = 1500
    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.2)
    simulation = setIterations(simulation, 750)
    simulation = setOutputConsole(simulation, CONSOLE_OUTPUT_OFF)
    simulation = setOutputCSV(simulation, CSV_OUTPUT_ON)

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

main()