small fix for last commit

WIAS-PDELib · Jan 16, 2024 · 9943bc9 · 9943bc9
1 parent 1219538
commit 9943bc9
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/src/solver_config.jl b/src/solver_config.jl
@@ -137,13 +137,18 @@ function SolverConfiguration(Problem::ProblemDescription, unknowns::Array{Unknow
 	end
 
 	## adjustments for using freedofs
-	if length(parameters[:restrict_dofs]) > 0
-		freedofs = Vector{Int}(parameters[:restrict_dofs][1])
-		for j = 2 : length(parameters[:restrict_dofs])
-			parameters[:restrict_dofs][j] .+= FES[j-1].ndofs
-			append!(freedofs, parameters[:restrict_dofs][j])
+	if haskey(parameters, :restrict_dofs)
+		if length(parameters[:restrict_dofs]) > 0
+			freedofs = Vector{Int}(parameters[:restrict_dofs][1])
+			for j = 2 : length(parameters[:restrict_dofs])
+				parameters[:restrict_dofs][j] .+= FES[j-1].ndofs
+				append!(freedofs, parameters[:restrict_dofs][j])
+			end
+			x_temp = deepcopy(b)
+		else
+			freedofs = []
+			x_temp = x
 		end
-		x_temp = deepcopy(b)
 	else
 		freedofs = []
 		x_temp = x