src/SVTK/SVTK_ImageWriterMgr.cxx

   1 // Copyright (C) 2007-2022  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, or (at your option) any later version.
   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 <vtkAlgorithm.h>
  24 #include <vtkImageData.h>
  25
  26 #include <QSemaphore>
  27
  28 #include <limits>
  29
  30
  31 #ifdef _DEBUG_
  32 static int MYDEBUG = 0;
  33 #else
  34 static int MYDEBUG = 0;
  35 #endif
  36
  37 //----------------------------------------------------------------------------
  38 SVTK_ImageWriterMgr
  39 ::SVTK_ImageWriterMgr()
  40 {
  41   int aMax = std::numeric_limits<int>::max() / 2;
  42   mySemaphore = new QSemaphore(aMax);
  43   mySemaphore->acquire( aMax );
  44   if(MYDEBUG) cout<<"SVTK_ImageWriterMgr::SVTK_ImageWriterMgr "<<
  45                 //"- total = "<<mySemaphore->total()<<
  46                 "; available = "<<mySemaphore->available()<<endl;
  47 }
  48
  49
  50 //----------------------------------------------------------------------------
  51 SVTK_ImageWriterMgr
  52 ::~SVTK_ImageWriterMgr()
  53 {
  54   Stop();
  55   delete mySemaphore;
  56 }
  57
  58
  59 //----------------------------------------------------------------------------
  60 void
  61 SVTK_ImageWriterMgr
  62 ::StartImageWriter(vtkAlgorithm *theAlgorithm,
  63                    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                          theAlgorithm,
  71                          theImageData,
  72                          theName,
  73                          theProgressive,
  74                          theQuality);
  75   myThreads.push_back(anImageWriter);
  76
  77   anImageWriter->start();
  78
  79 }
  80
  81
  82 //----------------------------------------------------------------------------
  83 void
  84 SVTK_ImageWriterMgr
  85 ::Stop()
  86 {
  87   if(MYDEBUG) cout<<"SVTK_ImageWriterMgr::Stop "<<
  88                 //"- total = "<<mySemaphore->total()<<
  89                 "; available = "<<mySemaphore->available()<<endl;
  90   if(MYDEBUG) cout<<"SVTK_ImageWriterMgr::Stop - *mySemaphore += "<<myThreads.size()<<endl;
  91   mySemaphore->acquire( (int)myThreads.size() ); //!< TODO: conversion from size_t to int
  92
  93   for(size_t anId = 0, anEnd = myThreads.size(); anId < anEnd; anId++){
  94     SVTK_ImageWriter* anImageWriter = myThreads[anId];
  95     anImageWriter->wait();
  96     delete anImageWriter;
  97   }
  98   myThreads.clear();
  99 }