/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2020 NKI/AVL, Netherlands Cancer Institute
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 *    list of conditions and the following disclaimer
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "cif++/cif++.hpp"
#include "cif++/symmetry.hpp"
#include "test-main.hpp"

#include <catch2/catch_test_macros.hpp>
#include <catch2/matchers/catch_matchers_floating_point.hpp>

#if defined(_MSC_VER)
# pragma warning(disable : 5054) // warning C5054: operator '&': deprecated between enumerations of different types
# pragma warning(disable : 4127) // conditional expression is constant
#endif

#include <Eigen/Eigen>

// --------------------------------------------------------------------

cif::file operator""_cf(const char *text, std::size_t length)
{
	struct membuf : public std::streambuf
	{
		membuf(char *text, std::size_t length)
		{
			this->setg(text, text, text + length);
		}
	} buffer(const_cast<char *>(text), length);

	std::istream is(&buffer);
	return cif::file(is);
}

// --------------------------------------------------------------------

// 3d tests

TEST_CASE("t1")
{
	// std::random_device rnd;
	// std::mt19937 gen(rnd());
	// std::uniform_real_distribution<float> dis(0, 1);

	// Quaternion q{ dis(gen), dis(gen), dis(gen), dis(gen) };
	// q = Normalize(q);

	// Quaternion q{ 0.1, 0.2, 0.3, 0.4 };
	cif::quaternion q{ 0.5, 0.5, 0.5, 0.5 };
	q = normalize(q);

	const auto &&[angle0, axis0] = cif::quaternion_to_angle_axis(q);

	std::vector<cif::point> p1{
		{ 16.979f, 13.301f, 44.555f },
		{ 18.150f, 13.525f, 43.680f },
		{ 18.656f, 14.966f, 43.784f },
		{ 17.890f, 15.889f, 44.078f },
		{ 17.678f, 13.270f, 42.255f },
		{ 16.248f, 13.734f, 42.347f },
		{ 15.762f, 13.216f, 43.724f }
	};

	auto p2 = p1;

	cif::center_points(p1);

	for (auto &p : p2)
		p.rotate(q);

	cif::center_points(p2);

	auto q2 = cif::align_points(p1, p2);

	const auto &&[angle, axis] = cif::quaternion_to_angle_axis(q2);

	CHECK_THAT(std::fmod(360 + angle, 360), Catch::Matchers::WithinRel(std::fmod(360 - angle0, 360), 0.01));

	for (auto &p : p1)
		p.rotate(q2);

	auto rmsd = cif::RMSd(p1, p2);

	REQUIRE(rmsd < 1e-5);

	// std::cout << "rmsd: " << RMSd(p1, p2) << '\n';
}

TEST_CASE("t2")
{
	cif::point p[] = {
		{ 1, 1, 0 },
		{ 2, 1, 0 },
		{ 1, 2, 0 }
	};

	cif::point xp = cif::cross_product(p[1] - p[0], p[2] - p[0]);

	auto q = cif::construct_from_angle_axis(45, xp);

	auto &&[angle, axis] = cif::quaternion_to_angle_axis(q);

	CHECK_THAT(angle, Catch::Matchers::WithinRel(45.f, 0.01f));
}

TEST_CASE("t3")
{
	cif::point p[] = {
		{ 1, 1, 0 },
		{ 2, 1, 0 },
		{ 1, 2, 0 }
	};

	cif::point xp = cif::cross_product(p[1] - p[0], p[2] - p[0]);

	auto q = cif::construct_from_angle_axis(45, xp);

	auto v = p[1];
	v -= p[0];
	v.rotate(q);
	v += p[0];

	std::cout << v << '\n';

	double a = cif::angle(v, p[0], p[1]);

	CHECK_THAT(a, Catch::Matchers::WithinRel(45.f, 0.01f));
}

TEST_CASE("dh_q_0")
{
	cif::point axis(1, 0, 0);

	cif::point p(1, 1, 0);

	cif::point t[3] = {
		{ 0, 1, 0 },
		{ 0, 0, 0 },
		{ 1, 0, 0 }
	};

	auto a = cif::dihedral_angle(t[0], t[1], t[2], p);
	CHECK_THAT(a, Catch::Matchers::WithinRel(0.f, 0.01f));

	auto q = cif::construct_from_angle_axis(90, axis);

	p.rotate(q);

	REQUIRE(std::abs(p.m_x - 1.f) < 0.01f);
	REQUIRE(std::abs(p.m_y - 0.f) < 0.01f);
	REQUIRE(std::abs(p.m_z - 1.f) < 0.01f);

	a = cif::dihedral_angle(t[0], t[1], t[2], p);
	REQUIRE(std::abs(a - 90.f) < 0.01f);

	q = cif::construct_from_angle_axis(-90, axis);

	p.rotate(q);

	REQUIRE(std::abs(p.m_x - 1.f) < 0.01f);
	REQUIRE(std::abs(p.m_y - 1.f) < 0.01f);
	REQUIRE(std::abs(p.m_z - 0.f) < 0.01f);

	a = cif::dihedral_angle(t[0], t[1], t[2], p);
	REQUIRE(std::abs(a - 0.f) < 0.01f);
}

