1c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath//===================================================== 2c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// File : blitz_interface.hh 3c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Author : L. Plagne <laurent.plagne@edf.fr)> 4c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Copyright (C) EDF R&D, lun sep 30 14:23:30 CEST 2002 5c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Copyright (C) 2008 Gael Guennebaud <gael.guennebaud@inria.fr> 6c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath//===================================================== 7c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 8c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This program is free software; you can redistribute it and/or 9c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// modify it under the terms of the GNU General Public License 10c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// as published by the Free Software Foundation; either version 2 11c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// of the License, or (at your option) any later version. 12c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 13c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This program is distributed in the hope that it will be useful, 14c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// but WITHOUT ANY WARRANTY; without even the implied warranty of 15c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// GNU General Public License for more details. 17c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// You should have received a copy of the GNU General Public License 18c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// along with this program; if not, write to the Free Software 19c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 20c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 21c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#ifndef BLITZ_INTERFACE_HH 22c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#define BLITZ_INTERFACE_HH 23c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 24c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#include <blitz/blitz.h> 25c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#include <blitz/array.h> 26c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#include <blitz/vector-et.h> 27c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#include <blitz/vecwhere.h> 28c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#include <blitz/matrix.h> 29c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#include <vector> 30c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 31c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan KamathBZ_USING_NAMESPACE(blitz) 32c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 33c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathtemplate<class real> 34c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathclass blitz_interface{ 35c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 36c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathpublic : 37c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 38c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef real real_type ; 39c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 40c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef std::vector<real> stl_vector; 41c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef std::vector<stl_vector > stl_matrix; 42c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 43c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef blitz::Array<real, 2> gene_matrix; 44c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef blitz::Array<real, 1> gene_vector; 45c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// typedef blitz::Matrix<real, blitz::ColumnMajor> gene_matrix; 46c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// typedef blitz::Vector<real> gene_vector; 47c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 48c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline std::string name() { return "blitz"; } 49c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 50c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static void free_matrix(gene_matrix & A, int N){} 51c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 52c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static void free_vector(gene_vector & B){} 53c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 54c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void matrix_from_stl(gene_matrix & A, stl_matrix & A_stl){ 55c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath A.resize(A_stl[0].size(),A_stl.size()); 56c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for (int j=0; j<A_stl.size() ; j++){ 57c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for (int i=0; i<A_stl[j].size() ; i++){ 58c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath A(i,j)=A_stl[j][i]; 59c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 60c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 61c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 62c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 63c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void vector_from_stl(gene_vector & B, stl_vector & B_stl){ 64c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath B.resize(B_stl.size()); 65c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for (int i=0; i<B_stl.size() ; i++){ 66c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath B(i)=B_stl[i]; 67c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 68c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 69c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 70c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void vector_to_stl(gene_vector & B, stl_vector & B_stl){ 71c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for (int i=0; i<B_stl.size() ; i++){ 72c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath B_stl[i]=B(i); 73c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 74c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 75c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 76c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void matrix_to_stl(gene_matrix & A, stl_matrix & A_stl){ 77c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath int N=A_stl.size(); 78c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for (int j=0;j<N;j++){ 79c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath A_stl[j].resize(N); 80c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath for (int i=0;i<N;i++) 81c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath A_stl[j][i] = A(i,j); 82c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 83c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 84c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 85c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void matrix_matrix_product(const gene_matrix & A, const gene_matrix & B, gene_matrix & X, int N) 86c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 87c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath firstIndex i; 88c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath secondIndex j; 89c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath thirdIndex k; 90c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath X = sum(A(i,k) * B(k,j), k); 91c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 92c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 93c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void ata_product(const gene_matrix & A, gene_matrix & X, int N) 94c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 95c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath firstIndex i; 96c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath secondIndex j; 97c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath thirdIndex k; 98c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath X = sum(A(k,i) * A(k,j), k); 99c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 100c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 101c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void aat_product(const gene_matrix & A, gene_matrix & X, int N) 102c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 103c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath firstIndex i; 104c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath secondIndex j; 105c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath thirdIndex k; 106c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath X = sum(A(i,k) * A(j,k), k); 107c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 108c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 109c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void matrix_vector_product(gene_matrix & A, gene_vector & B, gene_vector & X, int N) 110c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 111c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath firstIndex i; 112c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath secondIndex j; 113c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath X = sum(A(i,j)*B(j),j); 114c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 115c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 116c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void atv_product(gene_matrix & A, gene_vector & B, gene_vector & X, int N) 117c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 118c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath firstIndex i; 119c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath secondIndex j; 120c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath X = sum(A(j,i) * B(j),j); 121c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 122c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 123c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void axpy(const real coef, const gene_vector & X, gene_vector & Y, int N) 124c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 125c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath firstIndex i; 126c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Y = Y(i) + coef * X(i); 127c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath //Y += coef * X; 128c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 129c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 130c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void copy_matrix(const gene_matrix & source, gene_matrix & cible, int N){ 131c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath cible = source; 132c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath //cible.template operator=<gene_matrix>(source); 133c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// for (int i=0;i<N;i++){ 134c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// for (int j=0;j<N;j++){ 135c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// cible(i,j)=source(i,j); 136c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// } 137c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// } 138c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 139c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 140c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath static inline void copy_vector(const gene_vector & source, gene_vector & cible, int N){ 141c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath //cible.template operator=<gene_vector>(source); 142c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath cible = source; 143c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 144c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 145c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 146c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 147c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#endif 148