/home/neoflo/smb4k/SERVEUR/Millie/trunk/src/operator/PeronaMalikFilter/NeighbourPeronaMalikFlow.cpp

Aller à la documentation de ce fichier.

/******************************************************************************
 *       __    _    _   _       _       _   _____                             *
 *       | \  / |  | | | |     | |     | |  | ___|                            *
 *       |  \/  |  | | | |     | |     | |  | |_                              *
 *       |      |  | | | |     | |     | |  |  _|                             *
 *       | |\/| |  | | | |__   | |__   | |  | |__                             *
 *       |_|  |_|  |_| |____|  |____|  |_|  |____|                            *
 * __________________________________________________________________________ *
 *                 Multifunctional Library For Image Processing               *
 *                                                                            *
 *                                                                            *
 *                                                                            *
 *      (c) Copyright 2007 by Humbert Florent                                 *
 *                                                                            *
 *      This program 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; only version 2 of the License.          *
 *                                                                            *
 *      This program 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 this program; if not, write to the Free Software           *
 *      Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA             *
 *      02111-1307, USA.                                                      *
 ******************************************************************************/

#include "NeighbourPeronaMalikFlow.hpp"



using namespace Millie;

/*
 * Cet algorithme est fortement inspiré de l'algorithme de résolution proposé par :
 *   Jan-Friedrich Ehlenbröker, Alexander Maier, Uwe Mönks, Stefan
 * Schwalowsky et Matthias Tobergte
 */

float NeighbourPeronaMalikFlow::computeFlow(const Image& image,
                                        int x,
                                        int y,
                                        int canal) const
{
  float current = image.getPixel(x,y, canal);
  int width = image.getWidth();
  int height = image.getHeight();


  int px = x-1;
  int nx = x+1;
  int py = y-1;
  int ny = y+1;
  if (px<0)
    px=0;
  if (nx>=width)
    nx=width-1;
  if (py<0)
    py=0;
  if (ny>=height)
    ny=height-1;

  float ixp = image.getPixel(px, y, canal);
  float ixn = image.getPixel(nx, y, canal);
  float iyp = image.getPixel(x, ny, canal);
  float iyn = image.getPixel(x, py, canal);

  float diffxn = computeDiffusion(current- ixn);
  float diffxp = computeDiffusion(current- ixp);
  float diffyn = computeDiffusion(current- iyn);
  float diffyp = computeDiffusion(current- ixp);

  float iNE = image.getPixel(nx, py, canal);
  float iSW = image.getPixel(px, ny, canal);
  float iNW = image.getPixel(px, py, canal);
  float iSE = image.getPixel(nx ,ny, canal);
  float diffNE = computeDiffusion(current  - iNE);
  float diffSW = computeDiffusion(current- iSW);
  float diffNW = computeDiffusion(current- iNW);
  float diffSE = computeDiffusion(current- iSE);


  float delta =   diffxn * (ixn - current)
                  + diffxp * (ixp - current)
                  + diffyp * (iyp - current)
                  + diffyn * (iyn - current)
                  +
                  0.5f * (
                    diffNE * (iNE- current)
                    +  diffSW * (iSW - current)
                    +  diffSE * (iSE - current)
                    +  diffNW * (iNW -current)
                  );


  return delta;

}



NeighbourPeronaMalikFlow::NeighbourPeronaMalikFlow(float lambda) :
    SimplePeronaMalikFlow(lambda)
{
}

NeighbourPeronaMalikFlow*  NeighbourPeronaMalikFlow::clone() const
{
  return new NeighbourPeronaMalikFlow(*this);
}

---