Skip to content
GitLab
Select Git revision
  • master default
  • 1.1.0
  • 1.0.6
  • 1.0.5
  • 1.0.4
  • 1.0.3
  • 1.0.2
  • 1.0.1
  • 1.0.0
Find file BlameHistoryPermalink
lsgrmGraphOperations.h 5.67 KiB
1 
#ifndef __LSGRM_GRAPH_OPERATIONS_H
2 
#define __LSGRM_GRAPH_OPERATIONS_H
3
4 
#include "lsgrmHeader.h"
5 
#include "grmGraphOperations.h"
6 
#include "otbVectorImage.h"
7 
#include "otbMultiChannelExtractROI.h"
8 
#include "itkGrayscaleFillholeImageFilter.h"
9
10 
namespace lsgrm
11 
{
12
13 
struct ProcessingTile
14 
{
15 
  long int rows[2]; // lower and upper rows (-1 means that the row has not be considered)
16 
  long int columns[2]; // lower and upper columns (-1 means that the row has not be considered)
17 
  long int tileNeighbors[8]; // tile Neighbors at (top, top right, right, bottom right, bottom, bottom left, left, top left)
18 
  long int margin[4]; // Is there a margin at top, left, bottom or right
19 
  otb::VectorImage<double>::RegionType region; // The image region
20
21 
  // Temporary files
22 
  std::string nodeFileName;
23 
  std::string edgeFileName;
24 
  std::string nodeMarginFileName;
25 
  std::string edgeMarginFileName;
26 
};
27
28 
// Read an image region
29 
template<class TSegmenter>
30 
typename TSegmenter::ImageType::Pointer ReadImageRegion(
31 
    typename TSegmenter::ImageType * inputPtr,
32 
    typename TSegmenter::ImageType::RegionType region);
33
34 
template<class TSegmenter>
35 
typename TSegmenter::LabelImageType::Pointer
36 
MergeAllGraphsAndAchieveSegmentation(
37 
    const typename TSegmenter::ParamType& params,
38 
    const float& threshold,
39 
    std::vector<ProcessingTile>& tiles,
40 
    const unsigned int nbTilesX,
41 
    const unsigned int nbTilesY,
42 
    const unsigned int imageWidth,
43 
    const unsigned int imageHeight,
44 
    const unsigned int imageBands,
45 
    unsigned int numberOfIterations);
46
47 
template<class TSegmenter>
48 
long long unsigned int RunFirstPartialSegmentation(
49 
    typename TSegmenter::ImageType * inputPtr,
50 
    const typename TSegmenter::ParamType& params,
51 
    const float& threshold,
52 
    const unsigned int niter,
53 
    const unsigned int niter2,
54 
    std::vector<ProcessingTile>& tiles,
55 
    const unsigned int nbTilesX,
56 
    const unsigned int nbTilesY,
57 
    const unsigned int tileWidth,
58 
    const unsigned int tileHeight,
59 
    bool& isFusion);
60
61 
template<class TSegmenter>
62 
long long unsigned int RunPartialSegmentation(
63 
    const typename TSegmenter::ParamType& params,
64 
    const float& threshold,
65 
    const unsigned int niter,
66 
    std::vector<ProcessingTile>& tiles,
67 
    const unsigned int nbTilesX,
68 
    const unsigned int nbTilesY,
69 
    const unsigned int imageWidth,
70 
    const unsigned int imageHeight,
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
const unsigned int imageBands, bool& isFusion); template<class TSegmenter> void RemoveUselessNodes(ProcessingTile& tile, typename TSegmenter::GraphType& graph, const unsigned int imageWidth, const unsigned int numberOfLayers); template<class TSegmenter> void UpdateNeighborsOfNoneDuplicatedNodes(std::unordered_map<long unsigned int, std::vector<typename TSegmenter::NodePointerType> >& borderPixelMap, const unsigned int imageWidth, const unsigned int imageHeight); template<class TSegmenter> void RemoveDuplicatedNodes(std::unordered_map<long unsigned int, std::vector<typename TSegmenter::NodePointerType> >& borderPixelMap, typename TSegmenter::GraphType& graph, const unsigned int imageWidth); template<class TSegmenter> void BuildBorderPixelMap(typename TSegmenter::GraphType& graph, ProcessingTile& tile, const unsigned int rowTile, const unsigned int colTile, const unsigned int nbTilesX, const unsigned int nbTilesY, std::unordered_map<long unsigned int, std::vector<typename TSegmenter::NodePointerType> >& borderPixelMap, const unsigned int imageWidth); template<class TSegmenter> void InsertNodesFromTile(typename TSegmenter::GraphType& graph, ProcessingTile& tile, bool margin = true); template<class TSegmenter> void AddStabilityMargin(typename TSegmenter::GraphType& graph, std::vector<ProcessingTile>& tiles, const unsigned int row, const unsigned int col, const unsigned int nbTilesX, const unsigned int nbTilesY); template<class TSegmenter> void WriteStabilityMargin(std::unordered_map< typename TSegmenter::NodePointerType, unsigned int>& stabilityMargin, const std::string& nodesPath, const std::string& edgesPath); template<class TSegmenter> void ExtractStabilityMargin(std::unordered_map<typename TSegmenter::NodePointerType, unsigned int>& nodeMap, const unsigned int pmax); template<class TSegmenter> void ExploreDFS(typename TSegmenter::NodePointerType s, const unsigned int p, std::unordered_map<typename TSegmenter::NodePointerType, unsigned int>& Cb, const unsigned int pmax); template<class TSegmenter> void DetectBorderNodes(typename TSegmenter::GraphType& graph, ProcessingTile& tile, std::unordered_map<typename TSegmenter::NodePointerType, unsigned int>& borderNodeMaps, const unsigned int imageWidth, const unsigned int imageHeight); template<class TSegmenter>
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
void ReadGraph(TSegmenter& segmenter, const std::string& nodesPath, const std::string& edgesPath); template<class TSegmenter> void ReadGraph(typename TSegmenter::GraphType& graph, const std::string& nodesPath, const std::string& edgesPath); template<class TSegmenter> void WriteGraph(typename TSegmenter::GraphType& graph, const std::string& nodesFile, const std::string& edgesFile); template<class TSegmenter> void RemoveUnstableSegments(typename TSegmenter::GraphType& graph, ProcessingTile& tile, const unsigned int imageWidth); template<class TSegmenter> void RemoveEdgeToUnstableNode(typename TSegmenter::NodePointerType nodePtr); template<class TSegmenter> void RescaleGraph(typename TSegmenter::GraphType& graph, ProcessingTile& tile, const unsigned int rowTile, const unsigned int colTile, const unsigned int tileWidth, const unsigned int tileHeight, const unsigned int imageWidth); } #include "lsgrmGraphOperations.txx" #endif