Line data    Source code

       1             : /******************************************************************************
       2             :  *
       3             :  * Project:  GDAL
       4             :  * Purpose:  "filter" step of "vector pipeline"
       5             :  * Author:   Even Rouault <even dot rouault at spatialys.com>
       6             :  *
       7             :  ******************************************************************************
       8             :  * Copyright (c) 2024, Even Rouault <even dot rouault at spatialys.com>
       9             :  *
      10             :  * SPDX-License-Identifier: MIT
      11             :  ****************************************************************************/
      12             : 
      13             : #ifndef GDALALG_VECTOR_FILTER_INCLUDED
      14             : #define GDALALG_VECTOR_FILTER_INCLUDED
      15             : 
      16             : #include "gdalvectorpipelinestepalgorithm.h"
      17             : 
      18             : //! @cond Doxygen_Suppress
      19             : 
      20             : /************************************************************************/
      21             : /*                      GDALVectorFilterAlgorithm                       */
      22             : /************************************************************************/
      23             : 
      24             : class GDALVectorFilterAlgorithm /* non final */
      25             :     : public GDALVectorPipelineStepAlgorithm
      26             : {
      27             :   public:
      28             :     static constexpr const char *NAME = "filter";
      29             :     static constexpr const char *DESCRIPTION = "Filter a vector dataset.";
      30             :     static constexpr const char *HELP_URL = "/programs/gdal_vector_filter.html";
      31             : 
      32             :     explicit GDALVectorFilterAlgorithm(bool standaloneStep = false);
      33             : 
      34          12 :     bool IsNativelyStreamingCompatible() const override
      35             :     {
      36          12 :         return !m_updateExtent;
      37             :     }
      38             : 
      39             :   private:
      40             :     bool RunStep(GDALPipelineStepRunContext &ctxt) override;
      41             : 
      42             :     std::vector<std::string>
      43             :     CompleteWhere(const GDALAlgorithmArg &layerArg,
      44             :                   const std::string &currentValue) const;
      45             : 
      46             :     std::string m_activeLayer{};
      47             :     std::vector<double> m_bbox{};
      48             :     std::string m_where{};
      49             :     bool m_updateExtent = false;
      50             : };
      51             : 
      52             : /************************************************************************/
      53             : /*                 GDALVectorFilterAlgorithmStandalone                  */
      54             : /************************************************************************/
      55             : 
      56         130 : class GDALVectorFilterAlgorithmStandalone final
      57             :     : public GDALVectorFilterAlgorithm
      58             : {
      59             :   public:
      60          65 :     GDALVectorFilterAlgorithmStandalone()
      61          65 :         : GDALVectorFilterAlgorithm(/* standaloneStep = */ true)
      62             :     {
      63          65 :     }
      64             : 
      65             :     ~GDALVectorFilterAlgorithmStandalone() override;
      66             : };
      67             : 
      68             : //! @endcond
      69             : 
      70             : #endif /* GDALALG_VECTOR_FILTER_INCLUDED */