function validStiffElasP1(Num)
% function validStiffElasP1()
%  Validation function for the Assembly of the Stiffness Elasticity 
%  Matrix for P1-Lagrange finite elements.
%
%   The Stiffness Elasticity Matrix (K) is computed by functions StiffElasAssemblingP1{Version} where {Version} is one of
%     {'base', 'OptV0', 'OptV1','OptV2'}.
%     - Test 1: computation of the Elastic Stiffness Matrix using previous functions giving errors and cputimes
%     - Test 2: Approximation of the integral of epsilon(u)'*C*sigma(v) on unit square by 
%                    dot(K*U,V)
%       Functions u=(u1,u2) and v-v1,v2) are those defined in valid_FEMmatrices.
%       For a given mesh Th, we note q^i=(Th.q(1,i),Th.q(2,i)) and we have :
%         if Num in {0,2}    
%               U(2i-1)=u1(q^i), U(2i)=u2(q^i), V(2i-1)=v1(q^i), V(2i)=v2(q^i), 
%         if Num in {1,3}    
%               U(i)=u1(q^i), U(i+nq)=u2(q^i), V(i)=v1(q^i), V(i+nq)=v2(q^i), 
%     - Test 3: retrieve order 2 of `P_1`-Lagrange integration.
%
%   Num :
%    - 0 global alternate numbering with local alternate numbering (classical method), 
%    - 1 global block numbering with local alternate numbering,
%    - 2 global alternate numbering with local block numbering,
%    - 3 global block numbering with local block numbering.
%
% See also:
%   StiffElasAssemblingP1base, StiffElasAssemblingP1OptV0,
%   StiffElasAssemblingP1OptV1, StiffElasAssemblingP1OptV2, SquareMesh
% 
% author : Francois Cuvelier [2012-11-26]
%
% OptFEM2DP1 [V1.2b1] -    Copyright (C) 2013  CJS (LAGA)
%
%   This file is part of OptFEM2DP1.
%   OptFEM2DP1 is free software: you can redistribute it and/or modify
%   it under the terms of the GNU General Public License as published by
%   the Free Software Foundation, either version 3 of the License, or
%   (at your option) any later version.
%
%   OptFEM2DP1 is distributed in the hope that it will be useful,
%   but WITHOUT ANY WARRANTY; without even the implied warranty of
%   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
%   GNU General Public License for more details.
%
%   You should have received a copy of the GNU General Public License
%   along with this program.  If not, see <http://www.gnu.org/licenses/>.

  disp('********************************************')
  disp('*   Stiff Elas Assembling P1 validations   *')
  disp('********************************************')
  switch Num
  case 0
    s=sprintf('Global alternate numbering / local alternate numbering');
  case 1
    s=sprintf('Global block numbering / local alternate numbering');
  case 2
    s=sprintf('Global alternate numbering / local block numbering');
  case 3
    s=sprintf('Global block numbering / local block numbering');
  otherwise
    error('invalid Num value')
  end
  fprintf('  Numbering Choice : %s\n',s);
  
  
  
  Th=SquareMesh(20);

% TEST 1
  disp('-----------------------------------------')
  disp('  Test 1: Matrices errors and CPU times  ')
  disp('-----------------------------------------')
  tic();
  Mbase=StiffElasAssemblingP1base(Th.nq,Th.nme,Th.q,Th.me,Th.areas,1,1,Num);
  T(1)=toc();
  tic();
  MOptV0=StiffElasAssemblingP1OptV0(Th.nq,Th.nme,Th.q,Th.me,Th.areas,1,1,Num);
  T(2)=toc();
  Test1.error(1)=norm(Mbase-MOptV0,Inf);
  Test1.name{1}='StiffElasAssemblingP1OptV0';
  fprintf('    Error P1base vs OptV0 : %e\n',Test1.error(1))
  tic();
  MOptV1=StiffElasAssemblingP1OptV1(Th.nq,Th.nme,Th.q,Th.me,Th.areas,1,1,Num);
  T(3)=toc();
  Test1.error(2)=norm(Mbase-MOptV1,Inf);
  Test1.name{2}='StiffElasAssemblingP1OptV1';
  fprintf('    Error P1base vs OptV1 : %e\n',Test1.error(2))
  tic();
  MOptV2=StiffElasAssemblingP1OptV2(Th.nq,Th.nme,Th.q,Th.me,Th.areas,1,1,Num);
  T(4)=toc();
  Test1.error(3)=norm(Mbase-MOptV2,Inf);
  Test1.name{3}='StiffElasAssemblingP1OptV2';
  fprintf('    Error P1base vs OptV2 : %e\n',Test1.error(3))

  fprintf('    CPU times base (ref) : %3.4f (s)\n',T(1))
  fprintf('    CPU times OptV0       : %3.4f (s) - Speed Up X%3.3f\n',T(2),T(1)/T(2))
  fprintf('    CPU times OptV1       : %3.4f (s) - Speed Up X%3.3f\n',T(3),T(1)/T(3))
  fprintf('    CPU times OptV2       : %3.4f (s) - Speed Up X%3.3f\n',T(4),T(1)/T(4))
  checkTest1(Test1)

  


