RhodeCode

/* -*- C++ -*-
 *
 * This file is a part of LEMON, a generic C++ optimization library
 *
 * Copyright (C) 2003-2008
 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
 * (Egervary Research Group on Combinatorial Optimization, EGRES).
 *
 * Permission to use, modify and distribute this software is granted
 * provided that this copyright notice appears in all copies. For
 * precise terms see the accompanying LICENSE file.
 *
 * This software is provided "AS IS" with no warranty of any kind,
 * express or implied, and with no claim as to its suitability for any
 * purpose.
 *
 */

#ifndef LEMON_TEST_GRAPH_TEST_H
#define LEMON_TEST_GRAPH_TEST_H

#include <lemon/graph_utils.h>
#include "test_tools.h"

namespace lemon {

  template<class Graph>
  void checkGraphNodeList(const Graph &G, int cnt)
  {
    typename Graph::NodeIt n(G);
    for(int i=0;i<cnt;i++) {
      check(n!=INVALID,"Wrong Node list linking.");
      ++n;
    }
    check(n==INVALID,"Wrong Node list linking.");
    check(countNodes(G)==cnt,"Wrong Node number.");
  }

  template<class Graph>
  void checkGraphArcList(const Graph &G, int cnt)
  {
    typename Graph::ArcIt e(G);
    for(int i=0;i<cnt;i++) {
      check(e!=INVALID,"Wrong Arc list linking.");
      ++e;
    }
    check(e==INVALID,"Wrong Arc list linking.");
    check(countArcs(G)==cnt,"Wrong Arc number.");
  }

  template<class Graph>
  void checkGraphOutArcList(const Graph &G, typename Graph::Node n, int cnt)
  {
    typename Graph::OutArcIt e(G,n);
    for(int i=0;i<cnt;i++) {
      check(e!=INVALID,"Wrong OutArc list linking.");
      check(n==G.source(e),"Wrong OutArc list linking.");
      ++e;
    }
    check(e==INVALID,"Wrong OutArc list linking.");
    check(countOutArcs(G,n)==cnt,"Wrong OutArc number.");
  }

  template<class Graph>
  void checkGraphInArcList(const Graph &G, typename Graph::Node n, int cnt)
  {
    typename Graph::InArcIt e(G,n);
    for(int i=0;i<cnt;i++) {
      check(e!=INVALID,"Wrong InArc list linking.");
      check(n==G.target(e),"Wrong InArc list linking.");
      ++e;
    }
    check(e==INVALID,"Wrong InArc list linking.");
    check(countInArcs(G,n)==cnt,"Wrong InArc number.");
  }

  template<class Graph>
  void checkGraphEdgeList(const Graph &G, int cnt)
  {
    typename Graph::EdgeIt e(G);
    for(int i=0;i<cnt;i++) {
      check(e!=INVALID,"Wrong Edge list linking.");
      ++e;
    }
    check(e==INVALID,"Wrong Edge list linking.");
    check(countEdges(G)==cnt,"Wrong Edge number.");
  }

  template<class Graph>
  void checkGraphIncEdgeList(const Graph &G, typename Graph::Node n, int cnt)
  {
    typename Graph::IncEdgeIt e(G,n);
    for(int i=0;i<cnt;i++) {
      check(e!=INVALID,"Wrong IncEdge list linking.");
      check(n==G.u(e) || n==G.v(e),"Wrong IncEdge list linking.");
      ++e;
    }
    check(e==INVALID,"Wrong IncEdge list linking.");
    check(countIncEdges(G,n)==cnt,"Wrong IncEdge number.");
  }

  template <class Digraph>
  void checkDigraphIterators() {
    typedef typename Digraph::Node Node;
    typedef typename Digraph::NodeIt NodeIt;
    typedef typename Digraph::Arc Arc;
    typedef typename Digraph::ArcIt ArcIt;
    typedef typename Digraph::InArcIt InArcIt;
    typedef typename Digraph::OutArcIt OutArcIt;
  }

  template <class Graph>
  void checkGraphIterators() {
    checkDigraphIterators<Graph>();
    typedef typename Graph::Edge Edge;
    typedef typename Graph::EdgeIt EdgeIt;
    typedef typename Graph::IncEdgeIt IncEdgeIt;
  }

