routervisitor.hh

Go to the documentation of this file.

// -*- c-basic-offset: 4; related-file-name: "../../lib/routervisitor.cc" -*-
#ifndef CLICK_ROUTERVISITOR_HH
#define CLICK_ROUTERVISITOR_HH
#include <click/element.hh>
#include <click/bitvector.hh>
#include <click/pair.hh>
 
CLICK_DECLS
 
class RouterVisitor { public:
 
    RouterVisitor() {
    }
 
    virtual ~RouterVisitor() {
    }
 
    virtual bool visit(Element *e, bool isoutput, int port,
                       Element *from_e, int from_port, int distance);
 
    virtual int distance(Element *e, Element *from_e);
 
};
 
class ElementTracker : public RouterVisitor { public:
 
    ElementTracker(Router *router);
 
    const Vector<Element *> &elements() const {
        return _elements;
    }
    int size() const {
        return _elements.size();
    }
    Element *operator[](int i) const {
        return _elements[i];
    }
 
    typedef Vector<Element *>::const_iterator const_iterator;
    const_iterator begin() const {
        return _elements.begin();
    }
    const_iterator end() const {
        return _elements.end();
    }
 
    bool contains(const Element *e) const {
        return _reached[e->eindex()];
    }
 
    void insert(Element *e) {
        if (!_reached[e->eindex()]) {
            _reached[e->eindex()] = true;
            _elements.push_back(e);
        }
    }
 
    void clear() {
        _reached.clear();
        _elements.clear();
    }
 
  private:
 
    Bitvector _reached;
    Vector<Element *> _elements;
 
};
 
class ElementCastTracker : public ElementTracker { public:
 
    ElementCastTracker(Router *router, const String &name, bool stopAtFirst = true)
        : ElementTracker(router), _name(name), _continue(!stopAtFirst) {
    }
 
    bool visit(Element *e, bool isoutput, int port,
               Element *from_e, int from_port, int distance);
 
  private:
 
    String _name;
    bool _continue;
 
};
 
class ElementNeighborhoodTracker : public ElementTracker { public:
 
    ElementNeighborhoodTracker(Router *router, int diameter = 1)
        : ElementTracker(router), _diameter(diameter) {
    }
 
    bool visit(Element *e, bool isoutput, int port,
               Element *from_e, int from_port, int distance);
 
  private:
 
    int _diameter;
 
};
 
CLICK_ENDDECLS
#endif