graphics/dataGnuplot.hh

Go to the documentation of this file.

/* graphics of the data in Gnuplot
 */

#ifndef graphDataGnuplot_hh
#define graphDataGnuplot_hh

#include <string>
#include <fstream>
#include <memory>
#include "basics/typedefs.hh"
#include "basics/vectorsMatricesForward.hh"
#include "basics/exceptions.hh"
#include "space/postProcess.hh"
#include "space/formula.hh"
#include "space/function.hh"
#include "function/vector.hh"


namespace concepts {
  // forward declaration
  template<class F>
  class Formula;

  template<class F, class G>
  class ElementFormula;
}

namespace graphics {
  using concepts::Real;

  // ******************************************************* DataGnuplotCell **

  template<typename F>
  class DataGnuplotCell : public concepts::CellPostprocess<F> {
  public:
    DataGnuplotCell(std::ofstream* ofs,
        const concepts::Vector<Real>& solution,
        const concepts::ElementFunction<Real>* fun = 0);
    DataGnuplotCell(std::ofstream* ofs, const concepts::Formula<Real>& frm,
        const concepts::ElementFunction<Real>* fun = 0);
    DataGnuplotCell(std::ofstream* ofs,
        const concepts::ElementFormula<F>& frm,
        const concepts::ElementFunction<Real>* fun = 0);
    virtual ~DataGnuplotCell();
    virtual void operator() (const concepts::Element<F>& elm)
      throw(concepts::MissingFeature);
    virtual void operator() (const concepts::Cell& cell)
      throw(concepts::MissingFeature);
    std::ofstream* write() { return ofs_; }
    const concepts::Array<Real>& coeff() const { return coeff_; }
    const concepts::Formula<Real>* formula() const { return frm_.get(); }
    const concepts::ElementFormula<F>* elementFormula() const {
      return frmElm_;
    }
    const concepts::ElementFunction<F>* elementFunction() const {
      return fun_;
    }
  private:
    std::ofstream* ofs_;
    concepts::Array<Real> coeff_;
    const concepts::Vector<Real>* solution_;
    std::auto_ptr<const concepts::Formula<Real> > frm_;
    const concepts::ElementFormula<F>* frmElm_;
    const concepts::ElementFunction<F>* fun_;
  };

  // *********************************************************** DataGnuplot **

  template<typename T, typename S>
  void drawDataGnuplot(T& msh_spc, std::string filename, const S& sol,
           const concepts::ElementFunction<Real>* fun = 0) {
    std::ofstream ofs(filename.c_str());
    concepts::GlobalPostprocess<Real> postProcess(msh_spc);
    DataGnuplotCell<Real> gnu(&ofs, sol, fun);
    postProcess(gnu);
  }

} // namespace graphics

#endif // graphDataGnuplot_hh