summaryrefslogtreecommitdiff
path: root/kit/string_ref.h
blob: e7ea80923969a3183424d6f326720c315ab836cc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
#ifndef KIT_STRING_REF_H
#define KIT_STRING_REF_H

#include "array_ref.h"

#include <assert.h>
#include <string.h>

#ifdef __cplusplus
extern "C" {
#endif

typedef KIT_AR(char) kit_str_t;

#if defined(__GNUC__) || defined(__clang__)
#  pragma GCC diagnostic push
#  pragma GCC diagnostic ignored "-Wunused-function"
#  pragma GCC diagnostic ignored "-Wunknown-pragmas"
#  pragma GCC            push_options
#  pragma GCC            optimize("O3")
#endif

static kit_str_t kit_str(i64 size, char const *static_string) {
  kit_str_t s = { .size = size, .values = (char *) static_string };
  return s;
}

enum {
  KIT_STR_NOT_FOUND = -1,
};

static i64 kit_str_find(kit_str_t a, kit_str_t b) {
  assert(a.size >= 0 && (a.size == 0 || a.values != NULL) &&
         b.size >= 0 && (b.size == 0 || b.values != NULL));
  if (a.size < 0 || (a.size != 0 && a.values == NULL) || b.size < 0 ||
      (b.size != 0 && b.values == NULL))
    return -1;
  for (i64 index = 0; index + b.size <= a.size; index++)
    if (KIT_AR_EQUAL(kit_str(b.size, a.values + index), b))
      return index;
  return -1;
}

static i64 kit_str_find_back(kit_str_t a, kit_str_t b) {
  assert(a.size >= 0 && (a.size == 0 || a.values != NULL) &&
         b.size >= 0 && (b.size == 0 || b.values != NULL));
  if (a.size < 0 || (a.size != 0 && a.values == NULL) || b.size < 0 ||
      (b.size != 0 && b.values == NULL))
    return -1;
  for (i64 index = a.size - b.size; index >= 0; index--)
    if (KIT_AR_EQUAL(kit_str(b.size, a.values + index), b))
      return index;
  return -1;
}

static i64 kit_str_find_n(kit_str_t a, kit_str_t b, i64 n) {
  assert(a.size >= 0 && (a.size == 0 || a.values != NULL) &&
         b.size >= 0 && (b.size == 0 || b.values != NULL));
  if (a.size < 0 || (a.size != 0 && a.values == NULL) || b.size < 0 ||
      (b.size != 0 && b.values == NULL))
    return -1;
  i64 count = 0;
  for (i64 index = 0; index + b.size <= a.size; index++)
    if (KIT_AR_EQUAL(kit_str(b.size, a.values + index), b)) {
      if (count == n)
        return index;
      else
        count++;
    }
  return -1;
}

static i64 kit_str_find_back_n(kit_str_t a, kit_str_t b, i64 n) {
  assert(a.size >= 0 && (a.size == 0 || a.values != NULL) &&
         b.size >= 0 && (b.size == 0 || b.values != NULL));
  if (a.size < 0 || (a.size != 0 && a.values == NULL) || b.size < 0 ||
      (b.size != 0 && b.values == NULL))
    return -1;
  i64 count = 0;
  for (i64 index = a.size - b.size; index >= 0; index--)
    if (KIT_AR_EQUAL(kit_str(b.size, a.values + index), b)) {
      if (count == n)
        return index;
      else
        count++;
    }
  return -1;
}

//  Make a barbarian string for C standard library functions.
//  Not thread safe.
//  Use with caution.
//
static char *kit_make_bs(kit_str_t s) {
  static char buf[8][4096];
  static i32  index = 0;
  i64         n     = s.size;
  if (n > 4095)
    n = 4095;
  if (n > 0)
    memcpy(buf[index], s.values, n);
  buf[index][n] = '\0';
  char *result  = buf[index];
  index         = (index + 1) % 8;
  return result;
}

#if defined(__GNUC__) || defined(__clang__)
#  pragma GCC            pop_options
#  pragma GCC diagnostic pop
#endif

#define KIT_SZ(static_str_) \
  kit_str(sizeof(static_str_) - 1, (static_str_))

#define KIT_WRAP_BS(string_) kit_str(strlen(string_), (string_))

#define KIT_WRAP_STR(...) \
  kit_str((__VA_ARGS__).size, (__VA_ARGS__).values)

#ifdef __cplusplus
}
#endif

#define BS(...) kit_make_bs(KIT_WRAP_STR(__VA_ARGS__))
#define str_t kit_str_t
#define str kit_str
#define str_find kit_str_find
#define str_find_back kit_str_find_back
#define str_find_n kit_str_find_n
#define str_find_back_n kit_str_find_back_n
#define STR_NOT_FOUND KIT_STR_NOT_FOUND
#define SZ KIT_SZ
#define WRAP_BS KIT_WRAP_BS
#define WRAP_STR KIT_WRAP_STR

#endif