MeVisLab/Standard/Sources/ML/MLVesselGraph/mlSkeleton.h

Go to the documentation of this file.

// **InsertLicense** code
//----------------------------------------------------------------------------------

//----------------------------------------------------------------------------------
#ifndef __mlSkeleton_H
#define __mlSkeleton_H

// dll specific settings.
#include "mlVesselGraphSystem.h"

// graph includes
#include "Properties/mlPropertyContainer.h"
#include "mlPropertiesMacros.h"

// ml includes 
#include "mlBase.h"
#include "mlImageVector.h"
#include "mlVector3.h"
#include "mlMatrix4.h"
#include "mlSystemWarningsDisable.h"
#include <boost/shared_ptr.hpp>
#include "mlSystemWarningsRestore.h"

ML_START_NAMESPACE

class TreeNode;


class VESSELGRAPH_EXPORT Skeleton : public PropertyContainer
{
friend class Graph;
friend class VesselEdge;

public:

  enum LABELTYPE {TRACECOLOR,CONTINUOUS};

  Skeleton(const PropertyManager::Pointer& manager, const Vector3& pos=Vector3(0), float min=0, float max=0);

  Skeleton(const Skeleton& other);

  Skeleton& operator=(const Skeleton& other);

  virtual void copyProperties(const Skeleton* other);

  bool operator==(const Skeleton& other) const;
  bool operator!=(const Skeleton& other) const { return !(*this==other); }


  STATIC_GRAPH_PROPERTY(Vector3, pos, Pos)

  

  ImageVector voxelPos();

  void transform(const Matrix4& m);

  void convertToVoxel(Matrix4* mat);
  void convertToWorld(Matrix4* mat);


  STATIC_GRAPH_PROPERTY(double, minDistance, MinDistance);
  STATIC_GRAPH_PROPERTY(double, maxDistance, MaxDistance);
  STATIC_GRAPH_PROPERTY(double, area, Area);
  STATIC_GRAPH_PROPERTY(double, distanceLabel, DistanceLabel);
  STATIC_GRAPH_PROPERTY(double, freeLabel, FreeLabel);
  STATIC_GRAPH_PROPERTY(double, label, Label);



  STATIC_CASTED_GRAPH_PROPERTY(LABELTYPE, MLuint64, labelType, LabelType);
  STATIC_GRAPH_PROPERTY(MLuint64, enumLabel, EnumLabel);

  void setRGBA( float r, float g, float b, float a ) { 
      _colorRGBA[0] = r;
      _colorRGBA[1] = g;
      _colorRGBA[2] = b;
      _colorRGBA[3] = a;
  }  

  void setRGBA( float* rgba ){
    for ( size_t i=0; i<4; ++i ) { _colorRGBA[i] = rgba[i]; }
  }

  void getRGBA( float* r, float* g, float* b, float* a ) const {
      *r = _colorRGBA[0];
      *g = _colorRGBA[1];
      *b = _colorRGBA[2];
      *a = _colorRGBA[3];
  }

  float *getRGBA(void){
    return _colorRGBA;
  }

  virtual void addStateToTree(TreeNode* parent) const;
  ML_SET_ADDSTATE_VERSION(5);
  virtual void readStateFromTree(TreeNode* parent);

  //----------------------------------------------------------------------------

  //----------------------------------------------------------------------------
  size_t        getVoxelNum() const { return _vesselCoordinates.size()+1; }
  size_t        numVesselPos() const { return _vesselCoordinates.size(); }
  void          addVesselPos(const Vector3 &position) { _vesselCoordinates.push_back(position); }
  void          addVesselPos(const ImageVector &v) { _vesselCoordinates.push_back(Vector3(v.x, v.y, v.z)); }
  const Vector3 &vesselPos(size_t index) const { return _vesselCoordinates.at(index); }
  Vector3       &vesselPos(size_t index) { return _vesselCoordinates.at(index); }
  ImageVector   vesselVoxelPos(size_t index);
  void          clearVesselPos() { _vesselCoordinates.clear(); }

  static void setCopyVesselVoxels       ( bool flag ){ Skeleton::_copyVesselVoxels = flag; }
  static void setCopyVesselVoxelsDefault( void )     { Skeleton::_copyVesselVoxels = true; }

private:
  static bool _copyVesselVoxels;

  float _colorRGBA[4];

  std::vector<Vector3> _vesselCoordinates;

  ML_ABSTRACT_CLASS_HEADER(Skeleton)
};

ML_END_NAMESPACE

#endif // __mlSkeleton_H