save closest absolute dating with each reference

faustedition · Sep 19, 2018 · 68f14ab · 68f14ab
1 parent 5f60ee2
commit 68f14ab
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/graph.py b/graph.py
@@ -237,10 +237,16 @@ def _augment_details(self):
             ref.rank = self.closure.in_degree(ref)
             max_before_date = max((d for d, _ in self.closure.in_edges(ref) if isinstance(d, date)),
                                   default=EARLIEST - DAY)
+            max_abs_before_date = max((d for d, _ in self.dag.in_edges(ref) if isinstance(d, date)),
+                                  default=None)
             ref.earliest = max_before_date + DAY
+            ref.earliest_abs = max_abs_before_date + DAY if max_abs_before_date is not None else None
             min_after_date = min((d for _, d in self.closure.out_edges(ref) if isinstance(d, date)),
                                  default=LATEST + DAY)
+            min_abs_after_date = min((d for _, d in self.dag.out_edges(ref) if isinstance(d, date)),
+                                 default=None)
             ref.latest = min_after_date - DAY
+            ref.latest_abs = min_abs_after_date - DAY if min_abs_after_date is not None else None
 
 
 def adopt_orphans(graph: nx.MultiDiGraph):