Mask/include/Vec3.h

/*
	Class to represent a 3-space vector in both cartesian and spherical coordinates. Can perform vector
	addition, subtraction, and dot product. 

	--GWM Dec 2020
*/
#ifndef VEC3_H
#define VEC3_H

#include <cmath>

namespace Mask {

	class Vec3 {
	public:
		Vec3();
		Vec3(double x, double y, double z);
		~Vec3();
	
		void SetVectorCartesian(double x, double y, double z);
		void SetVectorSpherical(double r, double theta, double phi);
		inline double GetX() const { return m_data[0]; }
		inline double GetY() const { return m_data[1]; }
		inline double GetZ() const { return m_data[2]; }
		inline double GetRho() const { return std::sqrt(std::pow(m_data[0], 2.0) + std::pow(m_data[1], 2.0)); }
		inline double GetR() const { return std::sqrt(std::pow(m_data[0], 2.0) + std::pow(m_data[1], 2.0) + std::pow(m_data[2], 2.0)); }
		inline double GetTheta() const { return Atan2(GetRho(), GetZ()); }
		inline double GetPhi() const {
			double phi = Atan2(GetY(), GetX());
			if(phi < 0) phi += M_PI*2.0;
			return phi;
		}
	
		inline const double operator[](int index) const { return index>2 || index<0 ? 0.0 : m_data[index]; }
		inline Vec3& operator=(const Vec3& rhs) { SetVectorCartesian(rhs.GetX(), rhs.GetY(), rhs.GetZ()); return *this; }
		inline Vec3 operator+(const Vec3& rhs) const { return Vec3(this->GetX()+rhs.GetX(), this->GetY()+rhs.GetY(), this->GetZ()+rhs.GetZ()); }
		inline Vec3 operator-(const Vec3& rhs) const { return Vec3(this->GetX()-rhs.GetX(), this->GetY()-rhs.GetY(), this->GetZ()-rhs.GetZ()); }
	
	
		double Dot(const Vec3& rhs) const;
		Vec3 Cross(const Vec3& rhs) const;
	
	
	
	private:
	
		//Use instead of std::atan2. Better control over values close to x=0
		inline double Atan2(double y, double x) const {
			if(x != 0.0) return std::atan2(y, x);
			else if(y > 0.0) return M_PI/2.0;
			else if(y < 0.0) return -M_PI/2.0;
			else return 0.0;
		}
	
		double m_data[3];
	
	};

}

#endif