tboudreaux/SERiF
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(poly): changed lambda from fespace to scalar

Browse Source 
previously I had a lagrangian multipliers at every element; however, we are enforcing a global constraint so there need only be one lagrangian multiplier
This commit is contained in:
Emily Boudreaux
2025-02-20 15:36:46 -05:00
parent a0811dc0c4
commit ff299f8ce7
2 changed files with 5 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/poly/solver/private/polySolver.cpp
View File

@@ -20,7 +20,6 @@ PolySolver::PolySolver(double n, double order)
mesh(meshIO.GetMesh()),
feCollection(std::make_unique<mfem::H1_FECollection>(order, mesh.SpaceDimension())),
feSpace(std::make_unique<mfem::FiniteElementSpace>(&mesh, feCollection.get())),
lambdaFeSpace(std::make_unique<mfem::FiniteElementSpace>(&mesh, feCollection.get(), 1)), // Scalar space for lambda
compositeIntegrator(std::make_unique<polyMFEMUtils::CompositeNonlinearIntegrator>()),
nonlinearForm(std::make_unique<mfem::NonlinearForm>(feSpace.get())),
C(std::make_unique<mfem::LinearForm>(feSpace.get())),
@@ -73,15 +72,7 @@ void PolySolver::solve(){
// --- Combine DOFs (u and λ) into a single vector ---
int lambdaDofOffset = feSpace->GetTrueVSize(); // Get the size of θ space
int totalTrueDofs = lambdaDofOffset + lambdaFeSpace->GetTrueVSize();
std::cout << "Total True Dofs: " << totalTrueDofs << std::endl;
std::cout << "Lambda Dof Offset: " << lambdaDofOffset << std::endl;
std::cout << "Lambda True Dofs: " << lambdaFeSpace->GetTrueVSize() << std::endl;
std::cout << "U True Dofs: " << feSpace->GetTrueVSize() << std::endl;
if (totalTrueDofs != lambdaDofOffset + 1) {
throw std::runtime_error("The total number of true dofs is not equal to the sum of the lambda offset and the lambda space size");
}
int totalTrueDofs = lambdaDofOffset + 1;
mfem::Vector U(totalTrueDofs);
U = 0.0;
@@ -113,25 +104,17 @@ void PolySolver::solve(){
// --- Create the RHS of the augmented system ---
mfem::Vector B(totalTrueDofs);
B = 0.0;
// Set the constraint value (∫η(r) dΩ) in the last entry of B
// This sets the last entry to 1.0, this mighht be a problem later on...
mfem::ConstantCoefficient one(1.0);
mfem::LinearForm constraint_rhs(lambdaFeSpace.get());
constraint_rhs.AddDomainIntegrator(
new mfem::DomainLFIntegrator(*gaussianCoeff)
);
constraint_rhs.Assemble();
B[lambdaDofOffset] = constraint_rhs(0); // Get that single value for the rhs. Only one value because it's a scalar space
B[lambdaDofOffset] = 1.0;
// --- Solve the augmented system ---
newtonSolver.Mult(B, U);
// --- Extract the Solution ---
mfem::Vector u_sol_view(U.GetData(), lambdaDofOffset);
#pragma GCC diagnostic push // MFEM is using deprecated functions, I cant do anything about it so I will ignore the warning
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
u->SetData(u_sol_view);
#pragma GCC diagnostic pop
double lambda = U[lambdaDofOffset];
std::cout << "λ = " << lambda << std::endl;