src/work/jacint/j_graph.h
changeset 379 a5bff2813c4d
equal deleted inserted replaced
-1:000000000000 0:0a31c8b87faf
       
     1 // -*- mode:C++ -*-
       
     2 
       
     3 #ifndef J_GRAPH_H
       
     4 #define J_GRAPH_H
       
     5 
       
     6 #include <iostream>
       
     7 #include <vector>
       
     8 
       
     9 namespace hugo {
       
    10 
       
    11   class JGraph {
       
    12 
       
    13     struct NodeT 
       
    14     {
       
    15       int in, out;      
       
    16       NodeT() : in(), out() {}
       
    17     };
       
    18    
       
    19     struct EdgeT 
       
    20     {
       
    21       int head, tail, in, out;      
       
    22     };
       
    23 
       
    24 
       
    25     std::vector<NodeT> nodes;
       
    26     std::vector<EdgeT> edges;
       
    27     
       
    28 
       
    29     /*    template <typename Key> class DynMapBase
       
    30     {
       
    31     protected:
       
    32       SmartGraph* G; 
       
    33     public:
       
    34       virtual void add(const Key k) = NULL;
       
    35       virtual void erase(const Key k) = NULL;
       
    36       DynMapBase(SmartGraph &_G) : G(&_G) {}
       
    37       virtual ~DynMapBase() {}
       
    38       friend class SmartGraph;
       
    39       };
       
    40 
       
    41     template <typename T> class DynEdgeMap;
       
    42     template <typename T> class DynEdgeMap;
       
    43     */
       
    44 
       
    45   public:
       
    46   
       
    47     class TrivNodeIt;
       
    48     class TrivEdgeIt;
       
    49     class NodeIt;
       
    50     class EdgeIt;
       
    51     class OutEdgeIt;
       
    52     class InEdgeIt;
       
    53     
       
    54     /*  protected:
       
    55     std::vector<DynMapBase<NodeIt> * > dyn_node_maps;
       
    56     std::vector<DynMapBase<EdgeIt> * > dyn_edge_maps;
       
    57     */
       
    58 
       
    59     template <typename T> class NodeMap;
       
    60     template <typename T> class EdgeMap;
       
    61     
       
    62   public:
       
    63 
       
    64     JGraph() : nodes(1), edges(1) {
       
    65       edges[0].head=0; 
       
    66       edges[0].tail=0; 
       
    67       edges[0].in=0; //numEdges (numNodes is maintained in nodes[0].in)
       
    68       edges[0].out=0; 
       
    69     }    
       
    70 
       
    71 
       
    72     /*    ~SmartGraph()
       
    73     {
       
    74       for(std::vector<DynMapBase<NodeIt> * >::iterator i=dyn_node_maps.begin();
       
    75 	  i!=dyn_node_maps.end(); ++i) (**i).G=NULL;
       
    76       for(std::vector<DynMapBase<EdgeIt> * >::iterator i=dyn_edge_maps.begin();
       
    77 	  i!=dyn_edge_maps.end(); ++i) (**i).G=NULL;
       
    78 	  }*/
       
    79 
       
    80     int numNodes() const { return nodes[0].in; }
       
    81     int numEdges() const { return edges[0].in; } 
       
    82 
       
    83     TrivNodeIt tail(TrivEdgeIt e) const { return edges[e.n].tail; }
       
    84     TrivNodeIt head(TrivEdgeIt e) const { return edges[e.n].head; }
       
    85 
       
    86     /*    EachNodeIt& getFirst(EachNodeIt& v) const { 
       
    87       v=EachNodeIt(*this); return v; }
       
    88     EachEdgeIt& getFirst(EachEdgeIt& e) const { 
       
    89       e=EachEdgeIt(*this); return e; }
       
    90     OutEdgeIt& getFirst(OutEdgeIt& e, const NodeIt v) const { 
       
    91       e=OutEdgeIt(*this,v); return e; }
       
    92     InEdgeIt& getFirst(InEdgeIt& e, const NodeIt v) const { 
       
    93       e=InEdgeIt(*this,v); return e; }
       
    94     */
       
    95 
       
    96     NodeIt firstNode() const { 
       
    97       return NodeIt( numNodes() );
       
    98     }
       
    99     EdgeIt firstEdge() const { 
       
   100       return EdgeIt( numEdges() );
       
   101     }
       
   102     InEdgeIt firstIn(TrivNodeIt v) const { 
       
   103       return InEdgeIt(nodes[v.n].in);
       
   104     }
       
   105     OutEdgeIt firstOut(TrivNodeIt v) const { 
       
   106       return OutEdgeIt(nodes[v.n].out);
       
   107     }
       
   108 
       
   109 
       
   110    
       
   111     void next(NodeIt& it) const {--it.n;}
       
   112     void next(OutEdgeIt& it) const { it.n=edges[it.n].out; }
       
   113     void next(InEdgeIt& it) const { it.n=edges[it.n].in; }
       
   114     void next(EdgeIt& it) const {--it.n;}
       
   115     
       
   116     int id(TrivNodeIt v) const { return v.n; }
       
   117     int id(TrivEdgeIt e) const { return e.n; }
       
   118 
       
   119     TrivNodeIt addNode() {
       
   120       TrivNodeIt n; 
       
   121       n.n=++nodes[0].in;
       
   122       nodes.push_back(NodeT()); //FIXME
       
   123 
       
   124       /*      for(std::vector<DynMapBase<NodeIt> * >::iterator i=dyn_node_maps.begin();
       
   125 	      i!=dyn_node_maps.end(); ++i) (**i).add(n.n);*/
       
   126 
       
   127       return n;
       
   128     }
       
   129     
       
   130 
       
   131     TrivEdgeIt addEdge(TrivNodeIt u, TrivNodeIt v) {
       
   132       if ( u && v ) {
       
   133       TrivEdgeIt e; 
       
   134       e.n=++edges[0].in;
       
   135       edges.push_back(EdgeT()); //FIXME: Hmmm...
       
   136       edges[e.n].tail=u.n;   edges[e.n].head=v.n;
       
   137       edges[e.n].out=nodes[u.n].out;
       
   138       edges[e.n].in=nodes[v.n].in;
       
   139       nodes[u.n].out=nodes[v.n].in=e.n;
       
   140       return e;
       
   141       } 
       
   142       else return TrivEdgeIt();
       
   143 
       
   144 	       /*      for(std::vector<DynMapBase<EdgeIt> * >::iterator i=dyn_edge_maps.begin();
       
   145 		       i!=dyn_edge_maps.end(); ++i) (**i).add(e.n);*/
       
   146 
       
   147       
       
   148     }
       
   149 
       
   150 
       
   151     void clear() {
       
   152       nodes.resize(1);
       
   153       edges.resize(1); edges[0].in=0;
       
   154     }
       
   155 
       
   156 
       
   157     class TrivNodeIt {
       
   158       friend class JGraph;
       
   159       template <typename T> friend class NodeMap;
       
   160       
       
   161       friend class TrivEdgeIt;
       
   162       friend class OutEdgeIt;
       
   163       friend class InEdgeIt;
       
   164 
       
   165     protected:
       
   166       int n;
       
   167     public:
       
   168       TrivNodeIt() : n() {}
       
   169       TrivNodeIt(int number) {n=number;}
       
   170       bool operator==(const TrivNodeIt i) const {return n==i.n;}
       
   171       bool operator!=(const TrivNodeIt i) const {return n!=i.n;}
       
   172    
       
   173       operator bool() const  { return n!=0; }
       
   174    
       
   175  };
       
   176     
       
   177 
       
   178     class NodeIt : public TrivNodeIt {
       
   179       friend class JGraph;
       
   180     public:
       
   181       NodeIt() : TrivNodeIt() { }
       
   182       NodeIt(int i) : TrivNodeIt(i) { }
       
   183       operator bool() const  { return n!=0; }
       
   184     };
       
   185 
       
   186 
       
   187     class TrivEdgeIt {
       
   188       friend class JGraph;
       
   189       template <typename T> friend class EdgeMap;
       
   190       
       
   191       friend class TrivNodeIt;
       
   192       friend class NodeIt;
       
   193     protected:
       
   194       int n;
       
   195     public:
       
   196       TrivEdgeIt() : n() { }
       
   197       TrivEdgeIt(int number) {n=number;}
       
   198       bool operator==(const TrivEdgeIt i) const {return n==i.n;}
       
   199       bool operator!=(const TrivEdgeIt i) const {return n!=i.n;}
       
   200       operator bool() const { return n!=0; }
       
   201     
       
   202  };
       
   203     
       
   204 
       
   205     class EdgeIt : public TrivEdgeIt {
       
   206       friend class JGraph;
       
   207     public:
       
   208       EdgeIt() : TrivEdgeIt() { }
       
   209       EdgeIt(int i) : TrivEdgeIt(i) { }
       
   210       operator bool() const { return n!=0; } 
       
   211  };
       
   212     
       
   213 
       
   214     class OutEdgeIt : public TrivEdgeIt {
       
   215       friend class JGraph;
       
   216     public: 
       
   217       OutEdgeIt() : TrivEdgeIt() { }
       
   218       OutEdgeIt(int i) : TrivEdgeIt(i) { }
       
   219     };
       
   220     
       
   221 
       
   222     class InEdgeIt : public TrivEdgeIt {
       
   223       friend class JGraph;
       
   224     public: 
       
   225       InEdgeIt() : TrivEdgeIt() { }
       
   226       InEdgeIt(int i) : TrivEdgeIt(i) { }
       
   227     };
       
   228 
       
   229 
       
   230     // Map types
       
   231     
       
   232     template <typename T>
       
   233     class NodeMap {
       
   234       const JGraph& G; 
       
   235       std::vector<T> container;
       
   236     public:
       
   237       NodeMap(const JGraph& _G) : G(_G), container( G.numNodes()+1  ) { }
       
   238       NodeMap(const JGraph& _G, T a) : 
       
   239 	G(_G), container(G.numNodes()+1, a) { }
       
   240       void set(TrivNodeIt n, T a) { container[n.n]=a; }
       
   241       T get(TrivNodeIt n) const { return container[n.n]; }
       
   242       /*T& operator[](NodeIt n) { return container[n.n]; }
       
   243 	const T& operator[](NodeIt n) const { return container[n.n]; }*/
       
   244       void update() { container.resize(G.numNodes()+1); }
       
   245       void update(T a) { container.resize(G.numNodes()+1, a); }
       
   246     };
       
   247 
       
   248     template <typename T>
       
   249     class EdgeMap {
       
   250       const JGraph& G; 
       
   251       std::vector<T> container;
       
   252     public:
       
   253       EdgeMap(const JGraph& _G) : G(_G), container(G.numEdges()+1) { }
       
   254       EdgeMap(const JGraph& _G, T a) : 
       
   255 	G(_G), container(G.numEdges()+1, a) { }
       
   256       void set(TrivEdgeIt e, T a) { container[e.n]=a; }
       
   257       T get(TrivEdgeIt e) const { return container[e.n]; }
       
   258       /*T& operator[](EdgeIt e) { return container[e.n]; } 
       
   259 	const T& operator[](EdgeIt e) const { return container[e.n]; } */
       
   260       void update() { container.resize(G.numEdges()+1); }
       
   261       void update(T a) { container.resize(G.numEdges()+1, a); }
       
   262     };
       
   263 
       
   264     /*template <typename T> class DynNodeMap : public DynMapBase<NodeIt>
       
   265     {
       
   266       std::vector<T> container;
       
   267 
       
   268     public:
       
   269       typedef T ValueType;
       
   270       typedef NodeIt KeyType;
       
   271 
       
   272       DynNodeMap(JGraph &_G) :
       
   273 	DynMapBase<NodeIt>(_G), container(_G.maxNodeId())
       
   274       {
       
   275 	//FIXME: What if there are empty Id's?
       
   276 	//FIXME: Can I use 'this' in a constructor?
       
   277 	G->dyn_node_maps.push_back(this);
       
   278       }
       
   279       ~DynNodeMap()
       
   280       {
       
   281 	if(G) {
       
   282 	  std::vector<DynMapBase<NodeIt>* >::iterator i;
       
   283 	  for(i=G->dyn_node_maps.begin();
       
   284 	      i!=G->dyn_node_maps.end() && *i!=this; ++i) ;
       
   285 	  //if(*i==this) G->dyn_node_maps.erase(i); //FIXME: Way too slow...
       
   286 	  //A better way to do that: (Is this really important?)
       
   287 	  if(*i==this) {
       
   288 	    *i=G->dyn_node_maps.back();
       
   289 	    G->dyn_node_maps.pop_back();
       
   290 	  }
       
   291 	}
       
   292       }
       
   293 
       
   294       void add(const NodeIt k) 
       
   295       {
       
   296 	if(k.n>=container.size()) container.resize(k.n+1);
       
   297       }
       
   298       void erase(const NodeIt k)
       
   299       {
       
   300 	//FIXME: Please implement me.
       
   301       }
       
   302       
       
   303       void set(NodeIt n, T a) { container[n.n]=a; }
       
   304       T get(NodeIt n) const { return container[n.n]; }
       
   305       T& operator[](NodeIt n) { return container[n.n]; }
       
   306       const T& operator[](NodeIt n) const { return container[n.n]; }
       
   307 
       
   308       void update() {}    //Useless for DynMaps
       
   309       void update(T a) {}  //Useless for DynMaps
       
   310     };
       
   311     
       
   312     template <typename T> class DynEdgeMap : public DynMapBase<EdgeIt>
       
   313     {
       
   314       std::vector<T> container;
       
   315 
       
   316     public:
       
   317       typedef T ValueType;
       
   318       typedef EdgeIt KeyType;
       
   319 
       
   320       DynEdgeMap(JGraph &_G) :
       
   321 	DynMapBase<EdgeIt>(_G), container(_G.maxEdgeId())
       
   322       {
       
   323 	//FIXME: What if there are empty Id's?
       
   324 	//FIXME: Can I use 'this' in a constructor?
       
   325 	G->dyn_edge_maps.push_back(this);
       
   326       }
       
   327       ~DynEdgeMap()
       
   328       {
       
   329 	if(G) {
       
   330 	  std::vector<DynMapBase<EdgeIt>* >::iterator i;
       
   331 	  for(i=G->dyn_edge_maps.begin();
       
   332 	      i!=G->dyn_edge_maps.end() && *i!=this; ++i) ;
       
   333 	  //if(*i==this) G->dyn_edge_maps.erase(i); //Way too slow...
       
   334 	  //A better way to do that: (Is this really important?)
       
   335 	  if(*i==this) {
       
   336 	    *i=G->dyn_edge_maps.back();
       
   337 	    G->dyn_edge_maps.pop_back();
       
   338 	  }
       
   339 	}
       
   340       }
       
   341       
       
   342       void add(const EdgeIt k) 
       
   343       {
       
   344 	if(k.n>=int(container.size())) container.resize(k.n+1);
       
   345       }
       
   346       void erase(const EdgeIt k)
       
   347       {
       
   348 	//FIXME: Please implement me.
       
   349       }
       
   350       
       
   351       void set(EdgeIt n, T a) { container[n.n]=a; }
       
   352       T get(EdgeIt n) const { return container[n.n]; }
       
   353       T& operator[](EdgeIt n) { return container[n.n]; }
       
   354       const T& operator[](EdgeIt n) const { return container[n.n]; }
       
   355 
       
   356       void update() {}    //Useless for DynMaps
       
   357       void update(T a) {}  //Useless for DynMaps
       
   358       };*/
       
   359         
       
   360   };
       
   361 } //namespace hugo
       
   362 
       
   363 #endif //J_GRAPH_H