apfDynamicVector.h

Go to the documentation of this file.

/*
 * Copyright 2011 Scientific Computation Research Center
 *
 * This work is open source software, licensed under the terms of the
 * BSD license as described in the LICENSE file in the top-level directory.
 */

#ifndef APFDYNAMICVECTOR_H
#define APFDYNAMICVECTOR_H

#include "apfDynamicArray.h"
#include "apfVector.h"
#include <math.h>
#include <iostream>

namespace apf {

class DynamicVector;

class DynamicVector : public DynamicArray<double>
{
  public:
    DynamicVector() {}
    DynamicVector(std::size_t n):DynamicArray<double>(n) {}
    double operator()(std::size_t i) const
    {
      return (*this)[i];
    }
    double& operator()(std::size_t i)
    {
      return (*this)[i];
    } 
    DynamicVector& operator+=(DynamicVector const& b)
    {
      for (std::size_t i=0; i < size(); ++i)
        (*this)[i] += b[i];
      return *this;
    }
    DynamicVector& operator-=(DynamicVector const& b)
    {
      for (std::size_t i=0; i < size(); ++i)
        (*this)[i] -= b[i];
      return *this;
    }
    DynamicVector& operator*=(double s)
    {
      for (std::size_t i=0; i < size(); ++i)
        (*this)[i] *= s;
      return *this;
    }
    DynamicVector& operator/=(double s)
    {
      for (std::size_t i=0; i < size(); ++i)
        (*this)[i] /= s;
      return *this;
    }
    double operator*(DynamicVector const& b) const
    {
      double r=0;
      for (std::size_t i=0; i < size(); ++i)
        r += (*this)[i] * b[i];
      return r;
    }
    double getLength() {return sqrt((*this)*(*this));}
    void zero()
    {
      for (std::size_t i=0; i < size(); ++i)
        (*this)[i] = 0;
    }
};

template <std::size_t N>
inline DynamicVector fromVector(Vector<N> other)
{
  DynamicVector result(N);
  for(std::size_t ii = 0; ii < N; ii++)
    result[ii] = other[ii];
  return result;
}

} //namespace apf

std::ostream& operator<<(std::ostream& s, apf::DynamicVector const& x);

#endif