% TEST 2
  disp('-----------------------------------------------------')
  disp('  Test 2: Validations by integration on [0,1]x[0,1]  ')
  disp('-----------------------------------------------------')
  Test=valid_StiffElas();
  qf=Th.q;
  for kk=1:length(Test)
    KK=StiffElasAssemblingP1OptV2(Th.nq,Th.nme,Th.q,Th.me,Th.areas,Test(kk).lambda,Test(kk).mu,Num);
    u=Test(kk).u;
    v=Test(kk).v;
    switch Num
    case {0,2}
      U=[u{1}(qf(1,:),qf(2,:));u{2}(qf(1,:),qf(2,:))];
      U=U(:);
      V=[v{1}(qf(1,:),qf(2,:));v{2}(qf(1,:),qf(2,:))];
      V=V(:);
    case {1,3}
      U=[u{1}(qf(1,:),qf(2,:)) u{2}(qf(1,:),qf(2,:))]';
      V=[v{1}(qf(1,:),qf(2,:)) v{2}(qf(1,:),qf(2,:))]';
    end
    %whos
    Test(kk).error=abs(Test(kk).Stiff-U'*KK*V);
    fprintf('    functions %d : u(x,y)=(%s,%s), v(x,y)=(%s,%s),\n           -> Stiff error=%e\n', ...
            kk,Test(kk).cu{1},Test(kk).cu{2},Test(kk).cv{1},Test(kk).cv{2},Test(kk).error);
  end
  checkTest2(Test)
  
% TEST 3
  disp('--------------------------------')
  disp('  Test 3: Validations by order  ')
  disp('--------------------------------')
  n=length(Test);
  u=Test(n).u;
  v=Test(n).v;
  lambda=Test(n).lambda;
  mu=Test(n).mu;
  ExSol=Test(n).Stiff;

  for k=1:10  
    Th=SquareMesh(20*k+50);
    qf=Th.q;
    fprintf('    Matrix size : %d\n',Th.nq);
    h(k)=GetMaxLengthEdges(Th.q,Th.me);
    tic();
    M=StiffElasAssemblingP1OptV2(Th.nq,Th.nme,Th.q,Th.me,Th.areas,lambda,mu,Num);
    TT(k)=toc();
    switch Num
    case {0,2}
      U=[u{1}(qf(1,:),qf(2,:));u{2}(qf(1,:),qf(2,:))];
      U=U(:);
      V=[v{1}(qf(1,:),qf(2,:));v{2}(qf(1,:),qf(2,:))];
      V=V(:);
    case {1,3}
      U=[u{1}(qf(1,:),qf(2,:)) u{2}(qf(1,:),qf(2,:))]';
      V=[v{1}(qf(1,:),qf(2,:)) v{2}(qf(1,:),qf(2,:))]';
    end
    Error(k)=abs(ExSol-U'*M*V);
    fprintf('      StiffElasAssemblingP1OptV2 CPU times : %3.3f(s)\n',TT(k));
    fprintf('      Error                            : %e\n',Error(k));
  end

  figure(1)
  loglog(h,Error,'+-r',h,h*1.1*Error(1)/h(1),'-sm',h,1.1*Error(1)*(h/h(1)).^2,'-db')
  legend('Error','O(h)','O(h^2)')
  xlabel('h')
  title(sprintf('Test 3 : Stiff Elas. Matrix (Num=%d)',Num))
  checkTest3(h,Error)
  
  %figure(3)
  %spy(M)
  %title(sprintf('Matrix sparsity for %s numbering',s))
end

function checkTest1(Test)
  I=find(Test.error>1e-14);
  if isempty(I)
    disp('------------------------')
    disp('  Test 1 (results): OK')
    disp('------------------------')
  else
    disp('----------------------------')
    disp('  Test 1 (results): FAILED')
    disp('----------------------------')
  end
end

function checkTest2(Test)
  N=length(Test);
  cntFalse=0;
  for k=1:N
    if (Test(k).degree<=1)
      if (Test(k).error>1e-12)
        cntFalse=cntFalse+1;
      end
    end
  end
  if (cntFalse==0)
    disp('------------------------')
    disp('  Test 2 (results): OK')
    disp('------------------------')
  else
    disp('----------------------------')
    disp('  Test 2 (results): FAILED')
    disp('----------------------------')
  end
end

function checkTest3(h,error)
  % order 2
  P=polyfit(log(h),log(error),1);
  if abs(P(1)-2)<1e-3
    disp('------------------------')
    disp('  Test 3 (results): OK')
    fprintf('    -> found numerical order %f. Must be 2\n',P(1))
    disp('------------------------')
  else
    disp('----------------------------')
    disp('  Test 3 (results): FAILED')
    fprintf('    -> found numerical order %f. Must be 2\n',P(1))
    disp('----------------------------')
  end
end
