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)