mitkUSIGTLDevice.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 mitkUSIGTLDevice_h
#define mitkUSIGTLDevice_h
 
#include <mitkCommon.h>
#include <MitkUSExports.h>
#include <mitkUSDevice.h>
#include <mitkUSProbe.h>
#include <mitkIGTLClient.h>
#include <mitkIGTLDeviceSource.h>
#include <mitkIGTL2DImageDeviceSource.h>
#include <mitkIGTLTrackingDataDeviceSource.h>
#include <mitkIGTLMessageToUSImageFilter.h>
#include <mitkUSVideoDeviceCustomControls.h>
 
 
namespace itk {
  template<class T> class SmartPointer;
}
 
namespace mitk
{
  class USVideoDeviceCustomControls;
  class USAbstractControlInterface;
  class MITKUS_EXPORT USIGTLDevice : public mitk::USDevice
  {
  public:
    mitkClassMacro(USIGTLDevice, mitk::USDevice);
    // To open a device (Manufacturer, Model, Hostname, Port, IsServer)
    mitkNewMacro5Param(Self, std::string, std::string, std::string, int, bool);
 
    std::string GetDeviceClass() override;
    USImageSource::Pointer GetUSImageSource() override;
 
    USIGTLDevice(std::string manufacturer, std::string model, std::string host,
      int port, bool server);
 
    itk::SmartPointer<USAbstractControlInterface> GetControlInterfaceCustom() override;
 
    void UnregisterOnService();
 
    // Neu hinzugefügt:
    std::vector<mitk::USProbe::Pointer> GetAllProbes() override;
 
    void DeleteAllProbes() override;
 
    mitk::USProbe::Pointer GetCurrentProbe() override;
 
    void AddNewProbe(mitk::USProbe::Pointer probe) override;
 
    mitk::USProbe::Pointer GetProbeByName(std::string name) override;
 
    void RemoveProbeByName(std::string name) override;
 
    void SetDefaultProbeAsCurrentProbe() override;
 
    void SetCurrentProbe(std::string probename) override;
 
    void SetSpacing(double xSpacing, double ySpacing) override;
 
 
  protected:
    bool OnInitialization() override;
    bool OnConnection() override;
    bool OnDisconnection() override;
    bool OnActivation() override;
    bool OnDeactivation() override;
 
    void GenerateData() override;
 
  private:
    std::string m_Host;
    int m_Port;
    mitk::IGTLDevice::Pointer m_Device;
    mitk::IGTL2DImageDeviceSource::Pointer m_DeviceSource;
    mitk::IGTLTrackingDataDeviceSource::Pointer m_TransformDeviceSource;
    mitk::IGTLMessageToUSImageFilter::Pointer m_Filter;
 
    itk::SmartPointer<USVideoDeviceCustomControls> m_ControlInterfaceCustom;
 
    std::vector < mitk::USProbe::Pointer > m_Probes;
 
    mitk::USProbe::Pointer m_CurrentProbe;
  };
}  // namespace mitk
 
#endif