Source/bmalloc/bmalloc/SmallAllocator.h

   1 /*
   2  * Copyright (C) 2014 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #ifndef SmallAllocator_h
  27 #define SmallAllocator_h
  28
  29 #include "BAssert.h"
  30 #include "SmallChunk.h"
  31 #include "SmallLine.h"
  32
  33 namespace bmalloc {
  34
  35 // Helper object for allocating small objects.
  36
  37 class SmallAllocator {
  38 public:
  39     SmallAllocator();
  40     SmallAllocator(size_t);
  41
  42     bool isNull() { return !m_ptr; }
  43     SmallLine* line();
  44
  45     bool canAllocate() { return !!m_remaining; }
  46     void* allocate();
  47
  48     unsigned short objectCount();
  49     unsigned char derefCount();
  50
  51     void refill(SmallLine*);
  52     void clear();
  53
  54 private:
  55     char* m_ptr;
  56     unsigned short m_size;
  57     unsigned char m_remaining;
  58     unsigned char m_maxObjectCount;
  59 };
  60
  61 inline SmallAllocator::SmallAllocator()
  62     : m_ptr()
  63     , m_size()
  64     , m_remaining()
  65     , m_maxObjectCount()
  66 {
  67 }
  68
  69 inline SmallAllocator::SmallAllocator(size_t size)
  70     : m_ptr()
  71     , m_size(size)
  72     , m_remaining()
  73     , m_maxObjectCount(smallLineSize / size)
  74 {
  75 }
  76
  77 inline SmallLine* SmallAllocator::line()
  78 {
  79     return SmallLine::get(canAllocate() ? m_ptr : m_ptr - 1);
  80 }
  81
  82 inline void* SmallAllocator::allocate()
  83 {
  84     BASSERT(m_remaining);
  85     BASSERT(m_size >= SmallLine::minimumObjectSize);
  86
  87     --m_remaining;
  88     char* result = m_ptr;
  89     m_ptr += m_size;
  90     BASSERT(isSmall(result));
  91     return result;
  92 }
  93
  94 inline unsigned short SmallAllocator::objectCount()
  95 {
  96     return m_maxObjectCount - m_remaining;
  97 }
  98
  99 inline unsigned char SmallAllocator::derefCount()
 100 {
 101     return SmallLine::maxRefCount - objectCount();
 102 }
 103
 104 inline void SmallAllocator::refill(SmallLine* line)
 105 {
 106     BASSERT(!canAllocate());
 107     line->concurrentRef(SmallLine::maxRefCount);
 108     m_ptr = line->begin();
 109     m_remaining = m_maxObjectCount;
 110 }
 111
 112 inline void SmallAllocator::clear()
 113 {
 114     m_ptr = nullptr;
 115     m_remaining = 0;
 116 }
 117
 118 } // namespace bmalloc
 119
 120 #endif // SmallAllocator_h