Skip to content
GitLab
Select Git revision
  • master default
  • enhancements
Find file BlameHistoryPermalink
Forked from Cresson Remi / SimpleExtractionTools
Source project has a limited visibility.
otbVectorDataToLabelImageCustomFilter.h 5.94 KiB
1 
/*=========================================================================
2
3 
  Copyright (c) Remi Cresson (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 
#ifndef __gtbVectorDataToLabelImageCustomFilter_h
12 
#define __gtbVectorDataToLabelImageCustomFilter_h
13
14 
#include "itkImageToImageFilter.h"
15 
#include "itkImageSource.h"
16 
#include "otbMacro.h"
17 
#include "otbImageMetadataInterfaceFactory.h"
18 
#include "otbVectorData.h"
19
20 
#include "gdal.h"
21 
#include "ogr_api.h"
22 
#include <ogrsf_frmts.h>
23
24 
namespace otb {
25
26 
/** \class VectorDataToLabelImageCustomFilter
27 
 *  \brief Burn geometries from the specified VectorData into raster
28 
 *
29 
 *  This class handles burning several input VectorDatas into the
30 
 *  output raster.  The burn values are extracted from a field set by
31 
 *  the user.If no burning field is set, the "FID" is used by default.
32 
 *
33 
 *  A "Burn max value" mode overrides this and makes only the max value
34 
 *  burnt whenever the burning field.
35 
 *
36 
 *  Setting the output raster informations can be done in two ways by:
37 
 *    - Setting the Origin/Size/Spacing of the output image
38 
 *    - Using an existing image as support via SetOutputParametersFromImage(ImageBase)
39 
 *
40 
 *  OGRRegisterAll() method must have been called before applying filter.
41 
 *
42 
 * \ingroup SimpleExtractionTools
43 
 */
44 
template <class TVectorData, class TOutputImage  >
45 
class  ITK_EXPORT VectorDataToLabelImageCustomFilter :
46 
    public itk::ImageSource<TOutputImage>
47 
{
48 
public:
49 
  /** Standard class typedefs */
50 
  typedef VectorDataToLabelImageCustomFilter                           Self;
51 
  typedef itk::ImageSource<TOutputImage>     Superclass;
52 
  typedef itk::SmartPointer< Self >                           Pointer;
53 
  typedef itk::SmartPointer<const Self>                       ConstPointer;
54
55 
  /** Run-time type information (and related methods). */
56 
  itkTypeMacro(VectorDataToLabelImageCustomFilter, itk::ImageSource);
57
58 
  /** Method for creation through the object factory. */
59 
  itkNewMacro(Self);
60
61 
  typedef TOutputImage                                OutputImageType;
62 
  typedef typename OutputImageType::Pointer           OutputImagePointer;
63 
  typedef typename OutputImageType::SizeType          OutputSizeType;
64 
  typedef typename OutputImageType::IndexType         OutputIndexType;
65 
  typedef typename OutputImageType::SpacingType       OutputSpacingType;
66 
  typedef typename OutputImageType::PointType         OutputOriginType;
67 
  typedef typename OutputImageType::RegionType        OutputImageRegionType;
68 
  typedef typename OutputImageType::PixelType         OutputImagePixelType;
69 
  typedef typename OutputImageType::InternalPixelType OutputImageInternalPixelType;
70
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
/** VectorData typedefs*/ typedef TVectorData VectorDataType; typedef typename VectorDataType::DataTreeType DataTreeType; typedef typename DataTreeType::TreeNodeType InternalTreeNodeType; typedef typename DataTreeType::Pointer DataTreePointerType; typedef typename DataTreeType::ConstPointer DataTreeConstPointerType; typedef itk::ImageBase<OutputImageType::ImageDimension> ImageBaseType; /** Get Nth input VectorData */ const VectorDataType* GetInput(unsigned int idx); /** Method for adding VectorData to rasterize */ virtual void AddVectorData(const VectorDataType* vd); /** Set the size of the output image. */ itkSetMacro(OutputSize, OutputSizeType); /** Get the size of the output image. */ itkGetConstReferenceMacro(OutputSize, OutputSizeType); /** Burn mode */ itkSetMacro(BurnMaxValueMode, bool); itkGetMacro(BurnMaxValueMode, bool); /** Set the origin of the output image. * \sa GetOrigin() */ itkSetMacro(OutputOrigin, OutputOriginType); virtual void SetOutputOrigin(const double origin[2]); virtual void SetOutputOrigin(const float origin[2]); itkGetConstReferenceMacro(OutputOrigin, OutputOriginType); /** Set the spacing (size of a pixel) of the output image. * \sa GetSpacing() */ virtual void SetOutputSpacing(const OutputSpacingType& spacing); virtual void SetOutputSpacing(const double spacing[2]); virtual void SetOutputSpacing(const float spacing[2]); /** Set/Get Output Projection Ref */ itkSetStringMacro(OutputProjectionRef); itkGetStringMacro(OutputProjectionRef); itkSetStringMacro(BurnAttribute); itkGetStringMacro(BurnAttribute); /** Useful to set the output parameters from an existing image*/ void SetOutputParametersFromImage(const ImageBaseType * image); protected: virtual void GenerateData(); VectorDataToLabelImageCustomFilter(); virtual ~VectorDataToLabelImageCustomFilter() override {} virtual void GenerateOutputInformation(); void PrintSelf(std::ostream& os, itk::Indent indent) const; private: VectorDataToLabelImageCustomFilter(const Self&); //purposely not implemented void operator=(const Self&); //purposely not implemented OGRDataSource* m_OGRDataSourcePointer; // Vector Of OGRGeometyH std::vector< OGRGeometryH > m_SrcDataSetGeometries;