|
| 1 | +""" |
| 2 | + read_instance(dataset::Dataset, name::String) |
| 3 | +
|
| 4 | +Read the instance identified by `name` from a given `dataset` and return a `QPSReader.QPSData` object. |
| 5 | +
|
| 6 | +# See also |
| 7 | +
|
| 8 | +- [`Dataset`](@ref) |
| 9 | +""" |
| 10 | +function read_instance(dataset::Dataset, name::String) |
| 11 | + if dataset == MIPLIB2017 || dataset == MIPLIB2017Benchmark |
| 12 | + return read_miplib2017_instance(dataset, name) |
| 13 | + elseif dataset == Netlib |
| 14 | + return read_netlib_instance(name) |
| 15 | + end |
| 16 | +end |
| 17 | + |
| 18 | +function read_miplib2017_instance(dataset::Dataset, name::String) |
| 19 | + folder = if dataset == MIPLIB2017Benchmark |
| 20 | + datadep"miplib2017-benchmark" |
| 21 | + else |
| 22 | + datadep"miplib2017-collection" |
| 23 | + end |
| 24 | + name = lowercase(name) |
| 25 | + mps_gz_path = joinpath(folder, "$name.mps.gz") |
| 26 | + return read_mps(mps_gz_path) |
| 27 | +end |
| 28 | + |
| 29 | +function read_netlib_instance(name::String) |
| 30 | + name = uppercase(name) |
| 31 | + if name in ("BLEND", "DFL001", "FORPLAN", "GFRD-PNC", "SIERRA") |
| 32 | + # https://github.com/JuliaSmoothOptimizers/QPSReader.jl/issues/58 |
| 33 | + mpsformat = :fixed |
| 34 | + else |
| 35 | + mpsformat = :free |
| 36 | + end |
| 37 | + netlib_path = fetch_netlib() |
| 38 | + sif_path = joinpath(netlib_path, "$name.SIF") |
| 39 | + return read_mps(sif_path; mpsformat) |
| 40 | +end |
| 41 | + |
| 42 | +function read_mps(path::String; mpsformat::Symbol = :free) |
| 43 | + if endswith(path, ".mps.gz") |
| 44 | + contents = GZip.open(path, "r") do f |
| 45 | + read(f, String) |
| 46 | + end |
| 47 | + mps_path = string(tempname(), ".mps") |
| 48 | + open(mps_path, "w") do f |
| 49 | + write(f, contents) |
| 50 | + end |
| 51 | + else |
| 52 | + @assert endswith(path, ".mps") || endswith(path, ".SIF") |
| 53 | + mps_path = path |
| 54 | + end |
| 55 | + |
| 56 | + qps_data = with_logger(NullLogger()) do |
| 57 | + readqps(mps_path; mpsformat) |
| 58 | + end |
| 59 | + return qps_data |
| 60 | +end |
0 commit comments