Skip to content

Commit

Permalink
Support printing multiple lists
Browse files Browse the repository at this point in the history
  • Loading branch information
@Smit-create
Smit-create committed Jan 12, 2023
1 parent d6c75f8 commit f5ab20d
Showing 1 changed file with 167 additions and 130 deletions.
297 changes: 167 additions & 130 deletions src/libasr/pass/print_list.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -52,148 +52,185 @@ class PrintListVisitor
std::string rl_path;

public:
Vec<ASR::stmt_t*> print_pass_result_tmp;
PrintListVisitor(Allocator &al, const std::string &rl_path_)
: PassVisitor(al, nullptr), rl_path(rl_path_) {
pass_result.reserve(al, 1);
print_pass_result_tmp.reserve(al, 1);
}

void visit_Print(const ASR::Print_t &x) {
if (x.n_values == 1 &&
ASR::is_a<ASR::List_t>(*ASRUtils::expr_type(x.m_values[0]))) {
ASR::List_t *listC =
ASR::down_cast<ASR::List_t>(ASRUtils::expr_type(x.m_values[0]));

ASR::ttype_t *int_type = ASRUtils::TYPE(
ASR::make_Integer_t(al, x.base.base.loc, 4, nullptr, 0));
ASR::ttype_t *bool_type = ASRUtils::TYPE(
ASR::make_Logical_t(al, x.base.base.loc, 4, nullptr, 0));
ASR::ttype_t *str_type_len_0 = ASRUtils::TYPE(ASR::make_Character_t(
al, x.base.base.loc, 1, 0, nullptr, nullptr, 0));
ASR::ttype_t *str_type_len_1 = ASRUtils::TYPE(ASR::make_Character_t(
al, x.base.base.loc, 1, 1, nullptr, nullptr, 0));
ASR::ttype_t *str_type_len_2 = ASRUtils::TYPE(ASR::make_Character_t(
al, x.base.base.loc, 1, 2, nullptr, nullptr, 0));
ASR::expr_t *comma_space =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, x.base.base.loc, s2c(al, ", "), str_type_len_2));
ASR::expr_t *single_quote =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, x.base.base.loc, s2c(al, "'"), str_type_len_1));
ASR::expr_t *empty_str = ASRUtils::EXPR(ASR::make_StringConstant_t(
al, x.base.base.loc, s2c(al, ""), str_type_len_0));
ASR::expr_t *open_bracket =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, x.base.base.loc, s2c(al, "["), str_type_len_1));
ASR::expr_t *close_bracket =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, x.base.base.loc, s2c(al, "]"), str_type_len_1));

std::string list_iter_var_name;
ASR::symbol_t *list_iter_variable;
ASR::expr_t *list_iter_var;
{
list_iter_var_name =
current_scope->get_unique_name("__list_iterator");
list_iter_variable =
ASR::down_cast<ASR::symbol_t>(ASR::make_Variable_t(
al, x.base.base.loc, current_scope,
s2c(al, list_iter_var_name), nullptr, 0, ASR::intentType::Local,
nullptr, nullptr, ASR::storage_typeType::Default,
int_type, ASR::abiType::Source, ASR::accessType::Public,
ASR::presenceType::Required, false));

current_scope->add_symbol(list_iter_var_name,
list_iter_variable);
list_iter_var = ASRUtils::EXPR(
ASR::make_Var_t(al, x.base.base.loc, list_iter_variable));
}
void print_list_helper(ASR::expr_t *list_expr, ASR::expr_t *sep_expr,const Location &loc) {
ASR::List_t *listC =
ASR::down_cast<ASR::List_t>(ASRUtils::expr_type(list_expr));
ASR::ttype_t *int_type = ASRUtils::TYPE(
ASR::make_Integer_t(al, loc, 4, nullptr, 0));
ASR::ttype_t *bool_type = ASRUtils::TYPE(
ASR::make_Logical_t(al, loc, 4, nullptr, 0));
ASR::ttype_t *str_type_len_0 = ASRUtils::TYPE(ASR::make_Character_t(
al, loc, 1, 0, nullptr, nullptr, 0));
ASR::ttype_t *str_type_len_1 = ASRUtils::TYPE(ASR::make_Character_t(
al, loc, 1, 1, nullptr, nullptr, 0));
ASR::ttype_t *str_type_len_2 = ASRUtils::TYPE(ASR::make_Character_t(
al, loc, 1, 2, nullptr, nullptr, 0));
ASR::expr_t *comma_space =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, loc, s2c(al, ", "), str_type_len_2));
ASR::expr_t *single_quote =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, loc, s2c(al, "'"), str_type_len_1));
ASR::expr_t *empty_str = ASRUtils::EXPR(ASR::make_StringConstant_t(
al, loc, s2c(al, ""), str_type_len_0));
ASR::expr_t *open_bracket =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, loc, s2c(al, "["), str_type_len_1));
ASR::expr_t *close_bracket =
ASRUtils::EXPR(ASR::make_StringConstant_t(
al, loc, s2c(al, "]"), str_type_len_1));

