mitkIGTLMessageProvider.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 IGTLMESSAGEPROVIDER_H_HEADER_INCLUDED_
#define IGTLMESSAGEPROVIDER_H_HEADER_INCLUDED_

#include "mitkIGTLDevice.h"
#include "mitkIGTLDeviceSource.h"

//itk
#include "itkCommand.h"

namespace mitk {
  class MITKOPENIGTLINK_EXPORT IGTLMessageProvider : public IGTLDeviceSource
  {
  public:
    mitkClassMacro(IGTLMessageProvider, IGTLDeviceSource);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

    typedef itk::SimpleMemberCommand<mitk::IGTLMessageProvider> ProviderCommand;

    void Send(mitk::IGTLMessage::Pointer msg);

    void StartStreamingOfSource(mitk::IGTLMessageSource* src,
                                unsigned int fps);

    void StopStreamingOfSource(mitk::IGTLMessageSource* src);

    void StopStreamingOfAllSources();

    bool IsStreaming();

    itkGetMacro(StreamingTime, unsigned int);

    void Update() override;

  protected:
    IGTLMessageProvider();
    ~IGTLMessageProvider() override;

    void GenerateData() override;

    void CreateOutputs();

    void OnIncomingMessage() override;

    void OnIncomingCommand() override;

    void OnLostConnection() override;

    void ConnectTo( mitk::IGTLMessageSource* UpstreamFilter );

    void DisconnectFrom( mitk::IGTLMessageSource* UpstreamFilter );

    mitk::IGTLMessageSource::Pointer GetFittingSource(const char* requestedType);

    void InvokeStartStreamingEvent();

    void InvokeStopStreamingEvent();

  private:
    ProviderCommand::Pointer m_StreamingCommand;

    ProviderCommand::Pointer m_StopStreamingCommand;

    // * \brief Timer thread for generating a continuous time signal for the stream
    // *
    // * Everyt time the time is passed a time signal is invoked.
    // *
    // * \param pInfoStruct pointer to the mitkIGTLMessageProvider object
    // * \return
    // */
    //static ITK_THREAD_RETURN_TYPE TimerThread(void* pInfoStruct);

    //int                                       m_ThreadId;

    //bool                                      m_StopStreamingThread;

    //itk::SmartPointer<itk::MultiThreader>     m_MultiThreader;

    unsigned int                              m_StreamingTime;

    itk::SmartPointer<itk::FastMutexLock>     m_StreamingTimeMutex;

    //itk::SmartPointer<itk::FastMutexLock>     m_StopStreamingThreadMutex;

    bool                                      m_IsStreaming;

  };

  itkEventMacro(StreamingStartRequiredEvent, itk::AnyEvent);

  itkEventMacro(StreamingStopRequiredEvent, itk::AnyEvent);

} // namespace mitk
#endif /* MITKIGTLMESSAGEPROVIDER_H_HEADER_INCLUDED_ */