diff options
Diffstat (limited to 'src/brandes.cc')
| -rw-r--r-- | src/brandes.cc | 18 | 
1 files changed, 9 insertions, 9 deletions
diff --git a/src/brandes.cc b/src/brandes.cc index 5060a9d..7d12ec6 100644 --- a/src/brandes.cc +++ b/src/brandes.cc @@ -1,13 +1,13 @@ -#include <algorithm>  #include <fstream>  #include <iostream> -#include <thread> -#include <queue>  #include <mutex> +#include <queue> +#include <string> +#include <thread> -#include "parse.h" -#include "graph.h"  #include "dependency_calculator.h" +#include "graph.h" +#include "parse.h"  int threads;  std::string input_file; @@ -16,9 +16,10 @@ std::string output_file;  Graph graph;  std::unordered_map<int, double> betweenness;  std::queue<int> vertices_to_process; + +std::vector<std::thread> threads_list;  std::mutex queue_mutex;  std::mutex betweenness_mutex; -std::vector<std::thread> threads_list;  void parse_args(int argc, char *argv[]) {      if (argc < 4) { @@ -50,13 +51,12 @@ int next_vertex() {      if (vertices_to_process.empty()) {          return -1;      } -    int vertex; -    vertex = vertices_to_process.front(); +    int vertex = vertices_to_process.front();      vertices_to_process.pop();      return vertex;  } -void update_betweenness(int vertex, DependencyCalculator & dc) { +void update_betweenness(int vertex, DependencyCalculator& dc) {      std::lock_guard<std::mutex> lock(betweenness_mutex);      for (int v : graph.get_vertices()) {          if (v != vertex ) {  |