debugged getgeometry and getgeometryid

src/pg/sql/42_observatory_exploration.sql

@@ -0,0 +1,29 @@
+
+-- return a table that contains a string match based on input
+-- TODO: implement search for timespan
+
+CREATE OR REPLACE FUNCTION OBS_SearchTables(
+  search_term text,
+  time_span text DEFAULT '2009 - 2013'
+)
+RETURNS text[]
+As $$
+DECLARE
+  out_var text[];
+BEGIN
+
+  EXECUTE
+  'SELECT array_agg(tablename)
+FROM observatory.obs_table t JOIN observatory.obs_column_table ct
+  ON ct.table_id = t.id
+JOIN observatory.obs_column c
+  ON ct.column_id = c.id
+WHERE c.type ILIKE ''geometry''
+AND c.id = $1'
+  INTO out_var
+  USING search_term;
+
+  RETURN out_var;
+
+END;
+$$ LANGUAGE plpgsql;

src/pg/sql/44_observatory_geometries.sql

@@ -0,0 +1,159 @@
+-- Returns the polygon(s) that overlap with the input geometry.
+-- Input:
+-- :param geom geometry: input geometry
+-- :param geometry_level text: table to get polygon from (can be approximate name)
+-- :param use_literal boolean: use the literal table name (defaults to true)
+
+-- From an input point geometry, find the boundary which intersects with the centroid of the input geometry
+
+CREATE OR REPLACE FUNCTION Andy_OBS_GetGeometry(
+  geom geometry(Geometry, 4326),
+  geometry_level text DEFAULT '"us.census.tiger".census_tract', -- TODO: from a specified column id list (e.g., list of available catalog, see OBS_List)
+  time_span text DEFAULT '2009 - 2013')
+  RETURNS geometry(Geometry, 4326)
+AS $$
+DECLARE
+  boundary geometry(Geometry, 4326);
+  target_table text;
+  target_table_list text[];
+BEGIN
+
+  -- TODO: Check if SRID = 4326, if not transform?
+
+  -- if not a point, raise error
+  IF ST_GeometryType(geom) != 'ST_Point'
+  THEN
+    RAISE EXCEPTION 'Invalid geometry type (%), expecting ''ST_Point''', ST_GeometryType(geom);
+  END IF;
+
+  target_table_list := OBS_SearchTables(geometry_level, time_span);
+
+  -- if no tables are found, raise notice and return null
+  IF array_length(target_table_list, 1) IS NULL
+  THEN
+    RAISE NOTICE 'No boundaries found for ''%'' in ''%''', ST_AsText(geom), geometry_level;
+    RETURN NULL::geometry;
+  ELSE
+  -- else, choose first result
+    target_table = target_table_list[1];
+  END IF;
+
+  RAISE NOTICE 'target_table: %', target_table;
+
+  -- return the first boundary in intersections
+  EXECUTE format(
+    'SELECT t.the_geom
+     FROM observatory.%s As t
+     WHERE ST_Intersects($1, t.the_geom)
+     LIMIT 1', target_table)
+  INTO boundary
+  USING geom;
+
+  RETURN boundary;
+
+END;
+$$ LANGUAGE plpgsql;
+
+CREATE OR REPLACE FUNCTION ANDY_OBS_GetGeometryId(
+  geom geometry(Geometry, 4326),
+  geometry_level text DEFAULT '"us.census.tiger".census_tract',
+  time_span text DEFAULT '2009 - 2013'
+)
+RETURNS text
+AS $$
+DECLARE
+  output_id text;
+  target_table text;
+  target_table_list text[];
+BEGIN
+
+  -- If not point, raise error
+  IF ST_GeometryType(geom) != 'ST_Point'
+  THEN
+    RAISE EXCEPTION 'Error: Invalid geometry type (%), expecting ''ST_Point''', ST_GeometryType(geom);
+  END IF;
+
+  target_table_list := OBS_SearchTables(geometry_level, time_span);
+
+  -- if no tables are found, raise error
+  IF array_length(target_table_list, 1) IS NULL
+  THEN
+    RAISE NOTICE 'Error: No boundaries found for ''%''', geometry_level;
+    RETURN NULL::text;
+  ELSE
+    target_table = target_table_list[1];
+  END IF;
+
+  RAISE NOTICE 'target_table: %', target_table;
+
+  -- return name of geometry id column
+  EXECUTE format(
+    'SELECT t.geoid
+     FROM observatory.%s As t
+     WHERE ST_Intersects($1, t.the_geom)
+     LIMIT 1', target_table)
+  INTO output_id
+  USING geom;
+
+  RETURN output_id;
+
+END;
+$$ LANGUAGE plpgsql;
+
+--
+
+CREATE OR REPLACE FUNCTION OBS_GetGeometryById(
+  geom_ref text,      -- ex: '36047'
+  geometry_level text -- ex: '"us.census.tiger".county'
+)
+RETURNS geometry(geometry, 4326)
+AS $$
+DECLARE
+  boundary geometry;
+  target_table text;
+  geoid_colname text;
+  geom_colname text;
+BEGIN
+
+  EXECUTE
+  format(
+    $string$
+    SELECT geoid_ct.colname As geoid_colname,
+           tablename,
+           geom_ct.colname As geom_colname
+    FROM observatory.obs_column_table As geoid_ct,
+         observatory.obs_table As geom_t,
+         observatory.obs_column_table As geom_ct,
+         observatory.obs_column As geom_c
+    WHERE geoid_ct.column_id
+       IN (
+         SELECT source_id
+         FROM observatory.obs_column_to_column
+         WHERE reltype = 'geom_ref'
+           AND target_id = '%s'
+         )
+      AND geoid_ct.table_id = geom_t.id and
+          geom_t.id = geom_ct.table_id and
+          geom_ct.column_id = geom_c.id and
+          geom_c.type ilike 'geometry'
+    $string$, geometry_level
+  ) INTO geoid_colname, target_table, geom_colname;
+
+  IF target_table IS NULL
+  THEN
+    RAISE NOTICE 'No geometries found';
+    RETURN NULL::geometry;
+  END IF;
+
+  -- retrieve boundary
+  EXECUTE format(
+    'SELECT t.%s
+     FROM observatory.%s As t
+     WHERE t.%s = ''%s''
+     LIMIT 1', geom_colname, target_table, geoid_colname, geom_ref)
+  INTO boundary;
+
+  RETURN boundary;
+
+END;
+$$ LANGUAGE plpgsql;