  // Structure returned by addPetersen()
  template<class Digraph>
  struct PetStruct
  {
    // Vector containing the outer nodes
    std::vector<typename Digraph::Node> outer;
    // Vector containing the inner nodes
    std::vector<typename Digraph::Node> inner;
    // Vector containing the arcs of the inner circle
    std::vector<typename Digraph::Arc> incir;
    // Vector containing the arcs of the outer circle
    std::vector<typename Digraph::Arc> outcir;
    // Vector containing the chord arcs
    std::vector<typename Digraph::Arc> chords;
  };

  // Adds the reverse pair of all arcs to a digraph
  template<class Digraph>
  void bidirDigraph(Digraph &G)
  {
    typedef typename Digraph::Arc Arc;
    typedef typename Digraph::ArcIt ArcIt;

    std::vector<Arc> ee;

    for(ArcIt e(G);e!=INVALID;++e) ee.push_back(e);

    for(int i=0;i<int(ee.size());++i)
      G.addArc(G.target(ee[i]),G.source(ee[i]));
  }

  // Adds a Petersen digraph to G.
  // Returns the nodes and arcs of the generated digraph.
  template<typename Digraph>
  PetStruct<Digraph> addPetersen(Digraph &G,int num = 5)
  {
    PetStruct<Digraph> n;

    for(int i=0;i<num;i++) {
      n.outer.push_back(G.addNode());
      n.inner.push_back(G.addNode());
    }

    for(int i=0;i<num;i++) {
      n.chords.push_back(G.addArc(n.outer[i],n.inner[i]));
      n.outcir.push_back(G.addArc(n.outer[i],n.outer[(i+1) % num]));
      n.incir.push_back(G.addArc(n.inner[i],n.inner[(i+2) % num]));
    }

    return n;
  }

  // Checks the bidirectioned Petersen digraph
  template<class Digraph>
  void checkBidirPetersen(const Digraph &G, int num = 5)
  {
    typedef typename Digraph::NodeIt NodeIt;

    checkGraphNodeList(G, 2 * num);
    checkGraphArcList(G, 6 * num);

    for(NodeIt n(G);n!=INVALID;++n) {
      checkGraphInArcList(G, n, 3);
      checkGraphOutArcList(G, n, 3);
    }
  }

  // Structure returned by addUPetersen()
  template<class Graph>
  struct UPetStruct
  {
    // Vector containing the outer nodes
    std::vector<typename Graph::Node> outer;
    // Vector containing the inner nodes
    std::vector<typename Graph::Node> inner;
    // Vector containing the edges of the inner circle
    std::vector<typename Graph::Edge> incir;
    // Vector containing the edges of the outer circle
    std::vector<typename Graph::Edge> outcir;
    // Vector containing the chord edges
    std::vector<typename Graph::Edge> chords;
  };

  // Adds a Petersen graph to \c G.
  // Returns the nodes and edges of the generated graph.
  template<typename Graph>
  UPetStruct<Graph> addUPetersen(Graph &G,int num=5)
  {
    UPetStruct<Graph> n;

    for(int i=0;i<num;i++) {
      n.outer.push_back(G.addNode());
      n.inner.push_back(G.addNode());
    }

    for(int i=0;i<num;i++) {
      n.chords.push_back(G.addEdge(n.outer[i],n.inner[i]));
      n.outcir.push_back(G.addEdge(n.outer[i],n.outer[(i+1)%num]));
      n.incir.push_back(G.addEdge(n.inner[i],n.inner[(i+2)%num]));
    }
    
    return n;
  }

  // Checks the undirected Petersen graph
  template<class Graph>
  void checkUndirPetersen(const Graph &G, int num = 5)
  {
    typedef typename Graph::NodeIt NodeIt;

    checkGraphNodeList(G, 2 * num);
    checkGraphEdgeList(G, 3 * num);
    checkGraphArcList(G, 6 * num);

    for(NodeIt n(G);n!=INVALID;++n) {
      checkGraphIncEdgeList(G, n, 3);
    }
  }

  template <class Digraph>
  void checkDigraph() {
    const int num = 5;
    Digraph G;
    checkGraphNodeList(G, 0);
    checkGraphArcList(G, 0);
    addPetersen(G, num);
    bidirDigraph(G);
    checkBidirPetersen(G, num);
  }
  
  template <class Graph>
  void checkGraph() {
    const int num = 5;
    Graph G;
    checkGraphNodeList(G, 0);
    checkGraphEdgeList(G, 0);
    addUPetersen(G, num);
    checkUndirPetersen(G, num);
  }

} //namespace lemon

#endif