src/SVTK/SVTK_ImageWriterMgr.cxx

   1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 //  This library is free software; you can redistribute it and/or
   7 //  modify it under the terms of the GNU Lesser General Public
   8 //  License as published by the Free Software Foundation; either
   9 //  version 2.1 of the License.
  10 //
  11 //  This library is distributed in the hope that it will be useful,
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 //  Lesser General Public License for more details.
  15 //
  16 //  You should have received a copy of the GNU Lesser General Public
  17 //  License along with this library; if not, write to the Free Software
  18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22 #include "SVTK_ImageWriterMgr.h"
  23 #include "SVTK_ImageWriter.h"
  24
  25 #include <vtkImageData.h>
  26
  27 #include <QSemaphore>
  28
  29 #include <limits>
  30
  31
  32 #ifdef _DEBUG_
  33 static int MYDEBUG = 0;
  34 #else
  35 static int MYDEBUG = 0;
  36 #endif
  37
  38 //----------------------------------------------------------------------------
  39 SVTK_ImageWriterMgr
  40 ::SVTK_ImageWriterMgr()
  41 {
  42   int aMax = std::numeric_limits<int>::max() / 2;
  43   mySemaphore = new QSemaphore(aMax);
  44   mySemaphore->acquire( aMax );
  45   if(MYDEBUG) cout<<"SVTK_ImageWriterMgr::SVTK_ImageWriterMgr "<<
  46                 //"- total = "<<mySemaphore->total()<<
  47                 "; available = "<<mySemaphore->available()<<endl;
  48 }
  49
  50
  51 //----------------------------------------------------------------------------
  52 SVTK_ImageWriterMgr
  53 ::~SVTK_ImageWriterMgr()
  54 {
  55   Stop();
  56   delete mySemaphore;
  57 }
  58
  59
  60 //----------------------------------------------------------------------------
  61 void
  62 SVTK_ImageWriterMgr
  63 ::StartImageWriter(vtkImageData *theImageData,
  64                    const std::string& theName,
  65                    const int theProgressive,
  66                    const int theQuality)
  67 {
  68   SVTK_ImageWriter *anImageWriter =
  69     new SVTK_ImageWriter(mySemaphore,
  70                          theImageData,
  71                          theName,
  72                          theProgressive,
  73                          theQuality);
  74   myThreads.push_back(anImageWriter);
  75
  76   anImageWriter->start();
  77
  78 }
  79
  80
  81 //----------------------------------------------------------------------------
  82 void
  83 SVTK_ImageWriterMgr
  84 ::Stop()
  85 {
  86   if(MYDEBUG) cout<<"SVTK_ImageWriterMgr::Stop "<<
  87                 //"- total = "<<mySemaphore->total()<<
  88                 "; available = "<<mySemaphore->available()<<endl;
  89   if(MYDEBUG) cout<<"SVTK_ImageWriterMgr::Stop - *mySemaphore += "<<myThreads.size()<<endl;
  90   mySemaphore->acquire( myThreads.size() );
  91
  92   for(size_t anId = 0, anEnd = myThreads.size(); anId < anEnd; anId++){
  93     SVTK_ImageWriter* anImageWriter = myThreads[anId];
  94     anImageWriter->wait();
  95     delete anImageWriter;
  96   }
  97   myThreads.clear();
  98 }