formatting.h

Go to the documentation of this file.

// -*- c++ -*-
/* 
 * LSST Data Management System
 * Copyright 2008, 2009, 2010, 2011 LSST Corporation.
 * 
 * This product includes software developed by the
 * LSST Project (http://www.lsst.org/).
 *
 * 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 3 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 LSST License Statement and 
 * the GNU General Public License along with this program.  If not, 
 * see <http://www.lsstcorp.org/LegalNotices/>.
 */
#ifndef NDARRAY_formatting_h_INCLUDED
#define NDARRAY_formatting_h_INCLUDED

#include "ndarray/ExpressionBase.h"

#include <iostream>

namespace ndarray {
namespace detail {
template <typename Derived, int N = Derived::ND::value> class Formatter;
} // namespace detail

class FormatOptions {
    int _width;
    int _precision;
    std::ios_base::fmtflags _flags;
    std::string _delimiter;
    std::string _open;
    std::string _close;
public:
    
    explicit FormatOptions(
        int width = 8,
        int precision = 6,
        std::ios_base::fmtflags flags = std::ios_base::fmtflags(0),
        std::string const & delimiter = ", ",
        std::string const & open = "[",
        std::string const & close = "]"
    ) :
        _width(width),
        _precision(precision),
        _flags(flags),
        _delimiter(delimiter),
        _open(open),
        _close(close)
    {}

    template <typename Derived>
    void apply(std::ostream & os, ExpressionBase<Derived> const & expr) {
        detail::Formatter<Derived>::apply(*this,os,expr,0);
    }

    template <typename Derived, int N> friend class detail::Formatter;
};

template <typename Derived>
std::ostream & operator<<(std::ostream & os, ExpressionBase<Derived> const & expr) {
    FormatOptions options;
    options.apply(os,expr);
    return os;
}

namespace detail {

template <typename Derived, int N>
class Formatter {
public:
    static void apply(
        FormatOptions const & options,
        std::ostream & os, 
        ExpressionBase<Derived> const & expr,
        int level
    ) {
        os << options._open;
        if (!expr.empty()) {
            typename ExpressionBase<Derived>::Iterator const end = expr.end();
            typename ExpressionBase<Derived>::Iterator iter = expr.begin();
            Formatter<typename ExpressionBase<Derived>::Reference>::apply(options,os,*iter,level+1);
            for (++iter; iter != end; ++iter) {
                os << options._delimiter;
                os << std::endl << std::string(level,' ');
                Formatter<typename ExpressionBase<Derived>::Reference>::apply(options,os,*iter,level+1);
            }
        }
        os << options._close;
    }
};

template <typename Derived>
class Formatter<Derived,1> {
public:
    static void apply(
        FormatOptions const & options,
        std::ostream & os, 
        ExpressionBase<Derived> const & expr,
        int level
    ) {
        os << options._open;
        if (!expr.empty()) {
            typename ExpressionBase<Derived>::Iterator const end = expr.end();
            typename ExpressionBase<Derived>::Iterator iter = expr.begin();
            int precision = os.precision(options._precision);
            int width = os.width(options._width);
            std::ios_base::fmtflags flags = os.setf(options._flags,std::ios_base::floatfield);
            os << (*iter);
            for (++iter; iter != end; ++iter) {
                os << options._delimiter << (*iter);
            }
            os.precision(precision);
            os.width(width);
            os.setf(flags);
        }
        os << options._close;
    }
};

} // namespace detail
} // namespace ndarray

#endif // !NDARRAY_formatting_h_INCLUDED