Code refactor: add generic Cycles node infrastructure.
[blender-staging.git] / intern / cycles / graph / node_enum.h
1 /*
2  * Copyright 2011-2016 Blender Foundation
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #pragma once
18
19 #include "util_map.h"
20 #include "util_param.h"
21
22 CCL_NAMESPACE_BEGIN
23
24 /* Enum
25  *
26  * Utility class for enum values. */
27
28 struct NodeEnum {
29         bool empty() const { return left.empty(); }
30         void insert(const char *x, int y) {
31                 left[ustring(x)] = y;
32                 right[y] = ustring(x);
33         }
34
35         bool exists(ustring x) const { return left.find(x) != left.end(); }
36         bool exists(int y) const { return right.find(y) != right.end(); }
37
38         int operator[](const char *x) const { return left.find(ustring(x))->second; }
39         int operator[](ustring x) const { return left.find(x)->second; }
40         ustring operator[](int y) const { return right.find(y)->second; }
41
42 private:
43         unordered_map<ustring, int, ustringHash> left;
44         unordered_map<int, ustring> right;
45 };
46
47 CCL_NAMESPACE_END
48