TEST_CASE("dh_q_1")
{
	struct
	{
		float angle;
		cif::point pts[4];
	} tests[] = {
		{ -97.5f,
			{ { 68.8649979f, -7.34800005f, 54.3769989f },
				{ 68.1350021f, -8.18700027f, 53.6489983f },
				{ 68.7760239f, -9.07335377f, 52.7140236f },
				{ 68.9000015f, -10.3944235f, 53.2217026f } } },
		{ 80.3f,
			{ { 0.304512024f, 0.531184196f, 2.25860214f },
				{ 0.956512451f, 0.0321846008f, 1.07460022f },
				{ 0, 0, 0 },
				{ 0.21336633f, -1.09552193f, -0.878999829f } } },
		{ -97.5f,
			{ { 0.088973999f, 1.72535372f, 1.66297531f },
				{ -0.641021729f, 0.886353493f, 0.93497467f },
				{ 0, 0, 0 },
				{ 1.29433727f, -0.395142615f, 0.432300746f } } },
		{ -97.5f,
			{
				{ 0.088973999f, 1.72535372f, 1.66297531f },
				{ -0.641021729f, 0.886353493f, 0.93497467f },
				{ 0, 0, 0 },
				{ 1.33983064f, 0.384027064f, -0.275154471f },

			} }
	};

	for (auto &&[angle, pts] : tests)
	{
		auto q = cif::construct_for_dihedral_angle(pts[0], pts[1], pts[2], pts[3], angle, 1);

		pts[3].rotate(q, pts[2]);

		auto dh = cif::dihedral_angle(pts[0], pts[1], pts[2], pts[3]);
		CHECK_THAT(dh, Catch::Matchers::WithinRel(angle, 0.1f));
	}
}

// TEST_CASE("m2q_1")
// {
// 	for (std::size_t i = 0; i < cif::kSymopNrTableSize; ++i)
// 	{
// 		auto d = cif::kSymopNrTable[i].symop().data();

// 		Eigen::Matrix3f rot;

// 		rot << d[0], d[1], d[2], d[3], d[4], d[5], d[6], d[7], d[8];

// 		if (rot * rot.transpose() == Eigen::Matrix3f::Identity() and rot.determinant() == 1)
// 		{
// 			Eigen::Quaternionf qe(rot);
// 			auto q = normalize(cif::quaternion{ qe.w(), qe.x(), qe.y(), qe.z() });

// 			cif::point p1{ 1, 1, 1 };
// 			cif::point p2 = p1;
// 			p2.rotate(q);

// 			auto p3 = rot * Eigen::Vector3f{ p1.m_x, p1.m_y, p1.m_z };

// 			CHECK_THAT(p2.m_x, Catch::Matchers::WithinRel(p3[0], 0.01f));
// 			CHECK_THAT(p2.m_y, Catch::Matchers::WithinRel(p3[1], 0.01f));
// 			CHECK_THAT(p2.m_z, Catch::Matchers::WithinRel(p3[2], 0.01f));
// 		}
// 	}
// }

// --------------------------------------------------------------------

TEST_CASE("symm_1")
{
	cif::cell c(10, 10, 10);

	cif::point p{ 1, 1, 1 };

	cif::point f = fractional(p, c);

	CHECK_THAT(f.m_x, Catch::Matchers::WithinRel(0.1f, 0.01f));
	CHECK_THAT(f.m_y, Catch::Matchers::WithinRel(0.1f, 0.01f));
	CHECK_THAT(f.m_z, Catch::Matchers::WithinRel(0.1f, 0.01f));

	cif::point o = orthogonal(f, c);

	CHECK_THAT(o.m_x, Catch::Matchers::WithinRel(1.f, 0.01f));
	CHECK_THAT(o.m_y, Catch::Matchers::WithinRel(1.f, 0.01f));
	CHECK_THAT(o.m_z, Catch::Matchers::WithinRel(1.f, 0.01f));
}

TEST_CASE("symm_2")
{
	using namespace cif::literals;

	auto symop = "1_555"_symop;

	REQUIRE(symop.is_identity() == true);
}

