GeoRasterTableFunctions.h File Reference

#include <vector>
#include "QueryEngine/heavydbTypes.h"
#include "GeoRasterTableFunctions.cpp"

Include dependency graph for GeoRasterTableFunctions.h:

Go to the source code of this file.

Classes
struct	ComputeAgg< AggType >
struct	ComputeAgg< RasterAggType::COUNT >
struct	ComputeAgg< RasterAggType::MAX >
struct	ComputeAgg< RasterAggType::MIN >
struct	ComputeAgg< RasterAggType::SUM >
struct	ComputeAgg< RasterAggType::BOX_AVG >
struct	GeoRaster< T, Z >

Enumerations
enum	RasterAggType {   RasterAggType::COUNT, RasterAggType::MIN, RasterAggType::MAX, RasterAggType::SUM,   RasterAggType::AVG, RasterAggType::GAUSS_AVG, RasterAggType::BOX_AVG, RasterAggType::INVALID }

Functions
RasterAggType	get_raster_agg_type (const std::string &agg_type_str, const bool is_fill_agg)
template<typename T , typename Z >
TEMPLATE_NOINLINE int32_t	geo_rasterize_impl (TableFunctionManager &mgr, const Column< T > &input_x, const Column< T > &input_y, const Column< Z > &input_z, const RasterAggType raster_agg_type, const RasterAggType raster_fill_agg_type, const T bin_dim_meters, const bool geographic_coords, const int64_t neighborhood_fill_radius, const bool fill_only_nulls, Column< T > &output_x, Column< T > &output_y, Column< Z > &output_z)
template<typename T , typename Z >
TEMPLATE_NOINLINE int32_t	tf_geo_rasterize__cpu_template (TableFunctionManager &mgr, const Column< T > &input_x, const Column< T > &input_y, const Column< Z > &input_z, const TextEncodingNone &agg_type_str, const T bin_dim_meters, const bool geographic_coords, const int64_t neighborhood_fill_radius, const bool fill_only_nulls, Column< T > &output_x, Column< T > &output_y, Column< Z > &output_z)
template<typename T , typename Z >
TEMPLATE_NOINLINE int32_t	tf_geo_rasterize__cpu_template (TableFunctionManager &mgr, const Column< T > &input_x, const Column< T > &input_y, const Column< Z > &input_z, const TextEncodingNone &agg_type_str, const TextEncodingNone &fill_agg_type_str, const T bin_dim_meters, const bool geographic_coords, const int64_t neighborhood_fill_radius, const bool fill_only_nulls, Column< T > &output_x, Column< T > &output_y, Column< Z > &output_z)
template<typename T , typename Z >
TEMPLATE_NOINLINE int32_t	tf_geo_rasterize__cpu_template (TableFunctionManager &mgr, const Column< T > &input_x, const Column< T > &input_y, const Column< Z > &input_z, const TextEncodingNone &agg_type_str, const TextEncodingNone &fill_agg_type_str, const T bin_dim_meters, const bool geographic_coords, const int64_t neighborhood_fill_radius, const bool fill_only_nulls, const T x_min, const T x_max, const T y_min, const T y_max, Column< T > &output_x, Column< T > &output_y, Column< Z > &output_z)
template<typename T , typename Z >
TEMPLATE_NOINLINE int32_t	tf_geo_rasterize_slope__cpu_template (TableFunctionManager &mgr, const Column< T > &input_x, const Column< T > &input_y, const Column< Z > &input_z, const TextEncodingNone &agg_type_str, const T bin_dim_meters, const bool geographic_coords, const int64_t neighborhood_fill_radius, const bool fill_only_nulls, const bool compute_slope_in_degrees, Column< T > &output_x, Column< T > &output_y, Column< Z > &output_z, Column< Z > &output_slope, Column< Z > &output_aspect)

Enumeration Type Documentation

enum RasterAggType

strong

Enumerator
COUNT
MIN
MAX
SUM
AVG
GAUSS_AVG
BOX_AVG
INVALID

Definition at line 25 of file GeoRasterTableFunctions.h.

{ COUNT, MIN, MAX, SUM, AVG, GAUSS_AVG, BOX_AVG, INVALID };

Function Documentation

template<typename T , typename Z >

TEMPLATE_NOINLINE int32_t geo_rasterize_impl	(	TableFunctionManager &	mgr,
		const Column< T > &	input_x,
		const Column< T > &	input_y,
		const Column< Z > &	input_z,
		const RasterAggType	raster_agg_type,
		const RasterAggType	raster_fill_agg_type,
		const T	bin_dim_meters,
		const bool	geographic_coords,
		const int64_t	neighborhood_fill_radius,
		const bool	fill_only_nulls,
		Column< T > &	output_x,
		Column< T > &	output_y,
		Column< Z > &	output_z
	)

Definition at line 267 of file GeoRasterTableFunctions.h.

References GeoRaster< T, Z >::fill_bins_from_neighbors(), GeoRaster< T, Z >::outputDenseColumns(), and GeoRaster< T, Z >::setMetadata().

