15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/* Copyright (c) 2006, Google Inc.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * All rights reserved.
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * Redistribution and use in source and binary forms, with or without
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * modification, are permitted provided that the following conditions are
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * met:
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *     * Redistributions of source code must retain the above copyright
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * notice, this list of conditions and the following disclaimer.
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *     * Redistributions in binary form must reproduce the above
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * copyright notice, this list of conditions and the following disclaimer
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * in the documentation and/or other materials provided with the
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * distribution.
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *     * Neither the name of Google Inc. nor the names of its
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * contributors may be used to endorse or promote products derived from
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * this software without specific prior written permission.
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * ---
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) * Author: Maxim Lifantsev
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) */
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef BASE_STL_ALLOCATOR_H_
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define BASE_STL_ALLOCATOR_H_
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <config.h>
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stddef.h>   // for ptrdiff_t
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <limits>
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h"
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Generic allocator class for STL objects
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// that uses a given type-less allocator Alloc, which must provide:
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//   static void* Alloc::Allocate(size_t size);
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//   static void Alloc::Free(void* ptr, size_t size);
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// STL_Allocator<T, MyAlloc> provides the same thread-safety
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// guarantees as MyAlloc.
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Usage example:
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//   set<T, less<T>, STL_Allocator<T, MyAlloc> > my_set;
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// CAVEAT: Parts of the code below are probably specific
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         to the STL version(s) we are using.
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//         The code is simply lifted from what std::allocator<> provides.
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)template <typename T, class Alloc>
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class STL_Allocator {
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public:
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  typedef size_t     size_type;
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  typedef ptrdiff_t  difference_type;
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  typedef T*         pointer;
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  typedef const T*   const_pointer;
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  typedef T&         reference;
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  typedef const T&   const_reference;
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  typedef T          value_type;
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  template <class T1> struct rebind {
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    typedef STL_Allocator<T1, Alloc> other;
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  };
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  STL_Allocator() { }
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  STL_Allocator(const STL_Allocator&) { }
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  template <class T1> STL_Allocator(const STL_Allocator<T1, Alloc>&) { }
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ~STL_Allocator() { }
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  pointer address(reference x) const { return &x; }
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const_pointer address(const_reference x) const { return &x; }
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  pointer allocate(size_type n, const void* = 0) {
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    RAW_DCHECK((n * sizeof(T)) / sizeof(T) == n, "n is too big to allocate");
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return static_cast<T*>(Alloc::Allocate(n * sizeof(T)));
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  void deallocate(pointer p, size_type n) { Alloc::Free(p, n * sizeof(T)); }
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  size_type max_size() const { return size_t(-1) / sizeof(T); }
885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  void construct(pointer p, const T& val) { ::new(p) T(val); }
905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  void construct(pointer p) { ::new(p) T(); }
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  void destroy(pointer p) { p->~T(); }
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // There's no state, so these allocators are always equal
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool operator==(const STL_Allocator&) const { return true; }
955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif  // BASE_STL_ALLOCATOR_H_
98