vil_region_finder.hxx

Go to the documentation of this file.

#ifndef vil_region_finder_hxx_
#define vil_region_finder_hxx_

#include "vil_region_finder.h"
#include <cassert>
#ifdef _MSC_VER
#  include <vcl_msvc_warnings.h>
#endif

// ---------------------------------------------------------------------------
//                                                                 constructor

template <class pix_type, class predicate_type>
vil_region_finder<pix_type, predicate_type>::
vil_region_finder( image_view const& in_image,
                   vil_region_finder_connectivity conn )
  : image_( in_image ),
    processed_( image_.ni(), image_.nj(), image_.nplanes() )
{
  init( conn );
}


// ---------------------------------------------------------------------------
//                                                                        init

template <class pix_type, class predicate_type>
void
vil_region_finder<pix_type, predicate_type>::
init( vil_region_finder_connectivity conn )
{
  static int const nbrs4_delta[4][2] = { { 1, 0}, { 0,-1}, {-1, 0}, { 0, 1} };
  static int const nbrs8_delta[8][2] = { { 1, 0}, { 1,-1}, { 0,-1}, {-1,-1},
                                         {-1, 0}, {-1, 1}, { 0, 1}, { 1, 1} };
  processed_.fill( false );
  switch ( conn ) {
   case vil_region_finder_4_conn:
    num_nbrs_ = 4;
    nbr_delta_ = nbrs4_delta;
    break;
   case vil_region_finder_8_conn:
    num_nbrs_ = 8;
    nbr_delta_ = nbrs8_delta;
    break;
   default:
    assert(!"Invalid connectivity type");
  }
}


// ---------------------------------------------------------------------------
//                                                             same int region

template <class pix_type, class predicate_type>
void
vil_region_finder<pix_type, predicate_type>::
same_int_region( unsigned i, unsigned j,
                 std::vector<unsigned>& ri,
                 std::vector<unsigned>& rj )
{
  // get the pixel intensity
  pix_type p = image_(i,j);

  // call the real function
  same_int_region( i, j, p, ri, rj );
}


// ---------------------------------------------------------------------------
//                                                             same int region

template <class pix_type, class predicate_type>
void
vil_region_finder<pix_type, predicate_type>::
same_int_region( unsigned i, unsigned j, pix_type p,
                 std::vector<unsigned>& ri,
                 std::vector<unsigned>& rj )
{
  // early stop if this pixel has already been processed
  if ( processed_(i,j) )
    return;

  // use ri, rj as storage space
  ri.resize( 0 );
  rj.resize( 0 );

  // mark the initial position
  processed_(i,j) = true;
  ri.push_back( i );
  rj.push_back( j );

  for ( unsigned cur_pos = 0; cur_pos<ri.size(); ++cur_pos )
  {
    // get pixel coordinate
    i = ri[cur_pos];
    j = rj[cur_pos];

    // examine the neighbors
    for ( unsigned c = 0; c < num_nbrs_; ++c )
    {
      unsigned nbr_i = (unsigned)( (signed)i + nbr_delta_[c][0] );
      unsigned nbr_j = (unsigned)( (signed)j + nbr_delta_[c][1] );

      if (nbr_i < image_.ni() &&
          nbr_j < image_.nj() &&
          !processed_(nbr_i, nbr_j) &&
          predi_( image_( nbr_i, nbr_j ),  p ) )
      {
        // add this pixel to current region
        processed_(nbr_i, nbr_j) = true;
        ri.push_back( nbr_i );
        rj.push_back( nbr_j );
      }
    }
  }
}


// ---------------------------------------------------------------------------
//                                                                       image

template <class pix_type, class predicate_type>
typename vil_region_finder<pix_type, predicate_type>::image_view const&
vil_region_finder<pix_type, predicate_type>::
image() const
{
  return image_;
}

template <class pix_type, class predicate_type>
vil_image_view<bool> const&
vil_region_finder<pix_type, predicate_type>::
boolean_region_image() const
{
  return processed_;
}

#endif // vil_region_finder_hxx_