62 lines
1.4 KiB
C++
62 lines
1.4 KiB
C++
/*
|
|
Copyright (c) 2016 Microsoft Corporation. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
|
|
Author: Leonardo de Moura
|
|
*/
|
|
#include "library/constants.h"
|
|
#include "library/vm/vm.h"
|
|
|
|
namespace lean {
|
|
static void to_string(vm_obj const & o, std::string & s) {
|
|
if (!is_simple(o)) {
|
|
to_string(cfield(o, 1), s);
|
|
s += static_cast<char>(cidx(cfield(o, 0)));
|
|
}
|
|
}
|
|
|
|
static std::string to_string(vm_obj const & o) {
|
|
std::string r;
|
|
to_string(o, r);
|
|
return r;
|
|
}
|
|
|
|
static vm_obj to_obj(std::string const & str) {
|
|
vm_obj r = mk_vm_simple(0);
|
|
for (unsigned i = 0; i < str.size(); i++) {
|
|
vm_obj args[2] = { mk_vm_simple(str[i]), r };
|
|
r = mk_vm_constructor(1, 2, args);
|
|
}
|
|
return r;
|
|
}
|
|
|
|
static void put_str(vm_state & s) {
|
|
vm_obj const & str = s.get(-1);
|
|
std::cout << to_string(str);
|
|
return s.push(mk_vm_unit());
|
|
}
|
|
|
|
static void put_nat(vm_state & s) {
|
|
vm_obj const & n = s.get(-1);
|
|
if (is_simple(n))
|
|
std::cout << cidx(n);
|
|
else
|
|
std::cout << to_mpz(n);
|
|
return s.push(mk_vm_unit());
|
|
}
|
|
|
|
static void get_line(vm_state & s) {
|
|
std::string str;
|
|
std::getline(std::cin, str);
|
|
return s.push(to_obj(str));
|
|
}
|
|
|
|
void initialize_vm_io() {
|
|
declare_vm_builtin(get_put_str_name(), 1, put_str);
|
|
declare_vm_builtin(get_put_nat_name(), 1, put_nat);
|
|
declare_vm_builtin(get_get_line_name(), 0, get_line);
|
|
}
|
|
|
|
void finalize_vm_io() {
|
|
}
|
|
}
|