1/*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8#ifndef SKSL_TERNARYEXPRESSION
9#define SKSL_TERNARYEXPRESSION
10
11#include "src/sksl/SkSLPosition.h"
12#include "src/sksl/ir/SkSLExpression.h"
13
14namespace SkSL {
15
16/**
17 * A ternary expression (test ? ifTrue : ifFalse).
18 */
19struct TernaryExpression : public Expression {
20 TernaryExpression(int offset, std::unique_ptr<Expression> test,
21 std::unique_ptr<Expression> ifTrue, std::unique_ptr<Expression> ifFalse)
22 : INHERITED(offset, kTernary_Kind, ifTrue->fType)
23 , fTest(std::move(test))
24 , fIfTrue(std::move(ifTrue))
25 , fIfFalse(std::move(ifFalse)) {
26 SkASSERT(fIfTrue->fType == fIfFalse->fType);
27 }
28
29 bool hasProperty(Property property) const override {
30 return fTest->hasProperty(property) || fIfTrue->hasProperty(property) ||
31 fIfFalse->hasProperty(property);
32 }
33
34 bool isConstantOrUniform() const override {
35 return fTest->isConstantOrUniform() && fIfTrue->isConstantOrUniform() &&
36 fIfFalse->isConstantOrUniform();
37 }
38
39 int nodeCount() const override {
40 return 1 + fTest->nodeCount() + fIfTrue->nodeCount() + fIfFalse->nodeCount();
41 }
42
43 std::unique_ptr<Expression> clone() const override {
44 return std::unique_ptr<Expression>(new TernaryExpression(fOffset, fTest->clone(),
45 fIfTrue->clone(),
46 fIfFalse->clone()));
47 }
48
49 String description() const override {
50 return "(" + fTest->description() + " ? " + fIfTrue->description() + " : " +
51 fIfFalse->description() + ")";
52 }
53
54 std::unique_ptr<Expression> fTest;
55 std::unique_ptr<Expression> fIfTrue;
56 std::unique_ptr<Expression> fIfFalse;
57
58 typedef Expression INHERITED;
59};
60
61} // namespace SkSL
62
63#endif
64