// SPDX-FileCopyrightText: Copyright © DUNE Project contributors, see file LICENSE.md in module root
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception
// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#include <dune/common/fmatrix.hh>
#include <dune/common/diagonalmatrix.hh>
#include <dune/istl/scaledidmatrix.hh>
#include <dune/istl/bcrsmatrix.hh>
#include <dune/istl/matrix.hh>
#include <dune/istl/io.hh>
#include "laplacian.hh"

/*  "tests" the writeMatrixToMatlabHelper and writeSVGMatrix methods by calling
 *  them for a Laplacian with a given BlockType and writing to cout.
 *  Actual functionality is not tested.
 */
template <class BlockType>
void testWriteMatrix(BlockType b=BlockType(0.0))
{
  typedef Dune::BCRSMatrix<BlockType> Matrix;

  Matrix A;
  setupLaplacian(A, 3);

  A[0][0] += b;
  writeMatrixToMatlabHelper(A, 0, 0, std::cout);
  writeSVGMatrix(std::cout, A);
}

/* uses the writeVectorToMatlab method, filled with dummy data */
template <class VectorType>
void testWriteVectorToMatlab()
{
  VectorType v;
  for (unsigned int i = 0; i < v.size(); ++i)
  {
    v[i] = i;
  }

  Dune::writeVectorToMatlabHelper(v, std::cout);
}

