isl_obj_str.c

   1 #include <string.h>
   2 #include <isl_obj_str.h>
   3
   4 __isl_give isl_str *isl_str_alloc(struct isl_ctx *ctx)
   5 {
   6         isl_str *str;
   7
   8         str = isl_alloc_type(ctx, struct isl_str);
   9         if (!str)
  10                 return NULL;
  11
  12         str->ctx = ctx;
  13         isl_ctx_ref(str->ctx);
  14         str->ref = 1;
  15
  16         str->s = NULL;
  17
  18         return str;
  19 }
  20
  21 __isl_give isl_str *isl_str_from_string(isl_ctx *ctx, __isl_take char *s)
  22 {
  23         isl_str *str;
  24
  25         if (!s)
  26                 return NULL;
  27
  28         str = isl_str_alloc(ctx);
  29         if (!str)
  30                 goto error;
  31
  32         str->s = s;
  33
  34         return str;
  35 error:
  36         free(s);
  37         return NULL;
  38 }
  39
  40 __isl_give isl_str *isl_str_copy(__isl_keep isl_str *str)
  41 {
  42         if (!str)
  43                 return NULL;
  44
  45         str->ref++;
  46         return str;
  47 }
  48
  49 void isl_str_free(__isl_take isl_str *str)
  50 {
  51         if (!str)
  52                 return;
  53
  54         if (--str->ref > 0)
  55                 return;
  56
  57         free(str->s);
  58         isl_ctx_deref(str->ctx);
  59         free(str);
  60 }
  61
  62 static void *isl_obj_str_copy(void *v)
  63 {
  64         return isl_str_copy((isl_str *)v);
  65 }
  66
  67 static void isl_obj_str_free(void *v)
  68 {
  69         isl_str_free((isl_str *)v);
  70 }
  71
  72 static __isl_give isl_printer *isl_obj_str_print(__isl_take isl_printer *p,
  73         void *v)
  74 {
  75         isl_str *str = (isl_str *)v;
  76
  77         p = isl_printer_print_str(p, "\"");
  78         p = isl_printer_print_str(p, str->s);
  79         p = isl_printer_print_str(p, "\"");
  80
  81         return p;
  82 }
  83
  84 __isl_give isl_str *isl_str_concat(__isl_take isl_str *str1,
  85         __isl_take isl_str *str2)
  86 {
  87         isl_str *str = NULL;
  88         size_t len1, len2;
  89
  90         if (!str1 || !str2)
  91                 goto error;
  92
  93         str = isl_str_alloc(str1->ctx);
  94         if (!str)
  95                 goto error;
  96
  97         len1 = strlen(str1->s);
  98         len2 = strlen(str2->s);
  99         str->s = isl_alloc_array(str1->ctx, char, len1 + len2 + 1);
 100         if (!str->s)
 101                 goto error;
 102
 103         memcpy(str->s, str1->s, len1);
 104         memcpy(str->s + len1, str2->s, len2);
 105         str->s[len1 + len2] = '\0';
 106
 107         isl_str_free(str1);
 108         isl_str_free(str2);
 109
 110         return str;
 111 error:
 112         isl_str_free(str1);
 113         isl_str_free(str2);
 114         isl_str_free(str);
 115         return NULL;
 116 }
 117
 118 static void *isl_obj_str_add(void *v1, void *v2)
 119 {
 120         isl_str *str1 = (isl_str *)v1;
 121         isl_str *str2 = (isl_str *)v2;
 122
 123         return isl_str_concat(str1, str2);
 124 }
 125
 126 struct isl_obj_vtable isl_obj_str_vtable = {
 127         isl_obj_str_copy,
 128         isl_obj_str_add,
 129         isl_obj_str_print,
 130         isl_obj_str_free
 131 };