add: trip modification intervention

Author: vijay-jaisankar

README.md

@@ -20,7 +20,6 @@ ce = trafficinterventions.ChangeEdges.ChangeEdges(fileName="sample.xml")
 
 # Sample Intervention
 ce.disallowAppendTypes(["bus"], ["-100"], "new_file.xml")
-
 ```
 ---
 
@@ -32,8 +31,21 @@ cl = trafficinterventions.ChangeLanes.ChangeLanes(fileName="sample.xml")
 
 # Sample Intervention
 ce.changePriorityLanes(["highway.cycleway"], 100, "new_file.xml")
+```
+---
+
+### Sample Usage : Trip Manipulation
+```py
+import trafficinterventions
 
+ct = trafficinterventions.ChangeTrips.ChangeTrips(fileName="sample.xml")
+
+# Sample Intervention
+ct.changeTripStartTime([3], 1.00, "new_file.xml")
 ```
+---
+
+
 ## Simulations
 
 ### Sample Usage: Speed Camera Placement

dist/trafficinterventions-2.0.1-py3-none-any.whl

@@ -0,0 +1,31 @@
+import trafficinterventions
+
+"""
+Note: If you wish to not rewrite the XML file, please add another parameter to the functions.
+"""
+
+"""
+Change Trips Functionality
+"""
+
+# Instantiating an object of the XML Parser for Trip Modification
+fileName = "sample3.xml" # Change the path of your file accordingly.
+ct = trafficinterventions.ChangeTrips.ChangeTrips(fileName=fileName)
+
+# Get Root Tag
+print(ct.getRootElementTag())
+
+# Get Unique Parent Tags
+print(ct.getUniqueParentTags())
+
+# Get all Trip IDs
+print(ct.getLaneTypes())
+
+# Get all Trip Information
+print(ct.getTripInformation())
+
+# Changing the start time of trip 3 to 1.00
+ct.changeTripStartTime(
+    [3],
+    1.00
+)

dist/trafficinterventions-2.0.1.tar.gz

@@ -1,6 +1,6 @@
 [metadata]
 name = trafficinterventions
-version = 2.0.0
+version = 2.0.1
 author = WSL, IIITB
 author_email = WSL1@iiitb.ac.in
 description = Python Package to perform simple Traffic Interventions and run traffic simulations.

docs/change_trips.py

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: trafficinterventions
-Version: 2.0.0
+Version: 2.0.1
 Summary: Python Package to perform simple Traffic Interventions and run traffic simulations.
 Home-page: https://github.com/WSL-IIITB/Traffic-Interventions
 Author: WSL, IIITB
@@ -37,7 +37,6 @@ ce = trafficinterventions.ChangeEdges.ChangeEdges(fileName="sample.xml")
 
 # Sample Intervention
 ce.disallowAppendTypes(["bus"], ["-100"], "new_file.xml")
-
 ```
 ---
 
@@ -49,8 +48,21 @@ cl = trafficinterventions.ChangeLanes.ChangeLanes(fileName="sample.xml")
 
 # Sample Intervention
 ce.changePriorityLanes(["highway.cycleway"], 100, "new_file.xml")
+```
+---
+
+### Sample Usage : Trip Manipulation
+```py
+import trafficinterventions
 
+ct = trafficinterventions.ChangeTrips.ChangeTrips(fileName="sample.xml")
+
+# Sample Intervention
+ct.changeTripStartTime([3], 1.00, "new_file.xml")
 ```
+---
+
+
 ## Simulations
 
 ### Sample Usage: Speed Camera Placement

setup.cfg

@@ -4,6 +4,7 @@ pyproject.toml
 setup.cfg
 src/trafficinterventions/ChangeEdges.py
 src/trafficinterventions/ChangeLanes.py
+src/trafficinterventions/ChangeTrips.py
 src/trafficinterventions/EmissionJunction.py
 src/trafficinterventions/SpeedCamera.py
 src/trafficinterventions/StressJunction.py

src/trafficinterventions.egg-info/PKG-INFO

@@ -0,0 +1,68 @@
+import xml.etree.ElementTree as ET
+
+"""
+    @Class : To perform trip interventions
+"""
+class ChangeTrips():
+    def __init__(self, fileName):
+        self._fileName = fileName
+        self._tree = ET.parse(self._fileName)
+        self._root = self._tree.getroot()
+
+    """
+        Get Head of XML Tree
+    """
+    def getRootElementTag(self):
+        return self._root.tag
+
+    """
+        Get different attribute types
+    """
+    def getUniqueParentTags(self):
+        d = {}
+        for child in self._root:
+            if (child.tag) not in d:
+                d[child.tag] = 1
+            else:
+                d[child.tag] += 1
+
+        return d.keys()
+
+    """
+        Get all Trip IDs
+    """
+    def getLaneTypes(self):
+        l = []
+        for child in self._root:
+            if child.tag == "trip":
+                l.append(child.attrib["id"])
+
+        return l
+
+    """
+        Change the start time of a trip
+    """
+    def changeTripStartTime(self, tripIDList, newStartTime, newFileName = None):
+        for child in self._root:
+            if child.tag == "trip" and newStartTime >= 0:
+                attributesDict = child.attrib
+                if int(attributesDict["id"]) in tripIDList:
+                    child.set("depart", str(newStartTime))
+
+        if newFileName is None:
+            self._tree.write(self._fileName)
+        else:
+            self._tree.write(str(newFileName))
+
+    """
+        Get Trip information
+    """
+    def getTripInformation(self):
+        l = []
+        for child in self._root:
+            if child.tag == "trip":
+                l.append(
+                    (child.attrib["id"],
+                    child.attrib)
+                )
+        return l

src/trafficinterventions.egg-info/SOURCES.txt

@@ -1,5 +1,6 @@
 from . import ChangeEdges
 from . import ChangeLanes
+from . import ChangeTrips
 from . import SpeedCamera
 from . import StressJunction
 from . import EmissionJunction