mitkUSImageLoggingFilter.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#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);

    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();
    ~USImageLoggingFilter() override;
    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_ */