%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%
% Extended Isogeometric analysis for two dimensional linear elastic
% fracture mechanics problems.
%
% Center crack infinite plate in tension
% Dirichlet BCs = exact displacements imposed with penalty method.
%
% Vinh Phu Nguyen,
% Delft University of Technology, The Netherlands
% nvinhphu@gmail.com
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

path(path,'../fem_util');
path(path,' ../C_files/');
path(path,' ../data/');
path(path,' ../meshing/');
path(path,' ../post-processing/');
path(path,' ../fem-functions/');
path(path,' ../analytical-solutions/');
path(path,' ../nurbs-geopdes/inst/');
path(path,' ../nurbs-util/');
path(path,' ../integration/');
path(path,' ./function');

close all
clear all

global p q controlPts weights element xCrack xTips crack_node
global uKnot vKnot noDofs levelSets split_nodes spit_elem noGPs1
global cont node

E0          = 1e7;  % Young modulus
nu0         = 0.3;  % Poisson ratio
stressState ='PLANE_STRAIN'; % either 'PLANE_STRAIN' or "PLANE_STRESS
sigmato     = 1e4;

vtuFile      = '../results/infiniteCrackModeI';
vtuCrackFile = '../results/infinieCrackModeI-cracked';

% COMPUTE ELASTICITY MATRIX
if ( strcmp(stressState,'PLANE_STRESS')  )      % Plane Strain case
    C=E0/(1-nu0^2)*[  1      nu0          0;
        nu0        1          0;
        0        0  (1-nu0)/2  ];
else                                            % Plane Strain case
    C=E0/(1+nu0)/(1-2*nu0)*[ 1-nu0      nu0        0;
        nu0    1-nu0        0;
        0        0  1/2-nu0 ];
end

tic;

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%
%%%  CAD input: control points and knot vectors
%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%edgeInfiniteCenterCrackC0Data
edgeCrack2D_data


noGPs         = p+1; % # of Gauss points along one direction
noGPs1        = p+1;

% find boundary nodes for boundary conditions

% bottomNodes =  find(controlPts(:,2)==0);
% rightNodes  =  find(controlPts(:,1)==L);
% topNodes    =  find(controlPts(:,2)==D);
% leftNodes   =  find(controlPts(:,1)==0);

bottomNodes = 1 : noPtsX;
topNodes    = noPtsX*(noPtsY-1)+1 : noCtrPts;
rightNodes  = noPtsX : noPtsX : noCtrPts;
leftNodes   = 1: noPtsX : noCtrPts;
% build boundary mesh

bottomPoints    = controlPts(bottomNodes,:);
rightPoints     = controlPts(rightNodes,:);
topPoints       = controlPts(topNodes,:);
leftPoints      = controlPts(leftNodes,:);

% Loop over elements along top edge 

[elRangeU,elConnU] = buildConnectivity(p,uKnot,Numx);
bottomEdgeMesh  = bottomNodes(elConnU);
topEdgeMesh  = topNodes(elConnU);

[elRangeV,elConnV] = buildConnectivity(q,vKnot,Numy);
rightEdgeMesh  = rightNodes(elConnV);
leftEdgeMesh  = leftNodes(elConnV);

% bottomEdgeMesh  = zeros(noElemsU,p+1);
% rightEdgeMesh   = zeros(noElemsV,q+1);
% topEdgeMesh     = zeros(noElemsU,p+1);
% leftEdgeMesh    = zeros(noElemsU,q+1);
% for i=1:noElemsU
%     bottomEdgeMesh(i,:) = bottomNodes(i:i+p);
%     topEdgeMesh(i,:)    = topNodes(i:i+p);
% end
% 
% for i=1:noElemsV
%     rightEdgeMesh(i,:) = rightNodes(i:i+q);
%     leftEdgeMesh(i,:)  = leftNodes(i:i+q);
% end

% Each split node is enriched by ONE function, H(x)
% Each tip node is enriched by FOUR functions, B_i(x)
% then, the total dofs is :
% total dof = numnode*nsd + numsplitnode*1*nsd + numstipnode*4*nsd
% here, two dimension, nsd = 2

noDofs = noDofs + size(split_nodes,1)*1*2 + ...
    size(tip_nodes,1)*4*2;

K = sparse(noDofs,noDofs);  % global stiffness matrix
u = zeros(noDofs,1);        % displacement vector
f = zeros(noDofs,1);        % external force vector

