add isl_token_get_val

Signed-off-by: Sven Verdoolaege <skimo@kotnet.org>

diff --git a/include/isl/stream.h b/include/isl/stream.h
index 3706bcd..58c03cc 100644 (file)
--- a/include/isl/stream.h
+++ b/include/isl/stream.h
@@ -14,6 +14,7 @@
 #include <isl/hash.h>
 #include <isl/aff_type.h>
 #include <isl/obj.h>
+#include <isl/val.h>
 
 #if defined(__cplusplus)
 extern "C" {
@@ -52,6 +53,7 @@ struct isl_token {
        } u;
 };
 
+__isl_give isl_val *isl_token_get_val(isl_ctx *ctx, struct isl_token *tok);
 void isl_token_free(struct isl_token *tok);
 
 struct isl_stream {

diff --git a/isl_stream.c b/isl_stream.c
index 68f6e52..7ead411 100644 (file)
--- a/isl_stream.c
+++ b/isl_stream.c
@@ -14,6 +14,7 @@
 #include <isl_stream_private.h>
 #include <isl/map.h>
 #include <isl/aff.h>
+#include <isl_val_private.h>
 
 struct isl_keyword {
        char                    *name;
@@ -80,6 +81,19 @@ struct isl_token *isl_token_new(isl_ctx *ctx,
        return tok;
 }
 
+/* Given a token of type ISL_TOKEN_VALUE, return the value it represents.
+ */
+__isl_give isl_val *isl_token_get_val(isl_ctx *ctx, struct isl_token *tok)
+{
+       if (!tok)
+               return NULL;
+       if (tok->type != ISL_TOKEN_VALUE)
+               isl_die(ctx, isl_error_invalid, "not a value token",
+                       return NULL);
+
+       return isl_val_int_from_isl_int(ctx, tok->u.v);
+}
+
 void isl_token_free(struct isl_token *tok)
 {
        if (!tok)