Added PointView spatial reference getter methods

Author: jimmysoda

source/pdal/capi/PointView.cpp

@@ -52,6 +52,56 @@ namespace pdal
 
 				return ptr;
 			}
+
+			size_t PDALGetPointViewProj4(PDALPointViewPtr view, char *proj, size_t size)
+			{
+				pdal::capi::PointView *wrapper = reinterpret_cast<pdal::capi::PointView *>(view);
+
+				size_t result = 0;
+
+				if (size > 0 && proj)
+				{
+					proj[0] = '\0';
+					proj[size-1] = '\0';
+
+					if (wrapper && *wrapper)
+					{
+						std::string s = (*wrapper)->spatialReference().getProj4();
+						std::strncpy(proj, s.c_str(), size - 1);
+						result = std::min(s.length(), size);
+					}
+				}
+
+				return result;
+			}
+
+			size_t PDALGetPointViewWkt(PDALPointViewPtr view, char *wkt, size_t size, bool pretty)
+			{
+				pdal::capi::PointView *wrapper = reinterpret_cast<pdal::capi::PointView *>(view);
+
+				size_t result = 0;
+
+				if (size > 0 && wkt)
+				{
+					wkt[0] = '\0';
+					wkt[size-1] = '\0';
+
+					if (wrapper && *wrapper)
+					{
+						std::string s = (*wrapper)->spatialReference().getWKT();
+
+						 if (pretty)
+						 {
+							 s = SpatialReference::prettyWkt(s);
+						 }
+
+						std::strncpy(wkt, s.c_str(), size - 1);
+						result = std::min(s.length(), size);
+					}
+				}
+
+				return result;
+			}
 		}
 	}
 }

source/pdal/capi/PointView.h

@@ -30,7 +30,9 @@ namespace pdal
 			 */
 			PDAL_C_API PDALPointViewPtr PDALClonePointView(PDALPointViewPtr view);
 
+			PDAL_C_API size_t PDALGetPointViewProj4(PDALPointViewPtr view, char *proj, size_t size);
 
+			PDAL_C_API size_t PDALGetPointViewWkt(PDALPointViewPtr view, char *wkt, size_t size, bool pretty);
 
 
 			/**

tests/pdal/capi/PointViewTest.c.in

@@ -150,6 +150,63 @@ TEST testPDALClonePointView(void)
 	PASS();
 }
 
+TEST testPDALGetPointViewProj4(void)
+{
+	PDALResetPointViewCollection(gPointViewCollection);
+	bool hasNext = PDALHasNextPointView(gPointViewCollection);
+	ASSERT_FALSE(!hasNext);
+
+	PDALPointViewPtr view = PDALGetNextPointView(gPointViewCollection);
+	ASSERT_FALSE(view == NULL);
+
+	char proj[1024];
+	size_t size = PDALGetPointViewProj4(NULL, proj, 1024);
+	ASSERT_FALSE(size > 0);
+	ASSERT_FALSE(proj[0] != '\0');
+
+	size = PDALGetPointViewProj4(view, NULL, 1024);
+	ASSERT_FALSE(size > 0);
+
+	size = PDALGetPointViewProj4(view, proj, 0);
+	ASSERT_FALSE(size > 0);
+
+	// TODO Enable when test incorporates a pipeline with projection info
+	// size = PDALGetPointViewProj4(view, proj, 1024);
+	// ASSERT_FALSE(size == 0 || size > 1024);
+	// ASSERT_FALSE(proj[0] == '\0');
+	// printf("\n%s\n", proj);
+
+	PASS();
+}
+
+TEST testPDALGetPointViewWkt(void)
+{
+	PDALResetPointViewCollection(gPointViewCollection);
+	bool hasNext = PDALHasNextPointView(gPointViewCollection);
+	ASSERT_FALSE(!hasNext);
+
+	PDALPointViewPtr view = PDALGetNextPointView(gPointViewCollection);
+	ASSERT_FALSE(view == NULL);
+
+	char wkt[1024];
+	size_t size = PDALGetPointViewWkt(NULL, wkt, 1024, false);
+	ASSERT_FALSE(size > 0);
+	ASSERT_FALSE(wkt[0] != '\0');
+
+	size = PDALGetPointViewWkt(view, NULL, 1024, false);
+	ASSERT_FALSE(size > 0);
+
+	size = PDALGetPointViewWkt(view, wkt, 0, false);
+	ASSERT_FALSE(size > 0);
+
+	// TODO Enable when test incorporates a pipeline with projection info
+	// size = PDALGetPointViewWkt(view, wkt, 1024, false);
+	// ASSERT_FALSE(size == 0 || size > 1024);
+	// ASSERT_FALSE(wkt[0] == '\0');
+	// printf("\n%s\n", wkt);
+
+	PASS();
+}
 
 GREATEST_SUITE(PointViewTest)
 {
@@ -160,6 +217,8 @@ GREATEST_SUITE(PointViewTest)
 	RUN_TEST(testPDALGetPointViewSize);
 	RUN_TEST(testPDALIsPointViewEmpty);
 	RUN_TEST(testPDALClonePointView);
+	RUN_TEST(testPDALGetPointViewProj4);
+	RUN_TEST(testPDALGetPointViewWkt);
 
 	SET_SETUP(NULL, NULL);
 	SET_TEARDOWN(NULL, NULL);