std::string list_iter_var_name;
ASR::symbol_t *list_iter_variable;
ASR::expr_t *list_iter_var;
{
list_iter_var_name =
current_scope->get_unique_name("__list_iterator");
list_iter_variable =
ASR::down_cast<ASR::symbol_t>(ASR::make_Variable_t(
al, loc, current_scope,
s2c(al, list_iter_var_name), nullptr, 0, ASR::intentType::Local,
nullptr, nullptr, ASR::storage_typeType::Default,
int_type, ASR::abiType::Source, ASR::accessType::Public,
ASR::presenceType::Required, false));

current_scope->add_symbol(list_iter_var_name,
list_iter_variable);
list_iter_var = ASRUtils::EXPR(
ASR::make_Var_t(al, loc, list_iter_variable));
}

ASR::expr_t *list_item = ASRUtils::EXPR(
ASR::make_ListItem_t(al, loc, list_expr,
list_iter_var, listC->m_type, nullptr));
ASR::expr_t *list_len = ASRUtils::EXPR(ASR::make_ListLen_t(
al, loc, list_expr, int_type, nullptr));
ASR::expr_t *constant_one = ASRUtils::EXPR(
ASR::make_IntegerConstant_t(al, loc, 1, int_type));
ASR::expr_t *list_len_minus_one =
ASRUtils::EXPR(ASR::make_IntegerBinOp_t(
al, loc, list_len, ASR::binopType::Sub,
constant_one, int_type, nullptr));
ASR::expr_t *compare_cond =
ASRUtils::EXPR(ASR::make_IntegerCompare_t(
al, loc, list_iter_var, ASR::cmpopType::Lt,
list_len_minus_one, bool_type, nullptr));

Vec<ASR::expr_t *> v1, v2, v3, v4;
v1.reserve(al, 1);
v3.reserve(al, 1);
v4.reserve(al, 1);
v1.push_back(al, open_bracket);
v3.push_back(al, close_bracket);
v4.push_back(al, comma_space);

if (ASR::is_a<ASR::Character_t>(*listC->m_type)) {
v2.reserve(al, 3);
v2.push_back(al, single_quote);
v2.push_back(al, list_item);
v2.push_back(al, single_quote);
} else {
v2.reserve(al, 1);
v2.push_back(al, list_item);
}

ASR::expr_t *list_item = ASRUtils::EXPR(
ASR::make_ListItem_t(al, x.base.base.loc, x.m_values[0],
list_iter_var, listC->m_type, nullptr));
ASR::expr_t *list_len = ASRUtils::EXPR(ASR::make_ListLen_t(
al, x.base.base.loc, x.m_values[0], int_type, nullptr));
ASR::expr_t *constant_one = ASRUtils::EXPR(
ASR::make_IntegerConstant_t(al, x.base.base.loc, 1, int_type));
ASR::expr_t *list_len_minus_one =
ASRUtils::EXPR(ASR::make_IntegerBinOp_t(
al, x.base.base.loc, list_len, ASR::binopType::Sub,
constant_one, int_type, nullptr));
ASR::expr_t *compare_cond =
ASRUtils::EXPR(ASR::make_IntegerCompare_t(
al, x.base.base.loc, list_iter_var, ASR::cmpopType::Lt,
list_len_minus_one, bool_type, nullptr));

Vec<ASR::expr_t *> v1, v2, v3, v4;
v1.reserve(al, 1);
v3.reserve(al, 1);
v4.reserve(al, 1);
v1.push_back(al, open_bracket);
v3.push_back(al, close_bracket);
v4.push_back(al, comma_space);

