[binary-operators-draft]: udpate get_column_byte_width to support uns…

…igned types
rapidsai · Oct 24, 2018 · f7c3d4f · jrhemstad · Oct 24, 2018 · f7c3d4f
1 parent 3a74ff8
commit f7c3d4f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/include/gdf/cffi/types.h b/include/gdf/cffi/types.h
@@ -18,17 +18,17 @@ typedef enum {
     GDF_INT16,
     GDF_INT32,
     GDF_INT64,
-    GDF_UINT8,
-    GDF_UINT16,
-    GDF_UINT32,
-    GDF_UINT64,
     GDF_FLOAT32,
     GDF_FLOAT64,
     GDF_DATE32,   	/**< int32_t days since the UNIX epoch */
     GDF_DATE64,   	/**< int64_t milliseconds since the UNIX epoch */
     GDF_TIMESTAMP,	/**< Exact timestamp encoded with int64 since UNIX epoch (Default unit millisecond) */
     GDF_CATEGORY,
     GDF_STRING,
+    GDF_UINT8,
+    GDF_UINT16,
+    GDF_UINT32,
+    GDF_UINT64,
     N_GDF_TYPES, 	/* additional types should go BEFORE N_GDF_TYPES */
 } gdf_dtype;
 

diff --git a/src/column.cpp b/src/column.cpp
@@ -72,6 +72,18 @@ gdf_error get_column_byte_width(gdf_column * col, int * width){
 	case GDF_TIMESTAMP :
 		*width = 8;
 		break;
+	case GDF_UINT8 :
+		*width = 1;
+		break;
+	case GDF_UINT16:
+    	*width = 2;
+		break;
+	case GDF_UINT32:
+    	*width = 4;
+		break;
+	case GDF_UINT64:
+		*width = 8;
+		break;
 	default :
 		*width = -1;
 		return GDF_UNSUPPORTED_DTYPE;