//===================================================== // File : init_matrix.hh // Author : L. Plagne <laurent.plagne@edf.fr)> // Copyright (C) EDF R&D, lun sep 30 14:23:19 CEST 2002 //===================================================== // // This program 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 2 // of the License, or (at your option) any later version. // // This program 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, write to the Free Software // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. // #ifndef INIT_MATRIX_HH #define INIT_MATRIX_HH // The Vector class must satisfy the following part of STL vector concept : // resize() method // [] operator for setting element // value_type defined template<double init_function(int,int), class Vector> BTL_DONT_INLINE void init_row(Vector & X, int size, int row){ X.resize(size); for (int j=0;j<X.size();j++){ X[j]=typename Vector::value_type(init_function(row,j)); } } // Matrix is a Vector of Vector // The Matrix class must satisfy the following part of STL vector concept : // resize() method // [] operator for setting rows template<double init_function(int,int),class Vector> BTL_DONT_INLINE void init_matrix(Vector & A, int size){ A.resize(size); for (int row=0; row<A.size() ; row++){ init_row<init_function>(A[row],size,row); } } template<double init_function(int,int),class Matrix> BTL_DONT_INLINE void init_matrix_symm(Matrix& A, int size){ A.resize(size); for (int row=0; row<A.size() ; row++) A[row].resize(size); for (int row=0; row<A.size() ; row++){ A[row][row] = init_function(row,row); for (int col=0; col<row ; col++){ double x = init_function(row,col); A[row][col] = A[col][row] = x; } } } #endif