// The MIT License (MIT)
//
// Copyright (c) 2018 Mateusz Pusz
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

#include <catch2/catch_test_macros.hpp>
#include <catch2/matchers/catch_matchers_floating_point.hpp>
#include <mp-units/compat_macros.h>
#include <mp-units/ext/format.h>
#ifdef MP_UNITS_IMPORT_STD
import std;
#else
#include <cmath>
#include <complex>
#include <sstream>
#endif
#ifdef MP_UNITS_MODULES
import mp_units;
#else
#include <mp-units/cartesian_tensor.h>
#endif

using namespace mp_units;
using Catch::Matchers::WithinRel;
using namespace std::complex_literals;

// element-type conversions follow the library's non-truncating rule (like a quantity rep), not the
// language narrowing rule: a floating-point target or a widening is implicit, while a
// floating-point -> integer element conversion is explicit (truncating).
static_assert(
  std::convertible_to<utility::cartesian_tensor<float>, utility::cartesian_tensor<double>>);  // widen: implicit
static_assert(
  std::convertible_to<utility::cartesian_tensor<double>, utility::cartesian_tensor<float>>);  // FP target: implicit
static_assert(
  std::convertible_to<utility::cartesian_tensor<int>, utility::cartesian_tensor<double>>);  // int->FP: implicit
static_assert(
  !std::convertible_to<utility::cartesian_tensor<double>, utility::cartesian_tensor<int>>);  // FP->int: explicit
static_assert(
  std::constructible_from<utility::cartesian_tensor<int>, utility::cartesian_tensor<double>>);  // ...but constructible

// the per-axis dimension is queryable at compile time, both as a value and as a call (tensor is NxN)
static_assert(utility::cartesian_tensor<double>::extent == 3);  // default N (the tensor is 3x3)
static_assert(utility::cartesian_tensor<double, 2>::extent == 2);
static_assert(utility::cartesian_tensor<double, 3>::extent() == 3);

// operations close at a single dimension N: mixing 2x2 and 3x3 (or a tensor and a mismatched vector)
// is ill-formed. Negative tests via named helper concepts asserted with static_assert (the
// `invalid_types` idiom).
namespace {
template<typename A, typename B>
concept addable = requires(A a, B b) { a + b; };
template<typename A, typename B>
concept subtractable = requires(A a, B b) { a - b; };
template<typename A, typename B>
concept inner_producible = requires(A a, B b) { inner_product(a, b); };
template<typename A, typename B>
concept double_dot_producible = requires(A a, B b) { scalar_product(a, b); };
template<typename A, typename B>
concept tensor_producible = requires(A a, B b) { tensor_product(a, b); };
}  // namespace
static_assert(!addable<utility::cartesian_tensor<double, 2>, utility::cartesian_tensor<double, 3>>);
static_assert(!subtractable<utility::cartesian_tensor<double, 2>, utility::cartesian_tensor<double, 3>>);
static_assert(!inner_producible<utility::cartesian_tensor<double, 2>, utility::cartesian_tensor<double, 3>>);
static_assert(!double_dot_producible<utility::cartesian_tensor<double, 2>, utility::cartesian_tensor<double, 3>>);
static_assert(
  !inner_producible<utility::cartesian_tensor<double, 2>, utility::cartesian_vector<double, 3>>);  // tensor . vector
static_assert(!tensor_producible<utility::cartesian_vector<double, 2>, utility::cartesian_vector<double, 3>>);
static_assert(!std::constructible_from<utility::cartesian_tensor<double, 3>, utility::cartesian_tensor<double, 2>>);

