source: src/base/index.hpp@ 2fad0e0

/**
 * @file   index.hpp
 * @author Julian Iseringhausen <isering@ins.uni-bonn.de>
 * @date   Mon Apr 18 12:19:49 2011
 *
 * @brief  Header file for the class VMG::Index.
 *
 */


#ifndef INDEX_HPP_
#define INDEX_HPP_

#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <iostream>

#include "base/stencil.hpp"

namespace VMG
{

class Vector;

class Index
{
public:
  Index(const Vector& vec);
  Index(const Stencil::iterator& iter);
  Index(int x, int y, int z) {i[0]=x;i[1]=y;i[2]=z;}
  Index(int val) {i[0]=val;i[1]=val;i[2]=val;}
  Index(int* arr) {i[0]=arr[0];i[1]=arr[1];i[2]=arr[2];}
  Index() {i[0]=0;i[1]=0;i[2]=0;}

  int& operator[](const int& index) {return i[index];}
  const int& operator[](const int& index) const {return i[index];}

  int& X() {return i[0];}
  int& Y() {return i[1];}
  int& Z() {return i[2];}

  const int& X() const {return i[0];}
  const int& Y() const {return i[1];}
  const int& Z() const {return i[2];}

  const Index Abs() {return Index(abs(i[0]), abs(i[1]), abs(i[2]));}
  int Length() const {return sqrt(i[0]*i[0]+i[1]*i[1]+i[2]*i[2]);}
  int Max() const {return std::max(std::max(i[0],i[1]),i[2]);}
  int Min() const {return std::min(std::min(i[0],i[1]),i[2]);}
  int Product() const {return i[0]*i[1]*i[2];}

  Index& Clamp(const Index& lower_bound, const Index& upper_bound)
  {
    for (int j=0; j<3; ++j) {
      i[j] = std::max(i[j], lower_bound[j]);
      i[j] = std::min(i[j], upper_bound[j]);
    }

    return *this;
  }

  int* vec() {return i;}
  const int* vec() const {return i;}

  Index& operator=(const Index& rhs)
  {
    i[0] = rhs.X();
    i[1] = rhs.Y();
    i[2] = rhs.Z();
    return *this;
  }

  Index& operator=(const int& rhs)
  {
    i[0] = rhs;
    i[1] = rhs;
    i[2] = rhs;
    return *this;
  }

  bool operator==(const Index& other) const
  {
    return (i[0]==other.X() && i[1]==other.Y() && i[2]==other.Z());
  }

  bool operator!=(const Index& other) const
  {
    return !(*this == other);
  }

  Index& operator+=(const Index& rhs)
  {
    i[0] += rhs.X();
    i[1] += rhs.Y();
    i[2] += rhs.Z();
    return *this;
  }

  Index& operator-=(const Index& rhs)
  {
    i[0] -= rhs.X();
    i[1] -= rhs.Y();
    i[2] -= rhs.Z();
    return *this;
  }

  Index& operator*=(const Index& rhs)
  {
    i[0] *= rhs.X();
    i[1] *= rhs.Y();
    i[2] *= rhs.Z();
    return *this;
  }

  Index& operator/=(const Index& rhs)
  {
    i[0] /= rhs.X();
    i[1] /= rhs.Y();
    i[2] /= rhs.Z();
    return *this;
  }

  Index& operator%=(const Index& rhs)
  {
    i[0] %= rhs.X();
    i[1] %= rhs.Y();
    i[2] %= rhs.Z();
    return *this;
  }

  Index& operator+=(const int& rhs)
  {
    i[0] += rhs;
    i[1] += rhs;
    i[2] += rhs;
    return *this;
  }

  Index& operator-=(const int& rhs)
  {
    i[0] -= rhs;
    i[1] -= rhs;
    i[2] -= rhs;
    return *this;
  }

  Index& operator*=(const int& rhs)
  {
    i[0] *= rhs;
    i[1] *= rhs;
    i[2] *= rhs;
    return *this;
  }

  Index& operator/=(const int& rhs)
  {
    i[0] /= rhs;
    i[1] /= rhs;
    i[2] /= rhs;
    return *this;
  }

  Index& operator%=(const int& rhs)
  {
    i[0] %= rhs;
    i[1] %= rhs;
    i[2] %= rhs;
    return *this;
  }

  const Index operator+(const Index& rhs) const
  {
    return Index(*this) += rhs;
  }

  const Index operator-(const Index& rhs) const
  {
    return Index(*this) -= rhs;
  }

  const Index operator*(const Index& rhs) const
  {
    return Index(*this) *= rhs;
  }

  const Index operator/(const Index& rhs) const
  {
    return Index(*this) /= rhs;
  }

  const Index operator%(const Index& rhs) const
  {
    return Index(*this) %= rhs;
  }

  const Index operator+(const int& rhs) const
  {
    return Index(*this) += rhs;
  }

  const Index operator-(const int& rhs) const
  {
    return Index(*this) -= rhs;
  }

  const Index operator*(const int& rhs) const
  {
    return Index(*this) *= rhs;
  }

  const Index operator/(const int& rhs) const
  {
    return Index(*this) /= rhs;
  }

  const Index operator%(const int& rhs) const
  {
    return Index(*this) %= rhs;
  }

  const Vector operator+(const Vector& rhs) const;
  const Vector operator-(const Vector& rhs) const;
  const Vector operator*(const Vector& rhs) const;
  const Vector operator/(const Vector& rhs) const;

  const Vector operator+(const vmg_float& rhs) const;
  const Vector operator-(const vmg_float& rhs) const;
  const Vector operator*(const vmg_float& rhs) const;
  const Vector operator/(const vmg_float& rhs) const;

private:
  int i[3];
};

inline const Index operator+(const int& lhs, const Index& rhs)
{
  return Index(lhs) += rhs;
}

inline const Index operator-(const int& lhs, const Index& rhs)
{
  return Index(lhs) -= rhs;
}

inline const Index operator*(const int& lhs, const Index& rhs)
{
  return Index(lhs) *= rhs;
}

inline const Index operator/(const int& lhs, const Index& rhs)
{
  return Index(lhs) /= rhs;
}

inline const Index operator%(const int& lhs, const Index& rhs)
{
  return Index(lhs) %= rhs;
}

bool operator<(const Index& lhs, const Index& rhs);
const Index Max(const Index& index1, const Index& index2);
std::ostream& operator<<(std::ostream& out, const Index& base);

}

#endif /* INDEX_HPP_ */