summaryrefslogtreecommitdiffstats
path: root/src/3rdparty/angle/src/compiler/translator/UnfoldShortCircuitAST.h
blob: 24c14a60e37fedb222cf926d0863bbd2235779c3 (plain)
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
//
// Copyright (c) 2002-2013 The ANGLE Project Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//
// UnfoldShortCircuitAST is an AST traverser to replace short-circuiting
// operations with ternary operations.
//

#ifndef COMPILER_UNFOLD_SHORT_CIRCUIT_AST_H_
#define COMPILER_UNFOLD_SHORT_CIRCUIT_AST_H_

#include "common/angleutils.h"
#include "compiler/translator/intermediate.h"

// This traverser identifies all the short circuit binary  nodes that need to
// be replaced, and creates the corresponding replacement nodes. However,
// the actual replacements happen after the traverse through updateTree().

class UnfoldShortCircuitAST : public TIntermTraverser
{
  public:
    UnfoldShortCircuitAST() { }

    virtual bool visitBinary(Visit visit, TIntermBinary *);

    void updateTree();

  private:
    struct NodeUpdateEntry
    {
        NodeUpdateEntry(TIntermNode *_parent,
                        TIntermNode *_original,
                        TIntermNode *_replacement)
            : parent(_parent),
              original(_original),
              replacement(_replacement) {}

        TIntermNode *parent;
        TIntermNode *original;
        TIntermNode *replacement;
    };

    // During traversing, save all the replacements that need to happen;
    // then replace them by calling updateNodes().
    std::vector<NodeUpdateEntry> replacements;

    DISALLOW_COPY_AND_ASSIGN(UnfoldShortCircuitAST);
};

#endif  // COMPILER_UNFOLD_SHORT_CIRCUIT_AST_H_