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         int i;
  52
  53         if (!str)
  54                 return;
  55
  56         if (--str->ref > 0)
  57                 return;
  58
  59         free(str->s);
  60         isl_ctx_deref(str->ctx);
  61         free(str);
  62 }
  63
  64 static void *isl_obj_str_copy(void *v)
  65 {
  66         return isl_str_copy((isl_str *)v);
  67 }
  68
  69 static void isl_obj_str_free(void *v)
  70 {
  71         isl_str_free((isl_str *)v);
  72 }
  73
  74 static __isl_give isl_printer *isl_obj_str_print(__isl_take isl_printer *p,
  75         void *v)
  76 {
  77         isl_str *str = (isl_str *)v;
  78
  79         p = isl_printer_print_str(p, "\"");
  80         p = isl_printer_print_str(p, str->s);
  81         p = isl_printer_print_str(p, "\"");
  82
  83         return p;
  84 }
  85
  86 __isl_give isl_str *isl_str_concat(__isl_take isl_str *str1,
  87         __isl_take isl_str *str2)
  88 {
  89         int i;
  90         isl_str *str = NULL;
  91         size_t len1, len2;
  92
  93         if (!str1 || !str2)
  94                 goto error;
  95
  96         str = isl_str_alloc(str1->ctx);
  97         if (!str)
  98                 goto error;
  99
 100         len1 = strlen(str1->s);
 101         len2 = strlen(str2->s);
 102         str->s = isl_alloc_array(str1->ctx, char, len1 + len2 + 1);
 103         if (!str->s)
 104                 goto error;
 105
 106         memcpy(str->s, str1->s, len1);
 107         memcpy(str->s + len1, str2->s, len2);
 108         str->s[len1 + len2] = '\0';
 109
 110         isl_str_free(str1);
 111         isl_str_free(str2);
 112
 113         return str;
 114 error:
 115         isl_str_free(str1);
 116         isl_str_free(str2);
 117         isl_str_free(str);
 118         return NULL;
 119 }
 120
 121 static void *isl_obj_str_add(void *v1, void *v2)
 122 {
 123         isl_str *str1 = (isl_str *)v1;
 124         isl_str *str2 = (isl_str *)v2;
 125
 126         return isl_str_concat(str1, str2);
 127 }
 128
 129 struct isl_obj_vtable isl_obj_str_vtable = {
 130         isl_obj_str_copy,
 131         isl_obj_str_add,
 132         isl_obj_str_print,
 133         isl_obj_str_free
 134 };