1//=====================================================
2// File   :  action_lu_solve.hh
3// Author :  L. Plagne <laurent.plagne@edf.fr)>
4// Copyright (C) EDF R&D,  lun sep 30 14:23:19 CEST 2002
5//=====================================================
6//
7// This program is free software; you can redistribute it and/or
8// modify it under the terms of the GNU General Public License
9// as published by the Free Software Foundation; either version 2
10// of the License, or (at your option) any later version.
11//
12// This program is distributed in the hope that it will be useful,
13// but WITHOUT ANY WARRANTY; without even the implied warranty of
14// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15// GNU General Public License for more details.
16// You should have received a copy of the GNU General Public License
17// along with this program; if not, write to the Free Software
18// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19//
20#ifndef ACTION_LU_SOLVE
21#define ACTION_LU_SOLVE
22#include "utilities.h"
23#include "STL_interface.hh"
24#include <string>
25#include "init/init_function.hh"
26#include "init/init_vector.hh"
27#include "init/init_matrix.hh"
28
29using namespace std;
30
31template<class Interface>
32class Action_lu_solve
33{
34
35public :
36
37  static inline std::string name( void )
38  {
39    return "lu_solve_"+Interface::name();
40  }
41
42  static double nb_op_base(int size){
43    return 2.0*size*size*size/3.0;  // questionable but not really important
44  }
45
46
47  static double calculate( int nb_calc, int size ) {
48
49    // STL matrix and vector initialization
50
51    typename Interface::stl_matrix A_stl;
52    typename Interface::stl_vector B_stl;
53    typename Interface::stl_vector X_stl;
54
55    init_matrix<pseudo_random>(A_stl,size);
56    init_vector<pseudo_random>(B_stl,size);
57    init_vector<null_function>(X_stl,size);
58
59    // generic matrix and vector initialization
60
61    typename Interface::gene_matrix A;
62    typename Interface::gene_vector B;
63    typename Interface::gene_vector X;
64
65    typename Interface::gene_matrix LU;
66
67    Interface::matrix_from_stl(A,A_stl);
68    Interface::vector_from_stl(B,B_stl);
69    Interface::vector_from_stl(X,X_stl);
70    Interface::matrix_from_stl(LU,A_stl);
71
72    // local variable :
73
74    typename Interface::Pivot_Vector pivot; // pivot vector
75    Interface::new_Pivot_Vector(pivot,size);
76
77    // timer utilities
78
79    Portable_Timer chronos;
80
81    // time measurement
82
83    chronos.start();
84
85    for (int ii=0;ii<nb_calc;ii++){
86
87      // LU factorization
88      Interface::copy_matrix(A,LU,size);
89      Interface::LU_factor(LU,pivot,size);
90
91      // LU solve
92
93      Interface::LU_solve(LU,pivot,B,X,size);
94
95    }
96
97    // Time stop
98
99    chronos.stop();
100
101    double time=chronos.user_time();
102
103    // check result :
104
105    typename Interface::stl_vector B_new_stl(size);
106    Interface::vector_to_stl(X,X_stl);
107
108    STL_interface<typename Interface::real_type>::matrix_vector_product(A_stl,X_stl,B_new_stl,size);
109
110    typename Interface::real_type error=
111      STL_interface<typename Interface::real_type>::norm_diff(B_stl,B_new_stl);
112
113    if (error>1.e-5){
114      INFOS("WRONG CALCULATION...residual=" << error);
115      STL_interface<typename Interface::real_type>::display_vector(B_stl);
116      STL_interface<typename Interface::real_type>::display_vector(B_new_stl);
117      exit(0);
118    }
119
120    // deallocation and return time
121
122    Interface::free_matrix(A,size);
123    Interface::free_vector(B);
124    Interface::free_vector(X);
125    Interface::free_Pivot_Vector(pivot);
126
127    return time;
128  }
129
130};
131
132
133#endif
134
135
136
137