MeVisLab/Standard/Sources/ML/MLCSO/CSOTools/CSOLiveWireGraph.h

Go to the documentation of this file.

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

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

#ifndef __CSOLiveWireGraph_H
#define __CSOLiveWireGraph_H

#include "MLCSOSystem.h"

#include "CSOLiveWireNode.h"
#include "CSOObjectHeap.h"


ML_START_NAMESPACE



class MLCSO_EXPORT CSOLiveWireGraph
{

public:

  CSOLiveWireGraph();
  CSOLiveWireGraph(float* imageData, unsigned int sizex, unsigned int sizey, const bool useCostsDirectly, float voxelSizeX=1.0f, float voxelSizeY=1.0f);

  ~CSOLiveWireGraph();

  void setStartPosition(VoxelPos startVoxelPos);

  void setWeights(float weightGradient, float weightLaplace, float weightDirection);

  void setLengthPenalty(float lengthPenalty) 
  {
    _lengthPenalty = lengthPenalty;
  }

  inline void setPathCostThreshold(bool useThreshold, float threshold) { _usePathCostThreshold = useThreshold; _pathConstThreshold = threshold; }

  void getShortestPathToStart(VoxelPos endPos, std::vector<VoxelPos>& intermediatePositions, int neighborhoodMode, bool reverse=true);

  void resetGraphAndHeap();

  void setMaxPathLength(float maxLength) 
  {
    _maxPathLength = maxLength;
  }

  void updateResultCosts();

  enum CSOLiveWireNeighborhoodMode 
  {
    CSO_LIVE_WIRE_4_NEIGHBORHOOD = 0, 
    CSO_LIVE_WIRE_8_NEIGHBORHOOD = 1  
  };

protected:


  /* MEMBER VARIABLES */

  std::vector< VoxelPos > _lastPathPositions;

  unsigned int _imageSizeX;
  unsigned int _imageSizeY;
  float _voxelSizeX;
  float _voxelSizeY;
  CSOLiveWireNode** _lwGraph;

  CSOObjectHeap<CSOLiveWireNode>* _nodeHeap;

  VoxelPos _startVoxelPos;

  int _offsetX[8];
  int _offsetY[8];

  float _maxGlobalGradient;
  float _weightGradient, _weightLaplace, _weightDirection;
  float _lengthPenalty;

  bool _usePathCostThreshold;
  float _pathConstThreshold;

  bool _useCostsDirectly;

  float _maxPathLength;

  float _directionLengths[8];

  /* DATA STRUCTURES */

  /* METHODS */

  void _computeCosts();

  float _getEdgeCosts(unsigned char direction, int posx, int posy);

  void _computeAdditionalCostsAndNormalizeValues();

  void _normalizeValues();
};



ML_END_NAMESPACE

#endif // __CSOLiveWireGraph_H