apis/examples_python/transport13_8py_source.html

import sys

from gams import GamsWorkspace

from transport_class import Transport


plants = ["Seattle", "San-Diego"]

markets = ["New-York", "Chicago", "Topeka"]

capacity = {"Seattle": 350.0, "San-Diego": 600.0}

demand = {"New-York": 325.0, "Chicago": 300.0, "Topeka": 275.0}

distance = {

    ("Seattle", "New-York"): 2.5,

    ("Seattle", "Chicago"): 1.7,

    ("Seattle", "Topeka"): 1.8,

    ("San-Diego", "New-York"): 2.5,

    ("San-Diego", "Chicago"): 1.8,

    ("San-Diego", "Topeka"): 1.4,

}


if __name__ == "__main__":

    sys_dir = sys.argv[1] if len(sys.argv) > 1 else None

    work_dir = sys.argv[2] if len(sys.argv) > 2 else None

    t = Transport(sys_dir, work_dir)


    for p in plants:

        t.i.add_record(p)

    for m in markets:

        t.j.add_record(m)

    for p in plants:

        t.a.add_record(p).value = capacity[p]

    for m in markets:

        t.b.add_record(m).value = demand[m]

    for k, v in iter(distance.items()):

        t.d.add_record(k).value = v


    t.f.add_record().value = 90

    t.opt.all_model_types = "cplex"

    t.run(output=sys.stdout)


    print(f"Objective: {t.z.first_record().level}")


    for rec in t.x:

        print(rec)