mitkUSImageLoggingFilter.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 MITKUSImageLoggingFiler_H_HEADER_INCLUDED_
#define MITKUSImageLoggingFiler_H_HEADER_INCLUDED_

// MITK
#include <MitkUSExports.h>
#include <mitkImageToImageFilter.h>
#include <mitkRealTimeClock.h>


namespace mitk {
  class MITKUS_EXPORT USImageLoggingFilter : public mitk::ImageToImageFilter
  {
  public:

    mitkClassMacro(USImageLoggingFilter, mitk::ImageToImageFilter);

    itkNewMacro(USImageLoggingFilter);

    virtual void GenerateData() override;

    void AddMessageToCurrentImage(std::string message);

    void SaveImages(std::string path, std::vector<std::string>& imageFilenames, std::string& csvFileName);

    void SaveImages(std::string path);

    bool SetImageFilesExtension(std::string extension);


  protected:
    USImageLoggingFilter();
    virtual ~USImageLoggingFilter();
    typedef std::vector<mitk::Image::Pointer> ImageCollection;
    mitk::RealTimeClock::Pointer m_SystemTimeClock;  

    //members for logging
    ImageCollection m_LoggedImages; 
    std::map<int, std::string> m_LoggedMessages; 
    std::vector<double> m_LoggedMITKSystemTimes; 
    std::string m_ImageExtension; 

  };
} // namespace mitk
#endif /* MITKUSImageSource_H_HEADER_INCLUDED_ */