#include <sstream>
#include <gtest/gtest.h>
#include <wrmath/geom/vector.h>

using namespace std;
using namespace wr;


TEST(Vector3Miscellaneous, ExtractionOperator3d)
{
	geom::Vector3d	vec {1.0, 2.0, 3.0};
	stringstream	vecBuffer;

	vecBuffer << vec;
	EXPECT_EQ(vecBuffer.str(), "<1, 2, 3>");
}


TEST(Vector3Miscellaneous, ExtractionOperator3f)
{
	geom::Vector3f	vec {1.0, 2.0, 3.0};
	stringstream	vecBuffer;

	vecBuffer << vec;
	EXPECT_EQ(vecBuffer.str(), "<1, 2, 3>");
}


TEST(Vector3Miscellaneous, SetEpsilon)
{
	geom::Vector3f	a {1.0, 1.0, 1.0};
	geom::Vector3f	b;

	a.setEpsilon(1.1);
	EXPECT_EQ(a, b);
}

TEST(Vector3FloatTests, Magnitude)
{
	geom::Vector3f	v3f {1.0, -2.0, 3.0};
	const float	expected = 3.74165738677394;

	EXPECT_FLOAT_EQ(v3f.magnitude(), expected);
}


TEST(Vector3FloatTests, Equality)
{
	geom::Vector3f	a {1.0, 2.0, 3.0};
	geom::Vector3f	b {1.0, 2.0, 3.0};
	geom::Vector3f	c {1.0, 2.0, 1.0};

	EXPECT_EQ(a, b);
	EXPECT_EQ(b, a);
	EXPECT_NE(a, c);
	EXPECT_NE(b, c);
}


TEST(Vector3FloatTests, Addition)
{
	geom::Vector3f	a {1.0, 2.0, 3.0};
	geom::Vector3f	b {4.0, 5.0, 6.0};
	geom::Vector3f	expected {5.0, 7.0, 9.0};

	EXPECT_EQ(a+b, expected);
}


TEST(Vector3FloatTests, Subtraction)
{
	geom::Vector3f	a {1.0, 2.0, 3.0};
	geom::Vector3f	b {4.0, 5.0, 6.0};
	geom::Vector3f	c {5.0, 7.0, 9.0};

	EXPECT_EQ(c-b, a);
}


TEST(Vector3FloatTests, ScalarMultiplication)
{
	geom::Vector3f	a {1.0, 2.0, 3.0};
	geom::Vector3f	expected {3.0, 6.0, 9.0};

	EXPECT_EQ(a * 3.0, expected);
}


TEST(Vector3FloatTests, ScalarDivision)
{
	geom::Vector3f	a {1.0, 2.0, 3.0};
	geom::Vector3f	b {3.0, 6.0, 9.0};

	EXPECT_EQ(b / 3.0, a);
}


TEST(Vector3FloatTests, DotProduct)
{
	geom::Vector3f	a {1.0, 2.0, 3.0};
	geom::Vector3f	b {4.0, 5.0, 6.0};

	EXPECT_FLOAT_EQ(a * b, 32.0);	
}


TEST(Vector3FloatTests, UnitVector)
{
	// Test values randomly generated and calculated with numpy.
	geom::Vector3f	vec3 {5.320264018493507, 5.6541812891273935, 1.9233435162644652};
	geom::Vector3f	unit {0.6651669556972103, 0.7069150218815566, 0.24046636539587804};
	geom::Vector3f	unit2;

	EXPECT_EQ(vec3.unitVector(), unit);
	EXPECT_FALSE(vec3.isUnitVector());
	EXPECT_TRUE(unit.isUnitVector());
	EXPECT_TRUE(unit2.isUnitVector());
}


