101 lines
3.3 KiB
C++
101 lines
3.3 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 <string>
|
|
#include <iostream>
|
|
#include "library/vm/vm.h"
|
|
#include "library/vm/vm_nat.h"
|
|
#include "library/vm/vm_string.h"
|
|
#include "library/vm/vm_ordering.h"
|
|
#include "library/vm/vm_list.h"
|
|
|
|
namespace lean {
|
|
struct vm_name : public vm_external {
|
|
name m_val;
|
|
vm_name(name const & v):m_val(v) {}
|
|
virtual ~vm_name() {}
|
|
virtual void dealloc() override { this->~vm_name(); get_vm_allocator().deallocate(sizeof(vm_name), this); }
|
|
virtual vm_external * ts_clone() { return new vm_name(m_val); }
|
|
virtual vm_external * clone() { return new (get_vm_allocator().allocate(sizeof(vm_name))) vm_name(m_val); }
|
|
};
|
|
|
|
bool is_name(vm_obj const & o) {
|
|
return is_external(o) && dynamic_cast<vm_name*>(to_external(o));
|
|
}
|
|
|
|
name const & to_name(vm_obj const & o) {
|
|
lean_assert(is_external(o));
|
|
lean_assert(dynamic_cast<vm_name*>(to_external(o)));
|
|
return static_cast<vm_name*>(to_external(o))->m_val;
|
|
}
|
|
|
|
vm_obj to_obj(name const & n) {
|
|
return mk_vm_external(new (get_vm_allocator().allocate(sizeof(vm_name))) vm_name(n));
|
|
}
|
|
|
|
vm_obj name_anonymous() {
|
|
return to_obj(name());
|
|
}
|
|
|
|
vm_obj name_mk_string(vm_obj const & s, vm_obj const & n) {
|
|
std::string str = to_string(s);
|
|
return to_obj(name(to_name(n), str.c_str()));
|
|
}
|
|
|
|
vm_obj name_mk_numeral(vm_obj const & num, vm_obj const & n) {
|
|
return to_obj(name(to_name(n), to_unsigned(num)));
|
|
}
|
|
|
|
unsigned name_cases_on(vm_obj const & o, buffer<vm_obj> & data) {
|
|
name const & n = to_name(o);
|
|
if (n.is_anonymous()) {
|
|
return 0;
|
|
} else if (n.is_string()) {
|
|
data.push_back(to_obj(std::string(n.get_string())));
|
|
data.push_back(to_obj(n.get_prefix()));
|
|
return 1;
|
|
} else {
|
|
data.push_back(mk_vm_nat(n.get_numeral()));
|
|
data.push_back(to_obj(n.get_prefix()));
|
|
return 2;
|
|
}
|
|
}
|
|
|
|
vm_obj name_has_decidable_eq(vm_obj const & o1, vm_obj const & o2) {
|
|
return mk_vm_bool(to_name(o1) == to_name(o2));
|
|
}
|
|
|
|
vm_obj name_cmp(vm_obj const & o1, vm_obj const & o2) {
|
|
return int_to_ordering(quick_cmp(to_name(o1), to_name(o2)));
|
|
}
|
|
|
|
vm_obj name_lex_cmp(vm_obj const & o1, vm_obj const & o2) {
|
|
return int_to_ordering(cmp(to_name(o1), to_name(o2)));
|
|
}
|
|
|
|
vm_obj name_append_after(vm_obj const & n, vm_obj const & i) {
|
|
return to_obj(to_name(n).append_after(force_to_unsigned(i, 0)));
|
|
}
|
|
|
|
vm_obj name_append(vm_obj const & n1, vm_obj const & n2) {
|
|
return to_obj(to_name(n1) + to_name(n2));
|
|
}
|
|
|
|
void initialize_vm_name() {
|
|
DECLARE_VM_BUILTIN(name({"name", "anonymous"}), name_anonymous);
|
|
DECLARE_VM_BUILTIN(name({"name", "mk_string"}), name_mk_string);
|
|
DECLARE_VM_BUILTIN(name({"name", "mk_numeral"}), name_mk_numeral);
|
|
DECLARE_VM_BUILTIN(name({"name", "has_decidable_eq"}), name_has_decidable_eq);
|
|
DECLARE_VM_BUILTIN(name({"name", "cmp"}), name_cmp);
|
|
DECLARE_VM_BUILTIN(name({"name", "lex_cmp"}), name_lex_cmp);
|
|
DECLARE_VM_BUILTIN(name({"name", "append_after"}), name_append_after);
|
|
DECLARE_VM_BUILTIN(name({"name", "append"}), name_append);
|
|
DECLARE_VM_CASES_BUILTIN(name({"name", "cases_on"}), name_cases_on);
|
|
}
|
|
|
|
void finalize_vm_name() {
|
|
}
|
|
}
|