clc;

clear;

%Input
E = 200e6;
A = 0.04;
L = 5;
P = 10;
q = 2;
%Solution
n = 10;
Le = L/n;
dof = 3*n-(n-1);
connect = zeros(n,3);
for i = 1:n;
connect(i,1:3) = [3*i-(i-1)-2,3*i-(i-1)-1,3*i-(i-1)];
end

klocal = A*E* [7/(3*Le) -8/(3*Le) 1/(3*Le);

-8/(3*Le) 16/(3*Le) -8/(3*Le);
1/(3*Le) -8/(3*Le) 7/(3*Le)];

rlocal = [q*Le/6; 2*q*Le/3; q*Le/6];

%Assemble global matrix

K = zeros(dof);
R = zeros(dof,1);
for i = 1:n;
K(connect (i,:),connect(i,:))= K(connect (i,:),connect(i,:)) + klocal;
R(connect (i,:),1) = R(connect (i,:),1) + rlocal;
end
K
R

%Point load,P at end of bar

R(dof) = R(dof) + P;

%Solve global displacement

U = zeros (dof,1);
U(2:dof) = K(2:dof,2:dof)\R(2:dof)

%FEM solution
%Internal force or reaction
b = zeros (3,n);
for i = 1:n;
b(:,i) = klocal*[U(connect(i,:))] - rlocal;
sig (:,i) = (klocal*[U(connect(i,:))] - rlocal)/A;
end