creaImageIOTreeHandlerImageAdder.h

Go to the documentation of this file.

/*
# ---------------------------------------------------------------------
#
# Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image 
#                        pour la Santé)
# Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
# Previous Authors : Laurent Guigues, Jean-Pierre Roux
# CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
#
#  This software is governed by the CeCILL-B license under French law and 
#  abiding by the rules of distribution of free software. You can  use, 
#  modify and/ or redistribute the software under the terms of the CeCILL-B 
#  license as circulated by CEA, CNRS and INRIA at the following URL 
#  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html 
#  or in the file LICENSE.txt.
#
#  As a counterpart to the access to the source code and  rights to copy,
#  modify and redistribute granted by the license, users are provided only
#  with a limited warranty  and the software's author,  the holder of the
#  economic rights,  and the successive licensors  have only  limited
#  liability. 
#
#  The fact that you are presently reading this means that you have had
#  knowledge of the CeCILL-B license and that you accept its terms.
# ------------------------------------------------------------------------
*/

#ifndef __creaImageIOTreeHandlerImageAdder_h_INCLUDED__
#define __creaImageIOTreeHandlerImageAdder_h_INCLUDED__

#include <creaImageIOTreeHandler.h>
#include <creaImageIOSynchron.h>
#include <creaImageIOImageReader.h>
//#include <creaImageIOImageWriter.h>

// Signal/slot mechanism for progress events
#include <boost/signal.hpp>
#include <boost/bind.hpp>
#if defined(USE_GDCM2)
#include "creaImageIODicomScanner.h"
#endif

namespace creaImageIO
{
  //=======================================================================

  class TreeHandlerImageAdder
  {
          
  public:
    //====================================================================
    TreeHandlerImageAdder(TreeHandler* tree);
    ~TreeHandlerImageAdder();
    void SetTreeHandler(TreeHandler* tree) { mTreeHandler = tree;}
        
        void SetSynchronizer(Synchronizer* s){mSynchronizer=s;}
        void SetCurrentDatabase(std::string cur){mCurrentDB=cur;}
    //====================================================================

    //====================================================================
    class Progress
    {
    public:
      Progress() { Reset(); }
      ~Progress() {}
      
      void Reset() 
      {
        mStop = false;
        mNumberScannedFiles = 0;
        mNumberScannedDirs = 0;
        mNumberHandledFiles = 0;
        mNumberAddedFiles = 0;
      }

      int GetNumberScannedFiles() const { return mNumberScannedFiles; }
      int GetNumberScannedDirs() const { return mNumberScannedDirs; }
      int GetNumberHandledFiles() const { return mNumberHandledFiles; }
      int GetNumberAddedFiles() const { return mNumberAddedFiles; }

      void IncNumberScannedFiles() { mNumberScannedFiles++; }
      void IncNumberScannedDirs() { mNumberScannedDirs++; }
      void IncNumberHandledFiles() { mNumberHandledFiles++; }
      void IncNumberAddedFiles() { mNumberAddedFiles++; }

      void SetStop() { mStop = true; }
      bool GetStop() const { return mStop; }

    private:
      bool mStop;
      int mNumberScannedFiles;
      int mNumberScannedDirs;
      int mNumberHandledFiles;
      int mNumberAddedFiles;
    };
    //=============================================

    //=============================================
    const Progress& GetProgress() const { return mProgress; }
    //=============================================

    //=============================================
    typedef boost::signal<void (Progress&)>  ProgressSignalType;
    typedef ProgressSignalType::slot_function_type ProgressCallbackType;
    //=============================================

   //==================================================================
    void ConnectProgressObserver(ProgressCallbackType callback);
   //==================================================================

    //====================================================================
    bool IsHandledFile( const std::string& filename);
    void AddFiles( const std::vector<std::string>& filename );
    void AddDirectory( const std::string& directory, 
                       bool recurse);


#if defined(USE_GDCM2)
         void AddDirectoryRecursorScanner(const std::string &dirpath, 
                                                   bool recursive,const std::string &addKey,  DicomImageScanner i_sc, bool b_loaded);
#endif

        void RemoveFile(tree::Node* node);
        void RemoveFiles(const std::vector<tree::Node*>& nodes);

    void remove( tree::Node* i_node);

        std::string Synchronize(bool repair, bool checkAttributes);
        void CheckSyncDirectory(const std::string &dirpath, 
                                                        bool recursive, 
                                                        bool repair,
                                                        bool checkAttributes,
                                                        std::vector<std::string> &i_ignorefiles,
                                                        std::vector<std::string> & attsModified,
                                                        std::vector<std::string> & newfiles);
        void CopyFiles(const std::vector<std::string>& filenames, const std::string directory  );
        
        void SaveAs(const std::vector<std::string>& filenames, std::vector<vtkImageData *> i_images);
        void FindNode(tree::Node* parent, int level, 
                const std::string& searchParam, 
                const std::string& searchVal, 
                tree::Node*& node);

        void FindNodePartial(tree::Node* parent, int level, const std::string& searchParam, const std::string& searchVal, tree::Node*& node);
 
        void CheckAttributes(bool repair, std::string& file, std::vector<std::string>& attsModified);
        void DeleteDriveFromMainDB(const std::string& drive);
        void DeleteDriveFromOtherDB(const std::string& drive);
        void EditField(tree::Node* node, const std::string& name, const std::string& key, const std::string& val);
        void GetAttributes(const std::vector<std::string>& params, 
          const std::string& filename, 
          std::vector<std::string>& results);

        const std::string isAttributeExist(const std::string i_attr);
  
  
 
    //====================================================================

  private:

    void AddFile( const std::string& filename );

    void AddDirectoryRecursor( const std::string& directory, 
                               bool recurse,
                           const std::string &addKey);

    TreeHandler* mTreeHandler;
        Synchronizer* mSynchronizer;
    ImageReader mReader;
        //ImageWriter mWriter;
        std::string mCurrentDB;
    
    Progress mProgress;
    ProgressSignalType mProgressSignal;


  };
  // EO class TreeHandlerImageAdder
  //=======================================================================


 


} // EO namespace creaImageIO

#include <iostream>
inline std::ostream& operator << ( std::ostream& o, 
                            const creaImageIO::TreeHandlerImageAdder::Progress& p)
{
  o << p.GetNumberScannedFiles() << " files - "
    << p.GetNumberScannedDirs() << " dirs - "
    << p.GetNumberHandledFiles() << " handled -"
    << p.GetNumberAddedFiles() << " added";
  return o;
}

// EOF
#endif