view src/llrb/stack.h @ 25:390cf0523ea7

add file
author Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
date Fri, 01 May 2015 05:35:10 +0900
parents
children 368306e1bfed
line wrap: on
line source

#include <string.h>

typedef struct {
    size_t size;
    int max;
    int num;
    void* data;
} stack, *stack_ptr;

stack_ptr stack_init(size_t size, int max) {
    stack_ptr stack_ptr;
    
    if ((stack_ptr = calloc(1, sizeof(stack))) == NULL)
        return NULL;
    
    if ((stack_ptr->data = calloc(max, size)) == NULL) {
        free(stack_ptr);
        return NULL;
    }

    stack_ptr->size = size;
    stack_ptr->max = max;
    stack_ptr->num = 0;

    return stack_ptr;
}

stack_ptr stack_realloc(stack_ptr stack_ptr, int max) {
    if (stack_ptr == NULL)
        return NULL;

    if ((stack_ptr->data = realloc(stack_ptr->data, stack_ptr->size*max)) == NULL)
        return NULL;

    stack_ptr->max = max;

    return stack_ptr;
}

void stack_free(stack_ptr stack_ptr) {
    if (stack_ptr != NULL && stack_ptr->data != NULL) {
        free(stack_ptr->data);
        free(stack_ptr);
    }
}
    
int stack_push(stack_ptr stack_ptr, void* data) {
    if (stack_ptr->max <= stack_ptr->num)
        return -1;

    memcpy((char*)stack_ptr->data+stack_ptr->num*stack_ptr->size,  data, stack_ptr->size);
    stack_ptr->num++;

    return 0;
}

int stack_pop(stack_ptr stack_ptr, void* data) {
    if (stack_ptr->num == 0)
        return -1;

    stack_ptr->num--;

    memcpy(data, (char*)stack_ptr->data+stack_ptr->num*stack_ptr->size, stack_ptr->size);

    return 0;
}

int isMax(const stack_ptr stack_ptr) {
    return stack_ptr->max<=stack_ptr->num;
}

int isEmpty(const stack_ptr stack_ptr) {
    return stack_ptr->num<=0;
}