45 lines
1.3 KiB
C++
45 lines
1.3 KiB
C++
#import "CostMatrixGenerator.h"
|
|
#import "MakePermutationMatrix.h"
|
|
#import "ReadFromFile.h"
|
|
#import "mpi.h"
|
|
|
|
#include <iostream>
|
|
#define MCW MPI_COMM_WORLD
|
|
int main(int argc, char* argv[])
|
|
{
|
|
int rank, size, citiesSize;
|
|
int data;
|
|
MPI_Init(&argc, &argv);
|
|
MPI_Comm_rank(MCW, &rank);
|
|
MPI_Comm_size(MCW, &size);
|
|
std::vector<std::vector<double>> matrix;
|
|
std::vector<City> cities;
|
|
if (rank == 0) {
|
|
std::cout << "Reading in file" << std::endl;
|
|
cities = ReadFromFile::ReadFile("../input");
|
|
citiesSize = cities.size();
|
|
//Make matrix
|
|
matrix = CostMatrixGenerator::GenerateCostMatrix(cities);
|
|
}
|
|
MPI_Bcast(&citiesSize, 1, MPI_INT, 0, MPI_COMM_WORLD);
|
|
if(!rank){
|
|
// reserve memory for vectors
|
|
matrix.resize(citiesSize);
|
|
for(int i = 0; i < citiesSize; i++) {
|
|
matrix[i].resize(citiesSize);
|
|
}
|
|
}
|
|
MPI_Bcast(&matrix[0][0], citiesSize*citiesSize, MPI_FLOAT, 0, MPI_COMM_WORLD);
|
|
|
|
if(!rank){
|
|
for(int i = 0; i < citiesSize; i++){
|
|
for(int j = 0; j < citiesSize; j++) {
|
|
std::cout<<"Process " << rank << " has value " << matrix[i][j];
|
|
}
|
|
}
|
|
}
|
|
|
|
// //Make permutations
|
|
// auto perms = MakePermutationMatrix::GetLowestCost(cities, rank, size, matrix);
|
|
return 0;
|
|
} |