URL: https://svn.lrde.epita.fr/svn/oln/trunk/milena/sandbox ChangeLog: 2009-05-15 Edwin Carlinet <carlinet@lrde.epita.fr> Move inim project to sandbox/inim. * edwin/inim/Makefile: Remove. * edwin/inim/inim.cc: Remove. * inim/2011/fredwin/Makefile: New. * inim/2011/fredwin/configure: New. * inim/2011/fredwin/inim.cc: New. --- trunk/milena/sandbox/inim/2011/fredwin/Makefile | 37 +++ trunk/milena/sandbox/inim/2011/fredwin/configure | 21 + trunk/milena/sandbox/inim/2011/fredwin/inim.cc | 273 +++++++++++++++++++++++ 3 files changed, 331 insertions(+) Index: trunk/milena/sandbox/inim/2011/fredwin/inim.cc =================================================================== --- trunk/milena/sandbox/inim/2011/fredwin/inim.cc (revision 0) +++ trunk/milena/sandbox/inim/2011/fredwin/inim.cc (revision 3800) @@ -0,0 +1,273 @@ +/* mln core */ +#include <mln/core/image/image2d.hh> +#include <mln/core/image/image_if.hh> +#include <mln/core/alias/neighb2d.hh> +#include <mln/core/alias/window2d.hh> +#include <mln/core/alias/w_window2d_int.hh> +#include <mln/core/routine/duplicate.hh> +#include <mln/core/var.hh> + +/* mln value */ +#include <mln/value/int_u8.hh> +#include <mln/value/int_u16.hh> + +/* Site set */ +#include <mln/core/site_set/p_array.hh> +#include <mln/level/sort_psites.hh> + +/* Component trees */ +#include <mln/morpho/tree/data.hh> +#include <mln/morpho/tree/compute_attribute_image.hh> +#include <mln/morpho/tree/propagate.hh> +#include <mln/morpho/tree/components.hh> +#include <mln/morpho/tree/filter.hh> + +/* Attributes */ +#include <mln/transform/distance_front.hh> +#include <mln/morpho/attribute/bbox.hh> +#include <mln/morpho/attribute/card.hh> +#include <mln/morpho/attribute/height.hh> +#include <mln/make/w_window2d_int.hh> + +/* io */ +#include <mln/io/pbm/load.hh> +#include <mln/io/pgm/save.hh> +#include <mln/io/ppm/save.hh> + +/* data & pw */ +#include <mln/core/concept/function.hh> +#include <mln/fun/p2v/ternary.hh> +#include <mln/data/fill.hh> +#include <mln/data/paste.hh> +#include <mln/pw/all.hh> + +/* labeling */ +#include <mln/level/stretch.hh> +#include <mln/value/label_8.hh> +#include <mln/labeling/blobs.hh> +#include <mln/labeling/colorize.hh> + +/* Draw debug */ +#include <mln/draw/box.hh> +#include <mln/literal/colors.hh> +#include <mln/level/convert.hh> + +/* std */ +#include <string> +#include <iostream> +#include <cmath> + +bool mydebug = false; + +void usage(char** argv) +{ + std::cerr << "usage: " << argv[0] << " input [--debug] [-n nbr_components | -t treshold] [-h height]" << std::endl; + abort(); +} + +void dsp(const std::string& str) +{ + std::cout << std::endl + << "*********************" << std::endl + << "** " << str << std::endl + << "*********************" << std::endl; +} + +template <typename P2V, typename G> +struct ratio_ : public mln::Function_p2v< ratio_<P2V, G> > +{ + typedef double result; + + ratio_(const P2V& f, const G& g) : + f_ (f), g_ (g) + { + } + + template <typename P> + double operator() (const P& p) const + { + mln_VAR(box, f_(p)); + double a = (double) (box.len(1)) / (double)(box.len(0)); + return a * std::log(g_(p)); + } + +protected: + const P2V& f_; + const G& g_; +}; + +template <typename P2V, typename G> +ratio_<P2V, G> ratio(const mln::Function_p2v<P2V>& f, const mln::Function_p2v<G>& g) +{ + return ratio_<P2V, G>(exact(f), exact(g)); +} + + + + +int main(int argc, char* argv[]) +{ + using namespace mln; + using value::int_u8; + using value::int_u16; + + std::string arg; + unsigned nb_components = 0; + unsigned height = 15; + double treshold = 0; + + if (argc < 2) + usage(argv); + + for (int i = 2; i < argc; i++) + { + arg = std::string(argv[i]); + if (arg == "--debug") + mydebug = true; + else if (arg == "-n" && i != argc) + nb_components = atoi(argv[++i]); + else if (arg == "-t" && i != argc) + treshold = atof(argv[++i]); + else if (arg == "-h" && i != argc) + height = atoi(argv[++i]); + else if (arg == "--trace") + trace::quiet = false; + else + usage (argv); + } + + + /* Image loadin' */ + typedef image2d<int_u8> I; + + image2d<bool> input_; + io::pbm::load(input_, argv[1]); + + if (mydebug) { + dsp("Distance geodesic"); + } + + /* Work on geodesic distance image */ + I input; + { + const int weights[9] = + { 10, 9, 10, + 1, 0, 1, + 10, 9, 10 }; + + w_window2d_int win; + mln::convert::from_to(weights, win); + input = transform::distance_front(input_, c8(), win, mln_max(int_u8)); + } + + if (mydebug) { + dsp("Component tree computation"); + } + + /* Component tree creation */ + typedef p_array< mln_site_(I) > S; + typedef morpho::tree::data<I,S> tree_t; + + S sorted_sites = level::sort_psites_decreasing(input); + tree_t tree(input, sorted_sites, c4()); + + + if (mydebug) { + dsp("Attribute image computation"); + } + + /* Compute Attribute On Image */ + typedef morpho::attribute::bbox<I> bbox_t; + typedef morpho::attribute::card<I> card_t; + typedef morpho::attribute::height<I> height_t; + typedef mln_ch_value_(I, double) A; + + mln_VAR(attr_image, morpho::tree::compute_attribute_image(bbox_t (), tree)); + mln_VAR(card_image, morpho::tree::compute_attribute_image(card_t (), tree)); + mln_VAR(height_image, morpho::tree::compute_attribute_image(height_t (), tree)); + + A a = duplicate(ratio(pw::value(attr_image), pw::value(card_image)) | attr_image.domain()); + + if (height) + morpho::tree::filter::filter(tree, a, pw::value(height_image) > pw::cst(height), 0.0); + + /* Run max accumulator */ + p_array< mln_psite_(A) > obj_array; // Array of object components. + + if (mydebug) { + std::stringstream s("Run max accumulator, look for ", + std::stringstream::out|std::stringstream::in| + std::stringstream::ate); + if (nb_components) + s << nb_components << " components."; + else if (treshold) + s << "components whose treshold > " << treshold; + else + s << "components util leaves are glutted"; + dsp(s.str()); + } + + if (nb_components) { + obj_array = morpho::tree::get_components(tree, a, nb_components); + } else if (treshold > 0) { + mln_VAR(predicate, pw::value(a) > pw::cst(treshold)); + obj_array = morpho::tree::get_components(tree, a, predicate); + } else { + obj_array = morpho::tree::get_components(tree, a); + } + + /* Print them */ + if (mydebug) { + dsp("Image Filtered Components"); + mln_fwd_piter_(p_array< mln_psite_(I) >) c(obj_array); + for_all(c) + std::cout << c; + } + + + if (mydebug) { + dsp("Create mask and propagate"); + } + + /* Now Back Propagate to component */ + typedef mln_ch_value_(I, bool) M; + M mask = morpho::tree::set_value_to_components(tree, obj_array, true, false); + + a = morpho::tree::propagate_components(a, tree, obj_array, 0); + + /* Labeling */ + typedef mln_ch_value_(I, value::label<8>) L; + typedef mln_ch_value_(I, value::rgb<8>) O; + value::label_8 nlabel; + L label = labeling::blobs(mask, c4(), nlabel); + O output = labeling::colorize(value::rgb8(), label, nlabel); + if (mydebug) + io::ppm::save(output, "label.pgm"); + + /* Now store output image image */ + O out, distance; + initialize(out, input); + distance = level::convert(value::rgb8 (), input); + data::fill(out, literal::black_t()); + data::paste(output | pw::value(input_), out); + + { + mln_piter_(p_array<mln_psite_(I)>) it(obj_array); + for_all(it) + { + std::cout << it << " :: " << attr_image(it).pmin() << " -> " << a(it) + << " (h: " << height_image(it) << ")" + << std::endl; + + draw::box(out, attr_image(it), literal::red_t ()); + draw::box(distance, attr_image(it), literal::red_t ()); + } + } + + if (mydebug) { + dsp("Mask input"); + } + + io::ppm::save(distance, "distance.pgm"); + io::ppm::save(out, "output.pgm"); +} Property changes on: trunk/milena/sandbox/inim/2011/fredwin/inim.cc ___________________________________________________________________ Added: svn:mergeinfo Index: trunk/milena/sandbox/inim/2011/fredwin/configure =================================================================== --- trunk/milena/sandbox/inim/2011/fredwin/configure (revision 0) +++ trunk/milena/sandbox/inim/2011/fredwin/configure (revision 3800) @@ -0,0 +1,21 @@ +#! /bin/bash + +for arg in "$@"; do + if [ "$arg" == "--debug" ]; then + debug=1; + elif [ "$arg" == "--release" ]; then + release=1; + elif [ "${arg#--mln_dir=}" ]; then + MLN_DIR=${arg#--mln_dir=} + fi +done + +rm -f makefile.rules +touch makefile.rules +if [ "$debug" ]; then + echo 'DEBUG = 1' >> makefile.rules +fi +if [ "$release" ]; then + echo 'RELEASE = 1' >> makefile.rules +fi +echo "MLN_DIR = $MLN_DIR" >> makefile.rules \ No newline at end of file Property changes on: trunk/milena/sandbox/inim/2011/fredwin/configure ___________________________________________________________________ Added: svn:executable + * Index: trunk/milena/sandbox/inim/2011/fredwin/Makefile =================================================================== --- trunk/milena/sandbox/inim/2011/fredwin/Makefile (revision 0) +++ trunk/milena/sandbox/inim/2011/fredwin/Makefile (revision 3800) @@ -0,0 +1,37 @@ +include makefile.rules +TARGET=inim + +inim_SRC=inim.cc +inim_OBJS=${inim_SRC:.cc=.o} + +OLENADIR=$(MLN_DIR)/.. +MILENADIR=$(OLENADIR)/milena + +CXXFLAGS=-I$(MILENADIR) -I$(MILENADIR)/sandbox/edwin -W -Wall + + +CXXFLAGS += $(if $(DEBUG), -g -ggdb, -DNDEBUG\ + $(if $(RELEASE), -O3, -O1)) + +CXX=g++ +LD=g++ +LDFLAGS= + +all: $(TARGET) + + +$(TARGET): $($(TARGET)_OBJS) $($(TARGET)_SRC) + $(LD) $(LDFLAGS) -o $@ $($(TARGET)_OBJS) + +%.o: %.cc + $(CXX) $(CXXFLAGS) -c $< + + +depend: + makedepend -- $(CXXFLAGS) -- -v $($(TARGET)_SRC) + +clean: + rm -f *.o $(TARGET) + #rm -f *.pbm + #find -name "*.pgm" \! -regex ".*/affiche2?.pgm" -delete +# DO NOT DELETE Property changes on: trunk/milena/sandbox/inim/2011/fredwin/Makefile ___________________________________________________________________ Added: svn:mergeinfo Index: trunk/milena/sandbox/edwin/inim/inim.cc (deleted) =================================================================== Index: trunk/milena/sandbox/edwin/inim/Makefile (deleted) ===================================================================