iterator.h

Go to the documentation of this file.

#ifndef ITERATOR_H
#define ITERATOR_H
 
#include <stdbool.h>
#include <stdlib.h>
#include "../utils/check_alloc.h"

typedef struct Iterator {
    void  *container;
    void  *begin;
    void* (*next)(struct Iterator*); 
    void  (*free)(struct Iterator*); 
    bool  (*done)(struct Iterator*); 
} Iterator;

static inline Iterator* iterator_create(
    void  *container,
    void* (*next)(Iterator*),
    void  (*free)(Iterator*),
    bool  (*done)(Iterator*)
) {
    Iterator* it = malloc(sizeof(Iterator));
    check_alloc(it);
    it->container = container;
    it->begin = container;
    it->next = next;
    it->free = free;
    it->done = done;
    return it;
}

static inline bool iterator_done(Iterator* it) {
    if (it->done != NULL) {
        return it->done(it);
    }
    return it->container == NULL;
}

static inline void* iterator_next(Iterator* it) {
    return it->next(it);
}

static inline void iterator_free(Iterator *it) {
    if (it->free != NULL) {
        it->free(it);
    } else {
        free(it);
    }
}
 
#endif