mirror of
https://github.com/johnkerl/miller.git
synced 2026-07-26 01:15:27 +00:00
52 lines
1.4 KiB
C
52 lines
1.4 KiB
C
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "string_builder.h"
|
|
#include "../lib/mlrutil.h"
|
|
#include "../lib/mlr_globals.h"
|
|
|
|
static void sb_enlarge(string_builder_t* psb);
|
|
|
|
// typedef struct _string_builder_t {
|
|
// int used_length;
|
|
// int alloc_length;
|
|
// char* buffer;
|
|
// } string_builder_t;
|
|
|
|
void sb_init(string_builder_t* psb, int alloc_length) {
|
|
if (alloc_length < 1) {
|
|
fprintf(stderr, "%s: string_builder alloc_length must be >= 1; got %d.\n",
|
|
MLR_GLOBALS.argv0, alloc_length);
|
|
exit(1);
|
|
}
|
|
psb->used_length = 0;
|
|
psb->alloc_length = alloc_length;
|
|
psb->buffer = mlr_malloc_or_die(alloc_length); // xxx malloc ...
|
|
}
|
|
|
|
void sb_append_char(string_builder_t* psb, char c) {
|
|
if (psb->used_length >= psb->alloc_length)
|
|
sb_enlarge(psb);
|
|
psb->buffer[psb->used_length++] = c;
|
|
}
|
|
|
|
void sb_append_string(string_builder_t* psb, char* s) {
|
|
for (char* p = s; *p; p++)
|
|
sb_append_char(psb, *p);
|
|
}
|
|
|
|
char* sb_finish(string_builder_t* psb) {
|
|
sb_append_char(psb, '\0');
|
|
char* rv = psb->buffer;
|
|
psb->used_length = 0;
|
|
psb->alloc_length = 0;
|
|
psb->buffer = NULL;
|
|
return rv;
|
|
}
|
|
|
|
static void sb_enlarge(string_builder_t* psb) {
|
|
int new_alloc_length = psb->alloc_length * 2;
|
|
char* new_buffer = mlr_malloc_or_die(new_alloc_length);
|
|
memcpy(new_buffer, psb->buffer, psb->used_length);
|
|
psb->alloc_length = new_alloc_length;
|
|
psb->buffer = new_buffer;
|
|
}
|