mitkOpenIGTLinkTrackingDevice.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 MITKOPENIGTLINKTRACKINGDEVICE_H_HEADER_INCLUDED_
#define MITKOPENIGTLINKTRACKINGDEVICE_H_HEADER_INCLUDED_

#include <mitkIGTConfig.h>
#include <mitkTrackingDevice.h>
#include <mitkOpenIGTLinkTrackingTool.h>
#include <mitkIGTLClient.h>
#include <mitkIGTLDeviceSource.h>
#include <mitkIGTLMessageToNavigationDataFilter.h>
#include <itkMultiThreader.h>
#include <igtlQuaternionTrackingDataMessage.h>
#include <igtlTrackingDataMessage.h>
#include <igtlTransformMessage.h>
#include "mitkIGTLTransformDeviceSource.h"

namespace mitk
{
  class MITKIGT_EXPORT OpenIGTLinkTrackingDevice : public TrackingDevice
  {
  public:
    mitkClassMacro(OpenIGTLinkTrackingDevice, TrackingDevice);
    itkFactorylessNewMacro(Self)
      itkCloneMacro(Self)

      
      void SetPortNumber(int portNumber);

    void SetHostname(std::string hostname);

    int GetPortNumber();

    std::string GetHostname();

    virtual bool StartTracking();

    virtual bool StopTracking();

    virtual bool OpenConnection();

    virtual bool CloseConnection();

    virtual unsigned int GetToolCount() const;

    TrackingTool* GetTool(unsigned int toolNumber)  const;

    bool DiscoverTools(int WaitingTime = 10000);

    mitk::TrackingTool* AddTool(const char* toolName, const char* fileName);

    bool IsDeviceInstalled();

    itkSetMacro(UpdateRate, int);               
    itkGetConstMacro(UpdateRate, int);          

  protected:
    OpenIGTLinkTrackingDevice();
    ~OpenIGTLinkTrackingDevice();

    bool InternalAddTool(OpenIGTLinkTrackingTool::Pointer tool);

    void UpdateTools();
    unsigned long m_MessageReceivedObserverTag;

    std::vector<OpenIGTLinkTrackingTool::Pointer> GetAllTools();

    //OpenIGTLink connection class
    mitk::IGTLClient::Pointer m_OpenIGTLinkClient;

    //OpenIGTLink pipeline
    mitk::IGTLTransformDeviceSource::Pointer m_IGTLDeviceSource;
    mitk::IGTLMessageToNavigationDataFilter::Pointer m_IGTLMsgToNavDataFilter;

    std::vector<OpenIGTLinkTrackingTool::Pointer> m_AllTools; 

    int m_UpdateRate; 

  private:
    enum TrackingMessageType
    {
      TDATA, TRANSFORM, QTDATA, UNKNOWN
    };

    mitk::OpenIGTLinkTrackingDevice::TrackingMessageType GetMessageTypeFromString(const char* messageTypeString);

    bool DiscoverToolsFromTData(igtl::TrackingDataMessage::Pointer msg);

    bool DiscoverToolsFromQTData(igtl::QuaternionTrackingDataMessage::Pointer msg);

    bool DiscoverToolsFromTransform();

    void AddNewToolForName(std::string name, int i);
  };
}//mitk
#endif /* MITKOpenIGTLinkTRACKINGDEVICE_H_HEADER_INCLUDED_ */