block-allocator.hpp

Go to the documentation of this file.

/* -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */
/*
 *  Main authors:
 *     Christian Schulte <schulte@gecode.org>
 *
 *  Copyright:
 *     Christian Schulte, 2004
 *
 *  Last modified:
 *     $Date: 2010-07-28 16:13:53 +0200 (Wed, 28 Jul 2010) $ by $Author: schulte $
 *     $Revision: 11292 $
 *
 *  This file is part of Gecode, the generic constraint
 *  development environment:
 *     http://www.gecode.org
 *
 *  Permission is hereby granted, free of charge, to any person obtaining
 *  a copy of this software and associated documentation files (the
 *  "Software"), to deal in the Software without restriction, including
 *  without limitation the rights to use, copy, modify, merge, publish,
 *  distribute, sublicense, and/or sell copies of the Software, and to
 *  permit persons to whom the Software is furnished to do so, subject to
 *  the following conditions:
 *
 *  The above copyright notice and this permission notice shall be
 *  included in all copies or substantial portions of the Software.
 *
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
 *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
 *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 */

namespace Gecode { namespace Support {

  template<class T, class A, int blocksize = 512>
  class BlockAllocator {
  private:
    A& a;
    class Block {
    public:
      T      b[blocksize];
      Block* next;
    };
    Block* b;
    T*     n;
    size_t _size;
    void allocate(void);
  public:
    BlockAllocator(A& a);
    ~BlockAllocator(void);
    A& allocator(void);
    T* operator ()(void);
    size_t size(void) const;
  };

  template<class T, class A, int blocksize = 512>
  class BlockClient {
  public:
    static void* operator new(size_t s, BlockAllocator<T,A,blocksize>& ba);
    static void  operator delete(void*, BlockAllocator<T,A,blocksize>& ba);
    static void  operator delete(void*);
  };



  template<class T, class A, int blocksize>
  forceinline
  BlockAllocator<T,A,blocksize>::BlockAllocator(A& a0) : a(a0) {
    b = static_cast<Block*>(a.ralloc(sizeof(Block)));
    b->next = NULL;
    n       = &b->b[blocksize];
    _size = sizeof(Block);
  }

  template<class T, class A, int blocksize>
  forceinline
  BlockAllocator<T,A,blocksize>::~BlockAllocator(void) {
    while (b != NULL) {
      Block* f = b; b = b->next;
      a.rfree(f,sizeof(Block));
    }
  }

  template<class T, class A, int blocksize>
  forceinline A&
  BlockAllocator<T,A,blocksize>::allocator(void) {
    return a;
  }

  template<class T, class A, int blocksize>
  forceinline T*
  BlockAllocator<T,A,blocksize>::operator ()(void) {
    T* t = --n;
    if (t == &b->b[0])
      allocate();
    return t;
  }

  template<class T, class A, int blocksize>
  void
  BlockAllocator<T,A,blocksize>::allocate(void) {
    // Allocate another block
    Block* nb = static_cast<Block*>(a.ralloc(sizeof(Block)));
    nb->next = b; b = nb;
    n = &nb->b[blocksize];
    _size += sizeof(Block);
  }

  template<class T, class A, int blocksize>
  forceinline size_t
  BlockAllocator<T,A,blocksize>::size(void) const {
    return _size;
  }



  template<class T, class A, int blocksize>
  forceinline void
  BlockClient<T,A,blocksize>::operator 
  delete(void*,BlockAllocator<T,A,blocksize>&) {
  }
  template<class T, class A, int blocksize>
  forceinline void
  BlockClient<T,A,blocksize>::operator delete(void*) {
  }
  template<class T, class A, int blocksize>
  forceinline void*
  BlockClient<T,A,blocksize>::operator new(size_t,
                                           BlockAllocator<T,A,blocksize>& ba) {
    return ba();
  }

}}

// STATISTICS: support-any