Skip to content
GitLab
Select Git revision
  • master default
  • enhancements
Find file BlameHistoryPermalink
Forked from Cresson Remi / SimpleExtractionTools
Source project has a limited visibility.
otbExtractGeom.cxx 5.18 KiB
1 
/*=========================================================================
2
3 
  Copyright (c) IRSTEA. All rights reserved.
4
5
6 
     This software is distributed WITHOUT ANY WARRANTY; without even
7 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
8 
     PURPOSE.  See the above copyright notices for more information.
9
10 
=========================================================================*/
11 
#include "itkFixedArray.h"
12 
#include "itkObjectFactory.h"
13
14 
// Elevation handler
15 
#include "otbWrapperElevationParametersHandler.h"
16 
#include "otbWrapperApplicationFactory.h"
17
18 
// Application engine
19 
#include "otbStandardFilterWatcher.h"
20 
#include "itkFixedArray.h"
21
22 
// Filter
23 
#include "otbMultiChannelExtractROI.h"
24 
#include "otbVectorDataToLabelImageCustomFilter.h"
25 
#include "itkMaskImageFilter.h"
26 
#include "otbVectorDataIntoImageProjectionFilter.h"
27 
#include "otbVectorDataProperties.h"
28 
#include "otbRegionComparator.h"
29
30 
using namespace std;
31
32 
namespace otb
33 
{
34
35 
namespace Wrapper
36 
{
37
38 
class ExtractGeom : public Application
39 
{
40 
public:
41 
  /** Standard class typedefs. */
42 
  typedef ExtractGeom                        Self;
43 
  typedef Application                   Superclass;
44 
  typedef itk::SmartPointer<Self>       Pointer;
45 
  typedef itk::SmartPointer<const Self> ConstPointer;
46
47 
  /** Standard macro */
48 
  itkNewMacro(Self);
49 
  itkTypeMacro(ExtractGeom, Application);
50
51 
  /** Filters */
52 
  typedef otb::MultiChannelExtractROI<FloatVectorImageType::InternalPixelType,
53 
      FloatVectorImageType::InternalPixelType> ExtractFilterType;
54
55 
  /** mask */
56 
  typedef bool TMaskPixelValueType;
57 
  typedef otb::Image<TMaskPixelValueType, 2> MaskImageType;
58 
  typedef itk::MaskImageFilter<FloatVectorImageType, MaskImageType,
59 
      FloatVectorImageType> MaskFilterType;
60
61 
  /* vector data filters */
62 
  typedef otb::VectorDataIntoImageProjectionFilter<VectorDataType,
63 
      FloatVectorImageType> VectorDataReprojFilterType;
64 
  typedef otb::VectorDataToLabelImageCustomFilter<VectorDataType,
65 
      MaskImageType> RasteriseFilterType;
66 
  typedef otb::VectorDataProperties<VectorDataType> VectorDataPropertiesType;
67
68 
  void DoInit()
69 
  {
70 
    SetName("ExtractGeom");
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
SetDescription("Perform geometric extraction"); // Documentation SetDocName("ExtractGeom"); SetDocLongDescription("This application performs geometric extraction"); SetDocLimitations("None"); SetDocAuthors("RemiCresson"); SetDocSeeAlso(" "); AddDocTag(Tags::Manip); AddParameter(ParameterType_InputImage, "in", "Input Image"); SetParameterDescription("in"," Input image."); AddParameter(ParameterType_InputVectorData, "shp", "Input Shapefile" ); SetParameterDescription("shp","Input Shapefile" ); AddParameter(ParameterType_OutputImage, "out", "Output image"); SetParameterDescription("out"," Output image."); AddRAMParameter(); // Doc example parameter settings SetDocExampleParameterValue("shp", "vecteur.shp"); SetDocExampleParameterValue("in", "QB_Toulouse_Ortho_XS.tif"); SetDocExampleParameterValue("out", "QB_Toulouse_Ortho_XS_decoup.tif uint16"); } void DoUpdateParameters() { // Nothing to do here : all parameters are independent } void DoExecute() { // Get inputs FloatVectorImageType::Pointer xs = GetParameterImage("in"); VectorDataType* shp = GetParameterVectorData("shp"); /* Reproject vector data */ vdReproj = VectorDataReprojFilterType::New(); vdReproj->SetInputVectorData(shp); vdReproj->SetInputImage(xs); vdReproj->Update(); /* Get VectorData bounding box */ vdProperties = VectorDataPropertiesType::New(); vdProperties->SetVectorDataObject(vdReproj->GetOutput()); vdProperties->ComputeBoundingRegion(); /* Compute intersecting region */ otb::RegionComparator<FloatVectorImageType, FloatVectorImageType> comparator; comparator.SetImage1(xs); FloatVectorImageType::RegionType roi = comparator.RSRegionToImageRegion(vdProperties->GetBoundingRegion()); roi.PadByRadius(1); // avoid extrapolation if (!roi.Crop(xs->GetLargestPossibleRegion())) { otbAppLogFATAL( << " Input VectorData is outside image !" ); return; } /* Rasterize vector data */ rasterizer = RasteriseFilterType::New(); rasterizer->AddVectorData(vdReproj->GetOutput());
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
rasterizer->SetOutputOrigin(xs->GetOrigin()); rasterizer->SetOutputSpacing(xs->GetSpacing()); rasterizer->SetOutputSize( xs->GetLargestPossibleRegion().GetSize()); rasterizer->SetBurnMaxValueMode(true); rasterizer->SetOutputProjectionRef( xs->GetProjectionRef()); rasterizer->Update(); /* Mask input image */ maskFilter = MaskFilterType::New(); maskFilter->SetInput(xs); maskFilter->SetMaskImage(rasterizer->GetOutput()); /* Extract ROI */ m_Filter = ExtractFilterType::New(); m_Filter->SetInput(maskFilter->GetOutput()); m_Filter->SetExtractionRegion(roi); SetParameterOutputImage("out", m_Filter->GetOutput()); } ExtractFilterType::Pointer m_Filter; VectorDataReprojFilterType::Pointer vdReproj; RasteriseFilterType::Pointer rasterizer; VectorDataPropertiesType::Pointer vdProperties; MaskFilterType::Pointer maskFilter; }; } } OTB_APPLICATION_EXPORT( otb::Wrapper::ExtractGeom )