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
|
//
// Copyright 2022 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/d3d/AggregateAssignStructsInSSBOs.h"
#include "compiler/translator/StaticType.h"
#include "compiler/translator/Symbol.h"
#include "compiler/translator/tree_util/IntermNode_util.h"
#include "compiler/translator/tree_util/IntermTraverse.h"
#include "compiler/translator/util.h"
namespace sh
{
namespace
{
class AggregateAssignStructsInSSBOsTraverser : public TIntermTraverser
{
public:
AggregateAssignStructsInSSBOsTraverser(TSymbolTable *symbolTable)
: TIntermTraverser(true, false, false, symbolTable)
{}
protected:
bool visitBinary(Visit visit, TIntermBinary *node) override
{
// Replace all assignments to structs in SSBOs with field-by-field asignments.
// TODO(anglebug.com/7362): this implementation only works for the simple case (assignment
// statement), not more complex cases such as assignment-as-expression or functions with
// side effects in the RHS.
const TStructure *s;
if (node->getOp() != EOpAssign)
{
return true;
}
else if (!IsInShaderStorageBlock(node->getLeft()))
{
return true;
}
else if (!(s = node->getLeft()->getType().getStruct()))
{
return true;
}
ASSERT(node->getRight()->getType().getStruct() == s);
auto *block = new TIntermBlock();
for (int i = 0; i < static_cast<int>(s->fields().size()); ++i)
{
auto *left = new TIntermBinary(EOpIndexDirectStruct, node->getLeft()->deepCopy(),
CreateIndexNode(i));
auto *right = new TIntermBinary(EOpIndexDirectStruct, node->getRight()->deepCopy(),
CreateIndexNode(i));
auto *assign = new TIntermBinary(TOperator::EOpAssign, left, right);
block->appendStatement(assign);
}
queueReplacement(block, OriginalNode::IS_DROPPED);
return false;
}
};
} // namespace
bool AggregateAssignStructsInSSBOs(TCompiler *compiler,
TIntermBlock *root,
TSymbolTable *symbolTable)
{
AggregateAssignStructsInSSBOsTraverser traverser(symbolTable);
root->traverse(&traverser);
return traverser.updateTree(compiler, root);
}
} // namespace sh
|