summaryrefslogtreecommitdiffstats
path: root/gfx/angle/checkout/src/compiler/translator/tree_ops/ForcePrecisionQualifier.cpp
blob: 00dd0131c64847f93a9e599aaed6bcbc8050b363 (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
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
//
// Copyright 2020 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.
//

#include "compiler/translator/tree_ops/ForcePrecisionQualifier.h"
#include "angle_gl.h"
#include "common/debug.h"
#include "compiler/translator/Compiler.h"
#include "compiler/translator/tree_util/IntermTraverse.h"
#include "compiler/translator/util.h"

namespace sh
{

namespace
{
class TPrecisionTraverser : public TIntermTraverser
{
  public:
    TPrecisionTraverser(TSymbolTable *symbolTable);

  protected:
    bool visitDeclaration(Visit visit, TIntermDeclaration *node) override;

    void overwriteVariablePrecision(TType *type) const;
};

TPrecisionTraverser::TPrecisionTraverser(TSymbolTable *symbolTable)
    : TIntermTraverser(true, true, true, symbolTable)
{}

void TPrecisionTraverser::overwriteVariablePrecision(TType *type) const
{
    if (type->getPrecision() == EbpHigh)
    {
        type->setPrecision(EbpMedium);
    }
}

bool TPrecisionTraverser::visitDeclaration(Visit visit, TIntermDeclaration *node)
{
    // Variable declaration.
    if (visit == PreVisit)
    {
        const TIntermSequence &sequence = *(node->getSequence());
        TIntermTyped *variable          = sequence.front()->getAsTyped();
        const TType &type               = variable->getType();
        TQualifier qualifier            = variable->getQualifier();

        // Don't modify uniform since it might be shared between vertex and fragment shader
        if (qualifier == EvqUniform)
        {
            return true;
        }

        // Visit the struct.
        if (type.isStructSpecifier())
        {
            const TStructure *structure = type.getStruct();
            const TFieldList &fields    = structure->fields();
            for (size_t i = 0; i < fields.size(); ++i)
            {
                const TField *field    = fields[i];
                const TType *fieldType = field->type();
                overwriteVariablePrecision((TType *)fieldType);
            }
        }
        else if (type.getBasicType() == EbtInterfaceBlock)
        {
            const TInterfaceBlock *interfaceBlock = type.getInterfaceBlock();
            const TFieldList &fields              = interfaceBlock->fields();
            for (const TField *field : fields)
            {
                const TType *fieldType = field->type();
                overwriteVariablePrecision((TType *)fieldType);
            }
        }
        else
        {
            overwriteVariablePrecision((TType *)&type);
        }
    }
    return true;
}
}  // namespace

bool ForceShaderPrecisionToMediump(TIntermNode *root, TSymbolTable *symbolTable, GLenum shaderType)
{
    if (shaderType != GL_FRAGMENT_SHADER)
    {
        return true;
    }

    TPrecisionTraverser traverser(symbolTable);
    root->traverse(&traverser);
    return true;
}

}  // namespace sh