如何从 BigQuery JavaScript UDF 为字符串化几何集合中的每个要素创建几何？

Question

我是 UDF 函数的新手，我创建了一个 BigQuery UDF，它采用多边形几何并用它创建点。我正在尝试绘制点密度图（将多边形+人口数转换为点）。我改编了this blog post 的代码。因为 bigQuery 没有记录变量的方法，所以我一直在测试 in this codepen。

我正处于该功能似乎正常工作的地步。输出是点的几何集合。它在 bigquery 文档中说 st_geogfromgeojson 可以接受几何集合。

我的 UDF 返回一个字符串化的几何集合。

但我不知道为什么st_geogfromgeojson 不起作用。我不知道我是否只是没有取消嵌套或什么。

CREATE TEMP FUNCTION myFunc(feature string, ethnicity_column FLOAT64, year INT64)
  RETURNS string
  LANGUAGE js
  OPTIONS (
    library=["https://storage.googleapis.com.../d3.js","https://storage.googleapis.com/.../turf.min.js","https://storage.googleapis.com/.../wellknown.js"]
  )
  AS
    """
          
        if (feature === undefined || feature === null) return;
       
        
        var feature_parsed = wellknown.parse(feature)
       
        const bounds = turf.bbox(feature_parsed);
        
        const populationData = Math.round(ethnicity_column / 10);
        if (!populationData) return;
       
         const x_min = bounds[0];
         const y_min = bounds[1];
         const x_max = bounds[2];
         const y_max = bounds[3];
  
        let hits = 0;
        let count = 0;
        const limit = populationData * 10; // limit test to 10x the population.
        let points = [];
        while (hits < populationData - 1 && count < limit) {
          const lat = y_min + Math.random() * (y_max - y_min);
          const lng = x_min + Math.random() * (x_max - x_min);
          const randomPoint = turf.point([lng, lat]);
          if (turf.booleanPointInPolygon(randomPoint, feature_parsed)) {
            points.push(randomPoint);
            hits++;
          }
          count++;
        }
        return JSON.stringify((turf.geometryCollection(points)));
        
       // return JSON.stringify(points)

    """;


SELECT ST_GEOGFROMGEOJSON(JSON_EXTRACT((myFunc(st_astext(geom), white_pop, 2018)),'$')) FROM `myteam.kyle_data.blockgroups_with_acs` 

但我总是遇到随机错误，比如我没有正确使用函数

我愿意接受所有建议。为简单起见，我返回一个字符串，但也许我需要使用 STRUCT。也许我应该从创建积分中删除草皮？我一定在这里遗漏了什么。

Answer 1

GeoJson 有两种不同的集合类型：

• 一个是GeometryCollection - 这是一个描述几何集合的几何，例如点和多边形的联合将是一个 GeometryCollection。
• 另一个是FeatureCollection - 特征集合、具有各种属性的对象，包括geometry（可以是几何集合或任何其他几何）和其他用户定义的属性。

turf.geometryCollection 似乎返回 Feature: https://www.npmjs.com/package/turf-geometrycollection（这就是它也接受 properties 参数的原因）。

ST_GEOGFROMGEOJSON构造一个几何体，它支持GeometryCollection，但不支持FeatureCollection或turf返回的单数Feature。

您可以从该功能中提取geometry，并将其传递给ST_GEOGFROMGEOJSON。我认为只使用 JSON 选择器 $.geometry 而不是 $ 就足够了。

Answer 2

解决它的两件事：

• 返回 array<String> 而不是字符串
• 在select 中交叉加入 UNNEST

CREATE TEMP FUNCTION myFunc(feature string, ethnicity_column FLOAT64)
  RETURNS array<String>
  LANGUAGE js
  OPTIONS (
    library=["https://storage.googleapis.com/../d3.js","https://storage.googleapis.com/.../turf.min.js","https://storage.googleapis.com/.../wellknown.js"]
  )
  AS
    """
        geopath = d3.geoPath()
        if (feature === undefined || feature === null) return;
        var feature_parsed = wellknown.parse(feature)
        const bounds = turf.bbox(feature_parsed);
        const populationData = Math.round(ethnicity_column / 10);
        if (!populationData) return;
         const x_min = bounds[0];
         const y_min = bounds[1];
         const x_max = bounds[2];
         const y_max = bounds[3];
        let hits = 0;
        let count = 0;
        const limit = populationData; // limit test to 10x the population.
        let points = [];
        while (hits < populationData - 1 && count < limit) {
          const lat = y_min + Math.random() * (y_max - y_min);
          const lng = x_min + Math.random() * (x_max - x_min);
          const randomPoint = turf.point([lng, lat]);
          if (turf.booleanPointInPolygon(randomPoint, feature_parsed)) {
            points.push('POINT ('+lng+' '+lat+')');
            hits++;
          }
          count++;
        }
        return points;
    """;
SELECT st_geogfromtext(points) as the_geom,  'white' as ethnicity  from (SELECT (myFunc(st_astext(geom), white_pop)) as points FROM `tableonmybq`) CROSS JOIN UNNEST(points) as points