C++ sgraph::get_vertex_field_id方法代码示例

void triple_apply_pagerank(sgraph& g, size_t& num_iter, double& total_pagerank, double& total_delta) {
  typedef sgraph_compute::sgraph_engine<flexible_type>::graph_data_type graph_data_type;
  typedef sgraph::edge_direction edge_direction;

  // initialize every vertex with core id kmin
  g.init_vertex_field(PAGERANK_COLUMN, reset_probability);
  g.init_vertex_field(PREV_PAGERANK_COLUMN, 1.0);
  g.init_vertex_field(DELTA_COLUMN, 0.0);

  // Initialize degree count
  sgraph_compute::sgraph_engine<flexible_type> ga;
  auto degrees = ga.gather(
          g,
          [=](const graph_data_type& center,
              const graph_data_type& edge,
              const graph_data_type& other,
              edge_direction edgedir,
              flexible_type& combiner) {
              combiner += 1;
          },
          flexible_type(0),
          edge_direction::OUT_EDGE);
  g.add_vertex_field(degrees, OUT_DEGREE_COLUMN);

  num_iter = 0;
  total_delta = 0.0;
  total_pagerank = 0.0;
  timer mytimer;

  // Triple apply
  double w = (1 - reset_probability);
  const size_t degree_idx = g.get_vertex_field_id(OUT_DEGREE_COLUMN);
  const size_t pr_idx = g.get_vertex_field_id(PAGERANK_COLUMN);
  const size_t old_pr_idx = g.get_vertex_field_id(PREV_PAGERANK_COLUMN);

  sgraph_compute::triple_apply_fn_type apply_fn =
    [&](sgraph_compute::edge_scope& scope) {
      auto& source = scope.source();
      auto& target = scope.target();
      scope.lock_vertices();
      target[pr_idx] += w * source[old_pr_idx] / source[degree_idx];
      scope.unlock_vertices();
    };

  table_printer table({{"Iteration", 0}, 
                                {"L1 change in pagerank", 0}});
  table.print_header();

  for (size_t iter = 0; iter < max_iterations; ++iter) {
    if(cppipc::must_cancel()) {
      log_and_throw(std::string("Toolkit cancelled by user."));
    }

    mytimer.start();
    ++num_iter;

    g.init_vertex_field(PAGERANK_COLUMN, reset_probability);

    sgraph_compute::triple_apply(g, apply_fn, {PAGERANK_COLUMN});

    // compute the change in pagerank
    auto delta = sgraph_compute::vertex_apply(
        g,
        flex_type_enum::FLOAT,
        [&](const std::vector<flexible_type>& vdata) {
          return std::abs((double)(vdata[pr_idx]) - (double)(vdata[old_pr_idx]));
        });

    // make the current pagerank the old pagerank
    g.copy_vertex_field(PAGERANK_COLUMN, PREV_PAGERANK_COLUMN);
    g.replace_vertex_field(delta, DELTA_COLUMN);

    total_delta = 
        sgraph_compute::vertex_reduce<double>(g, 
                               DELTA_COLUMN,
                               [](const flexible_type& v, double& acc) {
                                 acc += (flex_float)v;
                               },
                               [](const double& v, double& acc) {
                                 acc += v;
                               });


    table.print_row(iter+1, total_delta);

    // check convergence
    if (total_delta < threshold) {
      break;
    }
  } // end of pagerank iterations

  table.print_footer();

  // cleanup
  g.remove_vertex_field(PREV_PAGERANK_COLUMN);
  g.remove_vertex_field(OUT_DEGREE_COLUMN);
  total_pagerank =
      sgraph_compute::vertex_reduce<double>(g, 
                                     PAGERANK_COLUMN,
                                     [](const flexible_type& v, double& acc) {
//.........这里部分代码省略.........

/**
 * We start with every vertex having core_id = KMAX,
 * Each iteration, while the gather will +1 for neighbors whose core_id > CURRENT_K 
 * If the gather is > 0 and <= CURRENT_K, then we set the core_id to CURRENT_K (indicate its deleted).
 * And repeat...
 */
void triple_apply_kcore(sgraph& g) {
  typedef sgraph_compute::sgraph_engine<flexible_type>::graph_data_type graph_data_type;
  typedef sgraph::edge_direction edge_direction;

  // initialize every vertex with core id kmin
  g.init_vertex_field(CORE_ID_COLUMN, KMIN);
  g.init_vertex_field(DEGREE_COLUMN, 0);
  g.init_vertex_field(DELETED_COLUMN, 0);
  g.init_edge_field(DELETED_COLUMN, 0);

  // Initialize degree count
  sgraph_compute::sgraph_engine<flexible_type> ga;
  auto degrees = ga.gather(
          g,
          [=](const graph_data_type& center,
              const graph_data_type& edge,
              const graph_data_type& other,
              edge_direction edgedir,
              flexible_type& combiner) {
              combiner += 1;
          },
          flexible_type(0),
          edge_direction::ANY_EDGE);
  g.replace_vertex_field(degrees, DEGREE_COLUMN);

  // Initialize fields
  long vertices_left = g.num_vertices();
  std::atomic<long> num_vertices_changed;
  const size_t core_idx = g.get_vertex_field_id(CORE_ID_COLUMN);
  const size_t degree_idx = g.get_vertex_field_id(DEGREE_COLUMN);
  const size_t v_deleted_idx= g.get_vertex_field_id(DELETED_COLUMN);
  const size_t e_deleted_idx= g.get_edge_field_id(DELETED_COLUMN);

  // Triple apply
  sgraph_compute::triple_apply_fn_type apply_fn =
  [&](sgraph_compute::edge_scope& scope) {
    auto& source = scope.source();
    auto& target = scope.target();
    auto& edge = scope.edge();
    scope.lock_vertices();
    // edge is not deleted
    if (!edge[e_deleted_idx]) {
      // check source degree
      if (!source[v_deleted_idx] && source[degree_idx] <= CURRENT_K) {
        source[core_idx] = CURRENT_K;
        source[v_deleted_idx] = 1;
        num_vertices_changed++;
      }
      // check target degree
      if (!target[v_deleted_idx] && target[degree_idx] <= CURRENT_K) {
        target[core_idx] = CURRENT_K;
        target[v_deleted_idx] = 1;
        num_vertices_changed ++;
      }
      // delete the edge if either side is deleted
      if (source[v_deleted_idx] || target[v_deleted_idx]) {
        edge[e_deleted_idx] = 1;
        --source[degree_idx];
        --target[degree_idx];
        // We need to check again if the deletion of this edge 
        // causing either source or target vertex to be deleted.
        if (!source[v_deleted_idx] && source[degree_idx] <= CURRENT_K) {
          source[core_idx] = CURRENT_K;
          source[v_deleted_idx] = 1;
          num_vertices_changed++;
        }
        // check target degree
        if (!target[v_deleted_idx] && target[degree_idx] <= CURRENT_K) {
          target[core_idx] = CURRENT_K;
          target[v_deleted_idx] = 1;
          num_vertices_changed++;
        }
      }
    }
    scope.unlock_vertices();
  };

  for (CURRENT_K = KMIN; CURRENT_K < KMAX; ++CURRENT_K) {
    while (true) {
      if(cppipc::must_cancel()) {
        log_and_throw(std::string("Toolkit cancelled by user."));
      }
      num_vertices_changed = 0;
      sgraph_compute::triple_apply(g, apply_fn, {CORE_ID_COLUMN, DEGREE_COLUMN, DELETED_COLUMN}, {DELETED_COLUMN});
      if (num_vertices_changed == 0)
        break;
      vertices_left -= num_vertices_changed;
      if (CURRENT_K == 0 || num_vertices_changed == 0 || vertices_left == 0) {
        // we are done with the current core.
        break;
      }
      ASSERT_GT(vertices_left, 0);
    }
    logprogress_stream << "Finish computing core " << CURRENT_K << "\t Vertices left: "
//.........这里部分代码省略.........