#include "gtest/gtest.h"
#include <stdexcept>
#include <tug/Boundary.hpp>
#include <utility>
#include <vector>

using namespace std;
using namespace tug;

#include <gtest/gtest.h>

#define BOUNDARY_TEST(x) TEST(Boundary, x)

BOUNDARY_TEST(Element) {

  BoundaryElement boundaryElementClosed = BoundaryElement<double>();
  EXPECT_NO_THROW(BoundaryElement<double>());
  EXPECT_EQ(boundaryElementClosed.getType(), BC_TYPE_CLOSED);
  EXPECT_DOUBLE_EQ(boundaryElementClosed.getValue(), -1);
  EXPECT_THROW(boundaryElementClosed.setValue(0.2), std::invalid_argument);

  BoundaryElement boundaryElementConstant = BoundaryElement(0.1);
  EXPECT_NO_THROW(BoundaryElement(0.1));
  EXPECT_EQ(boundaryElementConstant.getType(), BC_TYPE_CONSTANT);
  EXPECT_DOUBLE_EQ(boundaryElementConstant.getValue(), 0.1);
  EXPECT_NO_THROW(boundaryElementConstant.setValue(0.2));
  EXPECT_DOUBLE_EQ(boundaryElementConstant.getValue(), 0.2);
}

BOUNDARY_TEST(Class) {
  Boundary<double> boundary1D(10);
  Boundary<double> boundary2D(10, 12);
  vector<BoundaryElement<double>> boundary1DVector(1, BoundaryElement(1.0));

  constexpr double inner_condition_value = -5;
  constexpr std::pair<bool, double> innerBoundary =
      std::make_pair(true, inner_condition_value);

  std::vector<std::pair<bool, double>> row_ibc(12, std::make_pair(false, -1));
  row_ibc[1] = innerBoundary;

  std::vector<std::pair<bool, double>> col_ibc(10, std::make_pair(false, -1));
  col_ibc[0] = innerBoundary;

  {
    EXPECT_EQ(boundary1D.getBoundarySide(BC_SIDE_LEFT).size(), 1);
    EXPECT_EQ(boundary1D.getBoundarySide(BC_SIDE_RIGHT).size(), 1);
    EXPECT_EQ(boundary1D.getBoundaryElementType(BC_SIDE_LEFT, 0),
              BC_TYPE_CLOSED);
    EXPECT_DEATH(boundary1D.getBoundarySide(BC_SIDE_TOP),
                 ".*BC_SIDE_LEFT .* BC_SIDE_RIGHT.*");
    EXPECT_DEATH(boundary1D.getBoundarySide(BC_SIDE_BOTTOM),
                 ".*BC_SIDE_LEFT .* BC_SIDE_RIGHT.*");
    EXPECT_NO_THROW(boundary1D.setBoundarySideClosed(BC_SIDE_LEFT));
    EXPECT_DEATH(boundary1D.setBoundarySideClosed(BC_SIDE_TOP),
                 ".*BC_SIDE_LEFT .* BC_SIDE_RIGHT.*");
    EXPECT_NO_THROW(boundary1D.setBoundarySideConstant(BC_SIDE_LEFT, 1.0));
    EXPECT_DOUBLE_EQ(boundary1D.getBoundaryElementValue(BC_SIDE_LEFT, 0), 1.0);
    EXPECT_DEATH(boundary1D.getBoundaryElementValue(BC_SIDE_LEFT, 2),
                 ".*Index is selected either too large or too small.*");
    EXPECT_EQ(boundary1D.getBoundaryElementType(BC_SIDE_LEFT, 0),
              BC_TYPE_CONSTANT);
    EXPECT_EQ(boundary1D.getBoundaryElement(BC_SIDE_LEFT, 0).getType(),
              boundary1DVector[0].getType());

    EXPECT_NO_THROW(boundary1D.setInnerBoundary(0, inner_condition_value));
    EXPECT_DEATH(boundary1D.setInnerBoundary(0, 0, inner_condition_value),
                 ".*only available for 2D grids.*");
    EXPECT_EQ(boundary1D.getInnerBoundary(0), innerBoundary);
    EXPECT_FALSE(boundary1D.getInnerBoundary(1).first);
  }

  {
    EXPECT_EQ(boundary2D.getBoundarySide(BC_SIDE_LEFT).size(), 10);
    EXPECT_EQ(boundary2D.getBoundarySide(BC_SIDE_RIGHT).size(), 10);
    EXPECT_EQ(boundary2D.getBoundarySide(BC_SIDE_TOP).size(), 12);
    EXPECT_EQ(boundary2D.getBoundarySide(BC_SIDE_BOTTOM).size(), 12);
    EXPECT_EQ(boundary2D.getBoundaryElementType(BC_SIDE_LEFT, 0),
              BC_TYPE_CLOSED);
    EXPECT_NO_THROW(boundary2D.getBoundarySide(BC_SIDE_TOP));
    EXPECT_NO_THROW(boundary2D.getBoundarySide(BC_SIDE_BOTTOM));
    EXPECT_NO_THROW(boundary2D.setBoundarySideClosed(BC_SIDE_LEFT));
    EXPECT_NO_THROW(boundary2D.setBoundarySideClosed(BC_SIDE_TOP));
    EXPECT_NO_THROW(boundary2D.setBoundarySideConstant(BC_SIDE_LEFT, 1.0));
    EXPECT_DOUBLE_EQ(boundary2D.getBoundaryElementValue(BC_SIDE_LEFT, 0), 1.0);
    EXPECT_DEATH(boundary2D.getBoundaryElementValue(BC_SIDE_LEFT, 12),
                 ".*too large or too small.*");
    EXPECT_EQ(boundary2D.getBoundaryElementType(BC_SIDE_LEFT, 0),
              BC_TYPE_CONSTANT);
    EXPECT_EQ(boundary2D.getBoundaryElement(BC_SIDE_LEFT, 0).getType(),
              boundary1DVector[0].getType());

    EXPECT_DEATH(boundary2D.setInnerBoundary(0, inner_condition_value),
                 ".* 1D .*");
    EXPECT_NO_THROW(boundary2D.setInnerBoundary(0, 1, inner_condition_value));
    EXPECT_EQ(boundary2D.getInnerBoundary(0, 1), innerBoundary);
    EXPECT_FALSE(boundary2D.getInnerBoundary(0, 2).first);

    EXPECT_EQ(boundary2D.getInnerBoundaryRow(0), row_ibc);
    EXPECT_EQ(boundary2D.getInnerBoundaryCol(1), col_ibc);
  }
}