forked from lcompilers/lpython
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdead_code_removal.cpp
More file actions
109 lines (92 loc) · 3.75 KB
/
Copy pathdead_code_removal.cpp
File metadata and controls
109 lines (92 loc) · 3.75 KB
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
#include <libasr/asr.h>
#include <libasr/containers.h>
#include <libasr/exception.h>
#include <libasr/asr_utils.h>
#include <libasr/asr_verify.h>
#include <libasr/pass/dead_code_removal.h>
#include <libasr/pass/pass_utils.h>
#include <vector>
#include <map>
#include <utility>
namespace LCompilers {
using ASR::down_cast;
using ASR::is_a;
class DeadCodeRemovalVisitor : public PassUtils::PassVisitor<DeadCodeRemovalVisitor>
{
private:
std::string rl_path;
public:
bool dead_code_removed;
DeadCodeRemovalVisitor(Allocator &al_, const std::string& rl_path_) : PassVisitor(al_, nullptr),
rl_path(rl_path_), dead_code_removed(false)
{
pass_result.reserve(al, 1);
}
void visit_If(const ASR::If_t& x) {
ASR::If_t& xx = const_cast<ASR::If_t&>(x);
transform_stmts(xx.m_body, xx.n_body);
transform_stmts(xx.m_orelse, xx.n_orelse);
ASR::expr_t* m_test_value = ASRUtils::expr_value(x.m_test);
bool m_test_bool;
if( ASRUtils::is_value_constant(m_test_value, m_test_bool) ) {
ASR::stmt_t** selected_part = nullptr;
size_t n_selected_part = 0;
if( m_test_bool ) {
selected_part = x.m_body;
n_selected_part = x.n_body;
} else {
selected_part = x.m_orelse;
n_selected_part = x.n_orelse;
}
for( size_t i = 0; i < n_selected_part; i++ ) {
pass_result.push_back(al, selected_part[i]);
}
dead_code_removed = true;
}
}
void visit_Select(const ASR::Select_t& x) {
ASR::Select_t& xx = const_cast<ASR::Select_t&>(x);
ASR::expr_t* m_test_value = ASRUtils::expr_value(x.m_test);
if( !ASRUtils::is_value_constant(m_test_value) ) {
return ;
}
for( size_t i = 0; i < x.n_body; i++ ) {
ASR::case_stmt_t* case_body = x.m_body[i];
switch (case_body->type) {
case ASR::case_stmtType::CaseStmt: {
ASR::CaseStmt_t* casestmt = ASR::down_cast<ASR::CaseStmt_t>(case_body);
transform_stmts(casestmt->m_body, casestmt->n_body);
xx.m_body[i] = (ASR::case_stmt_t*)(&(casestmt->base));
for( size_t j = 0; j < casestmt->n_test; j++ ) {
if( ASRUtils::is_value_equal(casestmt->m_test[j], x.m_test) ) {
for( size_t k = 0; k < casestmt->n_body; k++ ) {
pass_result.push_back(al, casestmt->m_body[k]);
}
return ;
}
}
break;
}
case ASR::case_stmtType::CaseStmt_Range: {
ASR::CaseStmt_Range_t* casestmt_range = ASR::down_cast<ASR::CaseStmt_Range_t>(case_body);
transform_stmts(casestmt_range->m_body, casestmt_range->n_body);
xx.m_body[i] = (ASR::case_stmt_t*)(&(casestmt_range->base));
if( ASRUtils::is_value_in_range(casestmt_range->m_start, casestmt_range->m_end, x.m_test) ) {
for( size_t k = 0; k < casestmt_range->n_body; k++ ) {
pass_result.push_back(al, casestmt_range->m_body[k]);
}
return ;
}
break;
}
}
}
}
};
void pass_dead_code_removal(Allocator &al, ASR::TranslationUnit_t &unit,
const LCompilers::PassOptions& pass_options) {
std::string rl_path = pass_options.runtime_library_dir;
DeadCodeRemovalVisitor v(al, rl_path);
v.visit_TranslationUnit(unit);
}
} // namespace LCompilers