src/SVTK/SVTK_ImageWriterMgr.cxx

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