Merge back the whole branches/hugo++ to trunk.
1 #ifndef HUGO_PREFLOW_PUSH_HH
2 #define HUGO_PREFLOW_PUSH_HH
8 //#include "pf_hiba.hh"
9 //#include <marci_list_graph.hh>
10 //#include <marci_graph_traits.hh>
12 #include <graph_wrapper.h>
13 //#include <reverse_bfs.hh>
19 template <typename Graph, typename T>
23 typedef typename Graph::Node Node;
24 typedef typename Graph::NodeIt NodeIt;
25 typedef typename Graph::Edge Edge;
26 typedef typename Graph::OutEdgeIt OutEdgeIt;
27 typedef typename Graph::InEdgeIt InEdgeIt;
28 typedef typename Graph::EdgeMap<T> CapacityType;
30 typedef ResGraphWrapper<const Graph,int,CapacityType,CapacityType> ResGraphType;
33 //---------------------------------------------
34 //Parameters of the algorithm
35 //---------------------------------------------
36 //Fully examine an active node until excess becomes 0
37 enum node_examination_t {examine_full, examine_to_relabel};
38 //No more implemented yet:, examine_only_one_edge};
39 node_examination_t node_examination;
40 //Which implementation to be used
41 enum implementation_t {impl_fifo, impl_highest_label};
42 //No more implemented yet:};
43 implementation_t implementation;
44 //---------------------------------------------
45 //Parameters of the algorithm
46 //---------------------------------------------
53 CapacityType &capacity;
59 //auxiliary variables for computation
60 //The number of the nodes
62 //A nodemap for the level
63 typename Graph::NodeMap<int> level;
64 //A nodemap for the excess
65 typename Graph::NodeMap<T> excess;
67 //Number of nodes on each level
68 vector<int> num_of_nodes_on_level;
70 //For the FIFO implementation
71 list<Node> fifo_nodes;
72 //For 'highest label' implementation
74 //int second_highest_active;
75 vector< list<Node> > active_nodes;
79 //Constructing the object using the graph, source, sink and capacity vector
84 typename Graph::EdgeMap<T> & _capacity)
85 : G(_G), s(_s), t(_t),
88 //Counting the number of nodes
89 //number_of_nodes(count(G.first<EachNodeIt>())),
90 number_of_nodes(G.nodeNum()),
94 // Default constructor: active_nodes()
96 //Simplest parameter settings
97 node_examination = examine_full;//examine_to_relabel;//
98 //Which implementation to be usedexamine_full
99 implementation = impl_highest_label;//impl_fifo;
102 num_of_nodes_on_level.resize(2*number_of_nodes-1);
103 num_of_nodes_on_level.clear();
105 switch(implementation){
106 case impl_highest_label :{
107 active_nodes.clear();
108 active_nodes.resize(2*number_of_nodes-1);
118 //Returns the value of a maximal flow
121 typename Graph::EdgeMap<T> getmaxflow(){
127 //For testing purposes only
128 //Lists the node_properties
129 void write_property_vector(typename Graph::NodeMap<T> a,
130 //node_property_vector<Graph, T> a,
131 char* prop_name="property"){
132 for(NodeIt i=G.template first<NodeIt>(); G.valid(i); G.next(i)) {
133 cout<<"Node id.: "<<G.id(i)<<", "<<prop_name<<" value: "<<a[i]<<endl;
138 //Modifies the excess of the node and makes sufficient changes
139 void modify_excess(const Node& a ,T v){
140 //T old_value=excess[a];
144 //This private procedure is supposed to modify the preflow on edge j
145 //by value v (which can be positive or negative as well)
146 //and maintain the excess on the head and tail
147 //Here we do not check whether this is possible or not
148 void modify_preflow(Edge j, const T& v){
155 modify_excess(G.head(j),v);
158 modify_excess(G.tail(j),-v);
162 //Gives the active node to work with
163 //(depending on the implementation to be used)
164 Node get_active_node(){
167 switch(implementation) {
168 case impl_highest_label : {
170 //First need to find the highest label for which there's an active node
171 while( highest_active>=0 && active_nodes[highest_active].empty() ){
175 if( highest_active>=0) {
178 Node a=active_nodes[highest_active].front();
179 active_nodes[highest_active].pop_front();
192 if( ! fifo_nodes.empty() ) {
193 Node a=fifo_nodes.front();
194 fifo_nodes.pop_front();
207 //Puts node 'a' among the active nodes
208 void make_active(const Node& a){
209 //s and t never become active
211 switch(implementation){
212 case impl_highest_label :
213 active_nodes[level[a]].push_back(a);
216 fifo_nodes.push_back(a);
222 //Update highest_active label
223 if (highest_active<level[a]){
224 highest_active=level[a];
229 //Changes the level of node a and make sufficent changes
230 void change_level_to(Node a, int new_value){
231 int seged = level[a];
232 level.set(a,new_value);
233 --num_of_nodes_on_level[seged];
234 ++num_of_nodes_on_level[new_value];
237 //Collection of things useful (or necessary) to do before running
241 //---------------------------------------
242 //Initialize parameters
243 //---------------------------------------
245 //Setting starting preflow, level and excess values to zero
246 //This can be important, if the algorithm is run more then once
247 for(NodeIt i=G.template first<NodeIt>(); G.valid(i); G.next(i)) {
250 for(OutEdgeIt j=G.template first<OutEdgeIt>(i); G.valid(j); G.next(j))
253 num_of_nodes_on_level[0]=number_of_nodes;
255 //---------------------------------------
256 //Initialize parameters
257 //---------------------------------------
260 //------------------------------------
261 //This is the only part that uses BFS
262 //------------------------------------
264 /*Reverse_bfs from t, to find the starting level.*/
266 change_level_to(t,0);
268 std::queue<Node> bfs_queue;
271 while (!bfs_queue.empty()) {
273 Node v=bfs_queue.front();
278 for(G.first(e,v); G.valid(e); G.next(e)) {
280 if ( level[w] == number_of_nodes && w != s ) {
282 //Node first=level_list[l];
283 //if ( G.valid(first) ) left.set(first,w);
284 //right.set(w,first);
286 change_level_to(w, l);
291 change_level_to(s,number_of_nodes);
292 //level.set(s,number_of_nodes);
295 //Setting starting level values using reverse bfs
296 reverse_bfs<Graph> rev_bfs(G,t);
298 //write_property_vector(rev_bfs.dist,"rev_bfs");
299 for(NodeIt i=G.template first<NodeIt>(); G.valid(i); G.next(i)) {
300 change_level_to(i,rev_bfs.dist(i));
301 //level.put(i,rev_bfs.dist.get(i));
304 //------------------------------------
305 //This is the only part that uses BFS
306 //------------------------------------
309 //Starting level of s
310 change_level_to(s,number_of_nodes);
311 //level.put(s,number_of_nodes);
314 //we push as much preflow from s as possible to start with
315 for(OutEdgeIt j=G.template first<OutEdgeIt>(s); G.valid(j); G.next(j)){
316 modify_preflow(j,capacity[j] );
317 make_active(G.head(j));
318 int lev=level[G.head(j)];
319 if(highest_active<lev){
323 //cout<<highest_active<<endl;
327 //If the preflow is less than the capacity on the given edge
328 //then it is an edge in the residual graph
329 bool is_admissible_forward_edge(Edge j, int& new_level){
331 if (capacity[j]>preflow[j]){
332 if(level[G.tail(j)]==level[G.head(j)]+1){
336 if (level[G.head(j)] < new_level)
337 new_level=level[G.head(j)];
343 //If the preflow is greater than 0 on the given edge
344 //then the edge reversd is an edge in the residual graph
345 bool is_admissible_backward_edge(Edge j, int& new_level){
348 if(level[G.tail(j)]==level[G.head(j)]-1){
353 if (level[G.tail(j)] < new_level)
354 new_level=level[G.tail(j)];
362 }; //class preflow_push
364 template<typename Graph, typename T>
365 T preflow_push<Graph, T>::run() {
367 //We need a residual graph
368 ResGraphType res_graph(G, preflow, capacity);
371 //write_property_vector(level,"level");
374 while (a=get_active_node(), G.valid(a)){
376 bool go_to_next_node=false;
378 while (!go_to_next_node){
380 //Initial value for the new level for the active node we are dealing with
381 int new_level=2*number_of_nodes;
384 //Out edges from node a
386 ResGraphType::OutEdgeIt j=res_graph.first(j,a);
387 while (res_graph.valid(j) && e){
388 if (is_admissible_forward_edge(j,new_level)){
389 v=min(e,res_graph.resCap(j));
391 //New node might become active
392 if (excess[res_graph.head(j)]==0){
393 make_active(res_graph.head(j));
395 res_graph.augment(j,v);
396 excess[res_graph.tail(j)] -= v;
397 excess[res_graph.head(j)] += v;
404 //Out edges from node a
406 OutEdgeIt j=G.template first<OutEdgeIt>(a);
407 while (G.valid(j) && e){
409 if (is_admissible_forward_edge(j,new_level)){
410 v=min(e,capacity[j] - preflow[j]);
412 //New node might become active
413 if (excess[G.head(j)]==0){
414 make_active(G.head(j));
423 InEdgeIt j=G.template first<InEdgeIt>(a);
424 while (G.valid(j) && e){
425 if (is_admissible_backward_edge(j,new_level)){
428 //New node might become active
429 if (excess[G.tail(j)]==0){
430 make_active(G.tail(j));
432 modify_preflow(j,-v);
440 //cout<<new_level<<" e: "<<e<<endl;
441 //cout<<G.id(a)<<" "<<new_level<<endl;
445 go_to_next_node=true;
447 else{//If there is still excess in node a
449 //change_level_to(a,new_level+1);
451 //Level remains empty
452 if (num_of_nodes_on_level[level[a]]==1){
453 change_level_to(a,number_of_nodes);
454 //go_to_next_node=True;
457 change_level_to(a,new_level+1);
464 switch(node_examination){
465 case examine_to_relabel:
468 go_to_next_node = true;
479 maxflow_value = excess[t];
480 return maxflow_value;
486 #endif //PREFLOW_PUSH_HH