66 lines
1.1 KiB
C++
66 lines
1.1 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#include "util/map.h"
|
|
#include "util/param.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
/* Enum
|
|
*
|
|
* Utility class for enum values. */
|
|
|
|
struct NodeEnum {
|
|
bool empty() const
|
|
{
|
|
return left.empty();
|
|
}
|
|
void insert(const char *x, const int y)
|
|
{
|
|
const ustring ustr_x(x);
|
|
|
|
left[ustr_x] = y;
|
|
right[y] = ustr_x;
|
|
}
|
|
|
|
bool exists(ustring x) const
|
|
{
|
|
return left.find(x) != left.end();
|
|
}
|
|
bool exists(const int y) const
|
|
{
|
|
return right.find(y) != right.end();
|
|
}
|
|
|
|
int operator[](const char *x) const
|
|
{
|
|
return left.find(ustring(x))->second;
|
|
}
|
|
int operator[](ustring x) const
|
|
{
|
|
return left.find(x)->second;
|
|
}
|
|
ustring operator[](int y) const
|
|
{
|
|
return right.find(y)->second;
|
|
}
|
|
|
|
unordered_map<ustring, int>::const_iterator begin() const
|
|
{
|
|
return left.begin();
|
|
}
|
|
unordered_map<ustring, int>::const_iterator end() const
|
|
{
|
|
return left.end();
|
|
}
|
|
|
|
private:
|
|
unordered_map<ustring, int> left;
|
|
unordered_map<int, ustring> right;
|
|
};
|
|
|
|
CCL_NAMESPACE_END
|