/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         https://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/

#include "itkFEMElement3DC0LinearHexahedronMembrane.h"

#include <iostream>

//
int
itkFEMElement3DMembraneTest(int, char *[])
{

  using ElementType = itk::fem::Element;
  using NodeType = ElementType::Node;

  NodeType::Pointer       n0, n1, n2, n3, n4, n5, n6, n7;
  ElementType::VectorType pt(3);

  n0 = NodeType::New();
  pt[0] = 0.;
  pt[1] = 0.;
  pt[2] = 0.;
  n0->SetCoordinates(pt);

  n1 = NodeType::New();
  pt[0] = 1.;
  pt[1] = 0.;
  pt[2] = 0.;
  n1->SetCoordinates(pt);

  n2 = NodeType::New();
  pt[0] = 1.;
  pt[1] = 1.;
  pt[2] = 0.;
  n2->SetCoordinates(pt);

  n3 = NodeType::New();
  pt[0] = 0.;
  pt[1] = 1.;
  pt[2] = 0.;
  n3->SetCoordinates(pt);

  n4 = NodeType::New();
  pt[0] = 0.;
  pt[1] = 0.;
  pt[2] = 1.;
  n4->SetCoordinates(pt);

  n5 = NodeType::New();
  pt[0] = 1.;
  pt[1] = 0.;
  pt[2] = 1.;
  n5->SetCoordinates(pt);

  n6 = NodeType::New();
  pt[0] = 1.;
  pt[1] = 1.;
  pt[2] = 1.;
  n6->SetCoordinates(pt);

  n7 = NodeType::New();
  pt[0] = 0.;
  pt[1] = 1.;
  pt[2] = 1.;
  n7->SetCoordinates(pt);

  using ElasticityType = itk::fem::MaterialLinearElasticity;
  auto m = ElasticityType::New();
  m->SetGlobalNumber(0);
  m->SetYoungsModulus(10000.0);
  m->SetCrossSectionalArea(0.02);
  m->SetMomentOfInertia(.004);
  m->SetPoissonsRatio(0.4);

  using ElementMembraneType = itk::fem::Element3DC0LinearHexahedronMembrane;
  auto e0 = ElementMembraneType::New();

  e0->SetGlobalNumber(0);
  e0->SetNode(0, n0);
  e0->SetNode(1, n1);
  e0->SetNode(2, n2);
  e0->SetNode(3, n3);
  e0->SetNode(4, n4);
  e0->SetNode(5, n5);
  e0->SetNode(6, n6);
  e0->SetNode(7, n7);
  if (dynamic_cast<ElasticityType *>(m.GetPointer()))
  {
    e0->SetMaterial(dynamic_cast<ElasticityType *>(m.GetPointer()));
  }

  ElementType::MatrixType D, Me;

  e0->GetMassMatrix(Me);
  e0->GetMaterialMatrix(D);
  std::cout << "Mass matrix: " << std::endl << Me << std::endl;
  std::cout << "Material matrix: " << std::endl << D << std::endl;
  std::cout << "#dof per node = " << e0->GetNumberOfDegreesOfFreedomPerNode() << std::endl;

  std::cout << "Test PASSED!" << std::endl;
  return EXIT_SUCCESS;
}
