lean2/src/kernel/formatter.cpp

/*
Copyright (c) 2014 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.

Author: Leonardo de Moura
*/
#include <utility>
#include "kernel/formatter.h"

namespace lean {
static std::function<void(std::ostream &, expr const & e)> * g_print = nullptr;

void set_print_fn(std::function<void(std::ostream &, expr const &)> const & fn) {
    if (g_print)
        delete g_print;
    g_print = new std::function<void(std::ostream &, expr const &)>(fn);
}

std::ostream & operator<<(std::ostream & out, expr const & e) {
    if (g_print) {
        (*g_print)(out, e);
    } else {
        throw exception("print function is not available, Lean was not initialized correctly");
    }
    return out;
}

void print(lean::expr const & a) { std::cout << a << std::endl; }

void initialize_formatter() {}

void finalize_formatter() {
    if (g_print)
        delete g_print;
}
}