BKE_outliner_treehash.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
#pragma once
 
#include <memory>
 
#include "BLI_map.hh"
 
struct BLI_mempool;
struct ID;
struct TreeStoreElem;
 
namespace blender::bke::outliner::treehash {
 
/* -------------------------------------------------------------------- */
 
class TreeStoreElemKey {
 public:
  ID *id = nullptr;
  short type = 0;
  short nr = 0;
 
  explicit TreeStoreElemKey(const TreeStoreElem &elem);
  TreeStoreElemKey(ID *id, short type, short nr);
 
  uint64_t hash() const;
  friend bool operator==(const TreeStoreElemKey &a, const TreeStoreElemKey &b);
};
 
/* -------------------------------------------------------------------- */
 
class TreeHash {
  Map<TreeStoreElemKey, std::unique_ptr<class TseGroup>> elem_groups_;
 
 public:
  ~TreeHash();
 
  static std::unique_ptr<TreeHash> create_from_treestore(BLI_mempool &treestore);
 
  void rebuild_from_treestore(BLI_mempool &treestore);
 
  void clear_used();
 
  void add_element(TreeStoreElem &elem);
  void remove_element(TreeStoreElem &elem);
 
  TreeStoreElem *lookup_unused(short type, short nr, ID *id) const;
 
  TreeStoreElem *lookup_any(short type, short nr, ID *id) const;
 
 private:
  TreeHash() = default;
 
  TseGroup *lookup_group(const TreeStoreElemKey &key) const;
  TseGroup *lookup_group(const TreeStoreElem &elem) const;
  TseGroup *lookup_group(short type, short nr, ID *id) const;
  void fill_treehash(BLI_mempool &treestore);
};
 
}  // namespace blender::bke::outliner::treehash