TEST(Vector3FloatTests, Angle)
{
	geom::Vector3f	a {0.3977933061361172, 8.053980094436525, 8.1287759943773};
	geom::Vector3f	b {9.817895298608196, 4.034166890407462, 4.37628316513266};
	geom::Vector3f	c {7.35, 0.221, 5.188};
	geom::Vector3f	d {2.751, 8.259, 3.985};
	
	EXPECT_FLOAT_EQ(a.angle(b), 0.9914540426033251);
	EXPECT_NEAR(c.angle(d), 1.052, 0.001);
}

TEST(Vector3FloatTests, ParallelOrthogonalVectors)
{
	geom::Vector3f	a {-2.029, 9.97, 4.172};
	geom::Vector3f	b {-9.231, -6.639, -7.245};
	geom::Vector3f	c {-2.328, -7.284, -1.214};
	geom::Vector3f	d {-1.821, 1.072, -2.94};
	geom::Vector3f	e {-2.0, 1.0, 3.0};
	geom::Vector3f	f {-6.0, 3.0, 9.0};
	geom::Vector3f	zeroVector {0.0, 0.0, 0.0};

	EXPECT_FALSE(a.isParallel(b));
	EXPECT_FALSE(a.isOrthogonal(b));

	EXPECT_FALSE(c.isParallel(d));
	EXPECT_TRUE(c.isOrthogonal(d));

	EXPECT_TRUE(e.isParallel(f));
	EXPECT_FALSE(e.isOrthogonal(f));

	EXPECT_TRUE(zeroVector.isZero());
	EXPECT_TRUE(c.isParallel(zeroVector));
	EXPECT_TRUE(c.isOrthogonal(zeroVector));
}


TEST(Vector3FloatTests, Projections)
{
	geom::Vector3f	a {4.866769214609107, 6.2356222686140566, 9.140878417029711};
	geom::Vector3f	b {6.135533104801077, 8.757851406697895, 0.6738031370548048};
	geom::Vector3f	c {4.843812341655318, 6.9140509888133055, 0.5319465962229454};
	geom::Vector3f	d {0.02295687295378901, -0.6784287201992489, 8.608931820806765};

	ASSERT_EQ(a.projectParallel(b), c);
	ASSERT_EQ(a.projectOrthogonal(b), d);
}


TEST(Vector3FloatTests, CrossProduct)
{
	geom::Vector3f	a {8.462, 7.893, -8.187};
	geom::Vector3f	b {6.984, -5.975, 4.778};
	geom::Vector3f	c {-11.2046, -97.6094, -105.685};

	c.setEpsilon(0.001);
	EXPECT_EQ(c, a.cross(b));
}


TEST(Vector3DoubleTests, Magnitude)
{
	geom::Vector3d	v3d{1.0, -2.0, 3.0};
	const double	expected = 3.74165738677394;

	EXPECT_DOUBLE_EQ(v3d.magnitude(), expected);
}


TEST(Vector3DoubleTests, Equality)
{
	geom::Vector3d	a {1.0, 2.0, 3.0};
	geom::Vector3d	b {1.0, 2.0, 3.0};
	geom::Vector3d	c {1.0, 2.0, 1.0};

	EXPECT_EQ(a, b);
	EXPECT_EQ(b, a);
	EXPECT_NE(a, c);
	EXPECT_NE(b, c);
}


TEST(Vector3DoubleTests, Addition)
{
	geom::Vector3d	a {1.0, 2.0, 3.0};
	geom::Vector3d	b {4.0, 5.0, 6.0};
	geom::Vector3d	expected {5.0, 7.0, 9.0};

	EXPECT_EQ(a+b, expected);
}


TEST(Vector3DoubleTests, Subtraction)
{
	geom::Vector3d	a {1.0, 2.0, 3.0};
	geom::Vector3d	b {4.0, 5.0, 6.0};
	geom::Vector3d	c {5.0, 7.0, 9.0};

	EXPECT_EQ(c-b, a);
}


TEST(Vector3DoubleTests, ScalarMultiplication)
{
	geom::Vector3d	a {1.0, 2.0, 3.0};
	geom::Vector3d	expected {3.0, 6.0, 9.0};

	EXPECT_EQ(a * 3.0, expected);
}


