- Solved the proliferation of type checks for different DAI_ENUM's in properties.cpp
[libdai.git] / include / dai / bipgraph.h
index d6e720e..b385476 100644 (file)
@@ -36,12 +36,13 @@ namespace dai {
 /// A BipartiteGraph represents the neighborhood structure of nodes in a bipartite graph.
 /** A bipartite graph has two types of nodes: type 1 and type 2. Edges can occur only between
  *  nodes of different type. Nodes are indexed by an unsigned integer, edges are indexed as
- *  a pair of unsigned integers (where the pair (a,b) means the b'th neighbor of the a'th node).
- *  The BipartiteGraph stores neighborhood structures as vectors of vectors of Neighbor entries.
+ *  a pair of unsigned integers, where the pair (a,b) means the b'th neighbor of the a'th node.
+ *  The BipartiteGraph stores neighborhood structures as vectors of vectors of Neighbor entries:
+ *  each node has a vector of Neighbor entries, which is also known as a Neighbors type.
  */
 class BipartiteGraph {
     public:
-        /// A Neighbor describes a neighboring node of some other node.
+        /// A Neighbor describes a neighboring node of some other node in a BipartiteGraph.
         /** Iterating over all neighbors of node n1 of type 1 can be done in the following way:
          *  \code
          *      foreach( const BipartiteGraph::Neighbor &n2, nb1(n1) ) {
@@ -60,7 +61,7 @@ class BipartiteGraph {
             unsigned node;
             /// Contains the "dual" iter
             unsigned dual;
-            /// Cast to unsigned returns node
+            /// Cast to unsigned returns node member
             operator unsigned () const { return node; }
             /// Default constructor
             Neighbor() {}
@@ -71,7 +72,7 @@ class BipartiteGraph {
         /// Neighbors is a vector of Neighbor entries; each node has an associated Neighbors variable, which describes its neighbors.
         typedef std::vector<Neighbor> Neighbors;
 
-        /// Edge is used as index of an edge: an Edge(a,b) corresponds to the edge between the a'th node and its b'th neighbor.
+        /// Edge is used as index of an edge: an Edge(a,b) corresponds to the edge between the a'th node and its b'th neighbor. It depends on the context whether the first node (with index a) is of type 1 or of type 2.
         typedef std::pair<size_t,size_t> Edge;
 
     private:
@@ -119,31 +120,75 @@ class BipartiteGraph {
         }
 
         /// Returns constant reference to the _i2'th neighbor of node i1 of type 1
-        const Neighbor & nb1( size_t i1, size_t _i2 ) const { return _nb1[i1][_i2]; }
+        const Neighbor & nb1( size_t i1, size_t _i2 ) const { 
+#ifdef DAI_DEBUG
+            assert( i1 < _nb1.size() );
+            assert( _i2 < _nb1[i1].size() );
+#endif
+            return _nb1[i1][_i2]; 
+        }
         /// Returns reference to the _i2'th neighbor of node i1 of type 1
-        Neighbor & nb1( size_t i1, size_t _i2 ) { return _nb1[i1][_i2]; }
+        Neighbor & nb1( size_t i1, size_t _i2 ) {
+#ifdef DAI_DEBUG
+            assert( i1 < _nb1.size() );
+            assert( _i2 < _nb1[i1].size() );
+#endif
+            return _nb1[i1][_i2]; 
+        }
 
         /// Returns constant reference to the _i1'th neighbor of node i2 of type 2
-        const Neighbor & nb2( size_t i2, size_t _i1 ) const { return _nb2[i2][_i1]; }
+        const Neighbor & nb2( size_t i2, size_t _i1 ) const { 
+#ifdef DAI_DEBUG
+            assert( i2 < _nb2.size() );
+            assert( _i1 < _nb2[i2].size() );
+#endif
+            return _nb2[i2][_i1]; 
+        }
         /// Returns reference to the _i1'th neighbor of node i2 of type 2
-        Neighbor & nb2( size_t i2, size_t _i1 ) { return _nb2[i2][_i1]; }
+        Neighbor & nb2( size_t i2, size_t _i1 ) { 
+#ifdef DAI_DEBUG
+            assert( i2 < _nb2.size() );
+            assert( _i1 < _nb2[i2].size() );
+#endif
+            return _nb2[i2][_i1]; 
+        }
 
         /// Returns constant reference to all neighbors of node i1 of type 1
-        const Neighbors & nb1( size_t i1 ) const { return _nb1[i1]; }
+        const Neighbors & nb1( size_t i1 ) const { 
+#ifdef DAI_DEBUG
+            assert( i1 < _nb1.size() );
+#endif
+            return _nb1[i1]; 
+        }
         /// Returns reference to all neighbors of node of i1 type 1
-        Neighbors & nb1( size_t i1 ) { return _nb1[i1]; }
+        Neighbors & nb1( size_t i1 ) { 
+#ifdef DAI_DEBUG
+            assert( i1 < _nb1.size() );
+#endif
+            return _nb1[i1]; 
+        }
 
         /// Returns constant reference to all neighbors of node i2 of type 2
-        const Neighbors & nb2( size_t i2 ) const { return _nb2[i2]; }
+        const Neighbors & nb2( size_t i2 ) const { 
+#ifdef DAI_DEBUG
+            assert( i2 < _nb2.size() );
+#endif
+            return _nb2[i2]; 
+        }
         /// Returns reference to all neighbors of node i2 of type 2
-        Neighbors & nb2( size_t i2 ) { return _nb2[i2]; }
+        Neighbors & nb2( size_t i2 ) { 
+#ifdef DAI_DEBUG
+            assert( i2 < _nb2.size() );
+#endif
+            return _nb2[i2]; 
+        }
 
         /// Returns number of nodes of type 1
         size_t nr1() const { return _nb1.size(); }
         /// Returns number of nodes of type 2
         size_t nr2() const { return _nb2.size(); }
         
-        /// Calculates the number of edges
+        /// Calculates the number of edges, using O(nr1()) time
         size_t nrEdges() const {
             size_t sum = 0;
             for( size_t i1 = 0; i1 < nr1(); i1++ )
@@ -152,14 +197,10 @@ class BipartiteGraph {
         }
         
         /// Add node of type 1 without neighbors.
-        void add1() {
-            _nb1.push_back( Neighbors() );
-        }
+        void add1() { _nb1.push_back( Neighbors() ); }
         
         /// Add node of type 2 without neighbors.
-        void add2() {
-            _nb2.push_back( Neighbors() );
-        }
+        void add2() { _nb2.push_back( Neighbors() ); }
 
         /// Add node of type 1 with neighbors specified by a range.
         /** The value_type of an NodeInputIterator should be a size_t, corresponding to
@@ -201,223 +242,59 @@ class BipartiteGraph {
             _nb2.push_back( nbs2new );
         }
 
-        /// Remove node of type 1 and all incident edges.
-        void erase1( size_t n1 ) {
-            assert( n1 < nr1() );
-            // Erase neighbor entry of node n1
-            _nb1.erase( _nb1.begin() + n1 );
-            // Adjust neighbor entries of nodes of type 2
-            for( size_t n2 = 0; n2 < nr2(); n2++ )
-                for( size_t iter = 0; iter < nb2(n2).size(); ) {
-                    if( nb2(n2, iter).node == n1 ) {
-                        // delete this entry, because it points to the deleted node
-                        nb2(n2).erase( nb2(n2).begin() + iter );
-                        // adjust all subsequent entries:
-                        // update their iter and the corresponding dual of the neighboring node of type 1
-                        for( size_t newiter = iter; newiter < nb2(n2).size(); newiter++ ) {
-                            nb2( n2, newiter ).iter = newiter;
-                            nb1( nb2(n2, newiter).node, nb2(n2, newiter).dual ).dual = newiter;
-                        }
-                    } else if( nb2(n2, iter).node > n1 ) {
-                        nb2(n2, iter).node--;
-                        iter++;
-                    } else
-                        iter++;
-                }
-        }
+        /// Remove node n1 of type 1 and all incident edges.
+        void erase1( size_t n1 );
+
+        /// Remove node n2 of type 2 and all incident edges.
+        void erase2( size_t n2 );
 
-        /// Remove node of type 2 and all incident edges.
-        void erase2( size_t n2 ) {
+        /// Add edge between node n1 of type 1 and node n2 of type 2.
+        /** If check == true, only adds the edge if it does not exist already.
+         */
+        void addEdge( size_t n1, size_t n2, bool check = true ) {
+            assert( n1 < nr1() );
             assert( n2 < nr2() );
-            // Erase neighbor entry of node n2
-            _nb2.erase( _nb2.begin() + n2 );
-            // Adjust neighbor entries of nodes of type 1
-            for( size_t n1 = 0; n1 < nr1(); n1++ )
-                for( size_t iter = 0; iter < nb1(n1).size(); ) {
-                    if( nb1(n1, iter).node == n2 ) {
-                        // delete this entry, because it points to the deleted node
-                        nb1(n1).erase( nb1(n1).begin() + iter );
-                        // adjust all subsequent entries:
-                        // update their iter and the corresponding dual of the neighboring node of type 2
-                        for( size_t newiter = iter; newiter < nb1(n1).size(); newiter++ ) {
-                            nb1( n1, newiter ).iter = newiter;
-                            nb2( nb1(n1, newiter).node, nb1(n1, newiter).dual ).dual = newiter;
-                        }
-                    } else if( nb1(n1, iter).node > n2 ) {
-                        nb1(n1, iter).node--;
-                        iter++;
-                    } else
-                        iter++;
-                }
+            bool exists = false;
+            if( check ) {
+                // Check whether the edge already exists
+                foreach( const Neighbor &nb2, nb1(n1) )
+                    if( nb2 == n2 ) {
+                        exists = true;
+                        break;
+                    }
+            }
+            if( !exists ) { // Add edge
+                Neighbor nb_1( _nb1[n1].size(), n2, _nb2[n2].size() );
+                Neighbor nb_2( nb_1.dual, n1, nb_1.iter );
+                _nb1[n1].push_back( nb_1 );
+                _nb2[n2].push_back( nb_2 );
+            }
         }
 
         /// Calculate second-order neighbors (i.e., neighbors of neighbors) of node n1 of type 1.
         /** If include == true, include n1 itself, otherwise exclude n1.
          */
-        std::vector<size_t> delta1( size_t n1, bool include = false ) const {
-            std::vector<size_t> result;
-            foreach( const Neighbor &n2, nb1(n1) )
-                foreach( const Neighbor &m1, nb2(n2) )
-                    if( include || (m1 != n1) )
-                        result.push_back( m1 );
-            // remove duplicates
-            std::vector<size_t>::iterator it = std::unique( result.begin(), result.end() );
-            result.erase( it, result.end() );
-            return result;
-        }
+        std::vector<size_t> delta1( size_t n1, bool include = false ) const;
 
         /// Calculate second-order neighbors (i.e., neighbors of neighbors) of node n2 of type 2.
         /** If include == true, include n2 itself, otherwise exclude n2.
          */
-        std::vector<size_t> delta2( size_t n2, bool include = false ) const {
-            std::vector<size_t> result;
-            foreach( const Neighbor &n1, nb2(n2) )
-                foreach( const Neighbor &m2, nb1(n1) )
-                    if( include || (m2 != n2) )
-                        result.push_back( m2 );
-            // remove duplicates
-            std::vector<size_t>::iterator it = std::unique( result.begin(), result.end() );
-            result.erase( it, result.end() );
-            return result;
-        }
+        std::vector<size_t> delta2( size_t n2, bool include = false ) const;
 
         /// Returns true if the graph is connected
-        bool isConnected() const {
-            if( nr1() == 0 ) {
-                return true;
-            } else {
-                std::vector<bool> incomponent1( nr1(), false );
-                std::vector<bool> incomponent2( nr2(), false );
-
-                incomponent1[0] = true;
-                bool found_new_nodes;
-                do {
-                    found_new_nodes = false;
-
-                    // For all nodes of type 2, check if they are connected with the (growing) component
-                    for( size_t n2 = 0; n2 < nr2(); n2++ )
-                        if( !incomponent2[n2] ) {
-                            foreach( const Neighbor &n1, nb2(n2) ) {
-                                if( incomponent1[n1] ) {
-                                    found_new_nodes = true;
-                                    incomponent2[n2] = true;
-                                    break;
-                                }
-                            }
-                        }
-
-                    // For all nodes of type 1, check if they are connected with the (growing) component
-                    for( size_t n1 = 0; n1 < nr1(); n1++ )
-                        if( !incomponent1[n1] ) {
-                            foreach( const Neighbor &n2, nb1(n1) ) {
-                                if( incomponent2[n2] ) {
-                                    found_new_nodes = true;
-                                    incomponent1[n1] = true;
-                                    break;
-                                }
-                            }
-                        }
-                } while( found_new_nodes );
-
-                // Check if there are remaining nodes (not in the component)
-                bool all_connected = true;
-                for( size_t n1 = 0; (n1 < nr1()) && all_connected; n1++ )
-                    if( !incomponent1[n1] )
-                        all_connected = false;
-                for( size_t n2 = 0; (n2 < nr2()) && all_connected; n2++ )
-                    if( !incomponent2[n2] )
-                        all_connected = false;
-
-                return all_connected;
-            }
-        }
+        bool isConnected() const;
 
         /// Returns true if the graph is a tree, i.e., if it is singly connected and connected.
         /** This is equivalent to whether for each pair of vertices in the graph, there exists
          *  a unique path in the graph that starts at the first and ends at the second vertex.
          */
-        bool isTree() const {
-            using namespace std;
-            vector<levelType> levels;
-
-            bool foundCycle = false;
-            size_t nr_1 = 0;
-            size_t nr_2 = 0;
-
-            if( nr1() == 0 || nr2() == 0 )
-                return true;
-            else {
-                levelType newLevel;
-                do {
-                    newLevel.ind1.clear();
-                    newLevel.ind2.clear();
-                    if( levels.size() == 0 ) {
-                        size_t n1 = 0;
-                        // add n1 to ind1
-                        newLevel.ind1 = vector<size_t>( 1, n1 );
-                        // add all neighbors of n1 to ind2
-                        newLevel.ind2.reserve( nb1(n1).size() );
-                        foreach( const Neighbor &n2, nb1(n1) )
-                            newLevel.ind2.push_back( n2 );
-                    } else {
-                        const levelType &prevLevel = levels.back();
-                        // build newLevel.ind1
-                        foreach( size_t n2, prevLevel.ind2 ) { // for all n2 in the previous level
-                            foreach( const Neighbor &n1, nb2(n2) ) { // for all neighbors n1 of n2
-                                if( find( prevLevel.ind1.begin(), prevLevel.ind1.end(), n1 ) == prevLevel.ind1.end() ) { // n1 not in previous level
-                                    if( find( newLevel.ind1.begin(), newLevel.ind1.end(), n1 ) != newLevel.ind1.end() )
-                                        foundCycle = true; // n1 already in new level: we found a cycle
-                                    else
-                                        newLevel.ind1.push_back( n1 ); // add n1 to new level
-                                }
-                                if( foundCycle )
-                                    break;
-                            }
-                            if( foundCycle )
-                                break;
-                        }
-                        // build newLevel.ind2
-                        foreach( size_t n1, newLevel.ind1 ) { // for all n1 in this level
-                            foreach( const Neighbor &n2, nb1(n1) ) { // for all neighbors n2 of n1
-                                if( find( prevLevel.ind2.begin(), prevLevel.ind2.end(), n2 ) == prevLevel.ind2.end() ) { // n2 not in previous level
-                                    if( find( newLevel.ind2.begin(), newLevel.ind2.end(), n2 ) != newLevel.ind2.end() )
-                                        foundCycle = true; // n2 already in new level: we found a cycle
-                                    else
-                                        newLevel.ind2.push_back( n2 ); // add n2 to new level
-                                }
-                                if( foundCycle )
-                                    break;
-                            }
-                            if( foundCycle )
-                                break;
-                        } 
-                    }
-                    levels.push_back( newLevel );
-                    nr_1 += newLevel.ind1.size();
-                    nr_2 += newLevel.ind2.size();
-                } while( ((newLevel.ind1.size() != 0) || (newLevel.ind2.size() != 0)) && !foundCycle );
-                if( nr_1 == nr1() && nr_2 == nr2() && !foundCycle )
-                    return true;
-                else
-                    return false;
-            }
-        }
+        bool isTree() const;
 
         /// Stream to output stream os in graphviz .dot syntax
-        void display( std::ostream& os ) const {
-            using namespace std;
-            os << "graph G {" << endl;
-            os << "node[shape=circle,width=0.4,fixedsize=true];" << endl;
-            for( size_t n1 = 0; n1 < nr1(); n1++ )
-                os << "\tx" << n1 << ";" << endl;
-            os << "node[shape=box,width=0.3,height=0.3,fixedsize=true];" << endl;
-            for( size_t n2 = 0; n2 < nr2(); n2++ )
-                os << "\ty" << n2 << ";" << endl;
-            for( size_t n1 = 0; n1 < nr1(); n1++ )
-                foreach( const Neighbor &n2, nb1(n1) )
-                    os << "\tx" << n1 << " -- y" << n2 << ";" << endl;
-            os << "}" << endl;
-        }
+        void display( std::ostream& os ) const;
+
+        /// Checks internal consistency
+        void check() const;
 };
 
 
@@ -429,11 +306,11 @@ void BipartiteGraph::create( size_t nr1, size_t nr2, EdgeInputIterator begin, Ed
     _nb2.resize( nr2 );
 
     for( EdgeInputIterator e = begin; e != end; e++ ) {
-        // Each edge yields a neighbor pair
-        Neighbor nb_1( _nb1[e->first].size(), e->second, _nb2[e->second].size() );
-        Neighbor nb_2( nb_1.dual, e->first, nb_1.iter );
-        _nb1[e->first].push_back( nb_1 );
-        _nb2[e->second].push_back( nb_2 );
+#ifdef DAI_DEBUG
+        addEdge( e->first, e->second, true );
+#else
+        addEdge( e->first, e->second, false );
+#endif
     }
 }