igtlPolyDataMessage.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   The OpenIGTLink Library
  Language:  C++
  Web page:  http://openigtlink.org/
 
  Copyright (c) Insight Software Consortium. All rights reserved.
 
  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.  See the above copyright notices for more information.
 
=========================================================================*/
 
#ifndef __igtlPolyDataMessage_h
#define __igtlPolyDataMessage_h
 
#include <string>
 
#include "igtlObject.h"
#include "igtlMacro.h"
#include "igtlMath.h"
#include "igtlMessageBase.h"
#include "igtlTypes.h"
 
namespace igtl
{
 
class IGTLCommon_EXPORT GetPolyDataMessage: public MessageBase
{
public:
  typedef GetPolyDataMessage            Self;
  typedef MessageBase                   Superclass;
  typedef SmartPointer<Self>            Pointer;
  typedef SmartPointer<const Self>      ConstPointer;
 
  igtlTypeMacro(igtl::GetPolyDataMessage, igtl::MessageBase);
  igtlNewMacro(igtl::GetPolyDataMessage);
 
protected:
  GetPolyDataMessage();
  ~GetPolyDataMessage() {};
protected:
  virtual int  CalculateContentBufferSize() { return 0; };
  virtual int  PackContent()        { AllocateBuffer(); return 1; };
  virtual int  UnpackContent()      { return 1; };
};
 
class IGTLCommon_EXPORT RTSPolyDataMessage : public MessageBase
{
public:
  typedef RTSPolyDataMessage            Self;
  typedef MessageBase                    Superclass;
  typedef SmartPointer<Self>             Pointer;
  typedef SmartPointer<const Self>       ConstPointer;
 
  igtlTypeMacro(igtl::RTSPolyDataMessage, igtl::MessageBase);
  igtlNewMacro(igtl::RTSPolyDataMessage);
 
  bool GetStatus() const;
  void SetStatus(bool status);
 
protected:
  RTSPolyDataMessage() : MessageBase() { this->m_SendMessageType = "RTS_POLYDATA"; };
  ~RTSPolyDataMessage() {};
 
protected:
  virtual int  CalculateContentBufferSize();
  virtual int  PackContent();
  virtual int  UnpackContent();
 
protected:
  igtl_uint8 m_Status;
};
 
class IGTLCommon_EXPORT StopPolyDataMessage: public MessageBase
{
public:
  typedef StopPolyDataMessage            Self;
  typedef MessageBase                    Superclass;
  typedef SmartPointer<Self>             Pointer;
  typedef SmartPointer<const Self>       ConstPointer;
  
  igtlTypeMacro(igtl::StopPolyDataMessage, igtl::MessageBase);
  igtlNewMacro(igtl::StopPolyDataMessage);
  
protected:
  StopPolyDataMessage() : MessageBase() { this->m_SendMessageType = "STP_POLYDATA"; };
  ~StopPolyDataMessage() {};
protected:
  virtual int  GetBodyPackSize() { return 0; };
  virtual int  PackBody()        { AllocatePack(); return 1; };
  virtual int  UnpackBody()      { return 1; };
};
 
class IGTLCommon_EXPORT StartPolyDataMessage : public MessageBase
{
public:
  typedef StartPolyDataMessage            Self;
  typedef MessageBase                    Superclass;
  typedef SmartPointer<Self>             Pointer;
  typedef SmartPointer<const Self>       ConstPointer;
 
  igtlTypeMacro(igtl::StartPolyDataMessage, igtl::MessageBase);
  igtlNewMacro(igtl::StartPolyDataMessage);
 
protected:
  StartPolyDataMessage() : MessageBase() { this->m_SendMessageType = "STT_POLYDATA"; };
  ~StartPolyDataMessage() {};
protected:
  virtual int  GetBodyPackSize() { return 0; };
  virtual int  PackBody() { AllocatePack(); return 1; };
  virtual int  UnpackBody() { return 1; };
};
  
  
  
// A class to manage a point array.
class IGTLCommon_EXPORT PolyDataPointArray : public Object {
 
 public:
 
  typedef std::vector<igtlFloat32> Point;
 
 public:
  typedef PolyDataPointArray        Self;
  typedef Object                    Superclass;
  typedef SmartPointer<Self>        Pointer;
  typedef SmartPointer<const Self>  ConstPointer;
 
  igtlTypeMacro(igtl::PolyDataPointArray, igtl::Object);
  igtlNewMacro(igtl::PolyDataPointArray);
  
 protected:
  PolyDataPointArray();
  ~PolyDataPointArray();
 
 public:
 
  void Clear();
 
  void SetNumberOfPoints(int n);
 
  int  GetNumberOfPoints();
 
  int  SetPoint(unsigned int id, igtlFloat32 * point);
 
  int  SetPoint(unsigned int id, igtlFloat32 x, igtlFloat32 y, igtlFloat32 z);
 
  int  AddPoint(igtlFloat32 * point);
 
  int  AddPoint(igtlFloat32 x, igtlFloat32 y, igtlFloat32 z);
 
  int  GetPoint(unsigned int id, igtlFloat32 & x, igtlFloat32 & y, igtlFloat32 & z);
 
  int  GetPoint(unsigned int id, igtlFloat32 * point);
 
