From 595e6f24628518b09fe699924b5303658405075e Mon Sep 17 00:00:00 2001
From: Felix <felixpetre@gmail.com>
Date: Tue, 14 Nov 2023 19:36:38 +0100
Subject: [PATCH] aded max_dur to add_activity

---
 matsim/writers.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/matsim/writers.py b/matsim/writers.py
index 991d00f..48a7de2 100644
--- a/matsim/writers.py
+++ b/matsim/writers.py
@@ -153,7 +153,7 @@ def end_plan(self):
         self._write_line('</plan>')
 
     def add_activity(self, type: str, x: float, y: float, facility_id: str = None,
-                     start_time: int = None, end_time: int = None, attributes: dict = None):
+                     start_time: int = None, end_time: int = None, max_dur: int = None, attributes: dict = None):
         self._require_scope(self.PLAN_SCOPE)
         self._write_indent()
         self._write('<activity ')
@@ -162,6 +162,7 @@ def add_activity(self, type: str, x: float, y: float, facility_id: str = None,
         if facility_id is not None: self._write(f'facility="{facility_id}" ')
         if start_time is not None: self._write(f'start_time="{self.time(start_time)}" ')
         if end_time is not None: self._write(f'end_time="{self.time(end_time)}" ')
+        if max_dur is not None: self._write(f'max_dur="{self.time(max_dur)}" ')
         if attributes:
             self._write('>\n')
             self.indent += 1