vectorFormula.hh

Go to the documentation of this file.

 
#ifndef vectorformula_hh
#define vectorformula_hh
 
#include <vector>
#include "toolbox/sharedPointer.hh"
#include "elementFormula.hh"
 
namespace concepts {
 
 
  // ********************************************************* VectorFormula **
 
  template<typename F, int dim>
  class VectorFormula : public Formula< Point<F, dim> >
  {
  public:
    typedef Point<F, dim> value_type;
    typedef RCP<const concepts::Formula<F> > EntryFormulaRCP;
 
    VectorFormula<F,dim>() : formulas_() 
    {
      for(int i=0; i < dim; ++i)
        formulas_.push_back(EntryFormulaRCP());
    }
 
    VectorFormula<F,dim>(const std::vector< EntryFormulaRCP >& formulas) 
     : formulas_(formulas)
    { 
      conceptsAssert(formulas_.size() == dim, concepts::Assertion());
    }
 
    Point<F, dim> operator() (const Real p, const Real t = 0.0) const
    {
      Point<F, dim> ret;
      for (int i=0; i < dim; ++i) {
        ret[i] = formulas_[i] ? (*formulas_[i])(p, t) : F(0);
      }
      return ret;
    }
 
    Point<F, dim> operator() (const Real2d& p, const Real t = 0.0) const
    {
      Point<F, dim> ret;
      for (int i=0; i < dim; ++i) {
        ret[i] = formulas_[i] ? (*formulas_[i])(p, t) : F(0);
      }
      return ret;
    }
 
    Point<F, dim> operator() (const Real3d& p, const Real t = 0.0) const
    {
      Point<F, dim> ret;
      for (int i=0; i < dim; ++i) {
        ret[i] = formulas_[i] ? (*formulas_[i])(p, t) : F(0);
      }
      return ret;
    }
 
    virtual VectorFormula<F,dim>* clone() const
    {
      return new VectorFormula<F, dim>(*this);
    }
 
#if 0
    VectorFormula<F,dim>( const VectorFormula<F,dim>& other) 
      : formulas_(other.formulas_)
    { }
#endif
 
    virtual ~VectorFormula() { }
 
    const std::vector< EntryFormulaRCP >& getFormulas() 
    {
      return formulas_;
    }
 
    EntryFormulaRCP& operator[](int i)
    {
      conceptsAssert(i < dim, concepts::Assertion());
 
      return formulas_[i];
    }
 
  protected:
    std::vector< EntryFormulaRCP > formulas_;
 
    virtual std::ostream& info(std::ostream& os) const {
      os << concepts::typeOf(*this) << "([";
      for (int i=0; i < dim; ++i) {
        if (formulas_[i])
          os << *formulas_[i];
        else
          os << "0";
        if (i < dim - 1)
          os << ", ";
      }
      return os << "])";
    }
  };
 
 
} // namespace concepts
 
 
#endif // vectorformula_hh