// A second-order Cartesian tensor as defined by ISO 80000-2:2019, 18.
TEST_CASE("cartesian_tensor operations", "[tensor]")
{
  SECTION("initialization and access")
  {
    SECTION("no arguments")
    {
      utility::cartesian_tensor<double> t{};
      for (std::size_t r = 0; r < 3; ++r)
        for (std::size_t c = 0; c < 3; ++c) REQUIRE(t(r, c) == 0);
    }

    SECTION("all arguments")
    {
      utility::cartesian_tensor t{1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0};
      REQUIRE(t(0, 0) == 1.0);
      REQUIRE(t(0, 2) == 3.0);
      REQUIRE(t(1, 1) == 5.0);
      REQUIRE(t(2, 0) == 7.0);
      REQUIRE(t(2, 2) == 9.0);
    }

    SECTION("convertible arguments")
    {
      utility::cartesian_tensor<double> t{1, 2, 3, 4, 5, 6, 7, 8, 9};
      REQUIRE(t(0, 0) == 1.0);
      REQUIRE(t(2, 2) == 9.0);
    }

    SECTION("conversion from another representation")
    {
      utility::cartesian_tensor t1{1, 2, 3, 4, 5, 6, 7, 8, 9};
      utility::cartesian_tensor<double> t2 = t1;
      REQUIRE(t2(1, 1) == 5.0);
    }
  }

  SECTION("addition and subtraction (2-18.2)")
  {
    utility::cartesian_tensor a{1, 2, 3, 4, 5, 6, 7, 8, 9};
    utility::cartesian_tensor b{9, 8, 7, 6, 5, 4, 3, 2, 1};

    SECTION("operator+")
    {
      auto r = a + b;
      for (std::size_t i = 0; i < 3; ++i)
        for (std::size_t j = 0; j < 3; ++j) REQUIRE(r(i, j) == 10);
    }

    SECTION("operator- and unary -")
    {
      auto r = a - a;
      for (std::size_t i = 0; i < 3; ++i)
        for (std::size_t j = 0; j < 3; ++j) REQUIRE(r(i, j) == 0);
      REQUIRE((-a)(0, 0) == -1);
      REQUIRE((-a)(2, 2) == -9);
    }

    SECTION("mixed representation (int + double)")
    {
      utility::cartesian_tensor<double> c{0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5};
      auto r = a + c;
      REQUIRE(r(0, 0) == 1.5);
    }
  }

  SECTION("scalar multiply/divide (2-18.3)")
  {
    utility::cartesian_tensor<double> t{1, 2, 3, 4, 5, 6, 7, 8, 9};

    SECTION("t * s")
    {
      auto r = t * 2.0;
      REQUIRE(r(0, 0) == 2.0);
      REQUIRE(r(2, 2) == 18.0);
    }

    SECTION("s * t")
    {
      auto r = 2.0 * t;
      REQUIRE(r(0, 0) == 2.0);
      REQUIRE(r(2, 2) == 18.0);
    }

    SECTION("t / s")
    {
      auto r = t / 2.0;
      REQUIRE(r(0, 0) == 0.5);
      REQUIRE(r(2, 2) == 4.5);
    }
  }

  SECTION("compound assignments")
  {
    utility::cartesian_tensor t{2.0, 4.0, 6.0, 8.0, 10.0, 12.0, 14.0, 16.0, 18.0};

    SECTION("operator+=")
    {
      utility::cartesian_tensor o{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0};
      t += o;
      REQUIRE(t(0, 0) == 3.0);
    }

    SECTION("operator-=")
    {
      utility::cartesian_tensor o{1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0};
      t -= o;
      REQUIRE(t(0, 0) == 1.0);
    }

    SECTION("operator*=")
    {
      t *= 0.5;
      REQUIRE(t(0, 0) == 1.0);
    }

    SECTION("operator/=")
    {
      t /= 2.0;
      REQUIRE(t(0, 0) == 1.0);
    }
  }

  SECTION("equality")
  {
    utility::cartesian_tensor a{1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0};
    utility::cartesian_tensor b{1, 2, 3, 4, 5, 6, 7, 8, 9};
    utility::cartesian_tensor c{1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.1};
    REQUIRE(a == b);
    REQUIRE(a != c);
  }

  SECTION("tensor (dyadic) product of two vectors (2-18.21)")
  {
    utility::cartesian_vector a{1, 2, 3};
    utility::cartesian_vector b{4, 5, 6};
    auto t = tensor_product(a, b);  // (a (x) b)_ij = a_i b_j
    REQUIRE(t(0, 0) == 4);
    REQUIRE(t(0, 1) == 5);
    REQUIRE(t(0, 2) == 6);
    REQUIRE(t(1, 0) == 8);
    REQUIRE(t(1, 1) == 10);
    REQUIRE(t(1, 2) == 12);
    REQUIRE(t(2, 0) == 12);
    REQUIRE(t(2, 1) == 15);
    REQUIRE(t(2, 2) == 18);
  }

  SECTION("inner product of two tensors (2-18.23)")
  {
    utility::cartesian_tensor a{1, 2, 3, 4, 5, 6, 7, 8, 9};
    utility::cartesian_tensor id{1, 0, 0, 0, 1, 0, 0, 0, 1};

    SECTION("T . I == T") { REQUIRE(inner_product(a, id) == a); }

    SECTION("T . T")
    {
      auto r = inner_product(a, a);
      // hand-computed A . A
      REQUIRE(r(0, 0) == 30);
      REQUIRE(r(0, 1) == 36);
      REQUIRE(r(0, 2) == 42);
      REQUIRE(r(1, 0) == 66);
      REQUIRE(r(1, 1) == 81);
      REQUIRE(r(1, 2) == 96);
      REQUIRE(r(2, 0) == 102);
      REQUIRE(r(2, 1) == 126);
      REQUIRE(r(2, 2) == 150);
    }
  }

  SECTION("inner product of a tensor and a vector (2-18.24)")
  {
    utility::cartesian_tensor a{1, 2, 3, 4, 5, 6, 7, 8, 9};
    utility::cartesian_vector v{1, 2, 3};
    auto r = inner_product(a, v);  // (T . a)_i = sum_j T_ij a_j
    REQUIRE(r[0] == 14);
    REQUIRE(r[1] == 32);
    REQUIRE(r[2] == 50);
  }

  SECTION("scalar (double-dot) product of two tensors (2-18.25)")
  {
    utility::cartesian_tensor a{1, 2, 3, 4, 5, 6, 7, 8, 9};
    REQUIRE(scalar_product(a, a) == 285);  // sum of squares 1..9
  }

  SECTION("Frobenius norm/magnitude")
  {
    utility::cartesian_tensor<double> a{1, 2, 3, 4, 5, 6, 7, 8, 9};
    REQUIRE_THAT(a.magnitude(), WithinRel(std::sqrt(285.0), 1e-12));
    REQUIRE_THAT(a.norm(), WithinRel(std::sqrt(285.0), 1e-12));
    REQUIRE_THAT(magnitude(a), WithinRel(std::sqrt(285.0), 1e-12));
    REQUIRE_THAT(norm(a), WithinRel(std::sqrt(285.0), 1e-12));
  }
}

