TinyVector.hpp

Go to the documentation of this file.

//  This file is part of ff3d - http://www.freefem.org/ff3d
//  Copyright (C) 2001, 2002, 2003 Stéphane Del Pino

//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation; either version 2, or (at your option)
//  any later version.

//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.

//  You should have received a copy of the GNU General Public License
//  along with this program; if not, write to the Free Software Foundation,
//  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  

//  $Id: TinyVector.hpp,v 1.10 2007/06/09 10:37:08 delpinux Exp $

#ifndef TINY_VECTOR_HPP
#define TINY_VECTOR_HPP

#include <cmath>
#include <typeinfo>
#include <Assert.hpp>

#include <Types.hpp>
#include <StreamCenter.hpp>

template <size_t N, typename T=real_t>
class TinyVector
{
public:
  typedef T ValueType;

  enum {
    Dimension = N               
  };

protected:
  T __x[N];

public:

  size_t size() const
  {
    return N;
  }

  inline T& operator[](const size_t& i)
  {
    ASSERT (i<N);
    return __x[i];
  }

  inline const T& operator[](const size_t& i) const
  {
    ASSERT (i<N);
    return __x[i];
  }

  inline const TinyVector<N, T>& operator = (const TinyVector<N, T>& V)
  {
    for (size_t i=0; i<N; i++)
      __x[i] = V.__x[i];
    return (*this);
  }

  inline const TinyVector<N, T>& operator = (const T& t)
  {
    for (size_t i=0; i<N; i++)
      __x[i] = t;
    return (*this);
  }

  inline TinyVector<N, T> operator*(const T& t) const
  {
    TinyVector<N, T> W;
    for (size_t i=0; i<N; i++)
      W.__x[i] = __x[i] * t;
    return W;
  }

  inline real_t operator*(const TinyVector<N,T>& V) const
  {
    ASSERT(typeid(T) == typeid(real_t));
    real_t s=0;
    for (size_t i=0; i<N; i++)
      s += __x[i] * V.__x[i];
    return s;
  }

  inline const TinyVector<N, T>& operator*=(const T& t)
  {
    for (size_t i=0; i<N; i++)
      __x[i] *= t;
    return (*this);
  }

  inline const TinyVector<N, T>& operator/=(const T& t)
  {
    const T temp = 1./t;
    this->operator*=(temp);
    return (*this);
  }

  inline TinyVector<N, T> operator-(const TinyVector<N, T>& V) const
  {
    TinyVector<N, T> v = 0;
    for (size_t i=0; i<N; i++)
      v[i] = __x[i] - V.__x[i];
    return v;
  }

  inline const TinyVector<N, T>& operator-=(const TinyVector<N, T>& V)
  {
    for (size_t i=0; i<N; i++)
      __x[i] -= V.__x[i];
    return (*this);
  }

  inline TinyVector<N, T> operator^(const TinyVector<N, T>& V) const
  {
    TinyVector<N, T> W=0;
    for (size_t i=0; i<N; i++)
      W[i] += __x[(i+1)%N] * V.__x[(i+2)%N] - __x[(i+2)%N] * V.__x[(i+1)%N];
    return W;
  }

  inline TinyVector<N, T> operator + (const TinyVector<N, T>& V) const
  {
    TinyVector<N, T> W(*this);
    for (size_t i=0; i<N; i++)
      W.__x[i] += V.__x[i];
    return W;
  }

  inline TinyVector<N, T>& operator += (const TinyVector<N, T>& V) {
    for (size_t i=0; i<N; i++)
      __x[i] += V.__x[i];
    return (*this);
  }

  inline bool operator==(const TinyVector<N, T>& V) const
  {
    for (size_t i=0; i<N; i++) {
      if (__x[i] != V.__x[i]) {
        return false;
      }
    }
    return true;
  }

