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_RETURNSTATEMENT
9#define SKSL_RETURNSTATEMENT
10
11#include "src/sksl/ir/SkSLExpression.h"
12#include "src/sksl/ir/SkSLStatement.h"
13
14namespace SkSL {
15
16/**
17 * A 'return' statement.
18 */
19struct ReturnStatement : public Statement {
20 ReturnStatement(int offset)
21 : INHERITED(offset, kReturn_Kind) {}
22
23 ReturnStatement(std::unique_ptr<Expression> expression)
24 : INHERITED(expression->fOffset, kReturn_Kind)
25 , fExpression(std::move(expression)) {}
26
27 int nodeCount() const override {
28 return 1 + fExpression->nodeCount();
29 }
30
31 std::unique_ptr<Statement> clone() const override {
32 if (fExpression) {
33 return std::unique_ptr<Statement>(new ReturnStatement(fExpression->clone()));
34 }
35 return std::unique_ptr<Statement>(new ReturnStatement(fOffset));
36 }
37
38 String description() const override {
39 if (fExpression) {
40 return "return " + fExpression->description() + ";";
41 } else {
42 return String("return;");
43 }
44 }
45
46 std::unique_ptr<Expression> fExpression;
47
48 typedef Statement INHERITED;
49};
50
51} // namespace SkSL
52
53#endif
54