outputMatlab.hh

Go to the documentation of this file.

 
#ifndef OutputMatlabSpace_hh
#define OutputMatlabSpace_hh
 
#include <iostream>
#include "basics/outputMatlab.hh"
 
using namespace std;
 
namespace concepts {
 
  // ********************************************************** outputMatlab **
 
  inline std::ostream& outputMatlab(std::ostream& os, const TMatrix<Real>& T)
  {
    typedef pair<int, Real> PID;
    typedef vector<PID> VPID;
    typedef map<int, VPID> M_I_VPID;
 
    M_I_VPID local_to_global;
 
    os << "[";
 
    uint cnt = 0;
    for(uint i=0; i < T.n(); ++i) {
      const TMatrix<Real>::Control* ctrl = T.control(i);
      uint size = ctrl->sz;
      int global_idx = ctrl->idx;
 
      for(uint j=0; j < size; ++j) {
        const TMatrix<Real>::Data* d = T.data(cnt);
        int local_idx = d->idx;
        Real weight = d->data;
 
        local_to_global[local_idx].push_back(PID(global_idx, weight));
 
        ++cnt;
      }
    }
 
    for(M_I_VPID::const_iterator it = local_to_global.begin();
        it != local_to_global.end(); ++it)
    {
 
 
      const VPID& targets = it->second;
      for(uint i=0; i < targets.size(); ++i)
      {
        os << it->first << " " << targets[i].first << " "
          << targets[i].second << "; ";
        if (i % 3 == 2)
          os << std::endl;        
      }
    }
 
    os << "]";
 
    return os;
  }
 
  inline std::ostream& outputMatlab(std::ostream& os, const TMatrixBase<Real>& T)
  {
    const TMatrix<Real>* Tm = dynamic_cast<const TMatrix<Real>*>(&T);
    if (Tm)
      outputMatlab(os, *Tm);
    else
      os << T;
    return os;
  }
 
  template<class F>
  inline std::ostream& outputMatlab(std::ostream& os,
                                    const ElementMatrix<F>& em)
  {
    const int N = em.n();
    const int M = em.m();
    os << "[" << std::endl;
    for (int i = 0; i < M; ++i) {
      for (int j = 0; j < N; ++j) {
        outputMatlab(os << std::setw(2), em(i,j));
        os << ((j == N-1) ? "" : " ");
      } // for j
      os << ((i == M-1) ? "" : "\n");
    } // for i
    return os << "]";
  }
 
} // namespace concepts
 
#endif // OutputMatlabSpace_hh