TEST_CASE("symm_3")
{
	using namespace cif::literals;

	cif::spacegroup sg(18);

	REQUIRE(sg.size() == 4UL);
	REQUIRE(sg.get_name() == "P 21 21 2");
}

TEST_CASE("symm_4")
{
	using namespace cif::literals;

	// based on 2b8h
	auto sg = cif::spacegroup(154); // p 32 2 1
	auto c = cif::cell(107.516f, 107.516f, 338.487f, 90.00f, 90.00f, 120.00f);

	cif::point a{ -8.688f, 79.351f, 10.439f };  // O6 NAG A 500
	cif::point b{ -35.356f, 33.693f, -3.236f }; // CG2 THR D 400
	cif::point sb(-6.916f, 79.34f, 3.236f);     // 4_565 copy of b

	CHECK_THAT(distance(a, sg(a, c, "1_455"_symop)), Catch::Matchers::WithinRel(static_cast<float>(c.get_a()), 0.01f));
	CHECK_THAT(distance(a, sg(a, c, "1_545"_symop)), Catch::Matchers::WithinRel(static_cast<float>(c.get_b()), 0.01f));
	CHECK_THAT(distance(a, sg(a, c, "1_554"_symop)), Catch::Matchers::WithinRel(static_cast<float>(c.get_c()), 0.01f));

	auto sb2 = sg(b, c, "4_565"_symop);
	CHECK_THAT(sb.m_x, Catch::Matchers::WithinRel(sb2.m_x, 0.01f));
	CHECK_THAT(sb.m_y, Catch::Matchers::WithinRel(sb2.m_y, 0.01f));
	CHECK_THAT(sb.m_z, Catch::Matchers::WithinRel(sb2.m_z, 0.01f));

	CHECK_THAT(distance(a, sb2), Catch::Matchers::WithinRel(7.42f, 0.01f));
}

// --------------------------------------------------------------------

TEST_CASE("symm_2bi3_1")
{
	cif::file f(gTestDir / "2bi3.cif.gz");
	f.front().set_validator(cif::validator_factory::instance().get("mmcif_pdbx.dic"));

	auto &db = f.front();
	cif::mm::structure s(db);

	cif::crystal c(db);

	auto struct_conn = db["struct_conn"];
	for (const auto &[asym1, seqid1, authseqid1, atomid1, symm1,
			 asym2, seqid2, authseqid2, atomid2, symm2,
			 dist] : struct_conn.find<std::string, int, std::string, std::string, std::string,
			 std::string, int, std::string, std::string, std::string,
			 float>(
			 cif::key("ptnr1_symmetry") != "1_555" or cif::key("ptnr2_symmetry") != "1_555",
			 "ptnr1_label_asym_id", "ptnr1_label_seq_id", "ptnr1_auth_seq_id", "ptnr1_label_atom_id", "ptnr1_symmetry",
			 "ptnr2_label_asym_id", "ptnr2_label_seq_id", "ptnr2_auth_seq_id", "ptnr2_label_atom_id", "ptnr2_symmetry",
			 "pdbx_dist_value"))
	{
		auto &r1 = s.get_residue(asym1, seqid1, authseqid1);
		auto &r2 = s.get_residue(asym2, seqid2, authseqid2);

		auto a1 = r1.get_atom_by_atom_id(atomid1);
		auto a2 = r2.get_atom_by_atom_id(atomid2);

		auto sa1 = c.symmetry_copy(a1.get_location(), cif::sym_op(symm1));
		auto sa2 = c.symmetry_copy(a2.get_location(), cif::sym_op(symm2));

		CHECK_THAT(cif::distance(sa1, sa2), Catch::Matchers::WithinAbs(dist, 0.5f));

		auto pa1 = a1.get_location();

		const auto &[d, p, so] = c.closest_symmetry_copy(pa1, a2.get_location());

		CHECK_THAT(p.m_x, Catch::Matchers::WithinAbs(sa2.m_x, 0.5f));
		CHECK_THAT(p.m_y, Catch::Matchers::WithinAbs(sa2.m_y, 0.5f));
		CHECK_THAT(p.m_z, Catch::Matchers::WithinAbs(sa2.m_z, 0.5f));

		CHECK_THAT(d, Catch::Matchers::WithinAbs(dist, 0.5f));
		REQUIRE(so.string() == symm2);
	}
}

