Chris@49: // Copyright (C) 2008-2012 NICTA (www.nicta.com.au) Chris@49: // Copyright (C) 2008-2012 Conrad Sanderson Chris@49: // Copyright (C) 2012 Ryan Curtin Chris@49: // Chris@49: // This Source Code Form is subject to the terms of the Mozilla Public Chris@49: // License, v. 2.0. If a copy of the MPL was not distributed with this Chris@49: // file, You can obtain one at http://mozilla.org/MPL/2.0/. Chris@49: Chris@49: Chris@49: //! \addtogroup arma_ostream Chris@49: //! @{ Chris@49: Chris@49: Chris@49: Chris@49: inline Chris@49: arma_ostream_state::arma_ostream_state(const std::ostream& o) Chris@49: : orig_flags (o.flags()) Chris@49: , orig_precision(o.precision()) Chris@49: , orig_width (o.width()) Chris@49: , orig_fill (o.fill()) Chris@49: { Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: inline Chris@49: void Chris@49: arma_ostream_state::restore(std::ostream& o) const Chris@49: { Chris@49: o.flags (orig_flags); Chris@49: o.precision(orig_precision); Chris@49: o.width (orig_width); Chris@49: o.fill (orig_fill); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: // Chris@49: // Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: std::streamsize Chris@49: arma_ostream::modify_stream(std::ostream& o, const eT* data, const uword n_elem) Chris@49: { Chris@49: o.unsetf(ios::showbase); Chris@49: o.unsetf(ios::uppercase); Chris@49: o.unsetf(ios::showpos); Chris@49: Chris@49: o.fill(' '); Chris@49: Chris@49: std::streamsize cell_width; Chris@49: Chris@49: bool use_layout_B = false; Chris@49: bool use_layout_C = false; Chris@49: Chris@49: for(uword i=0; i= eT(+100) ) Chris@49: || Chris@49: //( (is_signed::value == true) && (val <= eT(-100)) ) || Chris@49: //( (is_non_integral::value == true) && (val > eT(0)) && (val <= eT(+1e-4)) ) || Chris@49: //( (is_non_integral::value == true) && (is_signed::value == true) && (val < eT(0)) && (val >= eT(-1e-4)) ) Chris@49: ( Chris@49: cond_rel< is_signed::value >::leq(val, eT(-100)) Chris@49: ) Chris@49: || Chris@49: ( Chris@49: cond_rel< is_non_integral::value >::gt(val, eT(0)) Chris@49: && Chris@49: cond_rel< is_non_integral::value >::leq(val, eT(+1e-4)) Chris@49: ) Chris@49: || Chris@49: ( Chris@49: cond_rel< is_non_integral::value && is_signed::value >::lt(val, eT(0)) Chris@49: && Chris@49: cond_rel< is_non_integral::value && is_signed::value >::geq(val, eT(-1e-4)) Chris@49: ) Chris@49: ) Chris@49: { Chris@49: use_layout_C = true; Chris@49: break; Chris@49: } Chris@49: Chris@49: if( Chris@49: // (val >= eT(+10)) || ( (is_signed::value == true) && (val <= eT(-10)) ) Chris@49: (val >= eT(+10)) || ( cond_rel< is_signed::value >::leq(val, eT(-10)) ) Chris@49: ) Chris@49: { Chris@49: use_layout_B = true; Chris@49: } Chris@49: } Chris@49: Chris@49: if(use_layout_C == true) Chris@49: { Chris@49: o.setf(ios::scientific); Chris@49: o.setf(ios::right); Chris@49: o.unsetf(ios::fixed); Chris@49: o.precision(4); Chris@49: cell_width = 13; Chris@49: } Chris@49: else Chris@49: if(use_layout_B == true) Chris@49: { Chris@49: o.unsetf(ios::scientific); Chris@49: o.setf(ios::right); Chris@49: o.setf(ios::fixed); Chris@49: o.precision(4); Chris@49: cell_width = 10; Chris@49: } Chris@49: else Chris@49: { Chris@49: o.unsetf(ios::scientific); Chris@49: o.setf(ios::right); Chris@49: o.setf(ios::fixed); Chris@49: o.precision(4); Chris@49: cell_width = 9; Chris@49: } Chris@49: Chris@49: return cell_width; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! "better than nothing" settings for complex numbers Chris@49: template Chris@49: inline Chris@49: std::streamsize Chris@49: arma_ostream::modify_stream(std::ostream& o, const std::complex* data, const uword n_elem) Chris@49: { Chris@49: arma_ignore(data); Chris@49: arma_ignore(n_elem); Chris@49: Chris@49: o.unsetf(ios::showbase); Chris@49: o.unsetf(ios::uppercase); Chris@49: o.fill(' '); Chris@49: Chris@49: o.setf(ios::scientific); Chris@49: o.setf(ios::showpos); Chris@49: o.setf(ios::right); Chris@49: o.unsetf(ios::fixed); Chris@49: Chris@49: std::streamsize cell_width; Chris@49: Chris@49: o.precision(3); Chris@49: cell_width = 2 + 2*(1 + 3 + o.precision() + 5) + 1; Chris@49: Chris@49: return cell_width; Chris@49: } Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: std::streamsize Chris@49: arma_ostream::modify_stream(std::ostream& o, typename SpMat::const_iterator begin, const uword n_elem, const typename arma_not_cx::result* junk) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: arma_ignore(junk); Chris@49: Chris@49: o.unsetf(ios::showbase); Chris@49: o.unsetf(ios::uppercase); Chris@49: o.unsetf(ios::showpos); Chris@49: Chris@49: o.fill(' '); Chris@49: Chris@49: std::streamsize cell_width; Chris@49: Chris@49: bool use_layout_B = false; Chris@49: bool use_layout_C = false; Chris@49: Chris@49: for(typename SpMat::const_iterator it = begin; it.pos() < n_elem; ++it) Chris@49: { Chris@49: const eT val = *it; Chris@49: Chris@49: if( Chris@49: val >= eT(+100) || Chris@49: ( (is_signed::value == true) && (val <= eT(-100)) ) || Chris@49: ( (is_non_integral::value == true) && (val > eT(0)) && (val <= eT(+1e-4)) ) || Chris@49: ( (is_non_integral::value == true) && (is_signed::value == true) && (val < eT(0)) && (val >= eT(-1e-4)) ) Chris@49: ) Chris@49: { Chris@49: use_layout_C = true; Chris@49: break; Chris@49: } Chris@49: Chris@49: if( Chris@49: (val >= eT(+10)) || ( (is_signed::value == true) && (val <= eT(-10)) ) Chris@49: ) Chris@49: { Chris@49: use_layout_B = true; Chris@49: } Chris@49: } Chris@49: Chris@49: if(use_layout_C == true) Chris@49: { Chris@49: o.setf(ios::scientific); Chris@49: o.setf(ios::right); Chris@49: o.unsetf(ios::fixed); Chris@49: o.precision(4); Chris@49: cell_width = 13; Chris@49: } Chris@49: else Chris@49: if(use_layout_B == true) Chris@49: { Chris@49: o.unsetf(ios::scientific); Chris@49: o.setf(ios::right); Chris@49: o.setf(ios::fixed); Chris@49: o.precision(4); Chris@49: cell_width = 10; Chris@49: } Chris@49: else Chris@49: { Chris@49: o.unsetf(ios::scientific); Chris@49: o.setf(ios::right); Chris@49: o.setf(ios::fixed); Chris@49: o.precision(4); Chris@49: cell_width = 9; Chris@49: } Chris@49: Chris@49: return cell_width; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! "better than nothing" settings for complex numbers Chris@49: template Chris@49: inline Chris@49: std::streamsize Chris@49: arma_ostream::modify_stream(std::ostream& o, typename SpMat::const_iterator begin, const uword n_elem, const typename arma_cx_only::result* junk) Chris@49: { Chris@49: arma_ignore(begin); Chris@49: arma_ignore(n_elem); Chris@49: arma_ignore(junk); Chris@49: Chris@49: o.unsetf(ios::showbase); Chris@49: o.unsetf(ios::uppercase); Chris@49: o.fill(' '); Chris@49: Chris@49: o.setf(ios::scientific); Chris@49: o.setf(ios::showpos); Chris@49: o.setf(ios::right); Chris@49: o.unsetf(ios::fixed); Chris@49: Chris@49: std::streamsize cell_width; Chris@49: Chris@49: o.precision(3); Chris@49: cell_width = 2 + 2*(1 + 3 + o.precision() + 5) + 1; Chris@49: Chris@49: return cell_width; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: arma_ostream::print_elem_zero(std::ostream& o, const bool modify) Chris@49: { Chris@49: if(modify == true) Chris@49: { Chris@49: const std::streamsize orig_precision = o.precision(); Chris@49: Chris@49: o.precision(0); Chris@49: Chris@49: o << eT(0); Chris@49: Chris@49: o.precision(orig_precision); Chris@49: } Chris@49: else Chris@49: { Chris@49: o << eT(0); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! Print an element to the specified stream Chris@49: template Chris@49: arma_inline Chris@49: void Chris@49: arma_ostream::print_elem(std::ostream& o, const eT& x, const bool modify) Chris@49: { Chris@49: if(x != eT(0)) Chris@49: { Chris@49: o << x; Chris@49: } Chris@49: else Chris@49: { Chris@49: arma_ostream::print_elem_zero(o, modify); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! Print a complex element to the specified stream Chris@49: template Chris@49: inline Chris@49: void Chris@49: arma_ostream::print_elem(std::ostream& o, const std::complex& x, const bool modify) Chris@49: { Chris@49: if( (x.real() != T(0)) || (x.imag() != T(0)) || (modify == false) ) Chris@49: { Chris@49: std::ostringstream ss; Chris@49: ss.flags(o.flags()); Chris@49: //ss.imbue(o.getloc()); Chris@49: ss.precision(o.precision()); Chris@49: Chris@49: ss << '(' << x.real() << ',' << x.imag() << ')'; Chris@49: o << ss.str(); Chris@49: } Chris@49: else Chris@49: { Chris@49: o << "(0,0)"; Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! Print a matrix to the specified stream Chris@49: template Chris@49: inline Chris@49: void Chris@49: arma_ostream::print(std::ostream& o, const Mat& m, const bool modify) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: const arma_ostream_state stream_state(o); Chris@49: Chris@49: const std::streamsize cell_width = modify ? arma_ostream::modify_stream(o, m.memptr(), m.n_elem) : o.width(); Chris@49: Chris@49: const uword m_n_rows = m.n_rows; Chris@49: const uword m_n_cols = m.n_cols; Chris@49: Chris@49: if(m.is_empty() == false) Chris@49: { Chris@49: if(m_n_cols > 0) Chris@49: { Chris@49: if(cell_width > 0) Chris@49: { Chris@49: for(uword row=0; row < m_n_rows; ++row) Chris@49: { Chris@49: for(uword col=0; col < m_n_cols; ++col) Chris@49: { Chris@49: // the cell width appears to be reset after each element is printed, Chris@49: // hence we need to restore it Chris@49: o.width(cell_width); Chris@49: arma_ostream::print_elem(o, m.at(row,col), modify); Chris@49: } Chris@49: Chris@49: o << '\n'; Chris@49: } Chris@49: } Chris@49: else Chris@49: { Chris@49: for(uword row=0; row < m_n_rows; ++row) Chris@49: { Chris@49: for(uword col=0; col < m_n_cols-1; ++col) Chris@49: { Chris@49: arma_ostream::print_elem(o, m.at(row,col), modify); Chris@49: o << ' '; Chris@49: } Chris@49: Chris@49: arma_ostream::print_elem(o, m.at(row, m_n_cols-1), modify); Chris@49: o << '\n'; Chris@49: } Chris@49: } Chris@49: } Chris@49: } Chris@49: else Chris@49: { Chris@49: o << "[matrix size: " << m_n_rows << 'x' << m_n_cols << "]\n"; Chris@49: } Chris@49: Chris@49: o.flush(); Chris@49: stream_state.restore(o); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! Print a cube to the specified stream Chris@49: template Chris@49: inline Chris@49: void Chris@49: arma_ostream::print(std::ostream& o, const Cube& x, const bool modify) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: const arma_ostream_state stream_state(o); Chris@49: Chris@49: const std::streamsize cell_width = modify ? arma_ostream::modify_stream(o, x.memptr(), x.n_elem) : o.width(); Chris@49: Chris@49: if(x.is_empty() == false) Chris@49: { Chris@49: for(uword slice=0; slice < x.n_slices; ++slice) Chris@49: { Chris@49: o << "[cube slice " << slice << ']' << '\n'; Chris@49: o.width(cell_width); Chris@49: arma_ostream::print(o, x.slice(slice), false); Chris@49: o << '\n'; Chris@49: } Chris@49: } Chris@49: else Chris@49: { Chris@49: o << "[cube size: " << x.n_rows << 'x' << x.n_cols << 'x' << x.n_slices << "]\n"; Chris@49: } Chris@49: Chris@49: stream_state.restore(o); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: Chris@49: //! Print a field to the specified stream Chris@49: //! Assumes type oT can be printed, i.e. oT has std::ostream& operator<< (std::ostream&, const oT&) Chris@49: template Chris@49: inline Chris@49: void Chris@49: arma_ostream::print(std::ostream& o, const field& x) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: const arma_ostream_state stream_state(o); Chris@49: Chris@49: const std::streamsize cell_width = o.width(); Chris@49: Chris@49: const uword x_n_rows = x.n_rows; Chris@49: const uword x_n_cols = x.n_cols; Chris@49: Chris@49: if(x.is_empty() == false) Chris@49: { Chris@49: for(uword col=0; col Chris@49: inline Chris@49: void Chris@49: arma_ostream::print(std::ostream& o, const subview_field& x) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: const arma_ostream_state stream_state(o); Chris@49: Chris@49: const std::streamsize cell_width = o.width(); Chris@49: Chris@49: const uword x_n_rows = x.n_rows; Chris@49: const uword x_n_cols = x.n_cols; Chris@49: Chris@49: for(uword col=0; col Chris@49: inline Chris@49: void Chris@49: arma_ostream::print_dense(std::ostream& o, const SpMat& m, const bool modify) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: const arma_ostream_state stream_state(o); Chris@49: Chris@49: const uword m_n_rows = m.n_rows; Chris@49: const uword m_n_cols = m.n_cols; Chris@49: Chris@49: if(m.n_nonzero > 0) Chris@49: { Chris@49: const std::streamsize cell_width = modify ? modify_stream(o, m.begin(), m.n_nonzero) : o.width(); Chris@49: Chris@49: typename SpMat::const_iterator begin = m.begin(); Chris@49: Chris@49: if(m_n_cols > 0) Chris@49: { Chris@49: if(cell_width > 0) Chris@49: { Chris@49: // An efficient row_iterator would make this simpler and faster Chris@49: for(uword row=0; row < m_n_rows; ++row) Chris@49: { Chris@49: for(uword col=0; col < m_n_cols; ++col) Chris@49: { Chris@49: // the cell width appears to be reset after each element is printed, Chris@49: // hence we need to restore it Chris@49: o.width(cell_width); Chris@49: eT val = eT(0); Chris@49: for(typename SpMat::const_iterator it = begin; it.pos() < m.n_nonzero; ++it) Chris@49: { Chris@49: if(it.row() == row && it.col() == col) Chris@49: { Chris@49: val = *it; Chris@49: break; Chris@49: } Chris@49: } Chris@49: arma_ostream::print_elem(o,eT(val), modify); Chris@49: } Chris@49: Chris@49: o << '\n'; Chris@49: } Chris@49: } Chris@49: else Chris@49: { Chris@49: // An efficient row_iterator would make this simpler and faster Chris@49: for(uword row=0; row < m_n_rows; ++row) Chris@49: { Chris@49: for(uword col=0; col < m_n_cols; ++col) Chris@49: { Chris@49: eT val = eT(0); Chris@49: for(typename SpMat::const_iterator it = begin; it.pos() < m.n_nonzero; ++it) Chris@49: { Chris@49: if(it.row() == row && it.col() == col) Chris@49: { Chris@49: val = *it; Chris@49: break; Chris@49: } Chris@49: } Chris@49: arma_ostream::print_elem(o,eT(val), modify); Chris@49: o << ' '; Chris@49: } Chris@49: Chris@49: o << '\n'; Chris@49: } Chris@49: } Chris@49: } Chris@49: } Chris@49: else Chris@49: { Chris@49: if(m.n_elem == 0) Chris@49: { Chris@49: o << "[matrix size: " << m_n_rows << 'x' << m_n_cols << "]\n"; Chris@49: } Chris@49: else Chris@49: { Chris@49: eT tmp[1]; Chris@49: tmp[0] = eT(0); Chris@49: Chris@49: const std::streamsize cell_width = modify ? arma_ostream::modify_stream(o, &tmp[0], 1) : o.width(); Chris@49: Chris@49: for(uword row=0; row < m_n_rows; ++row) Chris@49: { Chris@49: for(uword col=0; col < m_n_cols; ++col) Chris@49: { Chris@49: o.width(cell_width); Chris@49: Chris@49: arma_ostream::print_elem_zero(o, modify); Chris@49: Chris@49: o << ' '; Chris@49: } Chris@49: Chris@49: o << '\n'; Chris@49: } Chris@49: } Chris@49: } Chris@49: Chris@49: o.flush(); Chris@49: stream_state.restore(o); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: arma_ostream::print(std::ostream& o, const SpMat& m, const bool modify) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: const arma_ostream_state stream_state(o); Chris@49: Chris@49: o.unsetf(ios::showbase); Chris@49: o.unsetf(ios::uppercase); Chris@49: o.unsetf(ios::showpos); Chris@49: o.unsetf(ios::scientific); Chris@49: o.setf(ios::right); Chris@49: o.setf(ios::fixed); Chris@49: o.precision(2); Chris@49: Chris@49: const uword m_n_nonzero = m.n_nonzero; Chris@49: Chris@49: o << "[matrix size: " << m.n_rows << 'x' << m.n_cols << "; n_nonzero: " << m_n_nonzero Chris@49: << "; density: " << ((m.n_elem > 0) ? (double(m_n_nonzero) / double(m.n_elem) * double(100)) : double(0)) Chris@49: << "%]\n\n"; Chris@49: Chris@49: if(modify == false) { stream_state.restore(o); } Chris@49: Chris@49: if(m_n_nonzero > 0) Chris@49: { Chris@49: const std::streamsize cell_width = modify ? modify_stream(o, m.begin(), m_n_nonzero) : o.width(); Chris@49: Chris@49: typename SpMat::const_iterator begin = m.begin(); Chris@49: Chris@49: while(begin != m.end()) Chris@49: { Chris@49: const uword row = begin.row(); Chris@49: Chris@49: // TODO: change the maximum number of spaces before and after each location to be dependent on n_rows and n_cols Chris@49: Chris@49: if(row < 10) { o << " "; } Chris@49: else if(row < 100) { o << " "; } Chris@49: else if(row < 1000) { o << " "; } Chris@49: else if(row < 10000) { o << " "; } Chris@49: else if(row < 100000) { o << ' '; } Chris@49: Chris@49: const uword col = begin.col(); Chris@49: Chris@49: o << '(' << row << ", " << col << ") "; Chris@49: Chris@49: if(col < 10) { o << " "; } Chris@49: else if(col < 100) { o << " "; } Chris@49: else if(col < 1000) { o << " "; } Chris@49: else if(col < 10000) { o << " "; } Chris@49: else if(col < 100000) { o << ' '; } Chris@49: Chris@49: if(cell_width > 0) { o.width(cell_width); } Chris@49: Chris@49: arma_ostream::print_elem(o, eT(*begin), modify); Chris@49: o << '\n'; Chris@49: Chris@49: ++begin; Chris@49: } Chris@49: Chris@49: o << '\n'; Chris@49: } Chris@49: Chris@49: o.flush(); Chris@49: stream_state.restore(o); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! @}