creaImageIOVtkImageReader.cpp

Go to the documentation of this file.

#include <creaImageIOVtkImageReader.h>
#include <vtkImageReader2.h>
#include <creaImageIOSystem.h>
#include "boost/filesystem/path.hpp"

namespace creaImageIO{
  //=====================================================================
  VtkImageReader::VtkImageReader(vtkImageReader2* r, 
                                       const std::string& name,
                                       const std::string& extensions)
    : mReader(r), mExtensions(extensions)
  {
    if (name.size() == 0) 
      {
                  const char *test =mReader->GetDescriptiveName();
                  if(test != "")
                  {
                        SetName ( "toto");// mReader->GetDescriptiveName());
                  }
      }
    else 
      {
        SetName ( name );
      }
    GimmickDebugMessage(5,"Constructing vtkImageReader : "<<GetName()
                        <<std::endl);
      
  }
  //=====================================================================
  
  //=====================================================================
  VtkImageReader::~VtkImageReader()
  {

    mReader->Delete();
  }
  //=====================================================================

  //=====================================================================
  bool VtkImageReader::CanRead(const std::string& filename)
  { 

    return (mReader->CanReadFile(filename.c_str())!=0);
/*        if(filename != "")
          {
                return (mReader->CanReadFile(filename.c_str())!=0);
          }
          else
          {
                  return false;
          }*/
  }
  //=====================================================================
  void VtkImageReader::getAttributes(const std::string filename,
                std::map <std::string , std::string> &infos, std::vector<std::string> i_attr)
   {
   }
  //=====================================================================
  vtkImageData* VtkImageReader::ReadImage(const std::string& filename)
  {
    vtkImageData* im = 0;
    try
      {
        mReader->SetFileName(filename.c_str());
        mReader->Update();
        im = vtkImageData::New();
        im->ShallowCopy(mReader->GetOutput());
      }
    catch (...)
      {
        if (im!=0) im->Delete();
        im = 0;
      }
    return im;
  }
  //=====================================================================
  
  //=====================================================================
  void SplitExtensionsString ( const std::string& str, 
                               const std::string& delimiters, 
                               std::vector<std::string>& tokens)
  {
    // Skip delimiters at beginning.
    std::string::size_type lastPos = str.find_first_not_of(delimiters, 0);
    // Find first delimiter.
    std::string::size_type pos     = str.find_first_of(delimiters, lastPos);
    
    while (std::string::npos != pos || std::string::npos != lastPos)
      {
        // Found a token, add it to the vector.
        // SPECIFIC : REMOVE INITIAL DOT (lastPos + 1)
        tokens.push_back(str.substr(lastPos+1, pos - lastPos));
        // Skip delimiters.  Note the "not_of"
        lastPos = str.find_first_not_of(delimiters, pos);
        // Find next delimiter
        pos = str.find_first_of(delimiters, lastPos);
      }
    
    }
  //=====================================================================
  
  //=====================================================================
  void VtkImageReader::PushBackExtensions(std::vector<std::string>& v)
  {
    std::string ext = mExtensions;
    if (ext.size()==0) ext = mReader->GetFileExtensions ();
    
    SplitExtensionsString(ext," ",v);
  }
  //=====================================================================
 


  //=====================================================================
  void VtkImageReader::ReadAttributes(const std::string& filename, 
                                      std::map<std::string,std::string>& attr)
  {
    GimmickMessage(2,"Reading attributes from '"<<filename<<std::endl);
    // Get image dimensions
    // How to get the image info without loading it in vtk ?
    mReader->SetFileName(filename.c_str());
    mReader->Update(); //OpenFile();
    int ext[6];
    mReader->GetDataExtent(ext);
    // Columns
    char cols[128];
    sprintf(cols,"%i",ext[1]-ext[0]);
    // Rows
    char rows[128];
    sprintf(rows,"%i",ext[3]-ext[2]);
    // Planes 
    char planes[128];
    sprintf(planes,"%i",ext[5]-ext[4]);
   
        std::map<std::string,std::string>::iterator i;
    if ( (i = attr.find("FullFileName")) != attr.end())
      {
        i->second = filename;
      }
    if ( (i = attr.find("D0004_1500")) != attr.end())
      {
        boost::filesystem::path full_path(filename);
        std::string f = full_path.leaf();
        i->second = f;
      }
    if ( (i = attr.find("D0028_0010")) != attr.end())
      {
        i->second = rows;
      }
    if ( (i = attr.find("D0028_0011")) != attr.end())
      {
        i->second = cols;
      }
    
    if ( (i = attr.find("D0028_0012")) != attr.end())
      {
                i->second = planes;
      }
          if ( (i = attr.find("FullFileDirectory")) != attr.end())
      {
         std::string::size_type last_pos = filename.find_last_of("//");
                 i->second = filename.substr(0, last_pos);
          }

    GimmickMessage(2,"Attributes map:"<<std::endl<<attr<<std::endl);
  }
  //=====================================================================

} // namespace creaImageIO