Referenced by tf_geo_rasterize__cpu_template().

                                                                  {
  GeoRaster<T, Z> geo_raster(input_x,
                             input_y,
                             input_z,
                             raster_agg_type,
                             bin_dim_meters,
                             geographic_coords,
                             true);

  geo_raster.setMetadata(mgr);

  if (neighborhood_fill_radius > 0) {
    geo_raster.fill_bins_from_neighbors(
        neighborhood_fill_radius, fill_only_nulls, raster_fill_agg_type);
  }

  return geo_raster.outputDenseColumns(mgr, output_x, output_y, output_z);
}

Here is the call graph for this function:

Here is the caller graph for this function:

RasterAggType get_raster_agg_type	(	const std::string &	agg_type_str,
		const bool	is_fill_agg
	)

Definition at line 27 of file GeoRasterTableFunctions.h.

References AVG, BOX_AVG, COUNT, GAUSS_AVG, INVALID, MAX, MIN, SUM, and to_upper().

Referenced by tf_geo_rasterize__cpu_template(), and tf_geo_rasterize_slope__cpu_template().

                                                          {
  const auto upper_agg_type_str = to_upper(agg_type_str);
  const static std::map<std::string, RasterAggType> agg_type_map = {
      {"COUNT", RasterAggType::COUNT},
      {"MIN", RasterAggType::MIN},
      {"MAX", RasterAggType::MAX},
      {"SUM", RasterAggType::SUM},
      {"AVG", RasterAggType::AVG},
      {"GAUSS_AVG", RasterAggType::GAUSS_AVG},
      {"BOX_AVG", RasterAggType::BOX_AVG}};
  const auto itr = agg_type_map.find(upper_agg_type_str);
  if (itr == agg_type_map.end()) {
    return RasterAggType::INVALID;
  }
  if (is_fill_agg && itr->second == RasterAggType::AVG) {
    return RasterAggType::GAUSS_AVG;
  } else if (!is_fill_agg && (itr->second == RasterAggType::BOX_AVG ||
                              itr->second == RasterAggType::GAUSS_AVG)) {
    // GAUSS_AVG and BOX_AVG are fill-only aggregates
    return RasterAggType::INVALID;
  }
  return itr->second;
}

Here is the call graph for this function:

Here is the caller graph for this function:

template<typename T , typename Z >

TEMPLATE_NOINLINE int32_t tf_geo_rasterize__cpu_template	(	TableFunctionManager &	mgr,
		const Column< T > &	input_x,
		const Column< T > &	input_y,
		const Column< Z > &	input_z,
		const TextEncodingNone &	agg_type_str,
		const T	bin_dim_meters,
		const bool	geographic_coords,
		const int64_t	neighborhood_fill_radius,
		const bool	fill_only_nulls,
		Column< T > &	output_x,
		Column< T > &	output_y,
		Column< Z > &	output_z
	)

Definition at line 311 of file GeoRasterTableFunctions.h.

References GAUSS_AVG, geo_rasterize_impl(), get_raster_agg_type(), TextEncodingNone::getString(), and INVALID.

                                                    {
  const auto raster_agg_type = get_raster_agg_type(agg_type_str, false);
  if (raster_agg_type == RasterAggType::INVALID) {
    const std::string error_msg =
        "Invalid Raster Aggregate Type: " + agg_type_str.getString();
    return mgr.ERROR_MESSAGE(error_msg);
  }
  return geo_rasterize_impl(mgr,
                            input_x,
                            input_y,
                            input_z,
                            raster_agg_type,
                            RasterAggType::GAUSS_AVG,
                            bin_dim_meters,
                            geographic_coords,
                            neighborhood_fill_radius,
                            fill_only_nulls,
                            output_x,
                            output_y,
                            output_z);
}

Here is the call graph for this function:

template<typename T , typename Z >

TEMPLATE_NOINLINE int32_t tf_geo_rasterize__cpu_template	(	TableFunctionManager &	mgr,
		const Column< T > &	input_x,
		const Column< T > &	input_y,
		const Column< Z > &	input_z,
		const TextEncodingNone &	agg_type_str,
		const TextEncodingNone &	fill_agg_type_str,
		const T	bin_dim_meters,
		const bool	geographic_coords,
		const int64_t	neighborhood_fill_radius,
		const bool	fill_only_nulls,
		Column< T > &	output_x,
		Column< T > &	output_y,
		Column< Z > &	output_z
	)

Definition at line 357 of file GeoRasterTableFunctions.h.

References geo_rasterize_impl(), get_raster_agg_type(), TextEncodingNone::getString(), and INVALID.

                                                    {
  const auto raster_agg_type = get_raster_agg_type(agg_type_str, false);
  if (raster_agg_type == RasterAggType::INVALID) {
    const std::string error_msg =
        "Invalid Raster Aggregate Type: " + agg_type_str.getString();
    return mgr.ERROR_MESSAGE(error_msg);
  }
  const auto raster_fill_agg_type = get_raster_agg_type(fill_agg_type_str, true);
  if (raster_fill_agg_type == RasterAggType::INVALID) {
    const std::string error_msg =
        "Invalid Raster Fill Aggregate Type: " + fill_agg_type_str.getString();
    return mgr.ERROR_MESSAGE(error_msg);
  }
  return geo_rasterize_impl(mgr,
                            input_x,
                            input_y,
                            input_z,
                            raster_agg_type,
                            raster_fill_agg_type,
                            bin_dim_meters,
                            geographic_coords,
                            neighborhood_fill_radius,
                            fill_only_nulls,
                            output_x,
                            output_y,
                            output_z);
}

