graph.h

Go to the documentation of this file.

#ifndef _GRAPH_H_
#define _GRAPH_H_

#include <cstdlib>
#include <fstream>
#include <iostream>
#include <iomanip>
#include <set>
#include <sstream>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>

#include "lab_graphs_random.h"
#include "edge.h"

typedef std::unordered_map<Vertex, Edge> EdgeMap;
typedef std::unordered_map<Vertex, EdgeMap> VertexMap;
typedef std::unordered_map<Vertex, std::string> VertexLabelMap;
typedef std::unordered_map<Vertex, std::string> VertexNameMap;

using std::cerr;
using std::cout;
using std::endl;
using std::vector;
using std::set;
using std::string;
using std::vector;
using std::pair;
using std::make_pair;

class Graph
{
    public:
        Graph(bool isWeighted);

        Graph(bool isWeighted, int numVertices, unsigned long seed);
        
        vector<Vertex> getAdjacent(Vertex v) const;

        Vertex getStartingVertex() const;
        
        vector<Vertex> getVertices() const;
        
        Edge getEdge(Vertex u, Vertex v) const;

        vector<Edge> getEdges() const;

        void setVertexName(Vertex v, string name);

        string getVertexName(Vertex v) const;

        void setVertexLabel(Vertex v, string label);

        string getVertexLabel(Vertex v) const;
        
        void setEdgeLabel(Vertex u, Vertex v, string label);

        string getEdgeLabel(Vertex u, Vertex v) const;

        int getEdgeWeight(Vertex u, Vertex v) const;
        
        Vertex insertVertex(string label = "");

        void removeVertex(Vertex v);
        
        bool insertEdge(Vertex u, Vertex v);

        void removeEdge(Vertex u, Vertex v);
        
        void setEdgeWeight(Vertex u, Vertex v, int weight);

        void initSnapshot(string title);

        void snapshot();

        void print() const;

        void savePNG(string title) const;
        
    private:

        VertexMap graph;

        VertexLabelMap vertexLabels;

        VertexNameMap vertexNames;

        Random random;

        bool weighted;
        int picNum;
        string picName;
        size_t vertexCounter;

        Vertex nextVertex();

        void assertExists(Vertex v, string functionName) const;

        void assertConnected(Vertex u, Vertex v, string functionName) const;

        void error(string message) const;
};

#endif