int main(int argc, char** argv)
{
  /* testing the writeMatrixToMatlabHelper method for BlockType=FieldMatrix with different field_types */
  testWriteMatrix<Dune::FieldMatrix<double,1,1> >();
  testWriteMatrix<Dune::FieldMatrix<double,1,2> >();
  //    testWriteMatrix<Dune::FieldMatrix<double,2,1> >(); // commented because setUpLaplacian cannot handle block_types with more rows than cols
  testWriteMatrix<Dune::FieldMatrix<double,4,7> >();
  //    testWriteMatrix<Dune::FieldMatrix<double,7,4> >(); // commented because setUpLaplacian cannot handle block_types with more rows than cols
  testWriteMatrix<Dune::FieldMatrix<double,2,2> >();

  testWriteMatrix<Dune::FieldMatrix<std::complex<double>,1,1> >(Dune::FieldMatrix<std::complex<double>,1,1>(std::complex<double>(0, 1)));
  testWriteMatrix<Dune::FieldMatrix<std::complex<double>,1,2> >(Dune::FieldMatrix<std::complex<double>,1,2>(std::complex<double>(0, 1)));
  //    testWriteMatrix<Dune::FieldMatrix<std::complex<double>,2,1> >(); // commented because setUpLaplacian cannot handle block_types with more rows than cols
  testWriteMatrix<Dune::FieldMatrix<std::complex<double>,4,7> >(Dune::FieldMatrix<std::complex<double>,4,7>(std::complex<double>(0, 1)));
  //    testWriteMatrix<Dune::FieldMatrix<std::complex<double>,7,4> >(); // commented because setUpLaplacian cannot handle block_types with more rows than cols
  testWriteMatrix<Dune::FieldMatrix<std::complex<double>,2,2> >(Dune::FieldMatrix<std::complex<double>,2,2>(std::complex<double>(0, 1)));

  testWriteMatrix<double>();
  testWriteMatrix<std::complex<double> >();

  /* testing the writeMatrixToMatlabHelper method for BlockType=[Diagonal|ScaledIdentity]Matrix with different field_types */
  testWriteMatrix<Dune::DiagonalMatrix<double,1> >();
  testWriteMatrix<Dune::ScaledIdentityMatrix<double,1> >();
  testWriteMatrix<Dune::DiagonalMatrix<double,2> >();
  testWriteMatrix<Dune::ScaledIdentityMatrix<double,2> >();

  testWriteMatrix<Dune::DiagonalMatrix<std::complex<double>,1> >(Dune::DiagonalMatrix<std::complex<double>,1>(std::complex<double>(0,1)));
  testWriteMatrix<Dune::ScaledIdentityMatrix<std::complex<double>,1> >(Dune::ScaledIdentityMatrix<std::complex<double>,1>(std::complex<double>(0,1)));
  testWriteMatrix<Dune::DiagonalMatrix<std::complex<double>,2> >(Dune::DiagonalMatrix<std::complex<double>,2>(std::complex<double>(0,1)));
  testWriteMatrix<Dune::ScaledIdentityMatrix<std::complex<double>,2> >(Dune::ScaledIdentityMatrix<std::complex<double>,2>(std::complex<double>(0,1)));


  /* testing the writeVectorToMatlabHelper method for FieldVector */
  testWriteVectorToMatlab<Dune::FieldVector<double,1> >();
  testWriteVectorToMatlab<Dune::FieldVector<float,5> >();
  testWriteVectorToMatlab<Dune::FieldVector<std::complex<double>,5> >();
  /* testing the writeVectorToMatlabHelper method for BlockVector */
  Dune::BlockVector<Dune::FieldVector<double,3> > v1 = {{1.0, 2.0, 3.0}};
  Dune::writeVectorToMatlabHelper(v1, std::cout);
  Dune::BlockVector<Dune::BlockVector<Dune::FieldVector<double,1> > > v2 = {{1.0, 2.0}, {3.0, 4.0, 5.0}, {6.0}};
  Dune::writeVectorToMatlabHelper(v2, std::cout);
  /* testing the writeVectorToMatlabHelper method for STL containers */
  testWriteVectorToMatlab<std::array<double,5> >();
  std::vector<double> v3;
  v3.push_back(1);
  v3.push_back(2);
  Dune::writeVectorToMatlabHelper(v3, std::cout);

  // Test the printmatrix and printSparseMatrix methods
  // BCRSMatrix
  {
    Dune::BCRSMatrix<double> matrix;
    setupLaplacian(matrix, 3);
    Dune::printmatrix(std::cout, matrix, "BCRSMatrix<double>", "--");
    Dune::printSparseMatrix(std::cout, matrix, "BCRSMatrix<double>", "--");
  }
  {
    Dune::BCRSMatrix<Dune::FieldMatrix<double,1,1> > matrix;
    setupLaplacian(matrix, 3);
    Dune::printmatrix(std::cout, matrix, "BCRSMatrix<FieldMatrix<double,1,1> >", "--");
    Dune::printSparseMatrix(std::cout, matrix, "BCRSMatrix<FieldMatrix<double,1,1> >", "--");
  }
  {
    Dune::BCRSMatrix<Dune::ScaledIdentityMatrix<double,1> > matrix;
    setupLaplacian(matrix, 3);
    Dune::printSparseMatrix(std::cout, matrix, "BCRSMatrix<ScaledIdentityMatrix<double,1> >", "--");
  }
  {
    Dune::BCRSMatrix<Dune::FieldMatrix<double,2,3> > matrix;
    setupLaplacian(matrix, 3);
    Dune::printmatrix(std::cout, matrix, "BCRSMatrix<FieldMatrix<double,2,3> >", "--");
    Dune::printSparseMatrix(std::cout, matrix, "BCRSMatrix<FieldMatrix<double,2,3> >", "--");
  }

  // Matrix
  {
    Dune::Matrix<double> matrix(3,3);
    matrix = 0;
    Dune::printmatrix(std::cout, matrix, "Matrix<double>", "--");
  }
  {
    Dune::Matrix<Dune::FieldMatrix<double,1,1> > matrix(3,3);
    matrix = 0;
    Dune::printmatrix(std::cout, matrix, "Matrix<FieldMatrix<double,1,1> >", "--");
  }
  {
    Dune::Matrix<Dune::FieldMatrix<double,2,3> > matrix(3,3);
    matrix = 0;
    Dune::printmatrix(std::cout, matrix, "Matrix<FieldMatrix<double,2,3> >", "--");
  }
}