TEST_CASE("symm_2bi3_1a")
{
	using namespace cif::literals;

	cif::file f(gTestDir / "2bi3.cif.gz");
	f.front().set_validator(cif::validator_factory::instance().get("mmcif_pdbx.dic"));

	auto &db = f.front();

	cif::crystal c(db);
	auto struct_conn = db["struct_conn"];
	auto atom_site = db["atom_site"];

	for (const auto &[asym1, seqid1, authseqid1, atomid1, symm1,
			 asym2, seqid2, authseqid2, atomid2, symm2,
			 dist] : struct_conn.find<std::string, std::optional<int>, std::string, std::string, std::string,
			 std::string, std::optional<int>, std::string, std::string, std::string,
			 float>(
			 cif::key("ptnr1_symmetry") != "1_555" or cif::key("ptnr2_symmetry") != "1_555",
			 "ptnr1_label_asym_id", "ptnr1_label_seq_id", "ptnr1_auth_seq_id", "ptnr1_label_atom_id", "ptnr1_symmetry",
			 "ptnr2_label_asym_id", "ptnr2_label_seq_id", "ptnr2_auth_seq_id", "ptnr2_label_atom_id", "ptnr2_symmetry",
			 "pdbx_dist_value"))
	{
		cif::point p1 = atom_site.find1<float, float, float>(
			"label_asym_id"_key == asym1 and "label_seq_id"_key == seqid1 and "auth_seq_id"_key == authseqid1 and "label_atom_id"_key == atomid1,
			"cartn_x", "cartn_y", "cartn_z");
		cif::point p2 = atom_site.find1<float, float, float>(
			"label_asym_id"_key == asym2 and "label_seq_id"_key == seqid2 and "auth_seq_id"_key == authseqid2 and "label_atom_id"_key == atomid2,
			"cartn_x", "cartn_y", "cartn_z");

		auto sa1 = c.symmetry_copy(p1, cif::sym_op(symm1));
		auto sa2 = c.symmetry_copy(p2, cif::sym_op(symm2));

		CHECK_THAT(cif::distance(sa1, sa2), Catch::Matchers::WithinAbs(dist, 0.5f));

		const auto &[d, p, so] = c.closest_symmetry_copy(p1, p2);

		CHECK_THAT(p.m_x, Catch::Matchers::WithinAbs(sa2.m_x, 0.5f));
		CHECK_THAT(p.m_y, Catch::Matchers::WithinAbs(sa2.m_y, 0.5f));
		CHECK_THAT(p.m_z, Catch::Matchers::WithinAbs(sa2.m_z, 0.5f));

		CHECK_THAT(d, Catch::Matchers::WithinAbs(dist, 0.5f));
		REQUIRE(so.string() == symm2);
	}
}

TEST_CASE("symm_3bwh_1")
{
	cif::file f(gTestDir / "3bwh.cif.gz");
	f.front().set_validator(cif::validator_factory::instance().get("mmcif_pdbx.dic"));

	auto &db = f.front();

	cif::crystal c(db);
	cif::mm::structure s(db);

	for (auto a1 : s.atoms())
	{
		for (auto a2 : s.atoms())
		{
			if (a1 == a2)
				continue;

			const auto &[d, p, so] = c.closest_symmetry_copy(a1.get_location(), a2.get_location());

			CHECK_THAT(d, Catch::Matchers::WithinAbs(distance(a1.get_location(), p), 0.5f));
		}
	}
}

TEST_CASE("symm_476d")
{
	cif::file f(gTestDir / "476d.cif.gz");
	f.front().set_validator(cif::validator_factory::instance().get("mmcif_pdbx.dic"));

	auto &db = f.front();
	cif::mm::structure s(db);

	cif::crystal c(db);

	auto struct_conn = db["struct_conn"];
	for (const auto &[asym1, seqid1, authseqid1, atomid1, symm1,
			 asym2, seqid2, authseqid2, atomid2, symm2,
			 dist] : struct_conn.find<std::string, int, std::string, std::string, std::string,
			 std::string, int, std::string, std::string, std::string,
			 float>(
			 cif::key("ptnr1_symmetry") != "1_555" or cif::key("ptnr2_symmetry") != "1_555",
			 "ptnr1_label_asym_id", "ptnr1_label_seq_id", "ptnr1_auth_seq_id", "ptnr1_label_atom_id", "ptnr1_symmetry",
			 "ptnr2_label_asym_id", "ptnr2_label_seq_id", "ptnr2_auth_seq_id", "ptnr2_label_atom_id", "ptnr2_symmetry",
			 "pdbx_dist_value"))
	{
		auto &r1 = s.get_residue(asym1, seqid1, authseqid1);
		auto &r2 = s.get_residue(asym2, seqid2, authseqid2);

		auto a1 = r1.get_atom_by_atom_id(atomid1);
		auto a2 = r2.get_atom_by_atom_id(atomid2);

		auto p1 = a1.get_location();
		auto p2 = a2.get_location();

		cif::sym_op so1(symm1);
		cif::sym_op so2(symm2);

		auto sa1 = c.symmetry_copy(p1, so1);
		auto sa2 = c.symmetry_copy(p2, so2);

		CHECK_THAT(cif::distance(sa1, sa2), Catch::Matchers::WithinAbs(dist, 0.01f));
	}
}