// The dimension N is a compile-time parameter; a planar model uses a 2x2 tensor and pays nothing for
// a third row/column. All second-order operations close at the matching dimension.
TEST_CASE("cartesian_tensor in two dimensions (2x2)", "[tensor]")
{
  SECTION("deduces a 2x2 tensor from four components")
  {
    utility::cartesian_tensor t{1.0, 2.0, 3.0, 4.0};
    static_assert(std::is_same_v<decltype(t), utility::cartesian_tensor<double, 2>>);
    REQUIRE(t(0, 0) == 1.0);
    REQUIRE(t(0, 1) == 2.0);
    REQUIRE(t(1, 0) == 3.0);
    REQUIRE(t(1, 1) == 4.0);
  }

  SECTION("tensor (dyadic) product of two 2D vectors -> 2x2 (2-18.21)")
  {
    utility::cartesian_vector a{1, 2};
    utility::cartesian_vector b{3, 4};
    auto t = tensor_product(a, b);  // (a (x) b)_ij = a_i b_j
    static_assert(std::is_same_v<decltype(t), utility::cartesian_tensor<int, 2>>);
    REQUIRE(t(0, 0) == 3);
    REQUIRE(t(0, 1) == 4);
    REQUIRE(t(1, 0) == 6);
    REQUIRE(t(1, 1) == 8);
  }

  SECTION("inner product of two 2x2 tensors (2-18.23)")
  {
    utility::cartesian_tensor a{1, 2, 3, 4};  // [[1, 2], [3, 4]]
    auto r = inner_product(a, a);
    REQUIRE(r(0, 0) == 7);
    REQUIRE(r(0, 1) == 10);
    REQUIRE(r(1, 0) == 15);
    REQUIRE(r(1, 1) == 22);
  }

  SECTION("inner product of a 2x2 tensor and a 2D vector (2-18.24)")
  {
    utility::cartesian_tensor a{1, 2, 3, 4};
    utility::cartesian_vector v{1, 2};
    auto r = inner_product(a, v);  // (T . a)_i = sum_j T_ij a_j
    static_assert(std::is_same_v<decltype(r), utility::cartesian_vector<int, 2>>);
    REQUIRE(r[0] == 5);
    REQUIRE(r[1] == 11);
  }

  SECTION("scalar (double-dot) product and Frobenius norm")
  {
    utility::cartesian_tensor a{1.0, 2.0, 3.0, 4.0};
    REQUIRE(scalar_product(a, a) == 30.0);  // sum of squares 1+4+9+16
    REQUIRE_THAT(a.magnitude(), WithinRel(std::sqrt(30.0), 1e-12));
  }

  SECTION("text output")
  {
    std::ostringstream os;
    utility::cartesian_tensor t{1, 2, 3, 4};
    os << t;
    CHECK(os.str() == "[[1, 2], [3, 4]]");
    CHECK(MP_UNITS_STD_FMT::format("{}", t) == os.str());
  }
}

