ec0ecf08da
Add a `const` version of `blender::listbase_to_vector`, and use it in `animrig::fcurves_for_action_slot()`. No functional changes.
129 lines
2.6 KiB
C++
129 lines
2.6 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#pragma once
|
|
|
|
/** \file
|
|
* \ingroup bli
|
|
*
|
|
* `blender::ListBaseWrapper` is a typed wrapper for the #ListBase struct. That makes it safer and
|
|
* more convenient to use in C++ in some cases. However, if you find yourself iterating over a
|
|
* linked list a lot, consider to convert it into a vector for further processing. This improves
|
|
* performance and debug-ability.
|
|
*/
|
|
|
|
#include "BLI_listbase.h"
|
|
#include "BLI_vector.hh"
|
|
|
|
#include "DNA_listBase.h"
|
|
|
|
namespace blender {
|
|
|
|
template<typename LB, typename T> class ListBaseWrapperTemplate {
|
|
private:
|
|
LB *listbase_;
|
|
|
|
public:
|
|
ListBaseWrapperTemplate(LB *listbase) : listbase_(listbase)
|
|
{
|
|
BLI_assert(listbase);
|
|
}
|
|
|
|
ListBaseWrapperTemplate(LB &listbase) : ListBaseWrapperTemplate(&listbase) {}
|
|
|
|
class Iterator {
|
|
private:
|
|
LB *listbase_;
|
|
T *current_;
|
|
|
|
public:
|
|
Iterator(LB *listbase, T *current) : listbase_(listbase), current_(current) {}
|
|
|
|
Iterator &operator++()
|
|
{
|
|
/* Some types store `next/prev` using `void *`, so cast is necessary. */
|
|
current_ = static_cast<T *>(current_->next);
|
|
return *this;
|
|
}
|
|
|
|
Iterator operator++(int)
|
|
{
|
|
Iterator iterator = *this;
|
|
++(*this);
|
|
return iterator;
|
|
}
|
|
|
|
bool operator!=(const Iterator &iterator) const
|
|
{
|
|
return current_ != iterator.current_;
|
|
}
|
|
|
|
T *operator*() const
|
|
{
|
|
return current_;
|
|
}
|
|
};
|
|
|
|
Iterator begin() const
|
|
{
|
|
return Iterator(listbase_, static_cast<T *>(listbase_->first));
|
|
}
|
|
|
|
Iterator end() const
|
|
{
|
|
return Iterator(listbase_, nullptr);
|
|
}
|
|
|
|
T *get(uint index) const
|
|
{
|
|
void *ptr = BLI_findlink(listbase_, index);
|
|
BLI_assert(ptr);
|
|
return static_cast<T *>(ptr);
|
|
}
|
|
|
|
int64_t index_of(const T *value) const
|
|
{
|
|
int64_t index = 0;
|
|
for (T *ptr : *this) {
|
|
if (ptr == value) {
|
|
return index;
|
|
}
|
|
index++;
|
|
}
|
|
BLI_assert(false);
|
|
return -1;
|
|
}
|
|
};
|
|
|
|
template<typename T> using ListBaseWrapper = ListBaseWrapperTemplate<ListBase, T>;
|
|
template<typename T> using ConstListBaseWrapper = ListBaseWrapperTemplate<const ListBase, const T>;
|
|
|
|
/**
|
|
* Convert a ListBase to a Vector.
|
|
*/
|
|
template<typename T> Vector<T *> listbase_to_vector(ListBase &list)
|
|
{
|
|
Vector<T *> vector;
|
|
|
|
for (T *item : ListBaseWrapper<T>(list)) {
|
|
vector.append(item);
|
|
}
|
|
return vector;
|
|
}
|
|
|
|
/**
|
|
* Convert a ListBase to a Vector.
|
|
*/
|
|
template<typename T> Vector<T *> listbase_to_vector(const ListBase &list)
|
|
{
|
|
Vector<T *> vector;
|
|
|
|
for (T *item : ConstListBaseWrapper<T>(list)) {
|
|
vector.append(item);
|
|
}
|
|
return vector;
|
|
}
|
|
|
|
} /* namespace blender */
|