Skip to content
GitLab
Select Git revision
  • master default
  • graph_cython
Find file BlameHistoryPermalink
lsgrmSplitter.txx 4.64 KiB
1 
#ifndef __LSGRM_SPLITTER_TXX
2 
#define __LSGRM_SPLITTER_TXX
3
4 
namespace lsgrm
5 
{
6 
	template <class TInputImage>
7 
	void SplitOTBImage(const std::string& inputImagePath, // Path to the input image
8 
					   const std::string& ouputDirectoryPath, // Path to the directory storing the tiles
9 
					   unsigned int& tileWidth, // width of the tile
10 
					   unsigned int& tileHeight, // height of the tile
11 
					   const unsigned int margin, // stability margin
12 
					   const unsigned int niter)
13 
	{
14
15 
		/* Some convenient typedefs to short the code. */
16 
		using ImageType = TInputImage;
17 
		using InternalPixelType = typename ImageType::InternalPixelType;
18 
		using Extractor = otb::MultiChannelExtractROI<InternalPixelType, InternalPixelType>;
19 
		using Reader = otb::ImageFileReader<ImageType>;
20 
		using Writer = otb::ImageFileWriter<ImageType>;
21
22 
		/* Read the input image and update information about its dimension */
23 
		auto reader = Reader::New();
24 
		reader->SetFileName(inputImagePath);
25 
		reader->UpdateOutputInformation();
26 
		auto imagePtr = reader->GetOutput();
27 
		auto imageWidth = imagePtr->GetLargestPossibleRegion().GetSize()[0];
28 
		auto imageHeight = imagePtr->GetLargestPossibleRegion().GetSize()[1];
29 
		auto imageBands = imagePtr->GetNumberOfComponentsPerPixel();
30
31 
		if(tileWidth + 2*margin >= imageWidth || tileHeight + 2* margin >= imageHeight)
32 
		{
33 
			std::cout << "One dimension of the tile is greater than the image" << std::endl;
34 
			exit(1);
35 
		}
36
37 
		// Determine final size of the tiles according to the dimension of the image
38 
		unsigned int k = std::ceil(imageWidth / tileWidth);
39 
		while(imageWidth % k)
40 
			k++;
41 
		tileWidth = imageWidth / k;
42 
		assert(tileWidth > 0);
43 
		k = std::ceil(imageHeight / tileHeight);
44 
		while(imageWidth % k)
45 
			k++;
46 
		tileHeight = imageHeight / k;
47 
		assert(tileHeight > 0);
48
49
50 
		/* Determine the number of tiles according to the tile dimension given by the user. */
51 
		const unsigned int nbTilesX = imageWidth / tileWidth + ( imageWidth % tileWidth > 0 ? 1 : 0);
52 
		const unsigned int nbTilesY = imageHeight / tileHeight + ( imageHeight % tileHeight > 0 ? 1 : 0);
53
54 
		/* Open text file to record general parameters for the LSGRM library. */
55 
		std::ofstream lsgrmLog(ouputDirectoryPath + "info.txt");
56 
		assert(lsgrmLog.good());
57 
		lsgrmLog << "Image width:"<< imageWidth << "\n"
58 
				 << "Image height:"<< imageHeight << "\n"
59 
				 << "Number of bands:"<< imageBands << "\n"
60 
				 << "Number of tiles according to X:" << nbTilesX << "\n"
61 
				 << "Number of tiles according to Y:" << nbTilesY << "\n"
62 
				 << "Tile width:" << tileWidth << "\n"
63 
				 << "Tile height:" << tileHeight << "\n"
64 
				 << "Stability Margin value:" << margin << "\n"
65 
				 << "Number of iterations: " << niter << "\n";
66
67 
		/* Local variables for the next loop */
68 
		unsigned int startX, startY; // Upper left coordinates of the tile.
69 
		unsigned int sizeX, sizeY; // Size of the tiles.
70
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
/* Loop over the tiles*/ for(unsigned int row = 0; row < nbTilesY; ++row) { for(unsigned int col = 0; col < nbTilesX; ++col) { startX = col * tileWidth; startY = row * tileHeight; sizeX = std::min(tileWidth, static_cast<unsigned int>(imageWidth - startX)); sizeY = std::min(tileHeight, static_cast<unsigned int>(imageHeight - startY)); lsgrmLog << "Tile " << row*nbTilesY + col << ":" << startX << "," << startY << "," << sizeX << "," << sizeY << ","; /* Margin at the top ? */ if( row > 0 ) { startY -= margin; sizeY += margin; lsgrmLog << "1,"; }else lsgrmLog << "0,"; /* Margin at the right */ if( col < nbTilesX - 1 ) { sizeX += margin; lsgrmLog << "1,"; } else lsgrmLog << "0,"; /* Margin at the bottom */ if( row < nbTilesY - 1) { sizeY += margin; lsgrmLog << "1,"; } else lsgrmLog << "0,"; /* Margin at the left */ if( col > 0 ) { startX -= margin; sizeX += margin; lsgrmLog << "1\n"; } else lsgrmLog << "0\n"; /* Extract the tile */ auto extractPtr = Extractor::New(); extractPtr->SetStartX(startX); extractPtr->SetStartY(startY); extractPtr->SetSizeX(sizeX); extractPtr->SetSizeY(sizeY); /* Write the tile to the ouput directory. */ auto writer = Writer::New(); writer->SetFileName(ouputDirectoryPath + "tile_" + std::to_string(row) + "_" + std::to_string(col) + ".tif"); /* OTB streaming pipeline */ extractPtr->SetInput(reader->GetOutput()); writer->SetInput(extractPtr->GetOutput()); writer->Update(); } // end for(unsigned int col = 0; col < nbTilesX; ++col)
141142143144145146147148
} // for(unsigned int row = 0; row < nbTilesY; ++row) /* Close the lsgrm info file. */ lsgrmLog.close(); } } // end of namespace lsgrm #endif