TEST_CASE("cartesian_tensor text output", "[tensor][fmt][ostream]")
{
  std::ostringstream os;

  SECTION("integral representation")
  {
    utility::cartesian_tensor t{1, 2, 3, 4, 5, 6, 7, 8, 9};
    os << t;

    SECTION("iostream") { CHECK(os.str() == "[[1, 2, 3], [4, 5, 6], [7, 8, 9]]"); }

    SECTION("fmt with default format {}") { CHECK(MP_UNITS_STD_FMT::format("{}", t) == os.str()); }
  }

  SECTION("floating-point representation")
  {
    utility::cartesian_tensor t{1.5, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0};
    os << t;

    SECTION("iostream") { CHECK(os.str() == "[[1.5, 2, 3], [4, 5, 6], [7, 8, 9]]"); }

    SECTION("fmt with default format {}") { CHECK(MP_UNITS_STD_FMT::format("{}", t) == os.str()); }
  }
}

TEST_CASE("cartesian_tensor with a complex representation", "[tensor][complex]")
{
  using c = std::complex<double>;

  SECTION("Hermitian Frobenius norm is a real scalar")
  {
    // only the (0,0) entry is non-zero: |T| = |3+4i| = 5
    utility::cartesian_tensor t{3. + 4.i, c{}, c{}, c{}, c{}, c{}, c{}, c{}, c{}};
    STATIC_CHECK(std::is_same_v<decltype(t.magnitude()), double>);
    REQUIRE_THAT(t.magnitude(), WithinRel(5.0, 1e-12));
    // sqrt(|1+i|^2 * 9) = sqrt(2 * 9) = sqrt(18)
    utility::cartesian_tensor u{1. + 1.i, 1. + 1.i, 1. + 1.i, 1. + 1.i, 1. + 1.i,
                                1. + 1.i, 1. + 1.i, 1. + 1.i, 1. + 1.i};
    REQUIRE_THAT(u.magnitude(), WithinRel(std::sqrt(18.0), 1e-12));
  }

  SECTION("scalar_product (double-dot) is sesquilinear, T : T is real and non-negative")
  {
    utility::cartesian_tensor a{1. + 1.i, c{}, c{}, c{}, c{}, c{}, c{}, c{}, c{}};
    // a : a = sum |a_ij|^2 = |1+i|^2 = 2
    REQUIRE(scalar_product(a, a) == c{2.0, 0.0});

    // Hermitian symmetry of the double-dot: <b, a> = conj(<a, b>)
    utility::cartesian_tensor b{1.i, c{}, c{}, c{}, c{}, c{}, c{}, c{}, c{}};
    REQUIRE(scalar_product(b, a) == std::conj(scalar_product(a, b)));
  }

  SECTION("inner product (matmul) uses plain complex arithmetic, no conjugation")
  {
    utility::cartesian_tensor id{c{1}, c{}, c{}, c{}, c{1}, c{}, c{}, c{}, c{1}};
    utility::cartesian_tensor a{1. + 1.i, 2. + 0.i, c{}, c{}, 3. - 1.i, c{}, c{}, c{}, 0. + 2.i};
    REQUIRE(inner_product(id, a) == a);  // I . A == A
  }

  SECTION("inner product with a complex vector -> complex vector")
  {
    utility::cartesian_tensor id{c{1}, c{}, c{}, c{}, c{1}, c{}, c{}, c{}, c{1}};
    utility::cartesian_vector v{1. + 1.i, 2. + 0.i, 0. + 3.i};
    REQUIRE(inner_product(id, v) == v);  // I . v == v
  }
}

