#include "representation.h"
#include <cstdlib>

representation::representation(int vertex, int max_weight, int probability)
{
	vertexcount = vertex;
	matrix = new int*[vertex];
	for (int i=0; i<vertex; i++)
		matrix[i] = new int[vertex];

	for (i=0; i<vertex; i++)
	{
		for (int j=0; j<vertex; j++)
			matrix[i][j] = 0;
	}

	// Total edges possible
	int t_edge = vertex * (vertex - 1);
	// Total edges possible with edge probability taken into effect
	int t_poss = t_edge * (probability / 180);
	for (i=0; i<=t_poss; i++)
	{	
		int picker2 = rand()%int(vertex);						// The following two lines generate which
		int picker1 = rand()%int(vertex-picker2)+picker2;		//  vertex's we are making an edge for

		matrix[picker1][picker2] = rand()%max_weight;				// generate the weight for this edge
		matrix[picker2][picker1] = matrix[picker1][picker2];	// create the reverse edge weight

		if (picker1 == picker2)										// ensure we don't allow a self edge
		{
			matrix[picker1][picker2] = 0;
			matrix[picker2][picker1] = 0;
			i--;
		}
	}
}

representation::~representation()
{
	for (int i = 0; i < vertexcount; i++)
		delete []matrix[i];
	delete matrix;

	return;
}

bool representation::isEdge(int source, int destination)
{
	return ((this->matrix[source][destination] != 0) && (source != destination));
}

int representation::getWeight(int source, int destination)
{
	return (this->matrix[source][destination]);
}