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
00027 #include <iostream>
00028
00029 #include <mln/core/image/image2d.hh>
00030 #include <mln/core/alias/window2d.hh>
00031 #include <mln/core/alias/neighb2d.hh>
00032
00033 #include <mln/value/int_u8.hh>
00034 #include <mln/value/int_u16.hh>
00035
00036 #include <mln/morpho/watershed/flooding.hh>
00037 #include <mln/data/transform.hh>
00038
00039 #include <mln/io/pgm/load.hh>
00040 #include <mln/io/pgm/save.hh>
00041
00042 #include <mln/util/timer.hh>
00043
00044 #include "tests/data.hh"
00045
00046 struct f_16_to_8 : mln::Function_v2v< f_16_to_8 >
00047 {
00048 typedef mln::value::int_u8 result;
00049 result operator()(const mln::value::int_u16& v) const
00050 {
00051 if (v == 0)
00052 return 0;
00053 return 1 + (v - 1) % 255;
00054 }
00055 };
00056
00057
00058 int main()
00059 {
00060 using namespace mln;
00061 using value::int_u8;
00062 using value::int_u16;
00063
00064 image2d<int_u8> input;
00065 io::pgm::load(input, MLN_IMG_DIR "/squares.pgm");
00066
00067 typedef int_u16 L;
00068 L n_basins;
00069 {
00070 util::timer t;
00071 t.start();
00072 image2d<L> output =
00073 morpho::watershed::impl::generic::flooding(input, c4(), n_basins);
00074 std::cout << "gen: " << t << std::endl;
00075 io::pgm::save(data::transform(output, f_16_to_8()),
00076 "flooding-tmp_ref.pgm");
00077 }
00078 {
00079 util::timer t;
00080 t.start();
00081 image2d<L> output =
00082 morpho::watershed::impl::flooding_fastest(input, c4(), n_basins);
00083 std::cout << "fast: " << t << std::endl;
00084 io::pgm::save(data::transform(output, f_16_to_8()),
00085 "flooding-tmp_out.pgm");
00086 }
00087 }