00001 #ifndef DDD_graph_h
00002 #define DDD_graph_h
00003
00004
00005 #include <vector>
00006 #include <map>
00007 #include <iostream>
00008
00009
00010
00011
00012
00013
00014 template <class N, class E>
00015 class graph
00016 {
00017
00018 public:
00019 class const_iterator
00020 {
00021 friend class graph<N,E>;
00022 public:
00023 typedef typename graph::index_type index_type;
00024 typedef typename graph::adj_list adj_list;
00025 typedef typename graph::edge_list edge_list;
00026
00027 struct value_type {
00028 friend class graph<N,E>::const_iterator;
00029 value_type(const graph & g, index_type a, index_type e)
00030 : gr_(g), a_(a), e_(e)
00031 { }
00032
00033 const N & from() const { return gr_.nodeData(a_); }
00034 const N & to() const { return gr_.nodeData(gr_.adjl_[a_][e_].first); }
00035 const E & edge() const { return gr_.edgeData(gr_.adjl_[a_][e_].second); }
00036
00037 private:
00038 const graph & gr_;
00039 index_type a_,e_;
00040 };
00041
00042 typedef value_type& reference;
00043 typedef value_type* pointer;
00044
00045 bool operator==(const const_iterator & i) const {
00046 return ( (vt_.a_ == i.vt_.a_) && (vt_.e_ == i.vt_.e_) ) ? true : false;
00047 }
00048
00049 bool operator!=(const const_iterator & i) const {
00050 return ( (vt_.a_ == i.vt_.a_) && (vt_.e_ == i.vt_.e_) ) ? false : true;
00051 }
00052
00053 void operator++() {
00054 while(vt_.gr_.size() > vt_.a_){
00055
00056 index_type i = vt_.gr_.adjl_[vt_.a_].size();
00057 if (i > vt_.e_+1) {
00058 ++vt_.e_;
00059 return;
00060 }
00061 vt_.e_=0;
00062 ++vt_.a_;
00063 while( vt_.gr_.size() > vt_.a_ ) {
00064 if ( vt_.gr_.adjl_[vt_.a_].size() ) {
00065 return;
00066 }
00067 ++vt_.a_;
00068 }
00069 }
00070 }
00071
00072 const value_type & operator*() const {
00073 return vt_;
00074 }
00075
00076 const value_type * operator->() const {
00077 return &vt_;
00078 }
00079
00080 private:
00081 explicit const_iterator(const graph & g)
00082 : vt_(g, 0, 0)
00083 { }
00084
00085 const_iterator(const graph & g, index_type ait, index_type eit)
00086 : vt_(g, ait, eit)
00087 { }
00088
00089 value_type vt_;
00090
00091 bool operator<(const const_iterator & i) const {
00092 return ( vt_.a_ < i.vt_.a_ ) && ( vt_.e_ < i.vt_.e_);
00093 }
00094
00095 bool operator>(const const_iterator & i) const {
00096 return ( vt_.a_ > i.vt_.a_ ) && ( vt_.e_ > i.vt_.e_);
00097 }
00098 };
00099
00100 void dump_graph() const;
00101
00102
00103 struct value_type {
00104 value_type(const N & n, const E & e) : first(n), second(e) { }
00105 const N & first;
00106 const E & second;
00107 N firstToValue() const { return first; }
00108 E secondToValue() const { return second; }
00109 };
00110
00111 typedef std::vector<double>::size_type index_type;
00112
00113
00114 typedef std::pair<index_type, index_type> edge_type;
00115
00116
00117 typedef std::vector<N> node_list;
00118 typedef std::vector<E> edge_store;
00119
00120
00121 typedef std::vector<edge_type> edge_list;
00122
00123
00124 typedef std::vector<edge_list> adj_list;
00125 typedef typename adj_list::iterator adj_iterator;
00126 typedef typename adj_list::const_iterator const_adj_iterator;
00127
00128
00129
00130 typedef std::map<N, index_type> indexer_type;
00131 typedef typename indexer_type::iterator indexer_iterator;
00132 typedef typename indexer_type::const_iterator const_indexer_iterator;
00133
00134
00135 typedef typename edge_list::iterator edge_iterator;
00136
00137 typedef typename edge_list::const_iterator const_edge_iterator;
00138
00139 typedef std::pair<edge_iterator,edge_iterator> edge_range;
00140
00141 typedef std::pair<const_edge_iterator, const_edge_iterator> const_edge_range;
00142
00143 typedef std::pair<index_type, bool> index_result;
00144
00145 public:
00146
00147 graph() : edges_(1) { }
00148 ~graph() { }
00149
00150
00151
00152
00153 index_type addNode(const N &);
00154
00155 void addEdge(const N & from, const N & to, const E & edge);
00156
00157
00158
00159
00160
00161
00162 inline index_result nodeIndex(const N &) const;
00163
00164
00165
00166
00167 inline edge_range edges(index_type nodeIndex);
00168 inline const_edge_range edges(index_type nodeIndex) const;
00169
00170
00171 inline edge_range edges(const N &);
00172 inline const_edge_range edges(const N &) const;
00173
00174 inline const N & nodeData(const edge_type &) const;
00175 inline const N & nodeData(index_type) const;
00176 inline const N & nodeData(const const_adj_iterator &) const;
00177
00178
00179 bool replace(const N & oldNode , const N & newNode );
00180
00181
00182 bool replaceEdge(const E & ldEdge, const E &newEdge );
00183
00184 const E & edgeData(index_type i) const { return edges_[i]; }
00185
00186
00187
00189 void clear();
00190
00191 const_iterator begin_iter() const { return const_iterator(*this); }
00192
00193 const_iterator end_iter() const { return const_iterator(*this, adjl_.size(),0); }
00194
00195 size_t edge_size() const { return edges_.size(); }
00196
00197
00198 adj_iterator begin() { return adjl_.begin(); }
00199 const_adj_iterator begin() const { return adjl_.begin(); }
00200 adj_iterator end() { return adjl_.end(); }
00201 const_adj_iterator end() const { return adjl_.end(); }
00202 typename adj_list::size_type size() const { return adjl_.size(); }
00203
00204
00205 void findRoots(edge_list &) const;
00206
00207
00208 void invert(graph & g) const;
00209
00210
00211 private:
00212
00213
00214 adj_list adjl_;
00215
00216
00217 node_list nodes_;
00218
00219
00220 edge_store edges_;
00221
00222
00223 indexer_type indexer_;
00224
00225
00226 edge_list emptyEdges_;
00227
00228 };
00229
00230
00231
00232 template<class N, class E>
00233 typename graph<N,E>::index_type graph<N,E>::addNode(const N & node)
00234 {
00235 index_type idx = indexer_.size() ;
00236 std::pair<indexer_iterator,bool> result
00237 = indexer_.insert(typename indexer_type::value_type(node,idx));
00238
00239 if ( result.second ) {
00240 nodes_.push_back(node);
00241 adjl_.push_back(edge_list());
00242 }
00243 else {
00244 idx = result.first->second;
00245 }
00246 return idx;
00247 }
00248
00249
00250 template<class N, class E>
00251 typename graph<N,E>::index_result graph<N,E>::nodeIndex(const N & node) const
00252 {
00253 typename indexer_type::const_iterator result = indexer_.find(node);
00254 index_type idx = 0;
00255 bool flag = false;
00256 if (result != indexer_.end()) {
00257 flag = true;
00258 idx = result->second;
00259 }
00260 return index_result(idx, flag);
00261 }
00262
00263
00264 template<class N, class E>
00265 void graph<N,E>::addEdge(const N & from, const N & to, const E & edge)
00266 {
00267 index_type iFrom = addNode(from);
00268 index_type iTo = addNode(to);
00269
00270 adjl_[iFrom].push_back(edge_type(iTo,edges_.size()));
00271 edges_.push_back(edge);
00272 }
00273
00274
00275 template<class N, class E>
00276 typename graph<N,E>::edge_range graph<N,E>::edges(index_type nodeIndex)
00277 {
00278 edge_list & edges = adjl_[nodeIndex];
00279 return edge_range(edges.begin(), edges.end());
00280 }
00281
00282
00283 template<class N, class E>
00284 typename graph<N,E>::const_edge_range graph<N,E>::edges(index_type nodeIndex) const
00285 {
00286 const edge_list & edges = adjl_[nodeIndex];
00287 return const_edge_range(edges.begin(), edges.end());
00288 }
00289
00290
00291 template<class N, class E>
00292 typename graph<N,E>::edge_range graph<N,E>::edges(const N & node)
00293 {
00294 index_result idxResult = nodeIndex(node);
00295 edge_range result(emptyEdges_.begin(),emptyEdges_.end());
00296 if (idxResult.second) {
00297 result = edges(idxResult.first);
00298 }
00299 return result;
00300 }
00301
00302
00303 template<class N, class E>
00304 typename graph<N,E>::const_edge_range graph<N,E>::edges(const N & node) const
00305 {
00306 index_result idxResult = nodeIndex(node);
00307 const_edge_range result(emptyEdges_.begin(),emptyEdges_.end());
00308 if (idxResult.second) {
00309 result = edges(idxResult.first);
00310 }
00311 return result;
00312 }
00313
00314
00315 template<class N, class E>
00316 const N & graph<N,E>::nodeData(const edge_type & edge) const
00317 {
00318 return nodes_[edge.first];
00319 }
00320
00321
00322 template<class N, class E>
00323 const N & graph<N,E>::nodeData(index_type i) const
00324 {
00325 return nodes_[i];
00326 }
00327
00328
00329 template<class N, class E>
00330 const N & graph<N,E>::nodeData(const const_adj_iterator & it) const
00331 {
00332 return nodes_[it-adjl_.begin()];
00333 }
00334
00335 template<class N, class E>
00336 void graph<N,E>::findRoots(edge_list & result) const
00337 {
00338 result.clear();
00339
00340 const_adj_iterator it = begin();
00341 const_adj_iterator ed = end();
00342 std::vector<bool> rootCandidate(size(), true);
00343
00344 for (; it != ed; ++it) {
00345 const edge_list & el = *it;
00346 typename edge_list::const_iterator el_it = el.begin();
00347 typename edge_list::const_iterator el_ed = el.end();
00348 for (; el_it != el_ed; ++el_it) {
00349 rootCandidate[el_it->first]=false;
00350
00351
00352
00353 }
00354 }
00355 std::vector<bool>::size_type v_sz = 0;
00356 std::vector<bool>::size_type v_ed = rootCandidate.size();
00357 for (; v_sz < v_ed; ++v_sz) {
00358 if (rootCandidate[v_sz]) {
00359
00360 result.push_back(edge_type(v_sz,0));
00361 }
00362 }
00363 }
00364
00365 template<class N, class E>
00366 bool graph<N,E>::replace(const N & oldNode, const N & newNode)
00367 {
00368 typename indexer_type::iterator it = indexer_.find(oldNode);
00369 if (it != indexer_.end()) {
00370 index_type oldIndex = it->second;
00371 nodes_[oldIndex]=newNode;
00372 indexer_[newNode]=oldIndex;
00373 indexer_.erase(it);
00374 }
00375 else throw(oldNode);
00376 return true;
00377 }
00378
00379 template<class N, class E>
00380 bool graph<N,E>::replaceEdge(const E & oldEdge, const E & newEdge)
00381 {
00382 typename edge_store::size_type it = 0;
00383 typename edge_store::size_type ed = edges_.size();
00384 bool result = false;
00385
00386 for (; it < ed; ++it) {
00387
00388 if ( edges_[it] == oldEdge ) {
00389
00390 result = true;
00391 edges_[it] = newEdge;
00392 break;
00393 }
00394 }
00395
00396 return result;
00397 }
00398
00399 template<class N, class E>
00400 void graph<N,E>::clear()
00401 {
00402 adjl_.clear();
00403 nodes_.clear();
00404 edges_.clear();
00405 indexer_.clear();
00406 }
00407
00408 template<class N, class E>
00409 void graph<N,E>::dump_graph() const
00410 {
00411
00412
00413
00414
00415
00416
00417
00418
00419 }
00420
00421
00422 template<class N, class E>
00423 void graph<N,E>::invert(graph<N,E> & g) const
00424 {
00425 adj_list::size_type it = 0;
00426 adj_list::size_type ed = adjl_.size();
00427
00428 for (; it < ed; ++it) {
00429 const edge_list & el = adjl_[it];
00430 edge_list::size_type eit = 0;
00431 edge_list::size_type eed = el.size();
00432
00433 for (; eit < eed; ++eit) {
00434 const edge_type & e = el[eit];
00435 g.addEdge(nodeData(e.first), nodeData(it), edgeData(e.second));
00436 }
00437 }
00438 }
00439
00440
00441 template<typename T> std::ostream & operator<<(std::ostream & o, const std::vector< std::vector<std::pair<T,T> > > v)
00442 {
00443 typedef typename std::vector<std::vector<std::pair<T,T> > > v_t;
00444 typedef typename std::vector<std::pair<T,T> > i_t;
00445
00446 typename v_t::const_iterator it(v.begin()), ed(v.end());
00447 for (; it != ed; ++it) {
00448 typename i_t::const_iterator iit(it->begin()), ied(it->end());
00449 for(; iit != ied; ++iit) {
00450 o << iit->first << ':' << iit->second << std::endl;
00451 }
00452 }
00453 return o;
00454 }
00455 #endif