gp_temp_alloc.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-2.1-or-later
/*
 * Copyright (C) 2009-2024 Cyril Hrubis <metan@ucw.cz>
 */
 
#ifndef CORE_GP_TEMP_ALLOC_H
#define CORE_GP_TEMP_ALLOC_H
 
#ifdef __linux__
# include <alloca.h>
#endif
#include <stdlib.h>
 
#include <core/gp_common.h>
 
#ifndef GP_ALLOCA_THRESHOLD
# define GP_ALLOCA_THRESHOLD 2048
#endif
 
struct gp_temp_alloc {
        void *buffer;
        size_t pos;
        size_t size;
};
 
#define GP_TEMP_ALLOC(size) ({                                        \
        ((size) > GP_ALLOCA_THRESHOLD) ? malloc(size) : alloca(size); \
})
 
#define gp_temp_alloc_create(name, bsize)                              \
        struct gp_temp_alloc name = {.size = (bsize), .pos = 0,       \
                                    .buffer = GP_TEMP_ALLOC(bsize)};
 
#define gp_temp_alloc_get(self, bsize) ({            \
        GP_ASSERT(self.pos + bsize <= self.size);    \
        size_t gp_pos__ = self.pos;                  \
        self.pos += bsize;                           \
        (void*)(((char*)(self.buffer)) + gp_pos__);  \
})
 
#define gp_temp_alloc_arr(self, type, len) \
        gp_temp_alloc_get(self, sizeof(type) * len)
 
#define gp_temp_alloc_free(self) \
do { \
        if (self.size > GP_ALLOCA_THRESHOLD) \
                free(self.buffer); \
} while (0)
 
#define gp_temp_alloc(size) GP_TEMP_ALLOC(size)
 
static inline void gp_temp_free(size_t size, void *ptr)
{
        if (size > GP_ALLOCA_THRESHOLD)
                free(ptr);
}
 
#endif /* CORE_GP_TEMP_ALLOC_H */