Add getter and setter for disallowedNextLinks

matsim-org · Oct 13, 2023 · b80a197 · b80a197
1 parent 57598c8
commit b80a197
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/matsim/src/main/java/org/matsim/core/network/NetworkUtils.java b/matsim/src/main/java/org/matsim/core/network/NetworkUtils.java
@@ -22,6 +22,8 @@
 
 import java.util.*;
 
+import javax.annotation.Nullable;
+
 import org.apache.commons.lang3.StringUtils;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
@@ -988,4 +990,29 @@ public static List<Node> getOriginalGeometry(Link link) {
 		result.add(link.getToNode());
 		return result;
 	}
+
+	private static final String DISALLOWED_NEXT_LINKS_ATTRIBUTE = "disallowedNextLinks";
+
+	@Nullable
+	public static DisallowedNextLinks getDisallowedNextLinks(Link link) {
+		return (DisallowedNextLinks) link.getAttributes().getAttribute(DISALLOWED_NEXT_LINKS_ATTRIBUTE);
+	}
+
+	public static DisallowedNextLinks getOrCreateDisallowedNextLinks(Link link) {
+		DisallowedNextLinks disallowedNextLinks = getDisallowedNextLinks(link);
+		if (disallowedNextLinks == null) {
+			disallowedNextLinks = new DisallowedNextLinks();
+			setDisallowedNextLinks(link, disallowedNextLinks);
+		}
+		return disallowedNextLinks;
+	}
+
+	public static void setDisallowedNextLinks(Link link, DisallowedNextLinks disallowedNextLinks) {
+		link.getAttributes().putAttribute(DISALLOWED_NEXT_LINKS_ATTRIBUTE, disallowedNextLinks);
+	}
+
+	public static boolean addDisallowedNextLinks(Link link, String mode, Collection<Id<Link>> linkIds) {
+		DisallowedNextLinks disallowedNextLinks = getOrCreateDisallowedNextLinks(link);
+		return disallowedNextLinks.addDisallowedLinkSequence(mode, linkIds);
+	}
 }