TEST(Vector3DoubleTests, ScalarDivision)
{
	geom::Vector3d	a {1.0, 2.0, 3.0};
	geom::Vector3d	b {3.0, 6.0, 9.0};

	EXPECT_EQ(b / 3.0, a);
}


TEST(Vector3DoubleTests, DotProduct)
{
	geom::Vector3d	a {1.0, 2.0, 3.0};
	geom::Vector3d	b {4.0, 5.0, 6.0};

	EXPECT_DOUBLE_EQ(a * b, 32.0);	
}


TEST(Vector3DoubleTests, UnitVector)
{
	// Test values randomly generated and calculated with numpy.
	geom::Vector3d	vec3 {5.320264018493507, 5.6541812891273935, 1.9233435162644652};
	geom::Vector3d	unit {0.6651669556972103, 0.7069150218815566, 0.24046636539587804};
	geom::Vector3d	unit2;

	EXPECT_EQ(vec3.unitVector(), unit);
	EXPECT_FALSE(vec3.isUnitVector());
	EXPECT_TRUE(unit.isUnitVector());
	EXPECT_TRUE(unit2.isUnitVector());
}


TEST(Vector3DoubleTests, Angle)
{
	geom::Vector3d	a {0.3977933061361172, 8.053980094436525, 8.1287759943773};
	geom::Vector3d	b {9.817895298608196, 4.034166890407462, 4.37628316513266};
	geom::Vector3d	c {7.35, 0.221, 5.188};
	geom::Vector3d	d {2.751, 8.259, 3.985};
	
	EXPECT_DOUBLE_EQ(a.angle(b), 0.9914540426033251);
	EXPECT_NEAR(c.angle(d), 1.052, 0.001);
}


TEST(Vector3DoubleTests, ParallelOrthogonalVectors)
{
	geom::Vector3d	a {-2.029, 9.97, 4.172};
	geom::Vector3d	b {-9.231, -6.639, -7.245};
	geom::Vector3d	c {-2.328, -7.284, -1.214};
	geom::Vector3d	d {-1.821, 1.072, -2.94};
	geom::Vector3d	e {-2.0, 1.0, 3.0};
	geom::Vector3d	f {-6.0, 3.0, 9.0};
	geom::Vector3d	zeroVector {0.0, 0.0, 0.0};

	EXPECT_FALSE(a.isParallel(b));
	EXPECT_FALSE(a.isOrthogonal(b));

	EXPECT_FALSE(c.isParallel(d));
	EXPECT_TRUE(c.isOrthogonal(d));

	EXPECT_TRUE(e.isParallel(f));
	EXPECT_FALSE(e.isOrthogonal(f));

	EXPECT_TRUE(zeroVector.isZero());
	EXPECT_TRUE(c.isParallel(zeroVector));
	EXPECT_TRUE(c.isOrthogonal(zeroVector));
}


TEST(Vector3DoubleTests, Projections)
{
	geom::Vector3d	a {4.866769214609107, 6.2356222686140566, 9.140878417029711};
	geom::Vector3d	b {6.135533104801077, 8.757851406697895, 0.6738031370548048};
	geom::Vector3d	c {4.843812341655318, 6.9140509888133055, 0.5319465962229454};
	geom::Vector3d	d {0.02295687295378901, -0.6784287201992489, 8.608931820806765};

	ASSERT_EQ(a.projectParallel(b), c);
	ASSERT_EQ(a.projectOrthogonal(b), d);
}


TEST(Vector3DoubleTests, CrossProduct)
{
	geom::Vector3d	a {8.462, 7.893, -8.187};
	geom::Vector3d	b {6.984, -5.975, 4.778};
	geom::Vector3d	c {-11.2046, -97.6094, -105.685};

	c.setEpsilon(0.001); // double trouble
	EXPECT_EQ(c, a.cross(b));
}



int
main(int argc, char **argv)
{
	::testing::InitGoogleTest(&argc, argv);
	return RUN_ALL_TESTS();
}