cg-hw1-part2/vec3.h

//=================================================================================================
// Written in 2016 by Peter Shirley <ptrshrl@gmail.com>
//
// To the extent possible under law, the author(s) have dedicated all copyright and related and
// neighboring rights to this software to the public domain worldwide. This software is distributed
// without any warranty.
//
// You should have received a copy (see file COPYING.txt) of the CC0 Public Domain Dedication along
// with this software. If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
//==================================================================================================

#ifndef VEC3H
#define VEC3H

#include <stdlib.h>

#include <cmath>
#include <iostream>
#include <vector>

class vec3 {
   public:
    vec3() {}
    vec3(float e0, float e1, float e2) {
        e[0] = e0;
        e[1] = e1;
        e[2] = e2;
    }
	vec3(const vec3 &v2){
        this->e[0] = v2.e[0];
        this->e[1] = v2.e[1];
        this->e[2] = v2.e[2];
	}
    float x() const { return e[0]; }
    float y() const { return e[1]; }
    float z() const { return e[2]; }
    float r() const { return e[0]; }
    float g() const { return e[1]; }
    float b() const { return e[2]; }

    inline vec3 operator+(const vec3 &v2) const { return vec3(x() + v2.x(), y() + v2.y(), z() + v2.z()); }
    inline void operator+=(const vec3 &v2) { e[0] = x() + v2.x(), e[1] = y() + v2.y(), e[2] = z() + v2.z(); }
    inline vec3 operator-(const vec3 &v2) const { return vec3(x() - v2.x(), y() - v2.y(), z() - v2.z()); }
    inline void operator-=(const vec3 &v2) { e[0] = x() - v2.x(), e[1] = y() - v2.y(), e[2] = z() - v2.z(); }

    inline vec3 operator*(const float t) const { return vec3(x() * t, y() * t, z() * t); }
    inline void operator*=(const float t) { e[0] = x() * t, e[1] = y() * t, e[2] = z() * t; }
    inline vec3 operator/(const float t) const { return vec3(x() / t, y() / t, z() / t); }
    inline void operator/=(const float t) { e[0] = x() / t, e[1] = y() / t, e[2] = z() / t; }

	inline bool operator==(const vec3 &v2) const {
		if(this->x() == v2.x() && this->y() == v2.y() && this->z() == v2.z()) return true;
        return false;
    }

    inline float length() const { return sqrt(e[0] * e[0] + e[1] * e[1] + e[2] * e[2]); }
    inline float squared_length() const { return e[0] * e[0] + e[1] * e[1] + e[2] * e[2]; }
    inline void make_unit_vector() {
        float len = length();
        e[0] = e[0] / len;
        e[1] = e[1] / len;
        e[2] = e[2] / len;
    }

    float e[3];
};

inline vec3 operator*(const float t, const vec3 &v) {
    return vec3(v.x() * t, v.y() * t, v.z() * t);
}

/*
for color calculation
*/
inline vec3 operator*(const vec3 &v1, const vec3 &v2) {
    return vec3(v1.e[0] * v2.e[0], v1.e[1] * v2.e[1], v1.e[2] * v2.e[2]);
}

inline std::ostream &operator<<(std::ostream &os, const vec3 &t) {
    os << "( " << t.x() << " , " << t.y() << " , " << t.z() << " )";
    return os;
}

inline float dot(const vec3 &v1, const vec3 &v2) {
    return v1.x() * v2.x() + v1.y() * v2.y() + v1.z() * v2.z();
}

inline vec3 cross(const vec3 &v1, const vec3 &v2) {
    return vec3(v1.y() * v2.z() - v1.z() * v2.y(),
                v1.x() * v2.z() - v1.z() * v2.x(),
                v1.x() * v2.y() - v1.y() * v2.x());
}

inline vec3 unit_vector(vec3 v) {
    return v / v.length();
}

#endif