TEST_CASE("symm-P_32_2_1_a")
{
	cif::cell c{ 80, 80, 120, 90, 90, 120 };
	cif::spacegroup sg{ 154 };

	cif::crystal crystal{ c, sg };

	cif::point a{ 1, 90, 1 };
	cif::point p1{ 2, 2, 2 };

	auto d = distance(a, p1);

	auto [d2, p, so] = crystal.closest_symmetry_copy(a, p1);

	std::cout << "d: " << d2 << " p: " << p << " so: " << so.string() << '\n';

	auto p2 = crystal.symmetry_copy(p1, so);
	auto d3 = distance(p2, a);

	CHECK_THAT(cif::distance(p2, p), Catch::Matchers::WithinAbs(0.f, 0.01f));
	CHECK_THAT(d3, Catch::Matchers::WithinAbs(d2, 0.01f));

	CHECK(d2 <= d);
}

TEST_CASE("symm-P_32_2_1")
{
	cif::cell c{ 82.162, 82.162, 135.202, 90, 90, 120 };
	cif::spacegroup sg{ 154 };

	cif::crystal crystal{ c, sg };

	cif::point a{ 1.73727,89.1813,11.1388 };
	cif::point p1{ -8.98574, 50.3861, -11.6447 };

	auto d = distance(a, p1);

	auto [d2, p, so] = crystal.closest_symmetry_copy(a, p1);

	std::cout << "d: " << d2 << " p: " << p << " so: " << so.string() << '\n';

	auto p2 = crystal.symmetry_copy(p1, so);
	auto d3 = distance(p2, a);

	CHECK_THAT(cif::distance(p2, p), Catch::Matchers::WithinAbs(0.f, 0.01f));
	CHECK_THAT(d3, Catch::Matchers::WithinAbs(d2, 0.01f));

	CHECK(d2 <= d);
}

TEST_CASE("volume_3bwh_1")
{
	cif::file f(gTestDir / "1juh.cif.gz");
	f.front().set_validator(cif::validator_factory::instance().get("mmcif_pdbx.dic"));

	auto &db = f.front();

	cif::crystal c(db);

	CHECK_THAT(c.get_cell().get_volume(), Catch::Matchers::WithinRel(741009.625f, 0.01f));
}

// --------------------------------------------------------------------

TEST_CASE("smallest_sphere-1")
{
	std::vector<cif::point> pts{
		{ 0.9295, 4.9006, 46.9706 },
		{ -0.1215, 5.5936, 46.0726 },
		{ -0.7975, 4.7046, 45.0796 },
		{ -1.4875, 3.5486, 45.7196 },
		{ -0.6535, 2.8816, 46.8186 },
		{ 0.3825, 3.5156, 47.4496 },
		{ 1.1995, 2.9206, 48.5286 },
		{ 0.8255, 2.0466, 49.4716 },
		{ 1.6625, 1.5036, 50.5176 },
		{ 1.1165, 0.6056, 51.3626 },
		{ 1.8325, -0.0064, 52.4656 },
		{ 1.1945, -0.9044, 53.2216 },
		{ 1.8135, -1.5534, 54.3566 },
		{ 1.0925, -2.4574, 55.0656 },
		{ 1.5205, -3.2204, 56.2476 },
		{ 1.1955, 5.8066, 48.1796 },
		{ 2.2495, 4.6896, 46.1796 },
		{ -1.2515, 1.5186, 47.1786 },
		{ 3.1385, 1.9106, 50.6166 },
		{ 3.2605, -1.1834, 54.7206 },
		{ 2.5975, -3.8554, 56.2096 },
		{ 0.7975, -3.2184, 57.2686 }
	};

	for (int i = 0; i < 1000; ++i)
	{
		auto [c, r] = cif::smallest_sphere_around_points(pts);
		CHECK_THAT(cif::distance(c, cif::point{ 0, 0.743099928, 51.1741028 }), Catch::Matchers::WithinAbs(0.f, 0.01f));
		CHECK_THAT(r, Catch::Matchers::WithinAbs(7.31248331f, 0.01f));
	}
}