namespace {
template<typename T>
concept tensor_embeddable = requires(T t) { embed(t); };
template<typename T>
concept tensor_projectable = requires(T t) { project(t); };
}  // namespace
// embed only lifts 2x2->3x3 and project only lowers 3x3->2x2 (each defined for one source dimension)
static_assert(tensor_embeddable<utility::cartesian_tensor<double, 2>> &&
              !tensor_embeddable<utility::cartesian_tensor<double, 3>>);
static_assert(tensor_projectable<utility::cartesian_tensor<double, 3>> &&
              !tensor_projectable<utility::cartesian_tensor<double, 2>>);

TEST_CASE("cartesian_tensor embed/project between 2x2 and 3x3", "[tensor]")
{
  SECTION("embed places the 2x2 in the top-left block, zero elsewhere")
  {
    utility::cartesian_tensor t3 = embed(utility::cartesian_tensor{1.0, 2.0, 3.0, 4.0});  // [[1, 2], [3, 4]]
    static_assert(std::is_same_v<decltype(t3), utility::cartesian_tensor<double, 3>>);
    REQUIRE(t3(0, 0) == 1.0);
    REQUIRE(t3(0, 1) == 2.0);
    REQUIRE(t3(1, 0) == 3.0);
    REQUIRE(t3(1, 1) == 4.0);
    REQUIRE(t3(0, 2) == 0.0);
    REQUIRE(t3(2, 0) == 0.0);
    REQUIRE(t3(2, 2) == 0.0);
  }

  SECTION("project keeps the top-left 2x2 block")
  {
    utility::cartesian_tensor t2 = project(utility::cartesian_tensor{1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0});
    static_assert(std::is_same_v<decltype(t2), utility::cartesian_tensor<double, 2>>);
    REQUIRE(t2(0, 0) == 1.0);
    REQUIRE(t2(0, 1) == 2.0);
    REQUIRE(t2(1, 0) == 4.0);
    REQUIRE(t2(1, 1) == 5.0);
  }

  SECTION("project . embed is the identity on 2x2")
  {
    utility::cartesian_tensor t2{1.0, 2.0, 3.0, 4.0};
    REQUIRE(project(embed(t2)) == t2);
  }
}