subVector.hh

Go to the documentation of this file.

 
#ifndef subvector_hh
#define subvector_hh
 
#include "vector.hh"
#include "space/spaceSet.hh"
#include "toolbox/set.hh"
 
namespace concepts {
 
  // ************************************************************* SubVector **
 
  template<class F>
  class SubVector : public Function<F> {
  public:
    SubVector(Vector<F>& f, const Set<IndexRange>& indices);
    SubVector(const Vector<F>& f,  const Set<IndexRange>& indices);
 
    virtual ~SubVector() {}
 
    virtual Function<F>& operator=(const Function<F>& fnc);
 
    /* Assignement operator from different number type
 
       This is extra, because virtual member templates don't exist.
 
       The compiler checks, if the assigment is possible, e.g.
         Cmplx = Real, possible
   Real = Cmplx, not possible
     */
    template<class H>
    Function<F>& operator=(const Function<H>& fnc);
 
    SubVector<F>& operator=(F c);
 
    virtual F& operator()(uint i) {
      conceptsAssert3(i < this->dim_, Assertion(), "i = " << i << ", n = " << this->dim_);
      conceptsAssert(v_, Assertion());
      return (*v_)(indices_[i]); }
    virtual F operator()(uint i) const {
      conceptsAssert3(i < this->dim_, Assertion(), "i = " << i << ", n = " << this->dim_);
      if (v_)
  return (*v_)(indices_[i]);
      return (*vc_)(indices_[i]);
    }
 
    virtual Function<F>& operator+=(const Function<F>& fnc);
    virtual Function<F>& operator+=(F c);
 
    virtual Function<F>& operator-=(const Function<F>& fnc);
    virtual Function<F>& operator-=(F c);
 
    virtual Function<F>& operator*=(F sc);
 
    virtual Function<F>& add(const Function<F>& fnc, F sc);
 
    uint n() const { return this->dim_; }
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  private:
    Vector<F>* v_;
    const Vector<F>* vc_;
    const Set<IndexRange> indices_;
  };
 
  template<class F>
  template<class H>
  Function<F>& SubVector<F>::operator=(const Function<H>& fnc) {
    conceptsAssert(v_, Assertion());
    uint j = 0;
    for (Set<IndexRange>::index_iterator i = indices_.index_begin();
         i != indices_.index_end(); )
      (*v_)(i++) = fnc(j++);
    return *this;
  }
 
} // namespace concepts
 
#endif // subvector_hh