graph_union.cc 1.95 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) 2007-2012 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
//
// 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.hh"
#include "graph_filtering.hh"

#include "graph_union.hh"

#include <boost/bind.hpp>

#include <boost/python/extract.hpp>

using namespace graph_tool;
using namespace boost;

30
typedef property_map_type::apply<int32_t,
Tiago Peixoto's avatar
Tiago Peixoto committed
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
                                 GraphInterface::vertex_index_map_t>::type
    vprop_t;

typedef property_map_type::apply<GraphInterface::edge_t,
                                 GraphInterface::edge_index_map_t>::type
    eprop_t;

struct get_pointers
{
    template <class List>
    struct apply
    {
        typedef typename mpl::transform<List,
                                        mpl::quote1<add_pointer> >::type type;
    };
};

48 49
python::tuple graph_union(GraphInterface& ugi, GraphInterface& gi,
                          boost::any avprop)
Tiago Peixoto's avatar
Tiago Peixoto committed
50
{
51
    vprop_t vprop = boost::any_cast<vprop_t>(avprop);
Tiago Peixoto's avatar
Tiago Peixoto committed
52 53 54 55 56 57
    eprop_t eprop(gi.GetEdgeIndex());
    run_action<graph_tool::detail::always_directed,mpl::true_>()
        (ugi, bind<void>(graph_tool::graph_union(),
                         _1, _2, vprop, eprop),
         get_pointers::apply<graph_tool::detail::always_directed>::type())
        (gi.GetGraphView());
58
    return python::make_tuple(avprop, boost::any(eprop));
Tiago Peixoto's avatar
Tiago Peixoto committed
59
}