superpose.hh

// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE)
//
// This file is part of Olena.
//
// Olena 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, version 2 of the License.
//
// Olena 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 Olena.  If not, see <http://www.gnu.org/licenses/>.
//
// As a special exception, you may use this file as part of a free
// software project without restriction.  Specifically, if other files
// instantiate templates or use macros or inline functions from this
// file, or you compile this file and link it with other files to produce
// an executable, this file does not by itself cause the resulting
// executable to be covered by the GNU General Public License.  This
// exception does not however invalidate any other reasons why the
// executable file might be covered by the GNU General Public License.

#ifndef MLN_MORPHO_WATERSHED_SUPERPOSE_HH
# define MLN_MORPHO_WATERSHED_SUPERPOSE_HH


# include <mln/core/concept/image.hh>
# include <mln/core/image/dmorph/image_if.hh>
# include <mln/data/convert.hh>
# include <mln/data/fill.hh>
# include <mln/value/rgb8.hh>
# include <mln/literal/colors.hh>
# include <mln/pw/all.hh>



namespace mln
{

  namespace morpho
  {

    namespace watershed
    {

      /*
      ** \param[in] input_    An image.
      ** \param[in] ws_ima_   The watershed transform of \p input_.
      ** \param[in] wsl_color The color used for the watershed line.
      */
      template <typename I, typename J>
      inline
      mln_ch_value(I,value::rgb8)
      superpose(const Image<I>& input_,
                const Image<J>& ws_ima_,
                const value::rgb8& wsl_color);


      /*
      ** \param[in] input_    An image.
      ** \param[in] ws_ima_   The watershed transform of \p input_.
      **
      ** The watershed line is drawn in red.
      */
      template <typename I, typename J>
      inline
      mln_ch_value(I,value::rgb8)
      superpose(const Image<I>& input,
                const Image<J>& ws_ima);


# ifndef MLN_INCLUDE_ONLY

      template <typename I, typename J>
      inline
      mln_ch_value(I,value::rgb8)
      superpose(const Image<I>& input_,
                const Image<J>& ws_ima_,
                const value::rgb8& wsl_color)
      {
        trace::entering("morpho::watershed::superpose");

        const I& input = exact(input_);
        const J& ws_ima = exact(ws_ima_);
        mln_precondition(input.is_valid());
        mln_precondition(ws_ima.is_valid());

        mln_ch_value(I,value::rgb8) output = data::convert(value::rgb8(), input);

        data::fill((output | (pw::value(ws_ima) == pw::cst(literal::zero))).rw(),
                   wsl_color);

        trace::exiting("morpho::watershed::superpose");
        return output;
      }


      template <typename I, typename J>
      inline
      mln_ch_value(I,value::rgb8)
      superpose(const Image<I>& input,
                const Image<J>& ws_ima)
      {
        return superpose(input, ws_ima, literal::red);
      }

# endif // ! MLN_INCLUDE_ONLY


    } // end of namespace mln::morpho::watershed

  } // end of namespace mln::morpho

} // end of namespace mln

#endif // ! MLN_MORPHO_WATERSHED_SUPERPOSE_HH