mitkAbstractFileReader.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef AbstractFileReader_H_HEADER_INCLUDED_C1E7E521
#define AbstractFileReader_H_HEADER_INCLUDED_C1E7E521

// Macro
#include <MitkCoreExports.h>

// MITK
#include <mitkBaseData.h>
#include <mitkIFileReader.h>
#include <mitkMimeType.h>

// Microservices
#include <usGetModuleContext.h>
#include <usServiceProperties.h>
#include <usServiceRegistration.h>

namespace us
{
  struct PrototypeServiceFactory;
}

namespace mitk
{
  class CustomMimeType;

  class MITKCORE_EXPORT AbstractFileReader : public mitk::IFileReader
  {
  public:
    virtual void SetInput(const std::string &location) override;

    virtual void SetInput(const std::string &location, std::istream *is) override;

    virtual std::string GetInputLocation() const override;

    virtual std::istream *GetInputStream() const override;

    MimeType GetRegisteredMimeType() const;

    virtual std::vector<itk::SmartPointer<BaseData>> Read() override = 0;

    virtual DataStorage::SetOfObjects::Pointer Read(mitk::DataStorage &ds) override;

    virtual ConfidenceLevel GetConfidenceLevel() const override;

    virtual Options GetOptions() const override;
    virtual us::Any GetOption(const std::string &name) const override;

    virtual void SetOptions(const Options &options) override;
    virtual void SetOption(const std::string &name, const us::Any &value) override;

    virtual void AddProgressCallback(const ProgressCallback &callback) override;

    virtual void RemoveProgressCallback(const ProgressCallback &callback) override;

    us::ServiceRegistration<IFileReader> RegisterService(us::ModuleContext *context = us::GetModuleContext());
    void UnregisterService();

  protected:
    class MITKCORE_EXPORT InputStream : public std::istream
    {
    public:
      InputStream(IFileReader *writer, std::ios_base::openmode mode = std::ios_base::in);
      ~InputStream();

    private:
      std::istream *m_Stream;
    };

    AbstractFileReader();
    ~AbstractFileReader();

    AbstractFileReader(const AbstractFileReader &other);

    explicit AbstractFileReader(const CustomMimeType &mimeType, const std::string &description);

    virtual us::ServiceProperties GetServiceProperties() const;

    virtual us::ServiceRegistration<CustomMimeType> RegisterMimeType(us::ModuleContext *context);

    void SetMimeType(const CustomMimeType &mimeType);

    const CustomMimeType *GetMimeType() const;

    void SetMimeTypePrefix(const std::string &prefix);
    std::string GetMimeTypePrefix() const;

    void SetDescription(const std::string &description);
    std::string GetDescription() const;

    void SetDefaultOptions(const Options &defaultOptions);
    Options GetDefaultOptions() const;

    void SetRanking(int ranking);
    int GetRanking() const;

    std::string GetLocalFileName() const;

    virtual void SetDefaultDataNodeProperties(DataNode *node, const std::string &filePath);

  private:
    AbstractFileReader &operator=(const AbstractFileReader &other);

    virtual mitk::IFileReader *Clone() const = 0;

    class Impl;
    std::unique_ptr<Impl> d;
  };

} // namespace mitk

#endif /* AbstractFileReader_H_HEADER_INCLUDED_C1E7E521 */