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 MITKIGTLDevice_H_HEADER_INCLUDED_
#define MITKIGTLDevice_H_HEADER_INCLUDED_

#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  // MITKIGTLDevice_H_HEADER_INCLUDED_