% We use fictitious nodes/control points to handle these
% additional dofs. At a H(x) enriched node, we add one fantom node and
% at tip enriched node, four fantom nodes are added. These fictitious nodes
% are numbered from the total number of true nodes, ie, from numnode+1 ...

pos    = zeros(noCtrPts,1);
nsnode = 0 ;
ntnode = 0 ;

for i = 1 : noCtrPts
    if (enrich_node(i) == 1)
        pos(i) = (noCtrPts + nsnode*1 + ntnode*4) + 1 ;
        nsnode = nsnode + 1 ;
    elseif (enrich_node(i) == 2)
        pos(i) = (noCtrPts + nsnode*1 + ntnode*4) + 1 ;
        ntnode = ntnode + 1 ;
    end
end

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%
%%% PROCESSING
%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%%% Assembling system of equation
%%% Stiffness matrix and external force vector

disp([num2str(toc,'%0.6g'),'  ASSEMBLING THE SYSTEM'])

% Loop over elements (knot spans)
tol=1e-8;
for e=1:numelem
    ni = inn(ien(e,1),1);% get NURBS coordinates
    nj = inn(ien(e,1),2);
    if (abs(uKnot(ni)-uKnot(ni+1))>tol)&&(abs(vKnot(nj)-vKnot(nj+1))>tol)
        xiE=[uKnot(ni), uKnot(ni+1)];
        etaE=[vKnot(nj), vKnot(nj+1)];
               
        sctr   = element(e,:);          %  element scatter vector
        sctrV  = elementV(e,:);         %  element scatter vector Q4
        pts    = controlPts(sctr,:);
        nn     = length(sctr);
   
    % -----------------------------------------------
    % Choose Gauss quadrature rules for elements
%         [W,Q] = gaussPointForEnrichedElement(e,noGPs,sctrV,levelSets,xTip,split_elem,...
%                                         tip_elem,tip_nodes);
        
    
    if     (ismember(e,split_elem)) % split element, 13 GPs/subtriangle
%         [W,Q] = discontQ4quad(7,levelSets(1,elementV(e,:),1));
        [W,Q] = quadrature(20,'GAUSS',2);
    elseif (ismember(e,tip_elem))   % tip element
%         [W,Q] = disTipQ4quad(7,levelSets(1,elementV(e,:)),node(sctrV,:),xTip);
        [W,Q] = quadrature(20,'GAUSS',2);
    elseif ( any(intersect(tip_nodes,sctr)) ~= 0)% having tip enriched nodes
        [W,Q] = quadrature(12,'GAUSS',2);
    else
        [W,Q] = quadrature(10,'GAUSS',2);
    end
    
    % Determine the position in the global matrix K
    
        sctrB = assembly(e,element,enrich_node,pos);
    
    % loop over Gauss points
    
        for gp=1:size(W,1)
            pt      = Q(gp,:);
        % compute coords in parameter space
            Xi   = ((xiE(2) - xiE(1))*pt(1)+xiE(2) + xiE(1))/2;
            Eta  = ((etaE(2)-etaE(1))*pt(2)+etaE(2)+etaE(1))/2;
            J2   = (xiE(2) - xiE(1))*(etaE(2)-etaE(1))/4;
            
