aboutsummaryrefslogtreecommitdiffstats
path: root/libyul/optimiser/SSAReverser.cpp
blob: 6548ebb5ce1d1a19cc5923c5e21db7e6fc26b484 (plain) (blame)
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
102
103
104
105
106
107
108
109
110
111
112
113
114
/*
    This file is part of solidity.

    solidity is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    solidity is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with solidity.  If not, see <http://www.gnu.org/licenses/>.
*/
#include <libyul/optimiser/SSAReverser.h>
#include <libyul/optimiser/Metrics.h>
#include <libyul/AsmData.h>
#include <libdevcore/CommonData.h>

using namespace std;
using namespace dev;
using namespace yul;

void SSAReverser::run(Block& _block)
{
    AssignmentCounter assignmentCounter;
    assignmentCounter(_block);
    SSAReverser{assignmentCounter}(_block);
}

void SSAReverser::operator()(Block& _block)
{
    walkVector(_block.statements);
    iterateReplacingWindow<2>(
        _block.statements,
        [&](Statement& _stmt1, Statement& _stmt2) -> boost::optional<vector<Statement>>
        {
            auto* varDecl = boost::get<VariableDeclaration>(&_stmt1);

            if (!varDecl || varDecl->variables.size() != 1 || !varDecl->value)
                return {};

            // Replaces
            //   let a_1 := E
            //   a := a_1
            // with
            //   a := E
            //   let a_1 := a
            if (auto* assignment = boost::get<Assignment>(&_stmt2))
            {
                auto* identifier = boost::get<Identifier>(assignment->value.get());
                if (
                    assignment->variableNames.size() == 1 &&
                    identifier &&
                    identifier->name == varDecl->variables.front().name
                )
                {
                    vector<Statement> result;
                    result.emplace_back(Assignment{
                        std::move(assignment->location),
                        assignment->variableNames,
                        std::move(varDecl->value)
                    });
                    result.emplace_back(VariableDeclaration{
                        std::move(varDecl->location),
                        std::move(varDecl->variables),
                        std::make_unique<Expression>(std::move(assignment->variableNames.front()))
                    });
                    return { std::move(result) };
                }
            }
            // Replaces
            //   let a_1 := E
            //   let a := a_1
            // with
            //   let a := E
            //   let a_1 := a
            else if (auto* varDecl2 = boost::get<VariableDeclaration>(&_stmt2))
            {
                auto* identifier = boost::get<Identifier>(varDecl2->value.get());
                if (
                    varDecl2->variables.size() == 1 &&
                    identifier &&
                    identifier->name == varDecl->variables.front().name && (
                        m_assignmentCounter.assignmentCount(varDecl2->variables.front().name) >
                        m_assignmentCounter.assignmentCount(varDecl->variables.front().name)
                   )
                )
                {
                    vector<Statement> result;
                    auto varIdentifier2 = std::make_unique<Expression>(Identifier{
                        varDecl2->variables.front().location,
                        varDecl2->variables.front().name
                    });
                    result.emplace_back(VariableDeclaration{
                        std::move(varDecl2->location),
                        std::move(varDecl2->variables),
                        std::move(varDecl->value)
                    });
                    result.emplace_back(VariableDeclaration{
                        std::move(varDecl->location),
                        std::move(varDecl->variables),
                        std::move(varIdentifier2)
                    });
                    return { std::move(result) };
                }
            }

            return {};
        }
    );
}