#include <memory>
#include "config.h"
#include "sha-base-framework/frame.h"
#include "sha-topology-optimization-3d/Boundary.h"
#include "sha-topology-optimization-3d/IrregularMesh.h"
#include "sha-topology-optimization-3d/Top3d.h"
#include "sha-topology-optimization-3d/Util.h"
void GenerateDensityFieldByRegularModel() {
  using namespace da;
  using namespace da::sha;

  fs_path base_path = WorkingAssetDirectoryPath();
  auto config_path  = base_path / "config_mech_regular.json";
  log::info("Algo is working on path '{}'", base_path.string());

  // load config from json file
  Config config;
  if (!config.loadFromJSON(config_path.string())) {
    spdlog::error("error on reading json file!");
    exit(-1);
  }
  // config.backUpConfig((WorkingResultDirectoryPath() / "config.json").string());

  // set topology parameters
  auto para      = std::make_shared<top::CtrlPara>();
  para->max_loop = config.max_loop;
  para->volfrac  = config.volfrac;
  para->penal    = config.penal;
  para->r_min    = config.r_min;

  // set material parameters
  double E             = config.YM;
  double Poisson_ratio = config.PR;
  auto material        = std::make_shared<top::Material>(E, Poisson_ratio);

  // set mesh(regular)
  int len_x = config.len_x;
  int len_y = config.len_y;
  int len_z = config.len_z;
  auto mesh = std::make_shared<top::Mesh>(len_x, len_y, len_z);

  // initialize Top3d
  top::Top3d top3d(para, material, mesh);

  // auxiliary class(Boundary) help to get boundary coordinates
  //      see the comments in Boundary.h for more information
  top::Boundary bdr(mesh);

  //      add Dirichlet boundary, see the comments in Top3d::AddDBC for more information
  for(auto &dbc:config.v_Dir){
      top3d.AddDBC(bdr.GetChosenCoordsByRelativeAlignedBox(dbc.box), dbc.dir);
  }
  //      add Neumann boundary, see the comments in Top3d::AddNBC for more information
  for(auto &nbc:config.v_Neu){
    top3d.AddNBC(bdr.GetChosenCoordsByRelativeAlignedBox(nbc.box),nbc.val);
  }


  // process topology optimization
  top::Tensor3d ten_rho = top3d.TopOptMainLoop();

  // extract txt or vtk
  write_tensor3d(da::WorkingResultDirectoryPath().string() + "/field_matrix.txt", ten_rho,mesh->GetOrigin(),mesh->GetOrigin()+mesh->GetLenBox());
  WriteTensorToVtk(da::WorkingResultDirectoryPath() / "field_matrix.vtk", ten_rho, mesh);
  // extract stress field
  auto v_tenosr=top3d.GetTensorOfStress(Eigen::Vector3d{0,1,2});
  write_tensor3d(da::WorkingResultDirectoryPath().string() + "/stressX_field_matrix.txt", v_tenosr[0],mesh->GetOrigin(),mesh->GetOrigin()+mesh->GetLenBox());
  write_tensor3d(da::WorkingResultDirectoryPath().string() + "/stressY_field_matrix.txt", v_tenosr[1],mesh->GetOrigin(),mesh->GetOrigin()+mesh->GetLenBox());
  write_tensor3d(da::WorkingResultDirectoryPath().string() + "/stressZ_field_matrix.txt", v_tenosr[2],mesh->GetOrigin(),mesh->GetOrigin()+mesh->GetLenBox());
  WriteTensorToVtk(da::WorkingResultDirectoryPath() / "stressX_field_matrix.vtk", v_tenosr[0], mesh);
  WriteTensorToVtk(da::WorkingResultDirectoryPath() / "stressY_field_matrix.vtk", v_tenosr[1], mesh);
  WriteTensorToVtk(da::WorkingResultDirectoryPath() / "stressZ_field_matrix.vtk", v_tenosr[2], mesh);
  if(!config.visualObjPath.empty()){
    // visual model with tensor
    //  rho
    {
      auto [mesh_obj,vertex_propty]=top::GetMeshVertexPropty(da::WorkingAssetDirectoryPath() / config.visualObjPath ,top3d.GetRhoFieldOneFilled(), bdr,true);
      std::string  vtk_path=(da::WorkingResultDirectoryPath()/"regular_with_rho.vtk").string();
      WriteTriVTK(vtk_path, mesh_obj.mat_coordinates, mesh_obj.mat_faces,{}, std::vector<double>(vertex_propty.data(),vertex_propty.data()+vertex_propty.size()));
      spdlog::info("write vtk with rho to: {}",vtk_path);
      top::WriteVectorXd((da::WorkingResultDirectoryPath()/"regular_rho.txt").string(), vertex_propty);
    }
    //  stressX
    {
      auto [mesh_obj,vertex_propty]=top::GetMeshVertexPropty(da::WorkingAssetDirectoryPath()/config.visualObjPath,v_tenosr[0], bdr,true);
      std::string  vtk_path=(da::WorkingResultDirectoryPath()/"regular_with_stressX.vtk").string();
      WriteTriVTK(vtk_path, mesh_obj.mat_coordinates, mesh_obj.mat_faces,{}, std::vector<double>(vertex_propty.data(),vertex_propty.data()+vertex_propty.size()));
      spdlog::info("write vtk with stressX to: {}",vtk_path);
      top::WriteVectorXd((da::WorkingResultDirectoryPath()/"regular_stressX.txt").string(), vertex_propty);
    }
    //  stressY
    {
      auto [mesh_obj,vertex_propty]=top::GetMeshVertexPropty(da::WorkingAssetDirectoryPath()/config.visualObjPath,v_tenosr[1], bdr,true);
      std::string  vtk_path=(da::WorkingResultDirectoryPath()/"regular_with_stressY.vtk").string();
      WriteTriVTK(vtk_path, mesh_obj.mat_coordinates, mesh_obj.mat_faces,{}, std::vector<double>(vertex_propty.data(),vertex_propty.data()+vertex_propty.size()));
      spdlog::info("write vtk with stressY to: {}",vtk_path);
      top::WriteVectorXd((da::WorkingResultDirectoryPath()/"regular_stressY.txt").string(), vertex_propty);
    }
    //  stressZ
    {
      auto [mesh_obj,vertex_propty]=top::GetMeshVertexPropty(da::WorkingAssetDirectoryPath()/config.visualObjPath,v_tenosr[2], bdr,true);
      std::string  vtk_path=(da::WorkingResultDirectoryPath()/"regular_with_stressZ.vtk").string();
      WriteTriVTK(vtk_path, mesh_obj.mat_coordinates, mesh_obj.mat_faces,{}, std::vector<double>(vertex_propty.data(),vertex_propty.data()+vertex_propty.size()));
      spdlog::info("write vtk with stressZ to: {}",vtk_path);
      top::WriteVectorXd((da::WorkingResultDirectoryPath()/"regular_stressZ.txt").string(), vertex_propty);
    }
  }
  

 
}