itkImageFileReaderKW.h

/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
#ifndef itkImageFileReaderKW_h
#define itkImageFileReaderKW_h
#include "itkImageFileReaderException.h"

#include "ITKIOImageBaseExport.h"

#include "itkImageIOBase.h"
#include "itkImageSource.h"
#include "itkMacro.h"
#include "itkImageRegion.h"
#include "itkDefaultConvertPixelTraits.h"
#include "itkSimpleDataObjectDecorator.h"

namespace itk
{

    template< typename TOutputImage,
            typename ConvertPixelTraits = DefaultConvertPixelTraits<
                    typename TOutputImage::IOPixelType > >
    class ITKIOImageBase_HIDDEN ImageFileReaderKW:public ImageSource< TOutputImage >
    {
    public:
        typedef ImageFileReaderKW             Self;
        typedef ImageSource< TOutputImage > Superclass;
        typedef SmartPointer< Self >        Pointer;

        itkNewMacro(Self);

        itkTypeMacro(ImageFileReaderKW, ImageSource);

        typedef typename TOutputImage::SizeType SizeType;

        typedef typename TOutputImage::IndexType IndexType;

        typedef typename TOutputImage::RegionType ImageRegionType;

        typedef typename TOutputImage::InternalPixelType OutputImagePixelType;

        itkSetGetDecoratedInputMacro(FileName, std::string);

        void  SetImageIO(ImageIOBase *imageIO);
        itkGetModifiableObjectMacro(ImageIO, ImageIOBase);

        itkSetMacro(UseStreaming, bool);
        itkGetConstReferenceMacro(UseStreaming, bool);
        itkBooleanMacro(UseStreaming);

    protected:
        ImageFileReaderKW();
        ~ImageFileReaderKW();
        void PrintSelf(std::ostream & os, Indent indent) const ITK_OVERRIDE;

        void DoConvertBuffer(void *buffer, size_t numberOfPixels);

        void TestFileExistanceAndReadability();

        virtual void GenerateOutputInformation(void) ITK_OVERRIDE;

        virtual void EnlargeOutputRequestedRegion(DataObject *output) ITK_OVERRIDE;

        virtual void GenerateData() ITK_OVERRIDE;

        ImageIOBase::Pointer m_ImageIO;

        bool m_UserSpecifiedImageIO; // keep track whether the
        // ImageIO is user specified

        bool m_UseStreaming;

    private:
        ITK_DISALLOW_COPY_AND_ASSIGN(ImageFileReaderKW);

        std::string m_ExceptionMessage;

        // The region that the ImageIO class will return when we ask to
        // produce the requested region.
        ImageIORegion m_ActualIORegion;
    };
} //namespace ITK

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkImageFileReaderKW.hxx"
#endif

#ifdef ITK_IO_FACTORY_REGISTER_MANAGER
#include "itkImageIOFactoryRegisterManager.h"
#endif

#endif // itkImageFileReaderKW_h