io.h

// -*- mode: c++; fill-column: 80; c-basic-offset: 2; indent-tabs-mode: nil -*-
/*
    Copyright (c) 2010 Juan Jose Garcia Ripoll

    Tensor is free software; you can redistribute it and/or modify it
    under the terms of the GNU Library 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 Library 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 TENSOR_IO_H
#define TENSOR_IO_H

#include <iostream>
#include <tensor/tensor.h>
#include <tensor/sparse.h>

namespace tensor {

template<typename elt_t>
std::ostream &operator<<(std::ostream &s, const Vector<elt_t> &t);

template<typename elt_t>
std::ostream &operator<<(std::ostream &s, const Tensor<elt_t> &t);

template<typename elt_t>
std::ostream &operator<<(std::ostream &s, const Tensor<elt_t> &t);

template<typename t, size_t n>
inline std::ostream &operator<<(std::ostream &s, const StaticVector<t,n> &v)
{
  return s << Vector<t>(v);
}

template<typename elt_t>
inline std::ostream &operator<<(std::ostream &s, const Sparse<elt_t> &t)
{
  return s << full(t);
}

/* The following is a template used for creating an object that displays
 * a tensor with a slightly more attractive representation. We need to
 * create a template and then specializations because templates do not do
 * implicit type coercion. */
template<typename elt_t>
class MatrixForm {
  const Tensor<elt_t> data;
  MatrixForm();
public:
  MatrixForm(const Tensor<elt_t> &t) : data(t) {}
  std::ostream &display(std::ostream &s) const;
};

template<typename elt_t>
inline std::ostream &operator<<(std::ostream &s, const MatrixForm<elt_t> &m) {
  return m.display(s);
}

const MatrixForm<double> matrix_form(const Tensor<double> &t);

const MatrixForm<cdouble> matrix_form(const Tensor<cdouble> &t);

const MatrixForm<double> matrix_form(const Sparse<double> &t);

const MatrixForm<cdouble> matrix_form(const Sparse<cdouble> &t);

} // namespace tensor

#include <tensor/detail/io.hpp>

#endif // !TENSOR_IO_H