ImageDicomStream.hpp

/* ***** BEGIN LICENSE BLOCK *****
 * FW4SPL - Copyright (C) IRCAD, 2009-2015.
 * Distributed under the terms of the GNU Lesser General Public License (LGPL) as
 * published by the Free Software Foundation.
 * ****** END LICENSE BLOCK ****** */

#ifndef __VTKGDCMIO_HELPER_IMAGEDICOMSTREAM_HPP__
#define __VTKGDCMIO_HELPER_IMAGEDICOMSTREAM_HPP__

#include <vector>

#include <boost/iostreams/stream.hpp>

#include <vtkSmartPointer.h>
#include <vtkGDCMImageReader.h>

#include <fwCore/base.hpp>

#include <fwMemory/stream/in/IFactory.hpp>

#include "vtkGdcmIO/config.hpp"

namespace vtkGdcmIO
{
namespace helper
{

//------------------------------------------------------------------------------

class ImageDicomInfo
{

public:

    typedef std::vector< std::string > SeriesFilesType;

    typedef SPTR ( ImageDicomInfo ) sptr;

    size_t m_buffSizeInBytes;

    SeriesFilesType m_seriesFiles;
};

//------------------------------------------------------------------------------

class ImageDicomSource
{
public:

    typedef char char_type;

    typedef ::boost::iostreams::source_tag category;

    ImageDicomSource( ImageDicomInfo::sptr dcmInfo );

    std::streamsize read(char* s, std::streamsize n);

private:

    bool readImage();

    ImageDicomInfo::sptr m_dcmInfo;

    size_t m_pos;

    vtkSmartPointer< vtkGDCMImageReader > m_reader;

    char * m_inputReader;

    bool m_success;
};

//------------------------------------------------------------------------------

class ImageDicomStream : public ::fwMemory::stream::in::IFactory
{

public:

    ImageDicomStream( ImageDicomInfo::sptr dcmInfo );

protected:

    SPTR(std::istream) get();

    ImageDicomInfo::sptr m_dcmInfo;
};

//------------------------------------------------------------------------------

} // namespace helper
} // namespace vtkGdcmIO

#endif // __VTKGDCMIO_HELPER_IMAGEDICOMSTREAM_HPP__