1 //===================================================== 2 // File : init_matrix.hh 3 // Author : L. Plagne <laurent.plagne (at) 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> 28 BTL_DONT_INLINE void init_row(Vector & X, int size, int row){ 29 30 X.resize(size); 31 32 for (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> 43 BTL_DONT_INLINE void init_matrix(Vector & A, int size){ 44 A.resize(size); 45 for (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> 51 BTL_DONT_INLINE void init_matrix_symm(Matrix& A, int size){ 52 A.resize(size); 53 for (int row=0; row<A.size() ; row++) 54 A[row].resize(size); 55 for (int row=0; row<A.size() ; row++){ 56 A[row][row] = init_function(row,row); 57 for (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
