feat(libmeanfield): centrifugal + pressure
This commit is contained in:
@@ -11,7 +11,7 @@ namespace mean_field::utils {
|
||||
mfem::Vector &x_ref
|
||||
) {
|
||||
const int dim = fem.mesh->Dimension();
|
||||
x_ref = x_phys_target;
|
||||
x_ref = x_phys_target;
|
||||
|
||||
mfem::Array<int> init_elem;
|
||||
mfem::Array<mfem::IntegrationPoint> init_ip;
|
||||
@@ -29,8 +29,10 @@ namespace mean_field::utils {
|
||||
mfem::Array<mfem::IntegrationPoint> origin_ip;
|
||||
fem.mesh->FindPoints(P_origin, origin_elem, origin_ip, false);
|
||||
|
||||
if (origin_elem.Size() > 0 && origin_elem[0] >= 0 && !fem.mapping->IsIdentity()) {
|
||||
mfem::ElementTransformation *T0 = fem.mesh->GetElementTransformation(origin_elem[0]);
|
||||
if (origin_elem.Size() > 0 && origin_elem[0] >= 0 &&
|
||||
fem.mapping->HasDisplacementField()) {
|
||||
mfem::ElementTransformation *T0 =
|
||||
fem.mesh->GetElementTransformation(origin_elem[0]);
|
||||
T0->SetIntPoint(&origin_ip[0]);
|
||||
|
||||
mfem::DenseMatrix J0(dim, dim), J0_inv(dim, dim);
|
||||
@@ -81,7 +83,8 @@ namespace mean_field::utils {
|
||||
|
||||
if (elem_ids.Size() == 0 || elem_ids[0] < 0) {
|
||||
find_failures++;
|
||||
if (find_failures > 10) return false;
|
||||
if (find_failures > 10)
|
||||
return false;
|
||||
|
||||
double norm = x_ref.Norml2();
|
||||
if (norm > 1e-15) {
|
||||
@@ -92,10 +95,11 @@ namespace mean_field::utils {
|
||||
continue;
|
||||
}
|
||||
|
||||
int elemID = elem_ids[0];
|
||||
int elemID = elem_ids[0];
|
||||
const mfem::IntegrationPoint &ip = ips[0];
|
||||
|
||||
mfem::ElementTransformation *T = fem.mesh->GetElementTransformation(elemID);
|
||||
mfem::ElementTransformation *T =
|
||||
fem.mesh->GetElementTransformation(elemID);
|
||||
T->SetIntPoint(&ip);
|
||||
|
||||
mfem::Vector current_x_phys(dim);
|
||||
@@ -135,7 +139,8 @@ namespace mean_field::utils {
|
||||
x_ref = x_ref_candidate;
|
||||
} else {
|
||||
find_failures++;
|
||||
if (find_failures > 10) return false;
|
||||
if (find_failures > 10)
|
||||
return false;
|
||||
if (double norm = x_ref.Norml2(); norm > 1e-15) {
|
||||
x_ref *= 0.5 * RADIUS / norm;
|
||||
} else {
|
||||
@@ -155,7 +160,8 @@ namespace mean_field::utils {
|
||||
const mapping::COORDINATE_SPACE rspace
|
||||
) {
|
||||
mfem::Vector x_search;
|
||||
if (vspace == mapping::COORDINATE_SPACE::PHYSICAL && fem.has_mapping()) {
|
||||
if (vspace == mapping::COORDINATE_SPACE::PHYSICAL &&
|
||||
fem.has_mapping()) {
|
||||
GetReferencePoint(fem, x, x_search);
|
||||
} else {
|
||||
x_search = x;
|
||||
@@ -171,18 +177,22 @@ namespace mean_field::utils {
|
||||
double local_val = 0.0;
|
||||
if (elem_ids.Size() > 0 && elem_ids[0] >= 0) {
|
||||
const double val = u.GetValue(elem_ids[0], ips[0]);
|
||||
if (rspace == mapping::COORDINATE_SPACE::PHYSICAL && !fem.has_mapping()) {
|
||||
MFEM_ABORT("Physical evaluation mode requested but no mapping provided. Check domain bounds and mapping setup.");
|
||||
if (rspace == mapping::COORDINATE_SPACE::PHYSICAL &&
|
||||
!fem.has_mapping()) {
|
||||
MFEM_ABORT(
|
||||
"Physical evaluation mode requested but no mapping "
|
||||
"provided. Check "
|
||||
"domain bounds and mapping setup."
|
||||
);
|
||||
}
|
||||
local_val = val;
|
||||
}
|
||||
|
||||
double global_val = 0.0;
|
||||
MPI_Allreduce(&local_val, &global_val, 1, MPI_DOUBLE, MPI_MAX, fem.H1_fes->GetComm());
|
||||
MPI_Allreduce(
|
||||
&local_val, &global_val, 1, MPI_DOUBLE, MPI_MAX, fem.mesh->GetComm()
|
||||
);
|
||||
return global_val;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
} // namespace mean_field::utils
|
||||
|
||||
Reference in New Issue
Block a user