max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

feat(library/compiler/emit_cpp): emit_cnstr

Browse source
This commit is contained in:
Leonardo de Moura 2019-01-30 15:47:38 -08:00
parent 9bd4a7e9b4
commit d63bbb3728
1 changed files with 23 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
src/library/compiler/emit_cpp.cpp
View file

@ -296,11 +296,7 @@ struct emit_fn_fn {
m_out << "0;\n";
}
void emit_arg(expr const & arg, bool & first) {
if (first)
first = false;
else
m_out << ", ";
void emit_arg(expr const & arg) {
if (is_fvar(arg))
emit_fvar(arg);
else
@ -308,9 +304,10 @@ struct emit_fn_fn {
}
void emit_args(unsigned sz, expr const * args) {
bool first = true;
for (unsigned i = 0; i < sz; i++)
emit_arg(args[i], first);
for (unsigned i = 0; i < sz; i++) {
if (i > 0) m_out << ", ";
emit_arg(args[i]);
}
}
void emit_apply(expr const & x, buffer<expr> const & args) {
@ -338,9 +335,25 @@ struct emit_fn_fn {
m_out << "0;\n"; // TODO(Leo)
}
void emit_cnstr(expr const & x, expr const & /* fn */, buffer<expr> const & /* args */) {
void emit_cnstr_scalar_size(unsigned num_usizes, unsigned num_bytes) {
if (num_usizes == 0)
m_out << num_bytes;
else if (num_bytes == 0)
m_out << "sizeof(size_t)*" << num_usizes;
else
m_out << "sizeof(size_t)*" << num_usizes << " + " << num_bytes;
}
void emit_cnstr(expr const & x, expr const & fn, buffer<expr> const & args) {
lean_assert(!args.empty());
unsigned cidx, num_usizes, num_bytes;
lean_verify(is_llnf_cnstr(fn, cidx, num_usizes, num_bytes));
emit_rhs(x);
m_out << "0;\n"; // TODO(Leo)
m_out << "lean::alloc_cnstr(" << cidx << ", " << args.size() << ", ";
emit_cnstr_scalar_size(num_usizes, num_bytes); m_out << ");\n";
for (unsigned i = 0; i < args.size(); i++) {
m_out << "lean::cnstr_set("; emit_fvar(x); m_out << ", " << i << ", "; emit_arg(args[i]); m_out << ");\n";
}
}
void emit_reset(expr const & x, expr const & /* fn */, buffer<expr> const & /* args */) {