Here is the call graph for this function:

template<typename T , typename Z >

TEMPLATE_NOINLINE int32_t tf_geo_rasterize__cpu_template	(	TableFunctionManager &	mgr,
		const Column< T > &	input_x,
		const Column< T > &	input_y,
		const Column< Z > &	input_z,
		const TextEncodingNone &	agg_type_str,
		const TextEncodingNone &	fill_agg_type_str,
		const T	bin_dim_meters,
		const bool	geographic_coords,
		const int64_t	neighborhood_fill_radius,
		const bool	fill_only_nulls,
		const T	x_min,
		const T	x_max,
		const T	y_min,
		const T	y_max,
		Column< T > &	output_x,
		Column< T > &	output_y,
		Column< Z > &	output_z
	)

Definition at line 411 of file GeoRasterTableFunctions.h.

References GeoRaster< T, Z >::fill_bins_from_neighbors(), get_raster_agg_type(), TextEncodingNone::getString(), INVALID, GeoRaster< T, Z >::outputDenseColumns(), and GeoRaster< T, Z >::setMetadata().

                                                    {
  const auto raster_agg_type = get_raster_agg_type(agg_type_str, false);
  if (raster_agg_type == RasterAggType::INVALID) {
    const std::string error_msg =
        "Invalid Raster Aggregate Type: " + agg_type_str.getString();
    return mgr.ERROR_MESSAGE(error_msg);
  }
  const auto raster_fill_agg_type = get_raster_agg_type(fill_agg_type_str, true);
  if (raster_fill_agg_type == RasterAggType::INVALID) {
    const std::string error_msg =
        "Invalid Raster Fill Aggregate Type: " + fill_agg_type_str.getString();
    return mgr.ERROR_MESSAGE(error_msg);
  }

  GeoRaster<T, Z> geo_raster(input_x,
                             input_y,
                             input_z,
                             raster_agg_type,
                             bin_dim_meters,
                             geographic_coords,
                             true,
                             x_min,
                             x_max,
                             y_min,
                             y_max);

  geo_raster.setMetadata(mgr);

  if (neighborhood_fill_radius > 0) {
    geo_raster.fill_bins_from_neighbors(
        neighborhood_fill_radius, fill_only_nulls, raster_fill_agg_type);
  }

  return geo_raster.outputDenseColumns(mgr, output_x, output_y, output_z);
}

Here is the call graph for this function:

template<typename T , typename Z >

TEMPLATE_NOINLINE int32_t tf_geo_rasterize_slope__cpu_template	(	TableFunctionManager &	mgr,
		const Column< T > &	input_x,
		const Column< T > &	input_y,
		const Column< Z > &	input_z,
		const TextEncodingNone &	agg_type_str,
		const T	bin_dim_meters,
		const bool	geographic_coords,
		const int64_t	neighborhood_fill_radius,
		const bool	fill_only_nulls,
		const bool	compute_slope_in_degrees,
		Column< T > &	output_x,
		Column< T > &	output_y,
		Column< Z > &	output_z,
		Column< Z > &	output_slope,
		Column< Z > &	output_aspect
	)

Definition at line 475 of file GeoRasterTableFunctions.h.

References GeoRaster< T, Z >::calculate_slope_and_aspect(), GeoRaster< T, Z >::fill_bins_from_neighbors(), GAUSS_AVG, get_raster_agg_type(), TextEncodingNone::getString(), INVALID, GeoRaster< T, Z >::outputDenseColumns(), and GeoRaster< T, Z >::setMetadata().

                                                               {
  const auto raster_agg_type = get_raster_agg_type(agg_type_str, false);
  if (raster_agg_type == RasterAggType::INVALID) {
    const std::string error_msg =
        "Invalid Raster Aggregate Type: " + agg_type_str.getString();
    return mgr.ERROR_MESSAGE(error_msg);
  }

  GeoRaster<T, Z> geo_raster(input_x,
                             input_y,
                             input_z,
                             raster_agg_type,
                             bin_dim_meters,
                             geographic_coords,
                             true);
  geo_raster.setMetadata(mgr);

  if (neighborhood_fill_radius > 0) {
    geo_raster.fill_bins_from_neighbors(
        neighborhood_fill_radius, fill_only_nulls, RasterAggType::GAUSS_AVG);
  }

  const size_t output_rows =
      geo_raster.outputDenseColumns(mgr, output_x, output_y, output_z);
  geo_raster.calculate_slope_and_aspect(
      output_slope, output_aspect, compute_slope_in_degrees);
  return output_rows;
}

Here is the call graph for this function: