/home/neoflo/smb4k/SERVEUR/Millie/trunk/src/multithread/operator/MTAreaOperator.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 "MTAreaOperator.hpp"
#include <pthread.h>
#include <vector>

using namespace Millie;

struct SharedArea
{
  int debut;
  Image * dest;
  const Image * image;
  LocalAreaOperator * localOp;
  int nbThreads;
};

/*
 * pthread nous oblige à utiliser une fonction sous la forme :
 *  void * fonction(void*)
 */
static void * areaOperatorMT(void * p_data)
{
  /*recupèration des arguments*/
  SharedArea & shared = *((SharedArea*)p_data);
  const Image & image = *shared.image;
  Image & dest = *shared.dest;
  LocalAreaOperator & localOp = *shared.localOp;

  float t;

  int leftpadding = localOp.getLeftPadding();
  int rightpadding = localOp.getRightPadding();
  int toppadding = localOp.getTopPadding();
  int bottompadding = localOp.getBottomPadding();
  int hauteur = image.getHeight();
  int largeur = image.getWidth();

  /*application de la'opération pour le thread particulier*/
  for(int canal=0; canal< image.getNumComponents(); canal++)
    for(int j = toppadding + shared.debut; j< hauteur - bottompadding; j+=shared.nbThreads)
    {
      for(int i = leftpadding; i< largeur - rightpadding; i++)
      {
        dest.setPixel(i - leftpadding, j - toppadding,
                      canal, localOp.computeLocalArea(image, i,j, canal));

      }
    }

  return NULL;
}




MTAreaOperator::MTAreaOperator(const LocalAreaOperator & areaOp, int nbthread)
    : AreaOperator(areaOp)
{
  if(nbthread<1)
    throw Millie::IllegalArgument("MTAreaOperator  : thread<1");

  _nbThread = nbthread;
}

void MTAreaOperator::compute(Image & dest, const Image& in)
{
  if(dest.getNumComponents() != in.getNumComponents())
    throw IllegalArgument("MTAreaOperator");

  int leftpadding = getLeftPadding();
  int rightpadding = getRightPadding();
  int toppadding = getTopPadding();
  int bottompadding = getBottomPadding();
  int hauteur = in.getHeight();
  int largeur = in.getWidth();

  dest.resize(largeur - (rightpadding + leftpadding), hauteur - (toppadding+bottompadding));


  std::vector<pthread_t> vectPthread;
  std::vector<SharedArea> vectSharedArea;

  vectPthread.resize(_nbThread);
  vectSharedArea.resize(_nbThread);

  for(int i=0; i<_nbThread;i++)
  {
    SharedArea s;
    s.debut = i;
    s.dest = &dest;
    s.image = &in;
    s.localOp = &getLocalAreaOperator();
    s.nbThreads = _nbThread;
    vectSharedArea[i] = s;
  }

  for(int i = 0; i<_nbThread;i++)
  {
    if(pthread_create (&vectPthread[i], NULL, areaOperatorMT,  &vectSharedArea[i]) == -1)
      throw ThreadCreationFailed("MTAreaOperator : thread creation failed");
  }

  for(int i = 0; i<_nbThread;i++)
  {
    if(pthread_join (vectPthread[i], NULL) == -1)
      throw RuntimeException("MTAreaOperator : join thread impossibl");
  }
}

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

---