# UnitCommitment.jl: Optimization Package for Security-Constrained Unit Commitment
# Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
# Released under the modified BSD license. See COPYING.md for more details.

using UnitCommitment
using JuMP
using Gurobi
using JSON
using Logging
using Printf
using LinearAlgebra

UnitCommitment._setup_logger()

function main()
    basename, suffix = split(ARGS[1], ".")
    solution_filename = "results/$basename.$suffix.sol.json"
    model_filename = "results/$basename.$suffix.mps.gz"
    BLAS.set_num_threads(4)
    total_time = @elapsed begin
        @info "Reading: $basename"
        time_read = @elapsed begin
            instance = UnitCommitment.read_benchmark(basename)
        end
        @info @sprintf("Read problem in %.2f seconds", time_read)
        model = UnitCommitment.build_model(
            instance = instance,
            optimizer = optimizer_with_attributes(
                Gurobi.Optimizer,
                "Threads" => 4,
                "Seed" => rand(1:1000),
            ),
            variable_names = true,
        )
        @info "Optimizing..."
        BLAS.set_num_threads(1)
        UnitCommitment.optimize!(
            model,
            UnitCommitment._XaQiWaTh19(time_limit = 3600.0),
        )
    end
    @info @sprintf("Total time was %.2f seconds", total_time)
    @info "Writing: $solution_filename"
    solution = UnitCommitment.solution(model)
    open(solution_filename, "w") do file
        return JSON.print(file, solution, 2)
    end
    @info "Verifying solution..."
    UnitCommitment.validate(instance, solution)
    @info "Exporting model..."
    return JuMP.write_to_file(model, model_filename)
end

main()