Changeset 137:6364b07f8cd4 in lemon-0.x for src/work

Timestamp:

03/01/04 12:33:10 (19 years ago)

Author:

marci

Branch:

default

Phase:

public

Convert:

svn:c9d7d8f5-90d6-0310-b91f-818b3a526b0e/lemon/trunk@191

Message:

ResGraph3::EdgeMap?<T>

File:

• src/work/edmonds_karp.hh (modified) (1 diff)

src/work/edmonds_karp.hh

@@ -453 +453 @@
     int id(NodeIt v) const { return G.id(v); }
 
-    template <typename S>
+    template <typename T>
     class NodeMap {
-      typename Graph::NodeMap<S> node_map; 
+      typename Graph::NodeMap<T> node_map; 
     public:
       NodeMap(const ResGraph3<Graph, Number, FlowMap, CapacityMap>& _G) : node_map(_G.G) { }
-      NodeMap(const ResGraph3<Graph, Number, FlowMap, CapacityMap>& _G, S a) : node_map(_G.G, a) { }
-      void set(NodeIt nit, S a) { node_map.set(nit, a); }
-      S get(NodeIt nit) const { return node_map.get(nit); }
+      NodeMap(const ResGraph3<Graph, Number, FlowMap, CapacityMap>& _G, T a) : node_map(_G.G, a) { }
+      void set(NodeIt nit, T a) { node_map.set(nit, a); }
+      T get(NodeIt nit) const { return node_map.get(nit); }
+    };
+
+    template <typename T>
+    class EdgeMap {
+      typename Graph::EdgeMap<T> forward_map, backward_map; 
+    public:
+      EdgeMap(const ResGraph3<Graph, Number, FlowMap, CapacityMap>& _G) : forward_map(_G.G), backward_map(_G.G) { }
+      EdgeMap(const ResGraph3<Graph, Number, FlowMap, CapacityMap>& _G, T a) : forward_map(_G.G, a), backward_map(_G.G, a) { }
+      void set(EdgeIt e, T a) { 
+        if (e.out_or_in) 
+          forward_map.set(e.out, a); 
+        else 
+          backward_map.set(e.in, a); 
+      }
+      T get(EdgeIt e) { 
+        if (e.out_or_in) 
+          return forward_map.get(e.out); 
+        else 
+          return backward_map.get(e.in); 
+      }
     };
 
   };
-
 
   template <typename Graph, typename Number, typename FlowMap, typename CapacityMap>