blob: 88c4129ee2d57c435861dde08055a046d605783c (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
/*
* Copyright 2016 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef SKSL_PROGRAMELEMENT
#define SKSL_PROGRAMELEMENT
#include "include/private/SkSLIRNode.h"
#include <memory>
namespace SkSL {
/**
* Represents a top-level element (e.g. function or global variable) in a program.
*/
class ProgramElement : public IRNode {
public:
enum class Kind {
kExtension = 0,
kFunction,
kFunctionPrototype,
kGlobalVar,
kInterfaceBlock,
kModifiers,
kStructDefinition,
kFirst = kExtension,
kLast = kStructDefinition
};
ProgramElement(int offset, Kind kind)
: INHERITED(offset, (int) kind) {
SkASSERT(kind >= Kind::kFirst && kind <= Kind::kLast);
}
Kind kind() const {
return (Kind) fKind;
}
/**
* Use is<T> to check the type of a program element.
* e.g. replace `el.kind() == ProgramElement::Kind::kExtension` with `el.is<Extension>()`.
*/
template <typename T>
bool is() const {
return this->kind() == T::kProgramElementKind;
}
/**
* Use as<T> to downcast program elements. e.g. replace `(Extension&) el` with
* `el.as<Extension>()`.
*/
template <typename T>
const T& as() const {
SkASSERT(this->is<T>());
return static_cast<const T&>(*this);
}
template <typename T>
T& as() {
SkASSERT(this->is<T>());
return static_cast<T&>(*this);
}
virtual std::unique_ptr<ProgramElement> clone() const = 0;
private:
using INHERITED = IRNode;
};
} // namespace SkSL
#endif
|
