You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
3.1 KiB
73 lines
3.1 KiB
#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);
|
|
}
|
|
}
|
|
}
|
|
|