| //===================================================== |
| // File : blitz_interface.hh |
| // Author : L. Plagne <laurent.plagne@edf.fr)> |
| // Copyright (C) EDF R&D, lun sep 30 14:23:30 CEST 2002 |
| // Copyright (C) 2008 Gael Guennebaud <gael.guennebaud@inria.fr> |
| //===================================================== |
| // |
| // 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 BLITZ_INTERFACE_HH |
| #define BLITZ_INTERFACE_HH |
| |
| #include <blitz/blitz.h> |
| #include <blitz/array.h> |
| #include <blitz/vector-et.h> |
| #include <blitz/vecwhere.h> |
| #include <blitz/matrix.h> |
| #include <vector> |
| |
| BZ_USING_NAMESPACE(blitz) |
| |
| template<class real> |
| class blitz_interface{ |
| |
| public : |
| |
| typedef real real_type ; |
| |
| typedef std::vector<real> stl_vector; |
| typedef std::vector<stl_vector > stl_matrix; |
| |
| typedef blitz::Array<real, 2> gene_matrix; |
| typedef blitz::Array<real, 1> gene_vector; |
| // typedef blitz::Matrix<real, blitz::ColumnMajor> gene_matrix; |
| // typedef blitz::Vector<real> gene_vector; |
| |
| static inline std::string name() { return "blitz"; } |
| |
| static void free_matrix(gene_matrix & A, int N){} |
| |
| static void free_vector(gene_vector & B){} |
| |
| static inline void matrix_from_stl(gene_matrix & A, stl_matrix & A_stl){ |
| A.resize(A_stl[0].size(),A_stl.size()); |
| for (int j=0; j<A_stl.size() ; j++){ |
| for (int i=0; i<A_stl[j].size() ; i++){ |
| A(i,j)=A_stl[j][i]; |
| } |
| } |
| } |
| |
| static inline void vector_from_stl(gene_vector & B, stl_vector & B_stl){ |
| B.resize(B_stl.size()); |
| for (int i=0; i<B_stl.size() ; i++){ |
| B(i)=B_stl[i]; |
| } |
| } |
| |
| static inline void vector_to_stl(gene_vector & B, stl_vector & B_stl){ |
| for (int i=0; i<B_stl.size() ; i++){ |
| B_stl[i]=B(i); |
| } |
| } |
| |
| static inline void matrix_to_stl(gene_matrix & A, stl_matrix & A_stl){ |
| int N=A_stl.size(); |
| for (int j=0;j<N;j++){ |
| A_stl[j].resize(N); |
| for (int i=0;i<N;i++) |
| A_stl[j][i] = A(i,j); |
| } |
| } |
| |
| static inline void matrix_matrix_product(const gene_matrix & A, const gene_matrix & B, gene_matrix & X, int N) |
| { |
| firstIndex i; |
| secondIndex j; |
| thirdIndex k; |
| X = sum(A(i,k) * B(k,j), k); |
| } |
| |
| static inline void ata_product(const gene_matrix & A, gene_matrix & X, int N) |
| { |
| firstIndex i; |
| secondIndex j; |
| thirdIndex k; |
| X = sum(A(k,i) * A(k,j), k); |
| } |
| |
| static inline void aat_product(const gene_matrix & A, gene_matrix & X, int N) |
| { |
| firstIndex i; |
| secondIndex j; |
| thirdIndex k; |
| X = sum(A(i,k) * A(j,k), k); |
| } |
| |
| static inline void matrix_vector_product(gene_matrix & A, gene_vector & B, gene_vector & X, int N) |
| { |
| firstIndex i; |
| secondIndex j; |
| X = sum(A(i,j)*B(j),j); |
| } |
| |
| static inline void atv_product(gene_matrix & A, gene_vector & B, gene_vector & X, int N) |
| { |
| firstIndex i; |
| secondIndex j; |
| X = sum(A(j,i) * B(j),j); |
| } |
| |
| static inline void axpy(const real coef, const gene_vector & X, gene_vector & Y, int N) |
| { |
| firstIndex i; |
| Y = Y(i) + coef * X(i); |
| //Y += coef * X; |
| } |
| |
| static inline void copy_matrix(const gene_matrix & source, gene_matrix & cible, int N){ |
| cible = source; |
| //cible.template operator=<gene_matrix>(source); |
| // for (int i=0;i<N;i++){ |
| // for (int j=0;j<N;j++){ |
| // cible(i,j)=source(i,j); |
| // } |
| // } |
| } |
| |
| static inline void copy_vector(const gene_vector & source, gene_vector & cible, int N){ |
| //cible.template operator=<gene_vector>(source); |
| cible = source; |
| } |
| |
| }; |
| |
| #endif |