printer.hpp

/*
isl-cpp: C++ bindings to the ISL (Integer Set Library)

Copyright (C) 2014  Jakob Leben <jakob.leben@gmail.com>

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

#ifndef ISL_CPP_PRINTER_INCLUDED
#define ISL_CPP_PRINTER_INCLUDED

#include "context.hpp"

#include <isl/printer.h>

#include <memory>
#include <iostream>

namespace isl {

class printer
{
public:
    enum format
    {
        isl_format = ISL_FORMAT_ISL,
        c_format = ISL_FORMAT_C,
        omega_format = ISL_FORMAT_OMEGA,
        polylib_format = ISL_FORMAT_POLYLIB,
        ext_polylib_format = ISL_FORMAT_EXT_POLYLIB,
        latex_format = ISL_FORMAT_LATEX
    };

    enum yaml_style
    {
      yaml_block_style = ISL_YAML_STYLE_BLOCK,
      yaml_flow_style = ISL_YAML_STYLE_FLOW
    };

    printer(const context & ctx):
        m_ctx(ctx)
    {
        m_printer = isl_printer_to_file(ctx.get(), stdout);
    }

    printer(const printer & other) = delete;

    ~printer()
    {
        isl_printer_free(m_printer);
    }

    isl_printer *get() const { return m_printer; }

    format current_format() const
    {
        return (format) isl_printer_get_output_format(m_printer);
    }

    void set_format(format f)
    {
        m_printer = isl_printer_set_output_format(m_printer, f);
    }

    void set_yaml_style(yaml_style s)
    {
      m_printer = isl_printer_set_yaml_style(m_printer, s);
    }

    template <typename T>
    void print( const T & );

    template <typename T>
    void print_each_in(const T &);

private:
    isl_printer *m_printer;
    context m_ctx;
};

}

#endif // ISL_CPP_PRINTER_INCLUDED