RouteTable.cpp

#include "RouteTable.h"

namespace forest {


RouteTable::RouteTable(int maxRtx1, fAdr_t myAdr1, ComtreeTable *ctt1)
                                      : maxRtx(maxRtx1), myAdr(myAdr1), ctt(ctt1) {
                rteMap = new HashMap<uint64_t,Vset,Hash::u64>(maxRtx,false);
                clMap  = new HashMap<uint64_t,Vset,Hash::u64>(maxRtx,false);
}
                
RouteTable::~RouteTable() { delete rteMap; delete clMap; }

int RouteTable::addRoute(comt_t comt, fAdr_t adr, int cLnk) {
                Vset links;
                int rtx = rteMap->put(rmKey(comt,adr),links);
                if (rtx == 0) return 0;
                if (cLnk == 0) return rtx;
                rteMap->getValue(rtx).insert(cLnk);
                // now update clMap
                uint64_t kee = cmKey(comt,cLnk);
                int x = clMap->find(kee);
                if (x == 0) {
                                Vset new_routes;
                                x = clMap->put(kee,new_routes);
                }
                Vset& routes = clMap->getValue(x);
                routes.insert(rtx);
                return rtx;
}

void RouteTable::removeRoute(int rtx) {
                if (!validRtx(rtx)) return;
                Vset& lset = rteMap->getValue(rtx);
                for (int cLnk = lset.first(); cLnk != 0; cLnk = lset.next(cLnk)) {
                                uint64_t kee = cmKey(getComtree(rtx),cLnk);
                                Vset& routes = clMap->get(kee);
                                routes.remove(rtx);
                                if (routes.size() == 0) clMap->remove(kee);
                }
                lset.clear();
                rteMap->remove(rteMap->getKey(rtx));
}

void RouteTable::purge(comt_t comt, int cLnk) {
                uint64_t kee = cmKey(comt,cLnk);
        Vset& routes = clMap->get(kee);
                for (int rtx = routes.first(); rtx != 0; rtx = routes.next(rtx)) {
                                Vset& lset = rteMap->getValue(rtx);
                                lset.remove(cLnk);
                                if (lset.size() == 0)
                                                rteMap->remove(rteMap->getKey(rtx));
        }
        routes.clear();
                clMap->remove(kee);
}

bool RouteTable::readRoute(istream& in) {
                int comt, lnk; fAdr_t adr;
        Util::skipBlank(in);
        if (!Util::readInt(in, comt) || !Forest::readForestAdr(in,adr)) 
                return false;
                int rtx = addRoute(comt,adr,0);
                if (rtx == 0) return false;
                if (Forest::mcastAdr(adr)) {
                                do {
                                                if (!Util::readInt(in,lnk)) {
                                                                removeRoute(rtx); return false;
                                                }
                                                int cLnk = ctt->getClnkNum(comt,lnk);
                                                if (cLnk == 0) return false;
                                                addLink(rtx,cLnk);
                                } while (Util::verify(in,','));
                } else {
                                if (!Util::readInt(in,lnk)) {
                                                removeRoute(rtx); return false;
                                }
                                Util::nextLine(in);
                                int cLnk = ctt->getClnkNum(comt,lnk);
                                if (cLnk == 0) return false;
                                setLink(rtx,cLnk);
                }
                Util::nextLine(in);
                return true;
}

bool RouteTable::read(istream& in) {
                int num;
                Util::skipBlank(in);
                if (!Util::readInt(in,num)) return false;
                Util::nextLine(in);
                for (int i = 1; i <= num; i++) {
                                if (!readRoute(in)) {
                                                cerr << "Error in route table entry # " << i << endl;
                                                return false;
                                }
                }
        return true;
}

string RouteTable::entry2string(int rtx) const {
                stringstream ss;
                ss << getComtree(rtx) << " "
                   << Forest::fAdr2string(getAddress(rtx)) << " ";
                if (noLinks(rtx)) { ss << "-\n"; return ss.str(); }
                Vset& lset = rteMap->getValue(rtx);
                for (int cLnk = lset.first(); cLnk != 0; cLnk = lset.next(cLnk)) {
                                if (cLnk != lset.first()) ss << ",";
                                int ctx = ctt->getComtIndex(getComtree(rtx));
                                ss << ctt->getLink(ctx, cLnk);
                }
                ss << endl;
                return ss.str();
}

string RouteTable::toString() const {
                stringstream ss;
                ss << rteMap->size() << endl;
                for (int rtx = firstRtx(); rtx != 0; rtx = nextRtx(rtx))
                ss << entry2string(rtx);
                return ss.str();
}

} // ends namespace