Tools/Math/LA_Vector.cpp

Go to the documentation of this file.

//------------------------------------------------------------------------------
#include <iomanip>
#include <assert.h>
#include <sstream>
#include <math.h>
#include "LA_Vector.h"
//------------------------------------------------------------------------------
namespace LA
{
//------------------------------------------------------------------------------
Vector::Vector() : n(0), content(NULL)
{
}
//------------------------------------------------------------------------------
Vector::Vector(int n) : n(0), content(NULL)
{
  create(n);
}
//------------------------------------------------------------------------------
Vector::Vector(int n, const double* content) : n(0), content(NULL)
{
  create(n, content);
}
//------------------------------------------------------------------------------
Vector::Vector(const Vector& v) : n(0), content(NULL)
{
  *this = v;
}
//------------------------------------------------------------------------------
Vector::Vector(const Vector3<double>& v) : n(0), content(NULL)
{
  *this = v;
}
//------------------------------------------------------------------------------
Vector::~Vector()
{
  destroy();
}
//------------------------------------------------------------------------------
void Vector::create(int n)
{
  destroy();
  this->n = n;
  if (n != 0)
  {
    content = new double[n];
    for (int i = 0; i < n; ++i)
      content[i] = 0.0;
  }
}
//------------------------------------------------------------------------------
void Vector::create(int n, const double* v)
{
  destroy();
  this->n = n;
  if (n != 0)
  {
    content = new double[n];
    for (int i = 0; i < n; ++i)
      content[i] = v[i];
  }
}
//------------------------------------------------------------------------------
Vector& Vector::operator=(const Vector& v)
{
  create(v.n, v.content);
  return *this;
}
//------------------------------------------------------------------------------
Vector& Vector::operator=(const Vector3<double>& v)
{
  create(3);
  content[0] = v.x;
  content[1] = v.y;
  content[2] = v.z;
  return *this;
}
//------------------------------------------------------------------------------
const double& Vector::operator[](int i) const
{
  assert((n > i) && (i >= 0));
  return content[i];
}
//------------------------------------------------------------------------------
double& Vector::operator[](int i)
{
  assert((n > i) && (i >= 0));
  return content[i];
}
//------------------------------------------------------------------------------
const double& Vector::operator()(int i) const
{
  assert((i >= 1) && (i <= n));
  return content[i-1];
}
//------------------------------------------------------------------------------
double& Vector::operator()(int i)
{
  assert((i >= 1) && (i <= n));
  return content[i-1];
}
//------------------------------------------------------------------------------
int Vector::dim() const
{
  return n;
}
//------------------------------------------------------------------------------
Vector Vector::sub(int index, int length) const
{
  assert((index >= 1) && ((index + length - 1) <= n));
  return Vector(length, &content[index-1]);
}
//------------------------------------------------------------------------------
void Vector::set_sub(int index, const Vector& subv)
{
  assert((index >= 1) && ((index + subv.n - 1) <= n));
  int i, j;
  for (i = 0, j = i + index - 1; i < subv.n; ++i, ++j)
    content[j] = subv.content[i];
}
//------------------------------------------------------------------------------
Vector& Vector::operator+=(const Vector& v)
{
  assert(n == v.n);
  for (int i = 0; i < n; ++i)
    content[i] += v.content[i];
  return *this;
}
//------------------------------------------------------------------------------
Vector& Vector::operator-=(const Vector& v)
{
  assert(n == v.n);
  for (int i = 0; i < n; ++i)
    content[i] -= v.content[i];
  return *this;
}
//------------------------------------------------------------------------------
Vector& Vector::operator*=(const double& s)
{
  for (int i = 0; i < n; ++i)
    content[i] *= s;
  return *this;
}
//------------------------------------------------------------------------------
Vector& Vector::operator/=(const double& s)
{
  for (int i = 0; i < n; ++i)
    content[i] /= s;
  return *this;
}
//------------------------------------------------------------------------------
std::string Vector::asString() const
{
  std::ostringstream os;
  os << *this;
  return os.str();
}
//------------------------------------------------------------------------------
double Vector::length() const
{
  double squaresum = 0.0;
  for(int i = 0; i < n; ++i)
    squaresum += content[i]*content[i];
  return sqrt(squaresum);
}
//------------------------------------------------------------------------------
void Vector::destroy()
{
  if (content != NULL)
  {
    delete [] content;
    content = NULL;
  }
}
//------------------------------------------------------------------------------
Vector operator+(const Vector& v1, const Vector& v2)
{
  Vector v(v1);
  v += v2;
  return v;
}
//------------------------------------------------------------------------------
Vector operator-(const Vector& v1, const Vector& v2)
{
  Vector v(v1);
  v -= v2;
  return v;
}
//------------------------------------------------------------------------------
Vector operator*(const double& s, const Vector& v)
{
  Vector res(v);
  res *= s;
  return res;
}
//------------------------------------------------------------------------------
Vector operator*(const Vector& v, const double& s)
{
  Vector res(v);
  res *= s;
  return res;
}
//------------------------------------------------------------------------------
Vector operator/(const Vector& v, const double& s)
{
  Vector res(v);
  res /= s;
  return res;
}
//------------------------------------------------------------------------------
double operator*(const Vector& v1, const Vector& v2)
{
  assert(v1.n == v2.n);
  double s = 0.0;
  for (int i = 0; i < v1.n; ++i)
    s += v1.content[i]*v2.content[i];
  return s;
}
//------------------------------------------------------------------------------
std::ostream& operator<<(std::ostream& out, const Vector& v)
{
  for (int i = 0; i < v.dim(); ++i)
  {
    out << "| "
        << std::setw(14) << std::setfill(' ') << v[i]
        << " |" << std::endl;
  }
  return out;
}
//------------------------------------------------------------------------------
Out& operator<<(Out& stream, const Vector& v)
{
  stream << v.n;
  for (int i = 0; i < v.n; ++i)
    stream << v.content[i];

  return stream;
}
//------------------------------------------------------------------------------
In& operator>>(In& stream, Vector& v)
{
  int n;
  stream >> n;
  v.create(n);
  for (int i = 0; i < n; ++i)
    stream >> v.content[i];

  return stream;
}
//------------------------------------------------------------------------------
}
//------------------------------------------------------------------------------

---