formula/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 << "VectorFormula([";
      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