vtkVector.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Visualization Toolkit
  Module:    vtkVector.h

  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/

#ifndef vtkVector_h
#define vtkVector_h

#include "vtkTuple.h"
#include "vtkObject.h" // for legacy macros

#include <cmath>   // For math functions

template<typename T, int Size>
class vtkVector : public vtkTuple<T, Size>
{
public:
  vtkVector()
  {
  }

  explicit vtkVector(const T& scalar) : vtkTuple<T, Size>(scalar)
  {
  }

  explicit vtkVector(const T* init) : vtkTuple<T, Size>(init)
  {
  }


  T SquaredNorm() const
  {
    T result = 0;
    for (int i = 0; i < Size; ++i)
    {
      result += this->Data[i] * this->Data[i];
    }
    return result;
  }

  double Norm() const
  {
    return sqrt(static_cast<double>(this->SquaredNorm()));
  }


  double Normalize()
  {
    const double norm(this->Norm());
    const double inv(1.0 / norm);
    for (int i = 0; i < Size; ++i)
    {
      this->Data[i] = static_cast<T>(this->Data[i] * inv);
    }
    return norm;
  }


  vtkVector<T, Size> Normalized() const
  {
    vtkVector<T, Size> temp(*this);
    temp.Normalize();
    return temp;
  }


  T Dot(const vtkVector<T, Size>& other) const
  {
    T result(0);
    for (int i = 0; i < Size; ++i)
    {
      result += this->Data[i] * other[i];
    }
    return result;
  }


  template<typename TR>
  vtkVector<TR, Size> Cast() const
  {
    vtkVector<TR, Size> result;
    for (int i = 0; i < Size; ++i)
    {
      result[i] = static_cast<TR>(this->Data[i]);
    }
    return result;
  }
};

// .NAME vtkVector2 - templated base type for storage of 2D vectors.
//
template<typename T>
class vtkVector2 : public vtkVector<T, 2>
{
public:
  vtkVector2()
  {
  }

  explicit vtkVector2(const T& scalar) : vtkVector<T, 2>(scalar)
  {
  }

  explicit vtkVector2(const T* init) : vtkVector<T, 2>(init)
  {
  }

  vtkVector2(const T& x, const T& y)
  {
    this->Data[0] = x;
    this->Data[1] = y;
  }


  void Set(const T& x, const T& y)
  {
    this->Data[0] = x;
    this->Data[1] = y;
  }

  void SetX(const T& x) { this->Data[0] = x; }

  const T& GetX() const { return this->Data[0]; }

  void SetY(const T& y) { this->Data[1] = y; }

  const T& GetY() const { return this->Data[1]; }


  bool operator<(const vtkVector2<T> &v) const
  {
    return (this->Data[0] < v.Data[0]) || (this->Data[0] == v.Data[0] && this->Data[1] < v.Data[1]);
  }
};

// .NAME vtkVector3 - templated base type for storage of 3D vectors.
//
template<typename T>
class vtkVector3 : public vtkVector<T, 3>
{
public:
  vtkVector3()
  {
  }

  explicit vtkVector3(const T& scalar) : vtkVector<T, 3>(scalar)
  {
  }

  explicit vtkVector3(const T* init) : vtkVector<T, 3>(init)
  {
  }

  vtkVector3(const T& x, const T& y, const T& z)
  {
    this->Data[0] = x;
    this->Data[1] = y;
    this->Data[2] = z;
  }


  void Set(const T& x, const T& y, const T& z)
  {
    this->Data[0] = x;
    this->Data[1] = y;
    this->Data[2] = z;
  }

  void SetX(const T& x) { this->Data[0] = x; }

  const T& GetX() const { return this->Data[0]; }

  void SetY(const T& y) { this->Data[1] = y; }

  const T& GetY() const { return this->Data[1]; }

  void SetZ(const T& z) { this->Data[2] = z; }

  const T& GetZ() const { return this->Data[2]; }


  vtkVector3<T> Cross(const vtkVector3<T>& other) const
  {
    vtkVector3<T> res;
    res[0] = this->Data[1] * other.Data[2] - this->Data[2] * other.Data[1];
    res[1] = this->Data[2] * other.Data[0] - this->Data[0] * other.Data[2];
    res[2] = this->Data[0] * other.Data[1] - this->Data[1] * other.Data[0];
    return res;
  }


  bool operator<(const vtkVector3<T> &v) const
  {
    return (this->Data[0] < v.Data[0]) || (this->Data[0] == v.Data[0] && this->Data[1] < v.Data[1]) ||
      (this->Data[0] == v.Data[0] && this->Data[1] == v.Data[1] && this->Data[2] < v.Data[2]);
  }
};

#define vtkVectorNormalized(vectorType, type, size) \
vectorType Normalized() const \
{ \
  return vectorType(vtkVector<type, size>::Normalized().GetData()); \
} \

#define vtkVectorDerivedMacro(vectorType, type, size) \
vtkVectorNormalized(vectorType, type, size) \
explicit vectorType(type s) : Superclass(s) {} \
explicit vectorType(const type *i) : Superclass(i) {} \
explicit vectorType(const vtkTuple<type, size> &o) : Superclass(o.GetData()) {} \
vectorType(const vtkVector<type, size> &o) : Superclass(o.GetData()) {} \



class vtkVector2i : public vtkVector2<int>
{
public:
  typedef vtkVector2<int> Superclass;
  vtkVector2i() {}
  vtkVector2i(int x, int y) : vtkVector2<int>(x, y) {}
  vtkVectorDerivedMacro(vtkVector2i, int, 2)
};

class vtkVector2f : public vtkVector2<float>
{
public:
  typedef vtkVector2<float> Superclass;
  vtkVector2f() {}
  vtkVector2f(float x, float y) : vtkVector2<float>(x, y) {}
  vtkVectorDerivedMacro(vtkVector2f, float, 2)
};

class vtkVector2d : public vtkVector2<double>
{
public:
  typedef vtkVector2<double> Superclass;
  vtkVector2d() {}
  vtkVector2d(double x, double y) : vtkVector2<double>(x, y) {}
  vtkVectorDerivedMacro(vtkVector2d, double, 2)
};

#define vtkVector3Cross(vectorType, type) \
vectorType Cross(const vectorType& other) const \
{ \
  return vectorType(vtkVector3<type>::Cross(other).GetData()); \
} \

class vtkVector3i : public vtkVector3<int>
{
public:
  typedef vtkVector3<int> Superclass;
  vtkVector3i() {}
  vtkVector3i(int x, int y, int z) : vtkVector3<int>(x, y, z) {}
  vtkVectorDerivedMacro(vtkVector3i, int, 3)
  vtkVector3Cross(vtkVector3i, int)
};

class vtkVector3f : public vtkVector3<float>
{
public:
  typedef vtkVector3<float> Superclass;
  vtkVector3f() {}
  vtkVector3f(float x, float y, float z) : vtkVector3<float>(x, y, z) {}
  vtkVectorDerivedMacro(vtkVector3f, float, 3)
  vtkVector3Cross(vtkVector3f, float)
};

class vtkVector3d : public vtkVector3<double>
{
public:
  typedef vtkVector3<double> Superclass;
  vtkVector3d() {}
  vtkVector3d(double x, double y, double z) : vtkVector3<double>(x, y, z) {}
  vtkVectorDerivedMacro(vtkVector3d, double, 3)
  vtkVector3Cross(vtkVector3d, double)
};

#endif // vtkVector_h
// VTK-HeaderTest-Exclude: vtkVector.h