def dest_capicity(self):
"""Generates capacities for links from transit to destination"""
for tn in range(1, self.n_transit_nodes+1):
for dn in range(1, self.n_destination_nodes+1):
capacity = Capacity(self.n_source_nodes, False)
capacity.create_constraint(tn, dn)
self.add_line(capacity)
def source_capicity(self):
"""Generates capacities for links from source to transit"""
for sn in range(1, self.n_source_nodes+1):
for tn in range(1, self.n_transit_nodes+1):
capacity = Capacity(self.n_destination_nodes, True)
capacity.create_constraint(sn, tn)
self.add_line(capacity)
