seqcache.c

Go to the documentation of this file.

/*
* $Id: seqcache.c 36197 2011-04-17 10:05:27Z schlaile $
 *
 * ***** BEGIN GPL LICENSE BLOCK *****
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * Peter Schlaile <peter [at] schlaile [dot] de> 2010
 *
 * ***** END GPL LICENSE BLOCK *****
 */

#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

#include "MEM_guardedalloc.h"
#include "MEM_CacheLimiterC-Api.h"

#include "DNA_sequence_types.h"
#include "BKE_sequencer.h"
#include "BLI_utildefines.h"
#include "BLI_ghash.h"
#include "BLI_mempool.h"
#include <pthread.h>

#include "IMB_imbuf.h"
#include "IMB_imbuf_types.h"

typedef struct seqCacheKey 
{
        struct Sequence * seq;
        SeqRenderData context;
        float cfra;
        seq_stripelem_ibuf_t type;
} seqCacheKey;

typedef struct seqCacheEntry
{
        ImBuf * ibuf;
        MEM_CacheLimiterHandleC * c_handle;
} seqCacheEntry;

static GHash * hash = NULL;
static MEM_CacheLimiterC * limitor = NULL;
static struct BLI_mempool * entrypool = NULL;
static struct BLI_mempool * keypool = NULL;
static int ibufs_in  = 0;
static int ibufs_rem = 0;

static unsigned int HashHash(const void *key_)
{
        const seqCacheKey *key = (seqCacheKey*) key_;
        unsigned int rval = seq_hash_render_data(&key->context);

        rval ^= *(unsigned int*) &key->cfra;
        rval += key->type;
        rval ^= ((intptr_t) key->seq) << 6;

        return rval;
}

static int HashCmp(const void *a_, const void *b_)
{
        const seqCacheKey * a = (seqCacheKey*) a_;
        const seqCacheKey * b = (seqCacheKey*) b_;

        if (a->seq < b->seq) {
                return -1;              
        }
        if (a->seq > b->seq) {
                return 1;
        }

        if (a->cfra < b->cfra) {
                return -1;
        }
        if (a->cfra > b->cfra) {
                return 1;
        }

        if (a->type < b->type) {
                return -1;
        }
        if (a->type > b->type) {
                return 1;
        }

        return seq_cmp_render_data(&a->context, &b->context);
}

static void HashKeyFree(void *key)
{
        BLI_mempool_free(keypool, key);
}

static void HashValFree(void *val)
{
        seqCacheEntry* e = (seqCacheEntry*) val;

        if (e->ibuf) {
                /* fprintf(stderr, "Removing: %p, cnt: %d\n", e->ibuf, 
                   e->ibuf->refcounter); */
                IMB_freeImBuf(e->ibuf);
                MEM_CacheLimiter_unmanage(e->c_handle);
                ibufs_rem++;
        }

        e->ibuf = NULL;
        e->c_handle = NULL;

        BLI_mempool_free(entrypool, e);
}

static void IMB_seq_cache_destructor(void * p)
{
        seqCacheEntry* e = (seqCacheEntry*) p;
        
        if (e && e->ibuf) {
                /* fprintf(stderr, "Removing: %p, cnt: %d\n", e->ibuf,
                   e->ibuf->refcounter); */
                IMB_freeImBuf(e->ibuf);
                ibufs_rem++;

                e->ibuf = NULL;
                e->c_handle = NULL;
        }
}

void seq_stripelem_cache_init(void)
{
        hash = BLI_ghash_new(HashHash, HashCmp, "seq stripelem cache hash");
        limitor = new_MEM_CacheLimiter( IMB_seq_cache_destructor );

        entrypool = BLI_mempool_create(sizeof(seqCacheEntry), 64, 64, 0);
        keypool = BLI_mempool_create(sizeof(seqCacheKey), 64, 64, 0);
}

void seq_stripelem_cache_destruct(void)
{
        if (!entrypool) {
                return;
        }
        BLI_ghash_free(hash, HashKeyFree, HashValFree);
        delete_MEM_CacheLimiter(limitor);
        BLI_mempool_destroy(entrypool);
        BLI_mempool_destroy(keypool);
}

void seq_stripelem_cache_cleanup(void)
{
        if (!entrypool) {
                seq_stripelem_cache_init();
        }

        /* fprintf(stderr, "Stats before cleanup: in: %d rem: %d\n",
           ibufs_in, ibufs_rem); */

        BLI_ghash_free(hash, HashKeyFree, HashValFree);
        hash = BLI_ghash_new(HashHash, HashCmp, "seq stripelem cache hash");

        /* fprintf(stderr, "Stats after cleanup: in: %d rem: %d\n",
           ibufs_in, ibufs_rem); */

}

struct ImBuf * seq_stripelem_cache_get(
        SeqRenderData context, struct Sequence * seq, 
        float cfra, seq_stripelem_ibuf_t type)
{
        seqCacheKey key;
        seqCacheEntry * e;

        if (!seq) {
                return NULL;
        }

        if (!entrypool) {
                seq_stripelem_cache_init();
        }

        key.seq = seq;
        key.context = context;
        key.cfra = cfra - seq->start;
        key.type = type;
        
        e = (seqCacheEntry*) BLI_ghash_lookup(hash, &key);

        if (e && e->ibuf) {
                IMB_refImBuf(e->ibuf);

                MEM_CacheLimiter_touch(e->c_handle);
                return e->ibuf;
        }
        return NULL;
}

void seq_stripelem_cache_put(
        SeqRenderData context, struct Sequence * seq, 
        float cfra, seq_stripelem_ibuf_t type, struct ImBuf * i)
{
        seqCacheKey * key;
        seqCacheEntry * e;

        if (!i) {
                return;
        }

        ibufs_in++;

        if (!entrypool) {
                seq_stripelem_cache_init();
        }

        key = (seqCacheKey*) BLI_mempool_alloc(keypool);

        key->seq = seq;
        key->context = context;
        key->cfra = cfra - seq->start;
        key->type = type;

        IMB_refImBuf(i);

        e = (seqCacheEntry*) BLI_mempool_alloc(entrypool);

        e->ibuf = i;
        e->c_handle = NULL;

        BLI_ghash_remove(hash, key, HashKeyFree, HashValFree);
        BLI_ghash_insert(hash, key, e);

        e->c_handle = MEM_CacheLimiter_insert(limitor, e);

        MEM_CacheLimiter_ref(e->c_handle);
        MEM_CacheLimiter_enforce_limits(limitor);
        MEM_CacheLimiter_unref(e->c_handle);
}