feat: add int and bool support to grey dilate and erosion

fastmorph/fastmorphops.cpp

@@ -114,7 +114,8 @@ py::array multilabel_erode(const py::array &labels, const uint64_t threads) {
 
 // assumes fortran order
 py::array grey_dilate(const py::array &labels, const uint64_t threads) {
-	int width = labels.dtype().itemsize();
+	py::dtype dt = labels.dtype();
+	int width = dt.itemsize();
 
 	const uint64_t sx = labels.shape()[0];
 	const uint64_t sy = labels.shape()[1];
@@ -134,17 +135,36 @@ py::array grey_dilate(const py::array &labels, const uint64_t threads) {
 	);\
 	return to_numpy(reinterpret_cast<int_t*>(output_ptr), sx, sy, sz);
 
-	if (width == 1) {
-		GREY_DILATE_HELPER(uint8_t)
-	}
-	else if (width == 2) {
-		GREY_DILATE_HELPER(uint16_t)
+	if (dt.kind() == 'i') {
+		if (width == 1) {
+			GREY_DILATE_HELPER(int8_t)
+		}
+		else if (width == 2) {
+			GREY_DILATE_HELPER(int16_t)
+		}
+		else if (width == 4) {
+			GREY_DILATE_HELPER(int32_t)
+		}
+		else {
+			GREY_DILATE_HELPER(int64_t)
+		}
 	}
-	else if (width == 4) {
-		GREY_DILATE_HELPER(uint32_t)
+	else if (dt.kind() == 'b') {
+		GREY_DILATE_HELPER(uint8_t)
 	}
 	else {
-		GREY_DILATE_HELPER(uint64_t)
+		if (width == 1) {
+			GREY_DILATE_HELPER(uint8_t)
+		}
+		else if (width == 2) {
+			GREY_DILATE_HELPER(uint16_t)
+		}
+		else if (width == 4) {
+			GREY_DILATE_HELPER(uint32_t)
+		}
+		else {
+			GREY_DILATE_HELPER(uint64_t)
+		}
 	}
 #undef GREY_DILATE_HELPER
 }
@@ -186,6 +206,9 @@ py::array grey_erode(const py::array &labels, const uint64_t threads) {
 			GREY_ERODE_HELPER(int64_t)
 		}
 	}
+	else if (dt.kind() == 'b') {
+		GREY_ERODE_HELPER(uint8_t)
+	}
 	else {
 		if (width == 1) {
 			GREY_ERODE_HELPER(uint8_t)