# HG changeset patch
# User Nicolas Saunier <nicolas.saunier@polymtl.ca>
# Date 1769463206 18000
# Node ID 9c1870244adf2212d53dbb829cdc5a52d5f299b5
# Parent  41469fdaa85e7ad73668a737569b341c5c3d1121# Parent  d828f3f76273423ff816d2a96f1416e7a28f1cd8
merged

diff -r 41469fdaa85e -r 9c1870244adf trafficintelligence/traffic_engineering.py
--- a/trafficintelligence/traffic_engineering.py	Mon Jan 26 16:33:13 2026 -0500
+++ b/trafficintelligence/traffic_engineering.py	Mon Jan 26 16:33:26 2026 -0500
@@ -261,18 +261,23 @@
 class TriangularFD(FundamentalDiagram):
     def __init__(self, vf = None, kc = None, kj = None, qmax = None, w = None):
         FundamentalDiagram.__init__(self,'Triangular')
-        if vf is not None and qmax is not None and kj is not None:
+        if vf is not None and kj is not None:
             self.vf=vf
-            self.qmax = qmax
             self.kj = kj
-            self.kc = qmax/vf
-            self.w = qmax/(self.kc-kj)
+            if qmax is not None:
+                self.qmax = qmax
+                self.kc = qmax/vf
+                self.w = qmax/(self.kc-kj)
+            elif kc is not None:
+                self.kc = kc
+                self.qmax = vf*kc
+                self.w = self.qmax/(self.kc-kj)
 
     def v(self, k):
         if k<self.kc:
             return self.vf
         else:
-            return self.vf*self.kc*(self.kj/k-1)/(self.kj-self.kc)
+            return self.qmax/(self.kj-self.kc)*(1-self.kj/k)
 
 def generateDensities(n, maxDensity):
     return stats.uniform.rvs(size=n)*maxDensity