watershed.h

Go to the documentation of this file.

/* Copyright (C) 2000-2013 CEA
 *
 * This software and supporting documentation were developed by
 *     bioPICSEL
 *     CEA/DSV/I²BM/MIRCen/LMN, Batiment 61,
 *     18, route du Panorama
 *     92265 Fontenay-aux-Roses
 *     France
 */

#ifndef BIOPROCESSING_WATERSHED_WATERSHED
#define BIOPROCESSING_WATERSHED_WATERSHED

//--- bioprocessing ----------------------------------------------------------
#include <bioprocessing/watershed/coustywatershed.h>  // Cousty implementation
//--- cartodata --------------------------------------------------------------
#include <cartodata/volume/volume.h>                       // carto::VolumeRef
//--- cartobase --------------------------------------------------------------
#include <cartobase/config/verbose.h>                        // carto::verbose
//--- std --------------------------------------------------------------------
#include <string>                                               // std::string
//----------------------------------------------------------------------------

namespace bio {

  template <typename T, typename L>
  class Watershed
  {
  public:
    Watershed():
      _implem("cousty"),
      _verbose(carto::verbose)
    {}

    Watershed( const std::string & implem ):
      _implem(implem),
      _verbose(carto::verbose)
    {}

    void setImplementation( const std::string & implem ) { _implem = implem; }
    void setVerbose( int verbose = 1 )  { _verbose = verbose; }
    void setQuiet()                     { setVerbose(0); }

    carto::VolumeRef<L> execute( carto::VolumeRef<T> in )
    {
      if( _implem == "cousty" )
      {
        CoustyWatershed<T,L> watershed;
        watershed.setVerbose( _verbose );
        if( in.getSizeX() == 1 )
          watershed.set2D();
        _watershed =  watershed.execute( in );
        _minima = watershed.getMinima();
        return _watershed;
      }
      else
      {
        std::cout << "unknown watershed implementation" << std::endl;
        throw;
      }
    }

    carto::VolumeRef<L> getMinima()
    {
      return _minima;
    }

  protected:
    std::string   _implem;
    int           _verbose;
    carto::VolumeRef<L> _watershed;
    carto::VolumeRef<L> _minima;
  };

} // namespace bio

#endif // BIOPROCESSING_WATERSHED_WATERSHED