src/SVTK/SVTK_ImageWriterMgr.cxx

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