graphics/dataMatlab.hh

Go to the documentation of this file.

/* graphics in Matlab
 */

#ifndef graphDataMatlab_hh
#define graphDataMatlab_hh

#include <string>
#include <fstream>
#include <sstream>
#include <memory>
#include <vector>
#include "basics/typedefs.hh"
#include "basics/vectorsMatricesForward.hh"
#include "basics/exceptions.hh"
#include "toolbox/elementGraphics.hh"
#include "toolbox/dynArray.hh"
#include "function/vector.hh"
#include "geometry/formula.hh"
#include "space/postProcess.hh"
#include "space/function.hh"
#include "basis.hh"

// debugging
#include "basics/debug.hh"

#define Matlab2Constr_D 0

namespace concepts {
  template<class F>
  class Vector;
}

namespace graphics {

  using concepts::Real;
  using concepts::Real2d;

  // ******************************************************** DataMatlabCell **

  class DataMatlabCell : public concepts::CellPostprocess<Real> {
  public:
    DataMatlabCell(std::ofstream* ofsmsh, std::ofstream* ofs1,
       std::ofstream* ofs2, std::ofstream* ofs3,
       uint dim, Real scl);

    DataMatlabCell(std::ofstream* ofs, uint dim, Real scl);

    virtual void operator()(const concepts::Element<Real>& elm)
      throw(concepts::MissingFeature);
    virtual void operator()(const concepts::Cell& cell)
      throw(concepts::MissingFeature);

    void write(std::ofstream** ofsmsh, std::ofstream** ofs1,
         std::ofstream** ofs2, std::ofstream** ofs3);
    void setVec(const concepts::Vector<Real>* sol);
    const concepts::Array<Real>& coeff() const {return coeff_;}
    const concepts::Vector<Real>* solution() const {return sol_;}
    Real offset() const {return offset_;}
    Real range() const {return range_;}
    uint& count() {return cnt_;}
    uint dim() const {return dim_;}
    Real scale() const {return scl_;}
    concepts::DynArray<uint>& map() {return *map_;}
  private:
    uint dim_;
    Real offset_, range_;
    std::ofstream *ofsmsh_, *ofs1_, *ofs2_, *ofs3_;
    concepts::Array<Real> coeff_;
    const concepts::Vector<Real>* sol_;
    uint cnt_;
    Real scl_;

    std::auto_ptr<concepts::DynArray<uint> > map_;
  };

  // ************************************************************ DataMatlab **

  class DataMatlab {
  public:
    DataMatlab(concepts::Space<Real>& spc, const std::string filename,
         uint dim = 2, Real scl = 1.0);
    ~DataMatlab() {closeStreams_();}

    void operator()(const concepts::Vector<Real>& sol);

  private:
    void createStreams_(std::string filename);
    void closeStreams_();

    concepts::Space<Real>* spc_;
    uint dim_;
    uint cnt_;
    std::auto_ptr<std::ofstream> ofsmsh_;
    std::auto_ptr<std::ofstream> ofs1_, ofs2_, ofs3_;

    concepts::GlobalPostprocess<Real> postProcess_;
    std::auto_ptr<DataMatlabCell> mtlb_;
  };

} // namespace graphics

#endif // graphDataMatlab_hh