gb/gb.h

877 lines
21 KiB
C
Raw Normal View History

// gb.h - v0.02 - public domain C helper library - no warranty implied; use at your own risk
2016-03-03 06:37:48 -05:00
// (Experimental) A C helper library geared towards game development
/*
LICENSE
This software is in the public domain. Where that dedication is not
recognized, you are granted a perpetual, irrevocable license to copy,
distribute, and modify this file as you see fit.
WARNING
- This library is _highly_ experimental and features may not work as expected.
- This also means that many functions are not documented.
CONTENTS
- Common Macros
- Assert
- Types
- Cast macro (easy grepping)
- Memory
- Custom Allocation
- gb_Allocator
- gb_Arena
- gb_Pool
- gb_String
TODO
- Mutex
- Atomics
- Semaphore
- Thread
- OS Types and Functions (File/IO/OS/etc.)
*/
/*
Version History:
0.02 - Minor fixes
2016-03-03 06:37:48 -05:00
0.01 - Initial Version
*/
#ifndef GB_INCLUDE_GB_H
#define GB_INCLUDE_GB_H
#if defined(__cplusplus)
extern "C" {
#endif
2016-03-03 06:37:48 -05:00
#include <stdarg.h>
#include <stddef.h>
2016-03-03 06:37:48 -05:00
// NOTE(bill): Because static means three different things in C/C++
// Great design(!)
#ifndef local_persist
#define global static
#define internal static
#define local_persist static
2015-12-17 17:56:29 -05:00
#endif
2016-03-03 06:37:48 -05:00
// NOTE(bill): If I put gb_inline, I _want_ it inlined
#ifndef gb_inline
#if defined(_MSC_VER)
2016-03-03 06:37:48 -05:00
#define gb_inline __forceinline
#else
2016-03-03 06:37:48 -05:00
#define gb_inline __attribute__ ((__always_inline__))
#endif
#endif
2016-03-03 06:37:48 -05:00
2015-12-14 19:18:52 -05:00
#if !defined(GB_NO_STDIO)
#include <stdio.h>
2015-12-14 19:18:52 -05:00
#endif
2016-03-03 06:37:48 -05:00
#ifndef GB_ASSERT
#include <assert.h>
#define GB_ASSERT(cond) assert(cond)
#define GB_ASSERT_MSG(cond, msg) assert((cond) && (msg))
#endif
#define GB_STATIC_ASSERT3(cond, msg) typedef char gb__static_assertion_##msg[(!!(cond))*2-1]
2016-03-03 06:37:48 -05:00
// NOTE(bill): Token pasting madness
#define GB_STATIC_ASSERT2(cond, line) GB_STATIC_ASSERT3(cond, static_assertion_at_line_##line)
#define GB_STATIC_ASSERT1(cond, line) GB_STATIC_ASSERT2(cond, line)
#define GB_STATIC_ASSERT(cond) GB_STATIC_ASSERT1(cond, __LINE__)
2015-12-14 19:18:52 -05:00
#if !defined(GB_NO_STDIO) && defined(_MSC_VER)
2016-03-03 06:37:48 -05:00
// snprintf_msvc
gb_inline int
gb__vsnprintf_compatible(char* buffer, size_t size, char const *format, va_list args)
{
int result = -1;
2016-03-03 06:37:48 -05:00
if (size > 0)
result = _vsnprintf_s(buffer, size, _TRUNCATE, format, args);
if (result == -1)
return _vscprintf(format, args);
return result;
}
2016-03-03 06:37:48 -05:00
gb_inline int
gb__snprintf_compatible(char* buffer, size_t size, char const *format, ...)
{
2016-03-03 06:37:48 -05:00
int result = -1;
va_list args;
va_start(args, format);
2016-03-03 06:37:48 -05:00
result = gb__vsnprintf_compatible(buffer, size, format, args);
va_end(args);
return result;
}
#if !defined(GB_DO_NOT_USE_MSVC_SPRINTF_FIX)
#define snprintf gb__snprintf_compatible
#define vsnprintf gb__vsnprintf_compatible
#endif /* GB_DO_NOT_USE_MSVC_SPRINTF_FIX */
2015-12-14 19:18:52 -05:00
#endif /* !defined(GB_NO_STDIO) */
2016-03-03 06:37:48 -05:00
#if defined(_MSC_VER)
2016-03-03 06:37:48 -05:00
typedef unsigned __int8 u8;
typedef signed __int8 s8;
typedef unsigned __int16 u16;
typedef signed __int16 s16;
typedef unsigned __int32 u32;
typedef signed __int32 s32;
typedef unsigned __int64 u64;
typedef signed __int64 s64;
#else
2015-12-14 19:18:52 -05:00
#include <stdint.h>
2016-03-03 06:37:48 -05:00
typedef uint8_t u8;
typedef int8_t s8;
typedef uint16_t u16;
typedef int16_t s16;
typedef uint32_t u32;
typedef int32_t s32;
typedef uint64_t u64;
typedef int64_t s64;
#endif
GB_STATIC_ASSERT(sizeof(s8) == 1);
GB_STATIC_ASSERT(sizeof(s16) == 2);
GB_STATIC_ASSERT(sizeof(s32) == 4);
GB_STATIC_ASSERT(sizeof(s64) == 8);
2016-03-03 06:37:48 -05:00
typedef size_t usize;
2016-01-01 14:33:06 -05:00
2016-03-03 06:37:48 -05:00
typedef uintptr_t uintptr;
typedef intptr_t intptr;
2016-01-01 14:33:06 -05:00
2016-03-03 06:37:48 -05:00
typedef float f32;
typedef double f64;
2016-01-01 14:33:06 -05:00
#if defined(_MSC_VER) && _MSC_VER < 1900
#ifndef false
#define false 0
#endif
#ifndef true
#define true 1
#endif
#else
#include <stdbool.h> // NOTE(bill): To get false/true
#endif
2016-01-01 14:33:06 -05:00
2016-03-03 06:37:48 -05:00
// Boolean Types
typedef s8 b8;
typedef s32 b32;
2016-01-01 14:33:06 -05:00
#ifndef U8_MIN
#define U8_MIN 0u
#define U8_MAX 0xffu
#define S8_MIN (-0x7f - 1)
#define S8_MAX 0x7f
#define U16_MIN 0u
#define U16_MAX 0xffffu
#define S16_MIN (-0x7fff - 1)
#define S16_MAX 0x7fff
#define U32_MIN 0u
#define U32_MAX 0xffffffffu
#define S32_MIN (-0x7fffffff - 1)
#define S32_MAX 0x7fffffff
#define U64_MIN 0ull
#define U64_MAX 0xffffffffffffffffull
#define S64_MIN (-0x7fffffffffffffffll - 1)
#define S64_MAX 0x7fffffffffffffffll
#endif
2016-03-03 06:37:48 -05:00
#ifndef COUNT_OF
#define COUNT_OF(x) (sizeof((x)) / sizeof(0[(x)]))
#endif
2016-03-03 06:37:48 -05:00
// NOTE(bill): Allows for easy grep of casts
// NOTE(bill): Still not as type safe as C++ static_cast, reinterpret_cast, and const_cast, but I don't need them
2015-12-17 17:56:29 -05:00
#ifndef cast
2016-03-03 06:37:48 -05:00
#define cast(x) (x)
2015-12-17 17:56:29 -05:00
#endif
2015-12-14 19:18:52 -05:00
2016-03-03 06:37:48 -05:00
#ifndef NULL
#define NULL ((void *)0)
#endif
2016-01-01 14:33:06 -05:00
2016-03-03 06:37:48 -05:00
#ifndef GB_UNUSED
#define GB_UNUSED(x) ((void)(sizeof(x)))
2016-01-01 14:33:06 -05:00
#endif
2016-03-03 06:37:48 -05:00
#ifndef gb_inline
#if defined(_MSC_VER)
#define gb_inline __forceinline
2016-01-01 14:33:06 -05:00
#else
2016-03-03 06:37:48 -05:00
#define gb_inline __attribute__ ((__always_inline__))
2015-12-17 17:56:29 -05:00
#endif
2015-12-14 19:18:52 -05:00
#endif
2015-12-17 17:56:29 -05:00
2016-03-03 06:37:48 -05:00
////////////////////////////////
// //
// Memory //
// //
////////////////////////////////
2016-03-03 06:37:48 -05:00
#include <string.h> // For memcpy/memmove/memset/etc.
#ifndef GB_IS_POWER_OF_TWO
#define GB_IS_POWER_OF_TWO(x) ((x) != 0) && !((x) & ((x)-1))
2015-12-17 17:56:29 -05:00
#endif
2016-03-03 06:37:48 -05:00
void *gb_align_forward(void *ptr, usize align);
void gb_zero_size(void *ptr, usize size);
#define gb_zero_struct(t) gb_zero_size((t), sizeof(*(t))) // NOTE(bill): Pass pointer of struct
#define gb_zero_array(a, count) gb_zero_size((a), sizeof((a)[0])*count)
#if defined(GB_IMPLEMENTATION)
gb_inline void *
gb_align_forward(void *ptr, usize align)
2015-12-17 07:26:24 -05:00
{
2016-03-03 06:37:48 -05:00
uintptr p;
usize modulo;
GB_ASSERT(GB_IS_POWER_OF_TWO(align));
2016-03-03 06:37:48 -05:00
p = cast(uintptr)ptr;
modulo = p % align;
if (modulo) p += (align - modulo);
return cast(void *)p;
}
2016-03-03 06:37:48 -05:00
gb_inline void gb_zero_size(void *ptr, usize size) { memset(ptr, 0, size); }
2016-03-03 06:37:48 -05:00
#endif // GB_IMPLEMENTATION
2016-03-03 06:37:48 -05:00
////////////////////////////////
// //
// Custom Allocation //
// //
////////////////////////////////
typedef enum gb_Allocation_Type
{
GB_ALLOCATION_TYPE_ALLOC,
GB_ALLOCATION_TYPE_FREE,
GB_ALLOCATION_TYPE_FREE_ALL,
GB_ALLOCATION_TYPE_RESIZE,
} gb_Allocation_Type;
#ifndef GB_ALLOCATOR_PROCEDURE
2016-03-03 06:37:48 -05:00
#define GB_ALLOCATOR_PROCEDURE(name) void *name(void *allocator_data, gb_Allocation_Type type, usize size, usize alignment, void *old_memory, usize old_size, u32 options)
#endif
2016-03-03 06:37:48 -05:00
typedef GB_ALLOCATOR_PROCEDURE(gb_Allocator_Procedure);
2016-03-03 06:37:48 -05:00
typedef struct gb_Allocator
2015-12-17 07:26:24 -05:00
{
2016-03-03 06:37:48 -05:00
gb_Allocator_Procedure *procedure;
void *data;
} gb_Allocator;
#ifndef GB_DEFAULT_ALIGNMENT
#define GB_DEFAULT_ALIGNMENT 8
#endif
2016-03-03 06:37:48 -05:00
gb_inline void *gb_alloc_align(gb_Allocator a, usize size, usize alignment) { return a.procedure(a.data, GB_ALLOCATION_TYPE_ALLOC, size, alignment, NULL, 0, 0); }
gb_inline void *gb_alloc(gb_Allocator a, usize size) { return gb_alloc_align(a, size, GB_DEFAULT_ALIGNMENT); }
gb_inline void gb_free(gb_Allocator a, void *ptr) { a.procedure(a.data, GB_ALLOCATION_TYPE_FREE, 0, 0, ptr, 0, 0); }
gb_inline void gb_free_all(gb_Allocator a) { a.procedure(a.data, GB_ALLOCATION_TYPE_FREE_ALL, 0, 0, NULL, 0, 0); }
gb_inline void *gb_resize(gb_Allocator a, void *ptr, usize new_size) { return a.procedure(a.data, GB_ALLOCATION_TYPE_RESIZE, new_size, 0, ptr, 0, 0); }
2016-03-03 06:37:48 -05:00
gb_inline void *gb_alloc_copy(gb_Allocator a, void* src, usize size) { return memcpy(gb_alloc(a, size), src, size); }
gb_inline void *gb_alloc_align_copy(gb_Allocator a, void* src, usize size, usize alignment) { return memcpy(gb_alloc_align(a, size, alignment), src, size); }
2016-03-03 06:37:48 -05:00
#define gb_alloc_struct(allocator, Type) (Type *)gb_alloc_align(allocator, sizeof(Type))
#define gb_alloc_array(allocator, Type, count) (Type *)gb_alloc(allocator, sizeof(Type) * (count))
2016-03-03 06:37:48 -05:00
typedef struct gb_Arena
2015-12-17 07:26:24 -05:00
{
2016-03-03 06:37:48 -05:00
gb_Allocator backing;
void *physical_start;
usize total_size;
usize total_allocated_count;
usize prev_allocated_count;
u32 temp_count;
} gb_Arena;
void gb_init_arena_from_memory(gb_Arena *arena, void *start, usize size);
void gb_init_arena_from_allocator(gb_Arena *arena, gb_Allocator backing, usize size);
void gb_free_arena(gb_Arena *arena);
gb_Allocator gb_make_arena_allocator(gb_Arena *arena);
GB_ALLOCATOR_PROCEDURE(gb_arena_allocator_procedure);
2016-03-03 06:37:48 -05:00
typedef struct gb_Temp_Arena_Memory
{
gb_Arena *arena;
usize original_count;
} gb_Temp_Arena_Memory;
2016-03-03 06:37:48 -05:00
gb_Temp_Arena_Memory gb_begin_temp_arena_memory(gb_Arena *arena);
void gb_end_temp_arena_memory(gb_Temp_Arena_Memory tmp_mem);
2016-03-03 06:37:48 -05:00
typedef struct gb_Pool
{
gb_Allocator backing;
2016-03-03 06:37:48 -05:00
void *physical_start;
void *free_list;
2016-03-03 06:37:48 -05:00
usize block_size;
usize block_align;
usize total_size;
} gb_Pool;
void gb_init_pool(gb_Pool *pool, gb_Allocator backing, usize num_blocks, usize block_size);
void gb_init_pool_align(gb_Pool *pool, gb_Allocator backing, usize num_blocks, usize block_size, usize block_align);
void gb_free_pool(gb_Pool *pool);
gb_Allocator gb_make_pool_allocator(gb_Pool *pool);
GB_ALLOCATOR_PROCEDURE(gb_pool_allocator_procedure);
2016-03-03 06:37:48 -05:00
#if defined(GB_IMPLEMENTATION)
gb_inline void
gb_init_arena_from_memory(gb_Arena *arena, void *start, usize size)
{
2016-03-03 06:37:48 -05:00
arena->backing.procedure = NULL;
arena->backing.data = NULL;
arena->physical_start = start;
arena->total_size = size;
arena->total_allocated_count = 0;
arena->temp_count = 0;
}
2016-03-03 06:37:48 -05:00
gb_inline void
gb_init_arena_from_allocator(gb_Arena *arena, gb_Allocator backing, usize size)
{
2016-03-03 06:37:48 -05:00
arena->backing = backing;
arena->physical_start = gb_alloc(backing, size);
arena->total_size = size;
arena->total_allocated_count = 0;
arena->temp_count = 0;
}
2016-03-03 06:37:48 -05:00
gb_inline void
gb_free_arena(gb_Arena *arena)
{
2016-03-03 06:37:48 -05:00
if (arena->backing.procedure) {
gb_free(arena->backing, arena->physical_start);
arena->physical_start = NULL;
}
}
2016-03-03 06:37:48 -05:00
gb_inline gb_Allocator
gb_make_arena_allocator(gb_Arena *arena)
{
2016-03-03 06:37:48 -05:00
gb_Allocator allocator;
allocator.procedure = gb_arena_allocator_procedure;
allocator.data = arena;
return allocator;
}
2016-03-03 06:37:48 -05:00
GB_ALLOCATOR_PROCEDURE(gb_arena_allocator_procedure)
{
2016-03-03 06:37:48 -05:00
gb_Arena *arena = cast(gb_Arena *)allocator_data;
GB_UNUSED(options);
GB_UNUSED(old_size);
2016-03-03 06:37:48 -05:00
switch (type) {
case GB_ALLOCATION_TYPE_ALLOC: {
void *ptr;
usize actual_size = size + alignment;
2016-03-03 06:37:48 -05:00
// NOTE(bill): Out of memory
if (arena->total_allocated_count + actual_size > cast(usize)arena->total_size)
return NULL;
2016-03-03 06:37:48 -05:00
ptr = gb_align_forward(cast(u8 *)arena->physical_start + arena->total_allocated_count, alignment);
arena->prev_allocated_count = arena->total_allocated_count;
arena->total_allocated_count += actual_size;
return ptr;
} break;
2016-03-03 06:37:48 -05:00
case GB_ALLOCATION_TYPE_FREE: {
// NOTE(bill): Free all at once
// NOTE(bill): Use Temp_Arena_Memory if you want to free a block
} break;
2016-03-03 06:37:48 -05:00
case GB_ALLOCATION_TYPE_FREE_ALL:
arena->total_allocated_count = 0;
break;
2016-03-03 06:37:48 -05:00
case GB_ALLOCATION_TYPE_RESIZE: {
// TODO(bill): Check if ptr is at the top
void *ptr = gb_alloc_align(gb_make_arena_allocator(arena), size, alignment);
memcpy(ptr, old_memory, size);
return ptr;
} break;
}
2016-03-03 06:37:48 -05:00
return NULL; // NOTE(bill): Default return value
}
2016-03-03 06:37:48 -05:00
gb_inline gb_Temp_Arena_Memory
gb_begin_temp_arena_memory(gb_Arena *arena)
2015-12-17 07:26:24 -05:00
{
2016-03-03 06:37:48 -05:00
gb_Temp_Arena_Memory tmp;
tmp.arena = arena;
tmp.original_count = arena->total_allocated_count;
arena->temp_count++;
return tmp;
}
2016-03-03 06:37:48 -05:00
gb_inline void
gb_end_temp_arena_memory(gb_Temp_Arena_Memory tmp)
{
GB_ASSERT(tmp.arena->total_allocated_count >= tmp.original_count);
GB_ASSERT(tmp.arena->temp_count > 0);
tmp.arena->total_allocated_count = tmp.original_count;
tmp.arena->temp_count--;
}
2016-03-03 06:37:48 -05:00
void
gb_init_pool(gb_Pool *pool, gb_Allocator backing, usize num_blocks, usize block_size)
2015-12-17 07:26:24 -05:00
{
2016-03-03 06:37:48 -05:00
gb_init_pool_align(pool, backing, num_blocks, block_size, GB_DEFAULT_ALIGNMENT);
}
2016-03-03 06:37:48 -05:00
void
gb_init_pool_align(gb_Pool *pool, gb_Allocator backing, usize num_blocks, usize block_size, usize block_align)
{
usize actual_block_size, pool_size, block_index;
u8 *data, *curr;
2016-03-03 06:37:48 -05:00
pool->backing = backing;
pool->block_size = block_size;
pool->block_align = block_align;
actual_block_size = block_size + block_align;
pool_size = num_blocks * actual_block_size;
data = cast(u8 *)gb_alloc_align(backing, pool_size, block_align);
2016-03-03 06:37:48 -05:00
// Init intrusive freelist
curr = data;
for (block_index = 0; block_index < num_blocks-1; block_index++) {
2016-03-03 06:37:48 -05:00
uintptr *next = cast(uintptr *)curr;
*next = cast(uintptr)curr + actual_block_size;
curr += actual_block_size;
}
{
uintptr *end = cast(uintptr*)curr;
*end = cast(uintptr)NULL;
}
2016-03-03 06:37:48 -05:00
pool->physical_start = data;
pool->free_list = data;
}
void
gb_free_pool(gb_Pool *pool)
{
if (pool->backing.procedure) {
gb_free(pool->backing, pool->physical_start);
}
}
2016-03-03 06:37:48 -05:00
gb_Allocator
gb_make_pool_allocator(gb_Pool *pool)
{
gb_Allocator allocator;
allocator.procedure = gb_pool_allocator_procedure;
allocator.data = pool;
return allocator;
}
2016-03-03 06:37:48 -05:00
GB_ALLOCATOR_PROCEDURE(gb_pool_allocator_procedure)
{
gb_Pool *pool = cast(gb_Pool *)allocator_data;
2016-03-03 06:37:48 -05:00
GB_UNUSED(options);
GB_UNUSED(old_size);
2016-03-03 06:37:48 -05:00
switch (type) {
case GB_ALLOCATION_TYPE_ALLOC: {
uintptr next_free;
void *ptr;
GB_ASSERT(size == pool->block_size);
GB_ASSERT(alignment == pool->block_align);
GB_ASSERT(pool->free_list != NULL);
2016-03-03 06:37:48 -05:00
next_free = *cast(uintptr *)pool->free_list;
ptr = pool->free_list;
pool->free_list = cast(void *)next_free;
pool->total_size += pool->block_size;
return ptr;
} break;
2016-03-03 06:37:48 -05:00
case GB_ALLOCATION_TYPE_FREE: {
uintptr *next;
if (old_memory == NULL) return NULL;
2016-03-03 06:37:48 -05:00
next = cast(uintptr *)old_memory;
*next = cast(uintptr)pool->free_list;
pool->free_list = old_memory;
pool->total_size -= pool->block_size;
} break;
2016-03-03 06:37:48 -05:00
case GB_ALLOCATION_TYPE_FREE_ALL: {
// TODO(bill):
} break;
2016-03-03 06:37:48 -05:00
case GB_ALLOCATION_TYPE_RESIZE: {
// NOTE(bill): Cannot resize
} break;
}
2016-03-03 06:37:48 -05:00
return NULL;
}
2016-03-03 06:37:48 -05:00
#endif // GB_IMPLEMENTATION
2016-03-03 06:37:48 -05:00
////////////////////////////////
// //
// gb_String - C Compatible //
// //
////////////////////////////////
// Pascal like strings in C
typedef char *gb_String;
#ifndef GB_STRING_SIZE
#define GB_STRING_SIZE
typedef u32 gb_String_Size;
#endif
2016-03-03 06:37:48 -05:00
// This is stored at the beginning of the string
// NOTE(bill): It is (2*sizeof(gb_String_Size) + 2*sizeof(void*)) (default: 16B (32bit), 24B (64bit))
// NOTE(bill): If you only need a small string, just use a standard c string
2015-12-17 07:26:24 -05:00
typedef struct gb_String_Header
{
2016-03-03 06:37:48 -05:00
gb_Allocator allocator;
gb_String_Size length;
gb_String_Size capacity;
} gb_String_Header;
2016-03-03 06:37:48 -05:00
#define GB_STRING_HEADER(str) (cast(gb_String_Header *)str - 1)
2016-03-03 06:37:48 -05:00
gb_String gb_string_make(gb_Allocator a, char const *str);
gb_String gb_string_make_length(gb_Allocator a, void const *str, gb_String_Size num_bytes);
void gb_string_free(gb_String str);
2016-03-03 06:37:48 -05:00
gb_String gb_string_duplicate(gb_Allocator a, gb_String const str);
2015-12-17 07:26:24 -05:00
gb_String_Size gb_string_length(gb_String const str);
gb_String_Size gb_string_capacity(gb_String const str);
gb_String_Size gb_string_available_space(gb_String const str);
void gb_string_clear(gb_String str);
2015-12-17 07:26:24 -05:00
gb_String gb_string_append_string(gb_String str, gb_String const other);
2016-03-03 06:37:48 -05:00
gb_String gb_string_append_string_length(gb_String str, void const *other, gb_String_Size num_bytes);
gb_String gb_string_append_cstring(gb_String str, char const *other);
2016-03-03 06:37:48 -05:00
gb_String gb_string_set(gb_String str, char const *cstr);
gb_String gb_string_make_space_for(gb_String str, gb_String_Size add_len);
2015-12-17 07:26:24 -05:00
gb_String_Size gb_string_allocation_size(gb_String const str);
2016-03-03 06:37:48 -05:00
b32 gb_strings_are_equal(gb_String const lhs, gb_String const rhs);
2016-03-03 06:37:48 -05:00
gb_String gb_string_trim(gb_String str, char const *cut_set);
gb_String gb_string_trim_space(gb_String str); /* Whitespace ` \t\r\n\v\f` */
2016-03-03 06:37:48 -05:00
#if defined(GB_IMPLEMENTATION)
2016-03-03 06:37:48 -05:00
gb_inline void gb__string_set_length(gb_String str, gb_String_Size len) { GB_STRING_HEADER(str)->length = len; }
gb_inline void gb__string_set_capacity(gb_String str, gb_String_Size cap) { GB_STRING_HEADER(str)->capacity = cap; }
2016-03-03 06:37:48 -05:00
gb_inline gb_String
gb_string_make(gb_Allocator a, char const *str)
{
gb_String_Size len = cast(gb_String_Size)(str ? strlen(str) : 0);
return gb_string_make_length(a, str, len);
}
2016-03-03 06:37:48 -05:00
gb_String
gb_string_make_length(gb_Allocator a, void const *init_str, gb_String_Size num_bytes)
{
gb_String_Size header_size = sizeof(gb_String_Header);
void *ptr = gb_alloc(a, header_size + num_bytes + 1);
if (ptr == NULL) {
return NULL;
} else {
gb_String str = cast(char *)ptr + header_size;
gb_String_Header *header = GB_STRING_HEADER(str);
// Zero all data first
if (!init_str) gb_zero_size(ptr, header_size + num_bytes + 1);
header->allocator = a;
header->length = num_bytes;
header->capacity = num_bytes;
if (num_bytes && init_str)
memcpy(str, init_str, num_bytes);
str[num_bytes] = '\0'; // Just in case
return str;
}
2016-03-03 06:37:48 -05:00
}
2016-03-03 06:37:48 -05:00
gb_inline void
gb_string_free(gb_String str)
{
if (str) {
gb_String_Header *header;
header = GB_STRING_HEADER(str);
gb_free(header->allocator, header);
}
}
2016-03-03 06:37:48 -05:00
gb_inline gb_String gb_string_duplicate(gb_Allocator a, gb_String const str) { return gb_string_make_length(a, str, gb_string_length(str)); }
2016-03-03 06:37:48 -05:00
gb_inline gb_String_Size gb_string_length(gb_String const str) { return GB_STRING_HEADER(str)->length; }
gb_inline gb_String_Size gb_string_capacity(gb_String const str) { return GB_STRING_HEADER(str)->capacity; }
2016-03-03 06:37:48 -05:00
gb_inline gb_String_Size
gb_string_available_space(gb_String const str)
2015-12-17 17:56:29 -05:00
{
2016-03-03 06:37:48 -05:00
gb_String_Header *h = GB_STRING_HEADER(str);
if (h->capacity > h->length)
return h->capacity - h->length;
return 0;
}
2016-03-03 06:37:48 -05:00
gb_inline void gb_string_clear(gb_String str) { gb__string_set_length(str, 0); str[0] = '\0'; }
2016-03-03 06:37:48 -05:00
gb_inline gb_String gb_string_append_string(gb_String str, gb_String const other) { return gb_string_append_string_length(str, other, gb_string_length(other)); }
2016-03-03 06:37:48 -05:00
gb_String
gb_string_append_string_length(gb_String str, void const *other, gb_String_Size other_len)
{
gb_String_Size curr_len = gb_string_length(str);
2016-03-03 06:37:48 -05:00
str = gb_string_make_space_for(str, other_len);
if (str == NULL)
return NULL;
2016-03-03 06:37:48 -05:00
memcpy(str + curr_len, other, other_len);
str[curr_len + other_len] = '\0';
gb__string_set_length(str, curr_len + other_len);
2016-03-03 06:37:48 -05:00
return str;
}
2016-03-03 06:37:48 -05:00
gb_inline gb_String
gb_string_append_cstring(gb_String str, char const *other)
{
2016-03-03 06:37:48 -05:00
return gb_string_append_string_length(str, other, cast(gb_String_Size)strlen(other));
}
gb_String
2016-03-03 06:37:48 -05:00
gb_string_set(gb_String str, char const *cstr)
{
2016-03-03 06:37:48 -05:00
gb_String_Size len = cast(gb_String_Size)strlen(cstr);
if (gb_string_capacity(str) < len) {
str = gb_string_make_space_for(str, len - gb_string_length(str));
if (str == NULL)
return NULL;
}
memcpy(str, cstr, len);
str[len] = '\0';
gb__string_set_length(str, len);
return str;
}
2016-03-03 06:37:48 -05:00
local_persist void *
gb__string_realloc(gb_Allocator a, void *ptr, gb_String_Size old_size, gb_String_Size new_size)
{
if (!ptr) return gb_alloc(a, new_size);
if (new_size < old_size)
new_size = old_size;
2016-03-03 06:37:48 -05:00
if (old_size == new_size) {
return ptr;
2016-03-03 06:37:48 -05:00
} else {
// TODO(bill): Use gb_resize here??
2016-03-03 06:37:48 -05:00
void *new_ptr = gb_alloc(a, new_size);
if (!new_ptr)
return NULL;
2016-03-03 06:37:48 -05:00
memcpy(new_ptr, ptr, old_size);
gb_free(a, ptr);
return new_ptr;
}
}
gb_String
gb_string_make_space_for(gb_String str, gb_String_Size add_len)
{
gb_String_Size available = gb_string_available_space(str);
2016-03-03 06:37:48 -05:00
// Return if there is enough space left
if (available >= add_len) {
return str;
} else {
gb_String_Size new_len = gb_string_length(str) + add_len;
void *ptr = GB_STRING_HEADER(str);
gb_String_Size old_size = sizeof(struct gb_String_Header) + gb_string_length(str) + 1;
gb_String_Size new_size = sizeof(struct gb_String_Header) + new_len + 1;
2016-03-03 06:37:48 -05:00
void *new_ptr = gb__string_realloc(GB_STRING_HEADER(str)->allocator, ptr, old_size, new_size);
if (new_ptr == NULL) return NULL;
2016-03-03 06:37:48 -05:00
str = cast(char *)(GB_STRING_HEADER(new_ptr) + 1);
gb__string_set_capacity(str, new_len);
2016-03-03 06:37:48 -05:00
return str;
}
}
2016-03-03 06:37:48 -05:00
gb_inline gb_String_Size
2015-12-17 07:26:24 -05:00
gb_string_allocation_size(gb_String const str)
{
gb_String_Size result = gb_string_capacity(str) + cast(gb_String_Size)sizeof(gb_String_Header);
return result;
}
2016-03-03 06:37:48 -05:00
gb_inline b32
2015-12-17 07:26:24 -05:00
gb_strings_are_equal(gb_String const lhs, gb_String const rhs)
{
2016-03-03 06:37:48 -05:00
gb_String_Size lhs_len, rhs_len, i;
lhs_len = gb_string_length(lhs);
rhs_len = gb_string_length(rhs);
if (lhs_len != rhs_len)
return false;
2016-03-03 06:37:48 -05:00
for (i = 0; i < lhs_len; i++) {
if (lhs[i] != rhs[i])
return false;
}
return true;
}
gb_String
2016-03-03 06:37:48 -05:00
gb_string_trim(gb_String str, char const *cut_set)
{
2016-03-03 06:37:48 -05:00
char *start, *end, *start_pos, *end_pos;
gb_String_Size len;
start_pos = start = str;
end_pos = end = str + gb_string_length(str) - 1;
while (start_pos <= end && strchr(cut_set, *start_pos))
start_pos++;
while (end_pos > start_pos && strchr(cut_set, *end_pos))
end_pos--;
2016-03-03 06:37:48 -05:00
len = cast(gb_String_Size)((start_pos > end_pos) ? 0 : ((end_pos - start_pos)+1));
if (str != start_pos)
memmove(str, start_pos, len);
str[len] = '\0';
gb__string_set_length(str, len);
return str;
}
2016-03-03 06:37:48 -05:00
gb_inline gb_String gb_string_trim_space(gb_String str) { return gb_string_trim(str, " \t\r\n\v\f"); }
#endif
2016-03-03 06:37:48 -05:00
////////////////////////////////
// //
// Unfinished code //
// //
////////////////////////////////
#if defined(__cplusplus)
}
2016-03-03 06:37:48 -05:00
#endif
2016-03-03 06:37:48 -05:00
#endif // GB_INCLUDE_GB_H