GroupIndexTable.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2024 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* GroupIndexTable.h                                           (C) 2000-2024 */
/*                                                                           */
/* Table de hachage entre un item et sa position dans la table.              */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_CORE_GROUPINDEXTABLE_H
#define ARCANE_CORE_GROUPINDEXTABLE_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/HashTable.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
class ItemGroupImpl;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
class ARCANE_CORE_EXPORT GroupIndexTableView
{
  friend class GroupIndexTable;
  typedef Int32 KeyTypeValue;
  typedef Int32 ValueType;
  typedef HashTraitsT<KeyTypeValue> KeyTraitsType;
  typedef KeyTraitsType::KeyTypeConstRef KeyTypeConstRef;
 
 public:
 
  ARCCORE_HOST_DEVICE ValueType operator[](Int32 i) const { return _lookup(i); }
  ARCCORE_HOST_DEVICE Int32 size() const { return m_key_buffer_span.size(); }
 
 private:
 
  SmallSpan<const KeyTypeValue> m_key_buffer_span;
  SmallSpan<const Int32> m_next_buffer_span;
  SmallSpan<const Int32> m_buckets_span;
  Int32 m_nb_bucket = 0;
 
 private:

  ARCCORE_HOST_DEVICE Int32 _lookup(KeyTypeConstRef id) const
  {
    return _lookupBucket(_hash(id), id);
  }
  ARCCORE_HOST_DEVICE Int32 _hash(KeyTypeConstRef id) const
  {
    return static_cast<Int32>(KeyTraitsType::hashFunction(id) % m_nb_bucket);
  }
  ARCCORE_HOST_DEVICE Integer _lookupBucket(Int32 bucket, KeyTypeConstRef id) const
  {
    for (Integer i = m_buckets_span[bucket]; i >= 0; i = m_next_buffer_span[i]) {
      if (m_key_buffer_span[i] == id)
        return i;
    }
    return -1;
  }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_CORE_EXPORT GroupIndexTable
: public HashTableBase
{
 public:
 
  typedef Int32 KeyTypeValue;
  typedef Int32 ValueType;
  typedef HashTraitsT<KeyTypeValue> KeyTraitsType;
  typedef KeyTraitsType::KeyTypeConstRef KeyTypeConstRef;
 
 public:
 
  explicit GroupIndexTable(ItemGroupImpl* group_impl);
 
 public:
 
  void update();
 
  void clear();
 
  void compact(const Int32ConstArrayView* info);
 
  ValueType operator[](Int32 i) const { return _lookup(i); }
 
  KeyTypeValue keyLocalId(Int32 i) const { return m_key_buffer[i]; }
 
  Int32 size() const { return m_key_buffer.size(); }
 
  GroupIndexTableView view() const
  {
    ARCANE_ASSERT((_initialized()), ("GroupIndexTable not initialized"));
    ARCANE_ASSERT((_checkIntegrity(false)), ("GroupIndexTable integrity failed"));
    return m_view;
  }
 
 private:

  Int32 _hash(KeyTypeConstRef id) const
  {
    ARCANE_ASSERT((_initialized()), ("GroupIndexTable not initialized"));
    return m_view._hash(id);
  }

  bool _hasKey(KeyTypeConstRef id) const;

  Int32 _lookupBucket(Int32 bucket, KeyTypeConstRef id) const
  {
    ARCANE_ASSERT((_initialized()), ("GroupIndexTable not initialized"));
    return m_view._lookupBucket(bucket, id);
  }

  Int32 _lookup(KeyTypeConstRef id) const
  {
    ARCANE_ASSERT((_checkIntegrity(false)), ("GroupIndexTable integrity failed"));
    return _lookupBucket(_hash(id), id);
  }

  bool _initialized() const;

  bool _checkIntegrity(bool full = true) const;
 
 private:
 
  ItemGroupImpl* m_group_impl = nullptr;
  UniqueArray<KeyTypeValue> m_key_buffer; 
  UniqueArray<Int32> m_next_buffer; 
  UniqueArray<Int32> m_buckets; 
  bool m_disable_check_integrity = false;
  GroupIndexTableView m_view;
 
 private:
 
  void _updateSpan();
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif