00001 // Copyright (C) 2007, 2008, 2009 EPITA Research and Development Laboratory (LRDE) 00002 // 00003 // This file is part of Olena. 00004 // 00005 // Olena is free software: you can redistribute it and/or modify it under 00006 // the terms of the GNU General Public License as published by the Free 00007 // Software Foundation, version 2 of the License. 00008 // 00009 // Olena is distributed in the hope that it will be useful, 00010 // but WITHOUT ANY WARRANTY; without even the implied warranty of 00011 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 00012 // General Public License for more details. 00013 // 00014 // You should have received a copy of the GNU General Public License 00015 // along with Olena. If not, see <http://www.gnu.org/licenses/>. 00016 // 00017 // As a special exception, you may use this file as part of a free 00018 // software project without restriction. Specifically, if other files 00019 // instantiate templates or use macros or inline functions from this 00020 // file, or you compile this file and link it with other files to produce 00021 // an executable, this file does not by itself cause the resulting 00022 // executable to be covered by the GNU General Public License. This 00023 // exception does not however invalidate any other reasons why the 00024 // executable file might be covered by the GNU General Public License. 00025 00028 00029 #include <vector> 00030 00031 #include <mln/core/alias/point2d.hh> 00032 00034 #include <mln/core/image/vertex_image.hh> 00035 #include <mln/core/var.hh> 00036 #include <mln/fun/i2v/array.hh> 00037 #include <mln/util/graph.hh> 00038 #include <mln/make/vertex_image.hh> 00039 00040 #include <mln/morpho/meyer_wst.hh> 00041 00042 00043 int main() 00044 { 00045 using namespace mln; 00046 00047 /*--------. 00048 | Graph. | 00049 `--------*/ 00050 00051 /* The graph is as follows: 00052 00053 0 1 2 3 4 00054 .----------- 00055 | 00056 0 | 0 2 00057 1 | \ / | 00058 2 | 1 | 00059 3 | \ | 00060 4 | 3-0 00061 00062 */ 00063 00064 // Points associated to vertices. 00065 typedef fun::i2v::array<point2d> fsite_t; 00066 fsite_t sites(5); 00067 sites(0) = point2d(0,0); // Point associated to vertex 0. 00068 sites(1) = point2d(2,2); // Point associated to vertex 1. 00069 sites(2) = point2d(0,4); // Point associated to vertex 2. 00070 sites(3) = point2d(4,3); // Point associated to vertex 3. 00071 sites(4) = point2d(4,4); // Point associated to vertex 4. 00072 00073 // Graph. 00074 util::graph g; 00075 00076 // Populate the graph with vertices. 00077 g.add_vertices(sites.size()); 00078 00079 // Populate the graph with edges. 00080 g.add_edge(0, 1); 00081 g.add_edge(1, 2); 00082 g.add_edge(1, 3); 00083 g.add_edge(3, 4); 00084 g.add_edge(4, 2); 00085 00086 /*-------------. 00087 | Graph image. | 00088 `-------------*/ 00089 00090 // Graph values. 00091 typedef fun::i2v::array<unsigned> viota_t; 00092 viota_t iota(g.v_nmax()); 00093 for (unsigned i = 0; i < iota.size(); ++i) 00094 iota(i) = 10 + i; 00095 00096 typedef vertex_image<point2d, unsigned, util::graph> ima_t; 00097 ima_t ima = make::vertex_image(g, sites, iota); 00098 00099 /*------. 00100 | WST. | 00101 `------*/ 00102 00103 typedef ima_t::nbh_t nbh_t; 00104 nbh_t nbh; 00105 00106 unsigned nbasins; 00107 ima_t wshed = morpho::meyer_wst(ima, nbh, nbasins); 00108 std::cout << "nbasins = " << nbasins << std::endl; 00109 00110 // Manual iteration over the domain of WSHED. 00111 mln_piter_(ima_t) pw(wshed.domain()); 00112 for_all (pw) 00113 std::cout << "wshed (" << pw << ") = " << wshed(pw) << std::endl; 00114 }