space/outputMatlab.hh

Go to the documentation of this file.

/* matlab output functions for object in directory space
 */

#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;
  }


} // namespace concepts

#endif // OutputMatlabSpace_hh