#include <memory>
#include "config.h"
#include "sha-base-framework/frame.h"
#include "sha-topology-optimization-3d/Boundary.h"
#include "sha-topology-optimization-3d/HeatIrregularMesh.h"
#include "sha-topology-optimization-3d/HeatMesh.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_heat_arbitrary.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_heat_arbitrary.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;
  para->E_factor = 1e-3;

  // set material parameters
  auto material        = std::make_shared<top::Material>(config.thermal_conductivity);

  // set mesh(regular)
  auto mesh = std::make_shared<top::HeatIrregularMesh>(WorkingAssetDirectoryPath()/ config.arbitraryModel, WorkingAssetDirectoryPath() / config.chosenModel,config.relativeLengthOfVoxel);

  // 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.GetBoundaryCoordsInAbsoluteAlignedBox(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.GetBoundaryCoordsInAbsoluteAlignedBox(nbc.box),nbc.val);
  }

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

  // extract txt or vtk
  write_tensor3d(da::WorkingResultDirectoryPath().string() + "/heat_field_matrix.txt", ten_rho,mesh->GetOrigin(),mesh->GetOrigin()+mesh->GetLenBox());
  top::WriteTensorToVtk(da::WorkingResultDirectoryPath() / "heat_field_matrix.vtk", ten_rho, mesh);
  if(!config.visualObjPath.empty()){
    // visual model with tensor
    //  rho
    {
      auto [mesh_obj,vertex_propty]=top::GetMeshVertexPropty(da::WorkingAssetDirectoryPath() / config.visualObjPath ,top3d.GetRhoFieldOneFilled(), bdr);
      std::string  vtk_path=(da::WorkingResultDirectoryPath()/"heat_arbitrary_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()/"heat_arbitrary_rho.txt").string(), vertex_propty);
    }
  }
}