graph_katz.cc 2.54 KB
Newer Older
Tiago Peixoto's avatar
Tiago Peixoto committed
1
2
// graph-tool -- a general graph modification and manipulation thingy
//
Tiago Peixoto's avatar
Tiago Peixoto committed
3
// Copyright (C) 2006-2015 Tiago de Paula Peixoto <tiago@skewed.de>
Tiago Peixoto's avatar
Tiago Peixoto committed
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 3
// of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

#include "graph_filtering.hh"

#include <boost/python.hpp>

#include "graph.hh"
#include "graph_selectors.hh"
#include "graph_katz.hh"

using namespace std;
using namespace boost;
using namespace graph_tool;

void katz(GraphInterface& g, boost::any w, boost::any c, boost::any beta,
          long double alpha, double epsilon, size_t max_iter)
{
    if (!w.empty() && !belongs<writable_edge_scalar_properties>()(w))
        throw ValueException("edge property must be writable");
    if (!belongs<vertex_floating_properties>()(c))
        throw ValueException("centrality vertex property must be of floating point"
                             " value type");
    if (!beta.empty() && !belongs<vertex_floating_properties>()(beta))
        throw ValueException("personalization vertex property must be of floating point"
                             " value type");

42
    typedef ConstantPropertyMap<double, GraphInterface::edge_t> weight_map_t;
Tiago Peixoto's avatar
Tiago Peixoto committed
43
    typedef boost::mpl::push_back<writable_edge_scalar_properties, weight_map_t>::type
Tiago Peixoto's avatar
Tiago Peixoto committed
44
45
46
        weight_props_t;

    if(w.empty())
47
        w = weight_map_t(1.);
Tiago Peixoto's avatar
Tiago Peixoto committed
48

49
    typedef ConstantPropertyMap<double, GraphInterface::vertex_t> beta_map_t;
Tiago Peixoto's avatar
Tiago Peixoto committed
50
    typedef boost::mpl::push_back<vertex_floating_properties, beta_map_t>::type
Tiago Peixoto's avatar
Tiago Peixoto committed
51
52
53
        beta_props_t;

    if(beta.empty())
54
        beta = beta_map_t(1.);
Tiago Peixoto's avatar
Tiago Peixoto committed
55

56
57
58
    run_action<>()(g, std::bind(get_katz(), std::placeholders::_1, g.get_vertex_index(),
                                std::placeholders::_2, std::placeholders::_3,
                                std::placeholders::_4, alpha, epsilon, max_iter),
Tiago Peixoto's avatar
Tiago Peixoto committed
59
60
61
62
63
64
65
66
67
68
                   weight_props_t(),
                   vertex_floating_properties(),
                   beta_props_t())(w, c, beta);
}

void export_katz()
{
    using namespace boost::python;
    def("get_katz", &katz);
}