• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //=====================================================
2 // File   :  init_matrix.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 INIT_MATRIX_HH
21 #define INIT_MATRIX_HH
22 
23 // The Vector class must satisfy the following part of STL vector concept :
24 //            resize() method
25 //            [] operator for setting element
26 //            value_type defined
27 template<double init_function(int,int), class Vector>
init_row(Vector & X,int size,int row)28 BTL_DONT_INLINE void init_row(Vector & X, int size, int row){
29 
30   X.resize(size);
31 
32   for (unsigned int j=0;j<X.size();j++){
33     X[j]=typename Vector::value_type(init_function(row,j));
34   }
35 }
36 
37 
38 // Matrix is a Vector of Vector
39 // The Matrix class must satisfy the following part of STL vector concept :
40 //            resize() method
41 //            [] operator for setting rows
42 template<double init_function(int,int),class Vector>
init_matrix(Vector & A,int size)43 BTL_DONT_INLINE void init_matrix(Vector &  A, int size){
44   A.resize(size);
45   for (unsigned int row=0; row<A.size() ; row++){
46     init_row<init_function>(A[row],size,row);
47   }
48 }
49 
50 template<double init_function(int,int),class Matrix>
init_matrix_symm(Matrix & A,int size)51 BTL_DONT_INLINE void init_matrix_symm(Matrix&  A, int size){
52   A.resize(size);
53   for (unsigned int row=0; row<A.size() ; row++)
54     A[row].resize(size);
55   for (unsigned int row=0; row<A.size() ; row++){
56     A[row][row] = init_function(row,row);
57     for (unsigned int col=0; col<row ; col++){
58       double x = init_function(row,col);
59       A[row][col] = A[col][row] = x;
60     }
61   }
62 }
63 
64 #endif
65