%         Xi      = parent2ParametricSpace(xiE,pt(1));
%         Eta     = parent2ParametricSpace(etaE,pt(2));
%         J2      = jacobianPaPaMapping(xiE,etaE);
    
        % compute derivative of basis functions w.r.t parameter coord
        
            [N dRdxi dRdeta] = NURBS2DBasisDers([Xi; Eta],p,q,uKnot,vKnot,weights');
         
        % B matrix 
        
            [B,J1] = BMatrixXIGA(Xi,Eta,e,enrich_node,N,dRdxi,dRdeta);
            
        
        % Stiffness matrix
             
            K(sctrB,sctrB) = K(sctrB,sctrB) + B'*C*B*W(gp)*J1*J2;
        end
           
%         [N' dRdxi' dRdeta']
%         [N dRdxi dRdeta]=NURBS_Shape_2D_Ck(e,pt(1),pt(2),uKnot,vKnot,p,q,b_net);
%         [N' dRdxi' dRdeta']

    end
end

noElemsU=Numx;
noElemsV=Numy;
noElems =numelem;

%% Boundary conditions

disp([num2str(toc,4),'  APPLYING BOUNDARY CONDITIONS'])

% If penalty method is use, then one must modify the stiffness matrix and
% the nodal force vector
% $K_{ij}$ = Kij - alpha \int phi_i phi_j d \gamma_u
% fj       = fj  - alpha \int phi_i u_bar d \gamma_u


% Loop over elements along top edge

fu = zeros(noDofs,1);
k  = zeros(noDofs,noDofs);

[ff kk en] = penaltyBoundaryCondition(elRangeU,elConnU,Numx,...
    bottomEdgeMesh,bottomPoints,E0,nu0,stressState,...
    sigmato,xTip,seg,cracklength,pos,1);

fu   = fu + ff;
k    = k + kk;

[ff kk en] = penaltyBoundaryCondition(elRangeU,elConnU,Numx,...
    topEdgeMesh,topPoints,E0,nu0,stressState,...
    sigmato,xTip,seg,cracklength,pos,1);

fu   = fu + ff;
k    = k + kk;

[ff kk en] = penaltyBoundaryCondition(elRangeV,elConnV,Numy,...
    rightEdgeMesh,rightPoints,E0,nu0,stressState,...
    sigmato,xTip,seg,cracklength,pos,1);

fu   = fu + ff;
k    = k + kk;

clear kk ff
[W1,Q1] = quadrature(noGPs1, 'GAUSS', 1 );

%% left edge: natural BCs
% Loop over elements along left edge = noElemsV

for e=1:Numy
    xiE   = elRangeV(e,:); % [xi_i,xi_i+1]
    conn  = elConnV(e,:);
    sctrx = 2*leftEdgeMesh(e,:)-1;
    sctry = 2*leftEdgeMesh(e,:);
    
    % loop over Gauss points
    for gp=1:size(W1,1)
        xi      = Q1(gp,:);
        wt      = W1(gp);
        Xi      = 0.5 * ( ( xiE(2) - xiE(1) ) * xi + xiE(2) + xiE(1));
        J2      = 0.5 * ( xiE(2) - xiE(1) );
        
        [N dNdxi] = NURBS1DBasisDers(Xi,p,uKnot,weights);
        
        % compute the jacobian of physical and parameter domain mapping
        % then the derivative w.r.t spatial physical coordinates
        
        x        = N    *leftPoints(conn,:); % global coord of GP
        jacob1   = dNdxi*leftPoints(conn,:);
        J1       = norm (jacob1);
        
        [sigmaxx,sigmayy,sigmaxy] = exactStressModeI(x,E0,nu0,...
            sigmato,xTip,seg,cracklength);
        
        tx = -sigmaxx;
        ty = -sigmaxy;
        
        f(sctrx) = f(sctrx) + N' * tx * J1 * J2 * wt;
        f(sctry) = f(sctry) + N' * ty * J1 * J2 * wt;
    end
end
%}

% take the enrichment into account when compute the
% external force vector

% for e=1:noElemsV
%     sctr = leftEdgeMesh(e,:);
%     le   = length(sctr);
%
%     if (length(intersect(sctr,split_nodes))~=le)
%         continue
%     end
%
%     xiE   = elRangeV(e,:); % [xi_i,xi_i+1]
%     conn  = elConnV(e,:);
%     noFns = length(conn);
%     sctrx = 2*pos(sctr)-1;
%     sctry = 2*pos(sctr);
%
%     xCr    = reshape(xCrack(1,:,:),2,2);
%     Ne     = zeros(le,1);
%
%     % loop over Gauss points
%     for gp=1:size(W1,1)
%         xi      = Q1(gp,:);
%         wt      = W1(gp);
%         Xi      = 0.5 * ( ( xiE(2) - xiE(1) ) * xi + xiE(2) + xiE(1));
%         J2      = 0.5 * ( xiE(2) - xiE(1) );
%
%         N       = [];
%         dNdxi   = [];
%
%         % compute derivative of basis functions w.r.t parameter coord
%
%         for in=1:noFns
%             [Ni,dNi]  = NURBSbasis (conn(in),q,Xi,vKnot,weights);
%             N         = [N Ni];
%             dNdxi     = [dNdxi dNi];
%         end
%
%         % compute the jacobian of physical and parameter domain mapping
%         % then the derivative w.r.t spatial physical coordinates
%
%         x        = N    *leftPoints(conn,:); % global coord of GP
%         jacob1   = dNdxi*leftPoints(conn,:);
%         J1       = norm (jacob1);
%
%         [sigmaxx,sigmayy,sigmaxy] = exactStressModeI(x,E0,nu0,...
%                                    sigmato,xTip,seg,cracklength);
%
%         tx = -sigmaxx;
%         ty = -sigmaxy;
%
%         % Enrichment function, H(x) at global Gauss point
%         dist = signed_distance(xCr,x);
%         Hgp  = heaviside(dist);
%
%         % Enrichment function, H(x) at node "in"
%
%         for i=1:length(sctr)
%             nodeId = sctr(i);
%             dist   = signed_distance(xCr,controlPts(nodeId,:));
%             H      = heaviside(dist);H=0;
%             Ne(i)  = N(i)*(Hgp-H);
%         end
%
%         f(sctrx) = f(sctrx) + Ne * tx * J1 * J2 * wt;
%         f(sctry) = f(sctry) + Ne * ty * J1 * J2 * wt;
%     end
% end

% modified system of equations

alpha = 0.05*max(max(C)) ;           % penalty number
alpha = 1e10;
f = f - alpha*fu;
K = K - alpha*k;

clear fu k
%
% SOLVE SYSTEM
disp([num2str(toc,'%0.6g'),'  SOLVING THE SYSTEM'])

U=K\f;
% break
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%
%%% POST-PROCESSING
%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

Ux = U(1:2:noDofs);
Uy = U(2:2:noDofs);

% compute stresses and displacements at mesh vertices
disp([num2str(toc,'%0.6g'), '  COMPUTING STRESSES '])

plotStress;

% plot stress

stressComp=2;
figure
clf
hold on
plot_field(node+30*[dispX dispY],elementV,'Q4',sigmaYY);
hold on
colorbar
title('Stress in x direction')
axis off
plot_mesh(node+30*[dispX dispY],elementV,'Q4','g.-',1.2);

figure
clf
plot_field(node,elementV,'Q4',displacement(:,:,2));
hold on
colorbar
title('Displacement in y direction')
axis off

opts = struct('Color','rgb','Bounds','tight','FontMode','fixed','FontSize',20);
%exportfig(gcf,fileName,opts)

clear disp;

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%
disp([num2str(toc,'%0.6g'),'  SIFs COMPUTATION'])

computeInteractionIntegrals

% Compute SIFs from I integral

% Knum = I.*E0/(2*(1-nu0^2)) % plain strain

% Compute the exact SIFs

Kexact = sigmato*sqrt(pi*cracklength)


%%%%%%%%%%%%

for i = 1 : size(node,1)
    x = node(i,:) ;
    [ux1,uy1] = exactDispModeI(x,E0,nu0,stressState,sigmato,xTip,seg,cracklength);
    ux_exact(i) = ux1;
    uy_exact(i) = uy1;
    [sigmaxx,sigmayy,sigmaxy] = exactStressModeI(x,E0,nu0,...
        sigmato,xTip,seg,cracklength);
    eSigmaXX(i) = sigmaxx;
    eSigmaYY(i) = sigmayy;
    eSigmaXY(i) = sigmaxy;
    % von Mises stress
    eSigmaVM(i) = sqrt(sigmaxx^2+sigmayy^2-sigmaxx*sigmayy+3*sigmaxy^2);
end

vtuFile = 'infiniteCrackModeIExact';
VTKPostProcess(node,elementV,'Quad4',vtuFile,...
    [eSigmaXX' eSigmaYY' eSigmaXY' eSigmaVM'],[ux_exact' uy_exact'])

% ----------------------------------

% --------------------------------------------
% Plot both exact and numerical deformed shape
fac=3000000;
figure
hold on
h = plot(node(:,1)+fac*dispX,node(:,2)+fac*dispY,'rs');
set(h,'MarkerSize',7);
h = plot(node(:,1)+fac*ux_exact',node(:,2)+fac*uy_exact','b*');
set(h,'MarkerSize',7);
title('Exact and numerical deformed shape')
legend('XIGA','Exact')
axis equal
axis off
set(gcf,'color','white')
% --------------------------------------------

%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% post-processing for cracks

% crackedMeshNURBS

%%