  inline bool operator!=(const TinyVector<N, T>& V) const
  {
    for (size_t i=0; i<N; i++) {
      if (__x[i] != V.__x[i]) {
        return true;
      }
    }
    return false;
  }

  inline bool operator < (const TinyVector<N,T>& V) const
  {
    for (size_t i=0; i<N; ++i) {
      if(__x[i] != V.__x[i]) {
        return (__x[i] < V.__x[i]);
      }
    }
    return false;
  }

  inline friend TinyVector<N, T> sqrt(const TinyVector<N, T>& V)
  {
    using namespace std;
    TinyVector<N, T> W;
    for (size_t i=0; i<N; i++)
      W.__x[i] = sqrt(V.__x[i]);
    return W;
  }

  inline friend T Norm(const TinyVector<N, T>& V)
  {
    T t = 0;
    for (size_t i=0; i<N; i++) {
      t += V.__x[i]*V.__x[i];
    }

    using namespace std;
    return sqrt(t);
  }

  inline friend TinyVector<N, T> operator*(const T& a, const TinyVector<N, T>& V)
  {
    return (V*a);
  }

  inline friend std::ostream& operator<<(std::ostream& os,
                                         const TinyVector<N, T>& V)
  {
    ASSERT (N>0);
    os << '(';
    for (size_t i=0; i<N-1; i++)
      os << V[i] << ", ";
    os << V[N-1] << ')';
    return os;
  }

  TinyVector()
  {
    ;
  }

  TinyVector(const T& t)
  {
    for (size_t i=0; i<N; i++) {
      __x[i] = t;
    }
  }

  TinyVector(const TinyVector<N, T>& V)
  {
    for (size_t i=0; i<N; i++) 
      __x[i] = V.__x[i];
  }

  ~TinyVector()
  {
    ;
  }
};

template <class T>
class TinyVector<3, T>
{
  typedef T ValueType;

  enum {
    Dimension = 3               
  };

protected:
  T __x[3];

public:

  inline size_t size() const
  {
    return 3;
  }

  inline T& operator[](const size_t& i)
  {
    ASSERT (i<3);
    return __x[i];
  }

  inline const T& operator[](const size_t& i) const
  {
    ASSERT (i<3);
    return __x[i];
  }

  inline const TinyVector<3, T>& operator = (const TinyVector<3, T>& V)
  {
    for (size_t i=0; i<3; i++)
      __x[i] = V.__x[i];
    return (*this);
  }

  inline const TinyVector<3, T>& operator = (const T& t)
  {
    for (size_t i=0; i<3; i++)
      __x[i] = t;
    return (*this);
  }

  inline TinyVector<3, T> operator*(const T& t) const
  {
    TinyVector<3,T> W;
    for (size_t i=0; i<3; i++) {
      W.__x[i] = __x[i] * t;
    }
    return W;
  }

  inline real_t operator*(const TinyVector<3,T>& V) const
  {
    ASSERT(typeid(T) == typeid(real_t));
    real_t s=0;
    for (size_t i=0; i<3; i++)
      s += __x[i] * V.__x[i];
    return s;
  }

  inline TinyVector<3, T>& operator*=(const T& t)
  {
    for (size_t i=0; i<3; i++)
      __x[i] *= t;
    return (*this);
  }

  inline TinyVector<3, T>& operator/=(const T& t)
  {
    const T temp = 1./t;
    this->operator*=(temp);
    return (*this);
  }

  inline TinyVector<3, T> operator-(const TinyVector<3, T>& V) const
  {
    TinyVector<3, T> v = 0;
    for (size_t i=0; i<3; i++)
      v[i] = __x[i] - V.__x[i];
    return v;
  }

  inline TinyVector<3, T> operator-() const
  {
    return (TinyVector<3, T> (-__x[0],-__x[1],-__x[2]));
  }

  inline const TinyVector<3, T>& operator-=(const TinyVector<3, T>& V)
  {
    for (size_t i=0; i<3; i++)
      __x[i] -= V.__x[i];
    return (*this);
  }

