isl_ctx.c

   1 /*
   2  * Copyright 2008-2009 Katholieke Universiteit Leuven
   3  *
   4  * Use of this software is governed by the MIT license
   5  *
   6  * Written by Sven Verdoolaege, K.U.Leuven, Departement
   7  * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
   8  */
   9
  10 #include <isl_ctx_private.h>
  11 #include <isl/vec.h>
  12 #include <isl_options_private.h>
  13
  14 #define __isl_calloc(type,size)         ((type *)calloc(1, size))
  15 #define __isl_calloc_type(type)         __isl_calloc(type,sizeof(type))
  16
  17 /* Check that the result of an allocation ("p") is not NULL and
  18  * complain if it is.
  19  * The only exception is when allocation size ("size") is equal to zero.
  20  */
  21 static void *check_non_null(isl_ctx *ctx, void *p, size_t size)
  22 {
  23         if (p || size == 0)
  24                 return p;
  25         isl_die(ctx, isl_error_alloc, "allocation failure", return NULL);
  26 }
  27
  28 /* Call malloc and complain if it fails.
  29  * If ctx is NULL, then return NULL.
  30  */
  31 void *isl_malloc_or_die(isl_ctx *ctx, size_t size)
  32 {
  33         return ctx ? check_non_null(ctx, malloc(size), size) : NULL;
  34 }
  35
  36 /* Call calloc and complain if it fails.
  37  * If ctx is NULL, then return NULL.
  38  */
  39 void *isl_calloc_or_die(isl_ctx *ctx, size_t nmemb, size_t size)
  40 {
  41         return ctx ? check_non_null(ctx, calloc(nmemb, size), nmemb) : NULL;
  42 }
  43
  44 /* Call realloc and complain if it fails.
  45  * If ctx is NULL, then return NULL.
  46  */
  47 void *isl_realloc_or_die(isl_ctx *ctx, void *ptr, size_t size)
  48 {
  49         return ctx ? check_non_null(ctx, realloc(ptr, size), size) : NULL;
  50 }
  51
  52 void isl_handle_error(isl_ctx *ctx, enum isl_error error, const char *msg,
  53         const char *file, int line)
  54 {
  55         if (!ctx)
  56                 return;
  57
  58         isl_ctx_set_error(ctx, error);
  59
  60         switch (ctx->opt->on_error) {
  61         case ISL_ON_ERROR_WARN:
  62                 fprintf(stderr, "%s:%d: %s\n", file, line, msg);
  63                 return;
  64         case ISL_ON_ERROR_CONTINUE:
  65                 return;
  66         case ISL_ON_ERROR_ABORT:
  67                 fprintf(stderr, "%s:%d: %s\n", file, line, msg);
  68                 abort();
  69                 return;
  70         }
  71 }
  72
  73 static struct isl_options *find_nested_options(struct isl_args *args,
  74         void *opt, struct isl_args *wanted)
  75 {
  76         int i;
  77         struct isl_options *options;
  78
  79         if (args == wanted)
  80                 return opt;
  81
  82         for (i = 0; args->args[i].type != isl_arg_end; ++i) {
  83                 struct isl_arg *arg = &args->args[i];
  84                 void *child;
  85
  86                 if (arg->type != isl_arg_child)
  87                         continue;
  88
  89                 if (arg->offset == (size_t) -1)
  90                         child = opt;
  91                 else
  92                         child = *(void **)(((char *)opt) + arg->offset);
  93
  94                 options = find_nested_options(arg->u.child.child,
  95                                                 child, wanted);
  96                 if (options)
  97                         return options;
  98         }
  99
 100         return NULL;
 101 }
 102
 103 static struct isl_options *find_nested_isl_options(struct isl_args *args,
 104         void *opt)
 105 {
 106         return find_nested_options(args, opt, &isl_options_args);
 107 }
 108
 109 void *isl_ctx_peek_options(isl_ctx *ctx, struct isl_args *args)
 110 {
 111         if (!ctx)
 112                 return NULL;
 113         if (args == &isl_options_args)
 114                 return ctx->opt;
 115         return find_nested_options(ctx->user_args, ctx->user_opt, args);
 116 }
 117
 118 isl_ctx *isl_ctx_alloc_with_options(struct isl_args *args, void *user_opt)
 119 {
 120         struct isl_ctx *ctx = NULL;
 121         struct isl_options *opt = NULL;
 122         int opt_allocated = 0;
 123
 124         if (!user_opt)
 125                 return NULL;
 126
 127         opt = find_nested_isl_options(args, user_opt);
 128         if (!opt) {
 129                 opt = isl_options_new_with_defaults();
 130                 if (!opt)
 131                         goto error;
 132                 opt_allocated = 1;
 133         }
 134
 135         ctx = __isl_calloc_type(struct isl_ctx);
 136         if (!ctx)
 137                 goto error;
 138
 139         if (isl_hash_table_init(ctx, &ctx->id_table, 0))
 140                 goto error;
 141
 142         ctx->stats = isl_calloc_type(ctx, struct isl_stats);
 143         if (!ctx->stats)
 144                 goto error;
 145
 146         ctx->user_args = args;
 147         ctx->user_opt = user_opt;
 148         ctx->opt_allocated = opt_allocated;
 149         ctx->opt = opt;
 150         ctx->ref = 0;
 151
 152         isl_int_init(ctx->zero);
 153         isl_int_set_si(ctx->zero, 0);
 154
 155         isl_int_init(ctx->one);
 156         isl_int_set_si(ctx->one, 1);
 157
 158         isl_int_init(ctx->two);
 159         isl_int_set_si(ctx->two, 2);
 160
 161         isl_int_init(ctx->negone);
 162         isl_int_set_si(ctx->negone, -1);
 163
 164         isl_int_init(ctx->normalize_gcd);
 165
 166         ctx->n_cached = 0;
 167         ctx->n_miss = 0;
 168
 169         ctx->error = isl_error_none;
 170
 171         ctx->operations = 0;
 172         isl_ctx_set_max_operations(ctx, ctx->opt->max_operations);
 173
 174         return ctx;
 175 error:
 176         isl_args_free(args, user_opt);
 177         if (opt_allocated)
 178                 isl_options_free(opt);
 179         free(ctx);
 180         return NULL;
 181 }
 182
 183 struct isl_ctx *isl_ctx_alloc()
 184 {
 185         struct isl_options *opt;
 186
 187         opt = isl_options_new_with_defaults();
 188
 189         return isl_ctx_alloc_with_options(&isl_options_args, opt);
 190 }
 191
 192 void isl_ctx_ref(struct isl_ctx *ctx)
 193 {
 194         ctx->ref++;
 195 }
 196
 197 void isl_ctx_deref(struct isl_ctx *ctx)
 198 {
 199         isl_assert(ctx, ctx->ref > 0, return);
 200         ctx->ref--;
 201 }
 202
 203 /* Print statistics on usage.
 204  */
 205 static void print_stats(isl_ctx *ctx)
 206 {
 207         fprintf(stderr, "operations: %lu\n", ctx->operations);
 208 }
 209
 210 void isl_ctx_free(struct isl_ctx *ctx)
 211 {
 212         if (!ctx)
 213                 return;
 214         if (ctx->ref != 0)
 215                 isl_die(ctx, isl_error_invalid,
 216                         "isl_ctx freed, but some objects still reference it",
 217                         return);
 218
 219         if (ctx->opt->print_stats)
 220                 print_stats(ctx);
 221
 222         isl_hash_table_clear(&ctx->id_table);
 223         isl_blk_clear_cache(ctx);
 224         isl_int_clear(ctx->zero);
 225         isl_int_clear(ctx->one);
 226         isl_int_clear(ctx->two);
 227         isl_int_clear(ctx->negone);
 228         isl_int_clear(ctx->normalize_gcd);
 229         isl_args_free(ctx->user_args, ctx->user_opt);
 230         if (ctx->opt_allocated)
 231                 isl_options_free(ctx->opt);
 232         free(ctx->stats);
 233         free(ctx);
 234 }
 235
 236 struct isl_options *isl_ctx_options(isl_ctx *ctx)
 237 {
 238         if (!ctx)
 239                 return NULL;
 240         return ctx->opt;
 241 }
 242
 243 enum isl_error isl_ctx_last_error(isl_ctx *ctx)
 244 {
 245         return ctx->error;
 246 }
 247
 248 void isl_ctx_reset_error(isl_ctx *ctx)
 249 {
 250         ctx->error = isl_error_none;
 251 }
 252
 253 void isl_ctx_set_error(isl_ctx *ctx, enum isl_error error)
 254 {
 255         if (ctx)
 256                 ctx->error = error;
 257 }
 258
 259 void isl_ctx_abort(isl_ctx *ctx)
 260 {
 261         if (ctx)
 262                 ctx->abort = 1;
 263 }
 264
 265 void isl_ctx_resume(isl_ctx *ctx)
 266 {
 267         if (ctx)
 268                 ctx->abort = 0;
 269 }
 270
 271 int isl_ctx_aborted(isl_ctx *ctx)
 272 {
 273         return ctx ? ctx->abort : -1;
 274 }
 275
 276 int isl_ctx_parse_options(isl_ctx *ctx, int argc, char **argv, unsigned flags)
 277 {
 278         if (!ctx)
 279                 return -1;
 280         return isl_args_parse(ctx->user_args, argc, argv, ctx->user_opt, flags);
 281 }
 282
 283 /* Set the maximal number of iterations of "ctx" to "max_operations".
 284  */
 285 void isl_ctx_set_max_operations(isl_ctx *ctx, unsigned long max_operations)
 286 {
 287         if (!ctx)
 288                 return;
 289         ctx->max_operations = max_operations;
 290 }
 291
 292 /* Return the maximal number of iterations of "ctx".
 293  */
 294 unsigned long isl_ctx_get_max_operations(isl_ctx *ctx)
 295 {
 296         return ctx ? ctx->max_operations : 0;
 297 }
 298
 299 /* Reset the number of operations performed by "ctx".
 300  */
 301 void isl_ctx_reset_operations(isl_ctx *ctx)
 302 {
 303         if (!ctx)
 304                 return;
 305         ctx->operations = 0;
 306 }