hhvm/hphp/runtime/base/hphp_array-defs.h

/*
   +----------------------------------------------------------------------+
   | HipHop for PHP                                                       |
   +----------------------------------------------------------------------+
   | Copyright (c) 2010-2013 Facebook, Inc. (http://www.facebook.com)     |
   +----------------------------------------------------------------------+
   | This source file is subject to version 3.01 of the PHP license,      |
   | that is bundled with this package in the file LICENSE, and is        |
   | available through the world-wide-web at the following url:           |
   | http://www.php.net/license/3_01.txt                                  |
   | If you did not receive a copy of the PHP license and are unable to   |
   | obtain it through the world-wide-web, please send a note to          |
   | license@php.net so we can mail you a copy immediately.               |
   +----------------------------------------------------------------------+
*/

#ifndef incl_HPHP_HPHP_ARRAY_INL_H_
#define incl_HPHP_HPHP_ARRAY_INL_H_

#include "hphp/runtime/base/hphp_array.h"
#include "hphp/util/stacktrace_profiler.h"

namespace HPHP {
///////////////////////////////////////////////////////////////////////////////

inline void* ArrayData::modeAlloc(size_t nbytes) const {
  return m_allocMode == AllocationMode::smart ? smart_malloc(nbytes) :
         Util::safe_malloc(nbytes);
}

inline void* ArrayData::modeRealloc(void* ptr, size_t nbytes) const {
  return m_allocMode == AllocationMode::smart ? smart_realloc(ptr, nbytes) :
         Util::safe_realloc(ptr, nbytes);
}

inline void ArrayData::modeFree(void* ptr) const {
  return LIKELY(m_allocMode == AllocationMode::smart) ? smart_free(ptr) :
         free(ptr);
}

inline void HphpArray::initHash(size_t tableSize) {
  assert(HphpArray::ElmIndEmpty == -1);
  memset(m_hash, 0xffU, tableSize * sizeof(*m_hash));
  m_hLoad = 0;
}

inline ALWAYS_INLINE
HphpArray::ElmInd* HphpArray::findForNewInsert(size_t h0) const {
  assert(!isVector());
  size_t tableMask = m_tableMask;
  size_t probeIndex = h0 & tableMask;
  ElmInd* ei = &m_hash[probeIndex];
  return !validElmInd(*ei) ? ei :
         findForNewInsertLoop(tableMask, h0);
}

inline bool HphpArray::isTombstone(ssize_t pos) const {
  assert(size_t(pos) <= m_used);
  return isTombstone(m_data[pos].data.m_type);
}

inline void HphpArray::getElmKey(const Elm& e, TypedValue* out) {
  if (e.hasIntKey()) {
    out->m_data.num = e.ikey;
    out->m_type = KindOfInt64;
    return;
  }
  auto str = e.key;
  out->m_data.pstr = str;
  out->m_type = KindOfString;
  str->incRefCount();
}

template <bool withRef> inline ALWAYS_INLINE
void HphpArray::getArrayElm(ssize_t pos, TypedValue* valOut,
                            TypedValue* keyOut) const {
  assert(size_t(pos) < m_used);
  auto& elm = m_data[pos];
  if (withRef) {
    tvAsVariant(valOut) = withRefBind(tvAsVariant(&elm.data));
    if (LIKELY(keyOut != nullptr)) {
      DataType t = keyOut->m_type;
      uint64_t d = keyOut->m_data.num;
      if (isVector()) {
        keyOut->m_data.num = pos;
        keyOut->m_type = KindOfInt64;
      } else {
        HphpArray::getElmKey(elm, keyOut);
      }
      tvRefcountedDecRefHelper(t, d);
    }
  } else {
    TypedValue* cur = tvToCell(&elm.data);
    cellDup(*cur, *valOut);
    if (keyOut) {
      if (isVector()) {
        keyOut->m_data.num = pos;
        keyOut->m_type = KindOfInt64;
      } else {
        HphpArray::getElmKey(elm, keyOut);
      }
    }
  }
}

inline HphpArray* HphpArray::asVector(ArrayData* ad) {
  assert(ad->kind() == kVectorKind);
  assert(dynamic_cast<HphpArray*>(ad));
  auto a = static_cast<HphpArray*>(ad);
  assert(a->checkInvariants());
  return a;
}

inline const HphpArray* HphpArray::asVector(const ArrayData* ad) {
  assert(ad->kind() == kVectorKind);
  assert(dynamic_cast<const HphpArray*>(ad));
  auto a = static_cast<const HphpArray*>(ad);
  assert(a->checkInvariants());
  return a;
}

inline HphpArray* HphpArray::asHphpArray(ArrayData* ad) {
  assert(ad->kind() == kMixedKind);
  assert(dynamic_cast<HphpArray*>(ad));
  auto a = static_cast<HphpArray*>(ad);
  assert(a->checkInvariants());
  return a;
}

inline const HphpArray* HphpArray::asHphpArray(const ArrayData* ad) {
  assert(ad->kind() == kMixedKind);
  assert(dynamic_cast<const HphpArray*>(ad));
  auto a = static_cast<const HphpArray*>(ad);
  assert(a->checkInvariants());
  return a;
}

inline HphpArray* HphpArray::copyImpl() const {
  return isVector() ? copyVec() : copyGeneric();
}

///////////////////////////////////////////////////////////////////////////////
}

#endif // incl_HPHP_HPHP_ARRAY_INL_H_