Home | History | Annotate | Download | only in public
      1 /*
      2  * Copyright (C) 2009 Google 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 are
      6  * met:
      7  *
      8  *     * Redistributions of source code must retain the above copyright
      9  * notice, this list of conditions and the following disclaimer.
     10  *     * Redistributions in binary form must reproduce the above
     11  * copyright notice, this list of conditions and the following disclaimer
     12  * in the documentation and/or other materials provided with the
     13  * distribution.
     14  *     * Neither the name of Google Inc. nor the names of its
     15  * contributors may be used to endorse or promote products derived from
     16  * this software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     29  */
     30 
     31 #ifndef WebVector_h
     32 #define WebVector_h
     33 
     34 #include "WebCommon.h"
     35 
     36 #include <algorithm>
     37 
     38 namespace WebKit {
     39 
     40 // A simple vector class.
     41 //
     42 // Sample usage:
     43 //
     44 //   void Foo(WebVector<int>& result)
     45 //   {
     46 //       WebVector<int> data(10);
     47 //       for (size_t i = 0; i < data.size(); ++i)
     48 //           data[i] = ...
     49 //       result.swap(data);
     50 //   }
     51 //
     52 // It is also possible to assign from other types of random access
     53 // containers:
     54 //
     55 //   void Foo(const std::vector<std::string>& input)
     56 //   {
     57 //       WebVector<WebCString> cstrings = input;
     58 //       ...
     59 //   }
     60 //
     61 template <typename T>
     62 class WebVector {
     63 public:
     64     typedef T ValueType;
     65 
     66     ~WebVector()
     67     {
     68         destroy();
     69     }
     70 
     71     explicit WebVector(size_t size = 0)
     72     {
     73         initialize(size);
     74     }
     75 
     76     WebVector(const WebVector<T>& other)
     77     {
     78         initializeFrom(other.m_ptr, other.m_size);
     79     }
     80 
     81     template <typename C>
     82     WebVector(const C& other)
     83     {
     84         initializeFrom(other.size() ? &other[0] : 0, other.size());
     85     }
     86 
     87     WebVector& operator=(const WebVector& other)
     88     {
     89         if (this != &other)
     90             assign(other);
     91         return *this;
     92     }
     93 
     94     template <typename C>
     95     WebVector<T>& operator=(const C& other)
     96     {
     97         if (this != reinterpret_cast<const WebVector<T>*>(&other))
     98             assign(other);
     99         return *this;
    100     }
    101 
    102     template <typename C>
    103     void assign(const C& other)
    104     {
    105         assign(other.size() ? &other[0] : 0, other.size());
    106     }
    107 
    108     template <typename U>
    109     void assign(const U* values, size_t size)
    110     {
    111         destroy();
    112         initializeFrom(values, size);
    113     }
    114 
    115     size_t size() const { return m_size; }
    116     bool isEmpty() const { return !m_size; }
    117 
    118     T& operator[](size_t i)
    119     {
    120         WEBKIT_ASSERT(i < m_size);
    121         return m_ptr[i];
    122     }
    123     const T& operator[](size_t i) const
    124     {
    125         WEBKIT_ASSERT(i < m_size);
    126         return m_ptr[i];
    127     }
    128 
    129     bool contains(const T& value) const
    130     {
    131         for (size_t i = 0; i < m_size; i++) {
    132             if (m_ptr[i] == value)
    133                 return true;
    134         }
    135         return false;
    136     }
    137 
    138     T* data() { return m_ptr; }
    139     const T* data() const { return m_ptr; }
    140 
    141     void swap(WebVector<T>& other)
    142     {
    143         std::swap(m_ptr, other.m_ptr);
    144         std::swap(m_size, other.m_size);
    145     }
    146 
    147 private:
    148     void initialize(size_t size)
    149     {
    150         m_size = size;
    151         if (!m_size)
    152             m_ptr = 0;
    153         else {
    154             m_ptr = static_cast<T*>(::operator new(sizeof(T) * m_size));
    155             for (size_t i = 0; i < m_size; ++i)
    156                 new (&m_ptr[i]) T();
    157         }
    158     }
    159 
    160     template <typename U>
    161     void initializeFrom(const U* values, size_t size)
    162     {
    163         m_size = size;
    164         if (!m_size)
    165             m_ptr = 0;
    166         else {
    167             m_ptr = static_cast<T*>(::operator new(sizeof(T) * m_size));
    168             for (size_t i = 0; i < m_size; ++i)
    169                 new (&m_ptr[i]) T(values[i]);
    170         }
    171     }
    172 
    173     void destroy()
    174     {
    175         for (size_t i = 0; i < m_size; ++i)
    176             m_ptr[i].~T();
    177         ::operator delete(m_ptr);
    178     }
    179 
    180     T* m_ptr;
    181     size_t m_size;
    182 };
    183 
    184 } // namespace WebKit
    185 
    186 #endif
    187