00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026 #include <mln/core/image/image2d.hh>
00027 #include <mln/core/site_set/p_array.hh>
00028
00029 #include <mln/util/graph.hh>
00030
00031 #include <mln/value/label_8.hh>
00032
00033 #include <mln/make/p_edges_with_mass_centers.hh>
00034
00035 #include <mln/util/site_pair.hh>
00036
00037 int main()
00038 {
00039 using namespace mln;
00040
00041 typedef value::label_8 L;
00042
00043 L data[] = {
00044 1, 1, 1, 0, 2, 2, 2,
00045 1, 1, 1, 0, 2, 2, 2,
00046 1, 1, 1, 0, 2, 2, 2,
00047 0, 0, 0, 0, 0, 0, 0,
00048 3, 3, 3, 0, 4, 4, 4,
00049 3, 3, 3, 0, 4, 4, 4,
00050 3, 3, 3, 0, 4, 4, 4
00051 };
00052
00053 image2d<L> wst = make::image2d(data);
00054 typedef util::graph G;
00055 G g;
00056 g.add_vertices(5);
00057 g.add_edge(1,2);
00058 g.add_edge(1,3);
00059 g.add_edge(4,3);
00060 g.add_edge(4,2);
00061
00062 L nbasins = 4;
00063 typedef p_edges<G,fun::i2v::array<util::site_pair<point2d> > > pe_t;
00064 pe_t pe = make::p_edges_with_mass_centers(wst, g);
00065
00066 typedef util::site_pair<point2d> P;
00067 typedef p_array<P> arr_t;
00068 arr_t arr;
00069 arr.insert(P(point2d(1,1), point2d(1,5)));
00070 arr.insert(P(point2d(1,1), point2d(5,1)));
00071 arr.insert(P(point2d(5,1), point2d(5,5)));
00072 arr.insert(P(point2d(1,5), point2d(5,5)));
00073
00074 mln_piter_(pe_t) p1(pe);
00075 mln_piter_(arr_t) p2(arr);
00076 for_all_2(p1,p2)
00077 mln_assertion(p1 == p2);
00078 }