str_builder.h

Go to the documentation of this file.

 
#ifndef STR_BUILDER_H
#define STR_BUILDER_H
 
#include "base_macros.h"
#include "base_strings.h"
#include "base_types.h"
#include "memory/mem_allocator.h"
#include <stdarg.h>
#include <stdio.h>
 
C_LINKAGE_BEGIN


typedef struct StringBuilder {
    String8Node *first;  
    String8Node *last;   
    usize        node_count; 
    usize        total_size; 
} StringBuilder;

function inline void
sb_append(StringBuilder *sb, MemAllocator *alloc, String8 str)
{
    String8Node *node = MAKE_STRUCT(String8Node, alloc);
    ASSERT_NOT_NULL(node);
    node->string = str;
    node->next   = NULL;
 
    if (sb->last != NULL) {
        sb->last->next = node;
    } else {
        sb->first = node;
    }
    sb->last = node;
    sb->node_count += 1;
    sb->total_size += str.size;
}

function inline void
sb_append_cstr(StringBuilder *sb, MemAllocator *alloc, const char *cstr)
{
    sb_append(sb, alloc, str8_cstr(cstr));
}

function inline void
sb_append_fmt(StringBuilder *sb, MemAllocator *alloc, const char *fmt, ...)
{
    va_list args;
    va_start(args, fmt);
    int len = vsnprintf(NULL, 0, fmt, args);
    va_end(args);
 
    if (len <= 0) {
        return;
    }
 
    char *buf = (char *)mem_alloc(alloc, (usize)len + 1);
    ASSERT_NOT_NULL(buf);
 
    va_start(args, fmt);
    vsnprintf(buf, (usize)len + 1, fmt, args);
    va_end(args);
 
    sb_append(sb, alloc, str8((u8 *)buf, (usize)len));
}

function inline String8
sb_join(StringBuilder *sb, MemAllocator *out_alloc)
{
    if (sb->total_size == 0) {
        return (String8){0};
    }
 
    u8 *buf = (u8 *)mem_alloc(out_alloc, sb->total_size);
    ASSERT_NOT_NULL(buf);
 
    usize offset = 0;
    for (String8Node *n = sb->first; n != NULL; n = n->next) {
        MEM_COPY(buf + offset, n->string.str, n->string.size);
        offset += n->string.size;
    }
 
    return str8(buf, sb->total_size);
}

function inline void
sb_reset(StringBuilder *sb)
{
    sb->first      = NULL;
    sb->last       = NULL;
    sb->node_count = 0;
    sb->total_size = 0;
}

 
C_LINKAGE_END
 
#endif // STR_BUILDER_H