Skip to content
GitLab
Select Git revision
  • disassembled
  • master default
  • resumeopt
  • msbuild_portability
  • streaminggraphtolabelimage
  • change_output_mode
  • GRMBasedModule
  • v2.0
Find file BlameHistoryPermalink
lsgrmSplitter.txx 5.41 KiB
1 
#ifndef __LSGRM_SPLITTER_TXX
2 
#define __LSGRM_SPLITTER_TXX
3 
#include "lsgrmSplitter.h"
4
5 
namespace lsgrm
6 
{
7
8 
enum RelativePosition{
9 
  POS_TOP,
10 
  POS_RIGHT,
11 
  POS_BOTTOM,
12 
  POS_LEFT
13 
};
14
15 
enum NeighborhoodRelativePosition{
16 
  NBH_TOP,
17 
  NBH_TOP_RIGHT,
18 
  NBH_RIGHT,
19 
  NBH_BOTTOM_RIGHT,
20 
  NBH_BOTTOM,
21 
  NBH_BOTTOM_LEFT,
22 
  NBH_LEFT,
23 
  NBH_TOP_LEFT
24 
};
25
26 
template <class TInputImage>
27 
std::vector<ProcessingTile> SplitOTBImage(TInputImage * imagePtr, // input image
28 
    unsigned int& tileWidth, // width of the tile
29 
    unsigned int& tileHeight, // height of the tile
30 
    const unsigned int margin, // stability margin
31 
    unsigned int &nbTilesX,
32 
    unsigned int &nbTilesY,
33 
    std::string temporaryFilesPrefix)
34 
    {
35
36 
  std::vector<ProcessingTile> tiles;
37
38 
  // Image size
39 
  auto imageWidth = imagePtr->GetLargestPossibleRegion().GetSize()[0];
40 
  auto imageHeight = imagePtr->GetLargestPossibleRegion().GetSize()[1];
41
42 
  /* Local variables for the next loop */
43 
  unsigned int startX, startY; // Upper left coordinates of the tile.
44 
  unsigned int sizeX, sizeY; // Size of the tiles.
45
46 
  /* Loop over the tiles*/
47 
  tiles.assign(nbTilesX * nbTilesY, ProcessingTile());
48 
  unsigned int i = 0;
49 
  for(unsigned int row = 0; row < nbTilesY; ++row)
50 
    {
51 
    for(unsigned int col = 0; col < nbTilesX; ++col)
52 
      {
53
54 
      // Compute current tile start and size
55 
      startX = col * tileWidth;
56 
      startY = row * tileHeight;
57 
      sizeX = tileWidth;
58 
      sizeY = tileHeight;
59
60 
      // Current tile size might be different for right and bottom borders
61 
      if (col == nbTilesX -1)
62 
        {
63 
        sizeX += imageWidth % tileWidth;
64 
        }
65 
      if (row == nbTilesY -1)
66 
        {
67 
        sizeY += imageHeight % tileHeight;
68 
        }
69
70 
      /* Margin at the top ? */
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
if( row > 0 ) { tiles[i].margin[POS_TOP] = margin; tiles[i].rows[0] = row * tileHeight; tiles[i].tileNeighbors[NBH_TOP] = i - nbTilesX; } else { // Tile is on the top row --> no top margin tiles[i].margin[POS_TOP] = 0; tiles[i].rows[0] = 0; tiles[i].tileNeighbors[NBH_TOP] = -1; } /* Margin at the right */ if( col < nbTilesX - 1 ) { tiles[i].margin[POS_RIGHT] = margin; tiles[i].columns[1] = col * tileWidth + sizeX - 1; //sizeX tiles[i].tileNeighbors[NBH_RIGHT] = i+1; } else { // Tile is on the right column --> no right margin tiles[i].margin[POS_RIGHT] = 0; tiles[i].columns[1] = imageWidth - 1; tiles[i].tileNeighbors[NBH_RIGHT] = -1; } /* Margin at the bottom */ if( row < nbTilesY - 1) { tiles[i].margin[POS_BOTTOM] = margin; tiles[i].rows[1] = row * tileHeight + sizeY - 1; // sizeY tiles[i].tileNeighbors[NBH_BOTTOM] = i + nbTilesX; } else { // Tile is on the bottom --> no bottom margin tiles[i].margin[POS_BOTTOM] = 0; tiles[i].rows[1] = imageHeight - 1; tiles[i].tileNeighbors[NBH_BOTTOM] = -1; } /* Margin at the left */ if( col > 0 ) { tiles[i].margin[POS_LEFT] = margin; tiles[i].columns[0] = col * tileWidth; tiles[i].tileNeighbors[NBH_LEFT] = i-1; } else { // Tile is on the left --> no left margin tiles[i].margin[POS_LEFT] = 0; tiles[i].columns[0] = 0; tiles[i].tileNeighbors[NBH_LEFT] = -1; } /* Store the tile region */ typename TInputImage::IndexType index; index[0] = startX - tiles[i].margin[POS_LEFT]; index[1] = startY - tiles[i].margin[POS_TOP]; typename TInputImage::SizeType size; size[0] = sizeX + tiles[i].margin[POS_LEFT] + tiles[i].margin[POS_RIGHT]; size[1] = sizeY + tiles[i].margin[POS_TOP] + tiles[i].margin[POS_BOTTOM]; typename TInputImage::RegionType region(index, size); tiles[i].region = region; std::cout << "Tile " << i << ": start at " << index << " with size " << size << std::endl;
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181
/* Is there a neighbor at the rop right */ if(row > 0 && col < nbTilesX - 1) tiles[i].tileNeighbors[NBH_TOP_RIGHT] = i - nbTilesX + 1; else tiles[i].tileNeighbors[NBH_TOP_RIGHT] = -1; /* Is there a neighbor at the bottom right */ if(col < nbTilesX - 1 && row < nbTilesY - 1) tiles[i].tileNeighbors[NBH_BOTTOM_RIGHT] = i + nbTilesX + 1; else tiles[i].tileNeighbors[NBH_BOTTOM_RIGHT] = -1; /* Is there a neighbor at the bottom left */ if(row < nbTilesY - 1 && col > 0) tiles[i].tileNeighbors[NBH_BOTTOM_LEFT] = i + nbTilesX - 1; else tiles[i].tileNeighbors[NBH_BOTTOM_LEFT] = -1; /* Is there a neighbor at the top left */ if(col > 0 && row > 0) tiles[i].tileNeighbors[NBH_TOP_LEFT] = i - nbTilesX - 1; else tiles[i].tileNeighbors[NBH_TOP_LEFT] = -1; std::string suffix = std::to_string(row) + "_" + std::to_string(col) + ".bin"; tiles[i].nodeFileName = temporaryFilesPrefix + "_node_" + suffix; tiles[i].edgeFileName = temporaryFilesPrefix + "_edge_" + suffix; tiles[i].nodeMarginFileName = temporaryFilesPrefix + "_nodeMargin_" + suffix; tiles[i].edgeMarginFileName = temporaryFilesPrefix + "_edgeMargin_" + suffix; i++; } // end for(unsigned int col = 0; col < nbTilesX; ++col) } // for(unsigned int row = 0; row < nbTilesY; ++row) return tiles; } } // end of namespace lsgrm #endif