traffic-matrix.h

#ifndef TRAFFIC_MATRIX_H
#define TRAFFIC_MATRIX_H

#include "quantity.h"

#include <string>
#include <set>
#include <map>
#include <utility>

namespace fnss {

class TrafficMatrix {
public:
unsigned int size() const;

Quantity getFlow(const std::string &source, const std::string &destination) const;

Quantity getFlow(const std::pair<std::string, std::string> &nodes) const;

void setFlow(const std::string &source, const std::string &destination,
                                const Quantity &volume);

void setFlow(const std::pair<std::string, std::string> &nodes, const Quantity &volume);

std::set<std::pair<std::string, std::string> > getPairs() const;

private:
        typedef std::pair<std::string, std::string> matrixKeyType;
        typedef std::map<matrixKeyType, Quantity> matrixType;
        matrixType matrix;
};

} //namespace

#endif //TRAFFIC_MATRIX_H