  inline TinyVector<3, T> operator^(const TinyVector<3, T>& V) const
  {
    TinyVector<3, T> W(0);
    for (size_t i=0; i<3; i++)
      W[i] += __x[(i+1)%3] * V.__x[(i+2)%3] - __x[(i+2)%3] * V.__x[(i+1)%3];
    return W;
  }

  inline TinyVector<3, T> operator + (const TinyVector<3, T>& V) const
  {
    TinyVector<3, T> W(*this);
    for (size_t i=0; i<3; i++)
      W.__x[i] += V.__x[i];
    return W;
  }

  inline const TinyVector<3, T>& operator += (const TinyVector<3, T>& V)
  {
    for (size_t i=0; i<3; i++)
      __x[i] += V.__x[i];
    return (*this);
  }

  inline bool operator==(const TinyVector<3, T>& V) const
  {
    return ((__x[0]==V.__x[0])&&(__x[1]==V.__x[1])&&(__x[2]==V.__x[2]));
  }

  inline bool operator!=(const TinyVector<3, T>& V) const
  {
    return ((__x[0]!=V.__x[0])||(__x[1]!=V.__x[1])||(__x[2]!=V.__x[2]));
  }

  inline bool operator < (const TinyVector<3,T>& V) const
  {
    for (size_t i=0; i<3; ++i) {
      if(__x[i] != V.__x[i]) {
        return (__x[i] < V.__x[i]);
      }
    }
    return false;
  }

  inline friend TinyVector<3, T> sqrt(const TinyVector<3, T>& V)
  {
    using namespace std;
    TinyVector<3, T> W;
    for (size_t i=0; i<3; i++)
      W.__x[i] = sqrt(V.__x[i]);
    return W;
  }

  inline friend real_t Norm(const TinyVector<3, T>& V)
  {
    ASSERT(typeid(T) == typeid(real_t));
    real_t d = 0;
    for (size_t i=0; i<3; i++)
      d += V.__x[i]*V.__x[i];
    using namespace std;
    return sqrt(d);
  }

  inline friend TinyVector<3, T> operator*(const real_t& a,
                                           const TinyVector<3, T>& V)
  {
    return (V*a);
  }

  inline friend std::ostream& operator<<(std::ostream& os,
                                         const TinyVector<3, T>& V)
  {
    os << '(';
    for (size_t i=0; i<3-1; i++)
      os << V[i] << ", ";
    os << V[3-1] << ')';
    return os;
  }

  TinyVector (const T& x1, const T& x2, const T& x3)
  {
    __x[0] = x1;
    __x[1] = x2;
    __x[2] = x3;
  }

  TinyVector()
  {
    ;
  }

  TinyVector(const T& t)
  {
    for (size_t i=0; i<3; i++)
      __x[i] = t;
  }

  TinyVector(const TinyVector<3, T>& V)
  {
    for (size_t i=0; i<3; i++) 
      __x[i] = V.__x[i];
  }

  ~TinyVector()
  {
    ;
  }
};

template<typename T>
class TinyVector<1, T>
{
  typedef T ValueType;

  enum {
    Dimension =1                
  };
protected:
  T __x;
public:

  inline operator T&()
  {
    return __x;
  }

  inline operator const T&() const
  {
    return __x;
  }

  inline T& operator[](const size_t& i)
  {
    ASSERT(i==0);
    return __x;
  }

  inline const T& operator[](const size_t& i) const
  {
    ASSERT(i==0);
    return __x;
  }

  inline size_t size() const
  {
    return 1;
  }

  TinyVector(const T& y)
    : __x(y)
  {
    ;
  }

  TinyVector()
  {
    ;
  }


  TinyVector(const TinyVector<1, real_t>& V)
    : __x(V.__x)
  {
    ;
  }

  ~TinyVector()
  {
    ;
  }
};

#endif // TINY_VECTOR_HPP

---