MeVisLab/Standard/Sources/Inventor/SoCSO/CSOProcessor/CSOLiveWireProcessor/CSOLiveWireGraph.h

Go to the documentation of this file.

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

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

#ifndef __CSOLiveWireGraph_H
#define __CSOLiveWireGraph_H

#include "SoCSOSystem.h"

#include "CSOLiveWireNode.h"
#include <CSOTools/CSOObjectHeap.h>


ML_START_NAMESPACE



const float EULER_FACTOR = 1.0f / (6.0f * 2.718281828459f * 2.718281828459f); 

const float SQRT_2 = 1.414213562373f; 


const unsigned char DIR_N  = 0; 
const unsigned char DIR_E  = 1; 
const unsigned char DIR_S  = 2; 
const unsigned char DIR_W  = 3; 
const unsigned char DIR_NE = 4; 
const unsigned char DIR_SE = 5; 
const unsigned char DIR_SW = 6; 
const unsigned char DIR_NW = 7; 




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


class SOCSO_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 w1, float w2, float w3);

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

  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();

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 _w1, _w2, _w3;
  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