get_displacements.m

function displacements = get_displacements(GDof, prescribedDof, stiffness, force)
    % function to find solution in terms of global displacements
    activeDof = setdiff([1:GDof]', [prescribedDof]);
    U = stiffness(activeDof, activeDof)\force(activeDof);
    displacements=zeros(GDof, 1);
    displacements(activeDof)=U;
end