if (ASR::is_a<ASR::Character_t>(*listC->m_type)) {
v2.reserve(al, 3);
v2.push_back(al, single_quote);
v2.push_back(al, list_item);
v2.push_back(al, single_quote);
ASR::stmt_t *print_open_bracket = ASRUtils::STMT(
ASR::make_Print_t(al, loc, nullptr, v1.p, v1.size(),
nullptr, empty_str));
ASR::stmt_t *print_comma_space = ASRUtils::STMT(
ASR::make_Print_t(al, loc, nullptr, v4.p, v4.size(),
empty_str, empty_str));
ASR::stmt_t *print_item = ASRUtils::STMT(
ASR::make_Print_t(al, loc, nullptr, v2.p, v2.size(),
empty_str, empty_str));
ASR::stmt_t *print_close_bracket = ASRUtils::STMT(
ASR::make_Print_t(al, loc, nullptr, v3.p, v3.size(),
sep_expr, nullptr));

Vec<ASR::stmt_t *> if_body;
if_body.reserve(al, 1);
if_body.push_back(al, print_comma_space);

ASR::stmt_t *if_cond = ASRUtils::STMT(
ASR::make_If_t(al, loc, compare_cond, if_body.p,
if_body.size(), nullptr, 0));

ASR::do_loop_head_t loop_head;
Vec<ASR::stmt_t *> loop_body;
{
loop_head.loc = loc;
loop_head.m_v = list_iter_var;
loop_head.m_start = ASRUtils::EXPR(ASR::make_IntegerConstant_t(
al, loc, 0, int_type));
loop_head.m_end = list_len_minus_one;
loop_head.m_increment =
ASRUtils::EXPR(ASR::make_IntegerConstant_t(
al, loc, 1, int_type));

if (!ASR::is_a<ASR::List_t>(*listC->m_type)) {
loop_body.reserve(al, 2);
loop_body.push_back(al, print_item);
} else {
v2.reserve(al, 1);
v2.push_back(al, list_item);
print_list_helper(list_item, comma_space, loc);
loop_body.from_pointer_n_copy(al, print_pass_result_tmp.p, print_pass_result_tmp.size());
print_pass_result_tmp.n = 0;
}
loop_body.push_back(al, if_cond);
}

ASR::stmt_t *print_open_bracket = ASRUtils::STMT(
ASR::make_Print_t(al, x.base.base.loc, nullptr, v1.p, v1.size(),
nullptr, empty_str));
ASR::stmt_t *print_comma_space = ASRUtils::STMT(
ASR::make_Print_t(al, x.base.base.loc, nullptr, v4.p, v4.size(),
empty_str, empty_str));
ASR::stmt_t *print_item = ASRUtils::STMT(
ASR::make_Print_t(al, x.base.base.loc, nullptr, v2.p, v2.size(),
empty_str, empty_str));
ASR::stmt_t *print_close_bracket = ASRUtils::STMT(
ASR::make_Print_t(al, x.base.base.loc, nullptr, v3.p, v3.size(),
x.m_separator, x.m_end));

Vec<ASR::stmt_t *> if_body;
if_body.reserve(al, 1);
if_body.push_back(al, print_comma_space);

ASR::stmt_t *if_cond = ASRUtils::STMT(
ASR::make_If_t(al, x.base.base.loc, compare_cond, if_body.p,
if_body.size(), nullptr, 0));

ASR::do_loop_head_t loop_head;
Vec<ASR::stmt_t *> loop_body;
{
loop_head.loc = x.base.base.loc;
loop_head.m_v = list_iter_var;
loop_head.m_start = ASRUtils::EXPR(ASR::make_IntegerConstant_t(
al, x.base.base.loc, 0, int_type));
loop_head.m_end = list_len_minus_one;
loop_head.m_increment =
ASRUtils::EXPR(ASR::make_IntegerConstant_t(
al, x.base.base.loc, 1, int_type));

if (!ASR::is_a<ASR::List_t>(*listC->m_type)) {
loop_body.reserve(al, 2);
loop_body.push_back(al, print_item);
} else {
visit_Print(*ASR::down_cast<ASR::Print_t>(print_item));
loop_body.from_pointer_n_copy(al, pass_result.p, pass_result.size());
pass_result.n = 0;
}
loop_body.push_back(al, if_cond);
}
ASR::stmt_t *loop = ASRUtils::STMT(ASR::make_DoLoop_t(
al, loc, loop_head, loop_body.p, loop_body.size()));

ASR::stmt_t *loop = ASRUtils::STMT(ASR::make_DoLoop_t(
al, x.base.base.loc, loop_head, loop_body.p, loop_body.size()));
{
print_pass_result_tmp.push_back(al, print_open_bracket);
print_pass_result_tmp.push_back(al, loop);
print_pass_result_tmp.push_back(al, print_close_bracket);
}
}

void visit_Print(const ASR::Print_t &x) {
std::vector<ASR::expr_t*> print_tmp;
for (size_t i=0; i<x.n_values; i++) {
if (ASR::is_a<ASR::List_t>(*ASRUtils::expr_type(x.m_values[i]))) {
if (!print_tmp.empty()) {
Vec<ASR::expr_t*> tmp_vec;
tmp_vec.reserve(al, print_tmp.size());
for (auto &e: print_tmp) {
tmp_vec.push_back(al, e);
}
ASR::stmt_t *print_stmt = ASRUtils::STMT(
ASR::make_Print_t(al, x.base.base.loc, nullptr, tmp_vec.p, tmp_vec.size(),
x.m_separator, nullptr));
print_tmp.clear();
pass_result.push_back(al, print_stmt);

{
pass_result.push_back(al, print_open_bracket);
pass_result.push_back(al, loop);
pass_result.push_back(al, print_close_bracket);
}
print_list_helper(x.m_values[i], x.m_separator, x.base.base.loc);
for (size_t j=0; j<print_pass_result_tmp.n; j++)
pass_result.push_back(al, print_pass_result_tmp[j]);
print_pass_result_tmp.n = 0;
} else {
print_tmp.push_back(x.m_values[i]);
}
}
if (!print_tmp.empty()) {
Vec<ASR::expr_t*> tmp_vec;
tmp_vec.reserve(al, print_tmp.size());
for (auto &e: print_tmp) {
tmp_vec.push_back(al, e);
}
ASR::stmt_t *print_stmt = ASRUtils::STMT(
ASR::make_Print_t(al, x.base.base.loc, nullptr, tmp_vec.p, tmp_vec.size(),
x.m_separator, x.m_end));
print_tmp.clear();
pass_result.push_back(al, print_stmt);
}
}
};
Expand Down

0 comments on commit f5ab20d

Please sign in to comment.