 private:
  std::vector< Point > m_Data;
};
 
 
// The PolyDataCellArray class is used to pass vertices, lines, polygons, and triangle strips
class IGTLCommon_EXPORT PolyDataCellArray : public Object {
  
 public:
  enum {
    NULL_POINT = 0xFFFFFFFF,
  };
 
 public:
  typedef PolyDataCellArray         Self;
  typedef Object                    Superclass;
  typedef SmartPointer<Self>        Pointer;
  typedef SmartPointer<const Self>  ConstPointer;
  
  igtlTypeMacro(igtl::PolyDataCellArray, igtl::Object);
  igtlNewMacro(igtl::PolyDataCellArray);
 
 protected:
  PolyDataCellArray();
  ~PolyDataCellArray();
 
 public:
 
  void       Clear();
 
  igtlUint32 GetNumberOfCells();
 
  void       AddCell(int n, igtlUint32 * cell);
 
  void       AddCell(std::list<igtlUint32> cell);
 
  igtlUint32 GetTotalSize();
 
  igtlUint32 GetCellSize(unsigned int id);
 
  int        GetCell(unsigned int id, igtlUint32 * cell);
 
  int        GetCell(unsigned int id, std::list<igtlUint32>& cell);
 
 private:
  std::vector< std::list<igtlUint32> > m_Data;
};
 
 
class IGTLCommon_EXPORT PolyDataAttribute : public Object {
 public:
 
  enum {
    POINT_SCALAR = 0x00,
    POINT_VECTOR = 0x01,
    POINT_NORMAL = 0x02,
    POINT_TENSOR = 0x03,
    POINT_RGBA   = 0x04,
    CELL_SCALAR  = 0x10,
    CELL_VECTOR  = 0x11,
    CELL_NORMAL  = 0x12,
    CELL_TENSOR  = 0x13,
    CELL_RGBA    = 0x14,
  };
 
 public:
  typedef PolyDataAttribute         Self;
  typedef Object                    Superclass;
  typedef SmartPointer<Self>        Pointer;
  typedef SmartPointer<const Self>  ConstPointer;
  
  igtlTypeMacro(igtl::PolyDataAttribute, igtl::Object);
  igtlNewMacro(igtl::PolyDataAttribute);
 
 protected:
  PolyDataAttribute();
  ~PolyDataAttribute();
 
 public:
 
  void        Clear();
 
  int         SetType(int t, int n=1);
 
  igtlUint8   GetType() { return this->m_Type; };
 
  igtlUint32  GetNumberOfComponents();
 
  igtlUint32  SetSize(igtlUint32 size);
 
  igtlUint32  GetSize();
 
  void        SetName(const char * name);
 
  const char* GetName() { return this->m_Name.c_str(); };
 
  int         SetData(igtlFloat32 * data);
 
  int         GetData(igtlFloat32 * data);
 
  int         SetNthData(unsigned int n, igtlFloat32 * data);
 
  int         GetNthData(unsigned int n, igtlFloat32 * data);
 
 private:
 
  igtlUint8                m_Type;
 
  igtlUint8                m_NComponents;
 
  igtlUint32               m_Size;
 
  std::string              m_Name;
 
  std::vector<igtlFloat32> m_Data;
 
};
 
 
class IGTLCommon_EXPORT PolyDataMessage: public MessageBase
{
public:
  typedef PolyDataMessage                Self;
  typedef MessageBase                    Superclass;
  typedef SmartPointer<Self>             Pointer;
  typedef SmartPointer<const Self>       ConstPointer;
 
  igtlTypeMacro(igtl::PolyDataMessage, igtl::MessageBase);
  igtlNewMacro(igtl::PolyDataMessage);
 
public:
 
  void Clear();
 
  igtlSetObjectMacro(Points,   PolyDataPointArray);
 
  igtlGetObjectMacro(Points,   PolyDataPointArray);
 
  igtlSetObjectMacro(Vertices, PolyDataCellArray);
 
  igtlGetObjectMacro(Vertices, PolyDataCellArray);
 
  igtlSetObjectMacro(Lines,    PolyDataCellArray);
 
  igtlGetObjectMacro(Lines,    PolyDataCellArray);
 
  igtlSetObjectMacro(Polygons, PolyDataCellArray);
 
  igtlGetObjectMacro(Polygons, PolyDataCellArray);
 
  igtlSetObjectMacro(TriangleStrips, PolyDataCellArray);
 
  igtlGetObjectMacro(TriangleStrips, PolyDataCellArray);
 
  void ClearAttributes();
 
  void AddAttribute(PolyDataAttribute * att);
 
  int  GetNumberOfAttributes();
 
  PolyDataAttribute * GetAttribute(unsigned int id);
 
protected:
  PolyDataMessage();
  ~PolyDataMessage();
 
protected:
 
  virtual int  CalculateContentBufferSize();
  virtual int  PackContent();
  virtual int  UnpackContent();
 
  PolyDataPointArray::Pointer m_Points;
 
  PolyDataCellArray::Pointer  m_Vertices;
 
  PolyDataCellArray::Pointer  m_Lines;
 
  PolyDataCellArray::Pointer  m_Polygons;
 
  PolyDataCellArray::Pointer  m_TriangleStrips;
 
  std::vector<PolyDataAttribute::Pointer> m_Attributes;
 
};
 
} // namespace igtl
 
#endif // _igtlPolyDataMessage_h