Herezh_dev/Resolin/Matrices_externes/MV++/mvblas_GR.h

195 lines
5.4 KiB
C
Raw Normal View History

2021-09-28 17:28:23 +02:00
/*+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++*/
/* */
/* */
/* MV++ Numerical Matrix/Vector C++ Library */
/* MV++ Version 1.5 */
/* */
/* R. Pozo */
/* National Institute of Standards and Technology */
/* */
/* NOTICE */
/* */
/* Permission to use, copy, modify, and distribute this software and */
/* its documentation for any purpose and without fee is hereby granted */
/* provided that this permission notice appear in all copies and */
/* supporting documentation. */
/* */
/* Neither the Institution (National Institute of Standards and Technology) */
/* nor the author makes any representations about the suitability of this */
/* software for any purpose. This software is provided ``as is''without */
/* expressed or implied warranty. */
/* */
/*+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++*/
// modification et ajout de gérard rio
#ifndef _MV_BLAS1_TPL_H_
#define _MV_BLAS1_TPL_H_
#include "Sortie.h"
//#include "Debug.h"
#include <math.h>
template <class TYPE>
int operator==(const MV_Vector<TYPE> &x,const MV_Vector<TYPE> &y)
{
int Nx = x.size();
if(Nx != y.size())
return 0;
for (int i=0;i<Nx;i++)
if (x(i) != y(i))
return 0;
return 1;
}
template <class TYPE>
int operator!=(const MV_Vector<TYPE> &x,const MV_Vector<TYPE> &y)
{
if (x == y)
return 0;
else
return 1;
}
template <class TYPE>
MV_Vector<TYPE>& operator*=(MV_Vector<TYPE> &x, const TYPE &a)
{
int N = x.size();
for (int i=0;i<N;i++)
x(i) *= a;
return x;
}
template <class TYPE>
MV_Vector<TYPE> operator*(const TYPE &a, const MV_Vector<TYPE> &x)
{
int N = x.size();
MV_Vector<TYPE> result(N);
for (int i=0;i<N;i++)
result(i) = x(i)*a;
return result;
}
template <class TYPE>
MV_Vector<TYPE> operator*(const MV_Vector<TYPE> &x, const TYPE &a)
{
// This is the other commutative case of vector*scalar.
// It should be just defined to be
// "return operator*(a,x);"
// but some compilers (e.g. Turbo C++ v.3.0) have trouble
// determining the proper template match. For the moment,
// we'll just duplicate the code in the scalar * MV_Vector
// case above.
int N = x.size();
MV_Vector<TYPE> result(N);
for (int i=0;i<N;i++)
result(i) = x(i)*a;
return result;
}
template <class TYPE>
MV_Vector<TYPE> operator+(const MV_Vector<TYPE> &x, const MV_Vector<TYPE> &y)
{
int N = x.size();
#ifdef MISE_AU_POINT
if (N != y.size())
{
cout << "Incompatible vector lengths in +." << endl;
Sortie(1);
}
#endif
MV_Vector<TYPE> result(N);
for (int i=0;i<N; i++)
result(i) = x(i) + y(i);
return result;
}
template <class TYPE>
MV_Vector<TYPE> operator-(const MV_Vector<TYPE> &x, const MV_Vector<TYPE> &y)
{
int N = x.size();
#ifdef MISE_AU_POINT
if (N != y.size())
{
cout << "Incompatible vector lengths in -." << endl;
Sortie(1);
}
#endif
MV_Vector<TYPE> result(N);
for (int i=0;i<N; i++)
result(i) = x(i) - y(i);
return result;
}
template <class TYPE>
MV_Vector<TYPE>& operator+=(MV_Vector<TYPE> &x, const MV_Vector<TYPE> &y)
{
int N = x.size();
#ifdef MISE_AU_POINT
if (N != y.size())
{
cout << "Incompatible vector lengths in -." << endl;
Sortie(1);
}
#endif
for (int i=0;i<N; i++)
x(i) += y(i);
return x;
}
template <class TYPE>
MV_Vector<TYPE>& operator-=(MV_Vector<TYPE> &x, const MV_Vector<TYPE> &y)
{
int N = x.size();
#ifdef MISE_AU_POINT
if (N != y.size())
{
cout << "Incompatible vector lengths in -." << endl;
Sortie(1);
}
#endif
for (int i=0;i<N; i++)
x(i) -= y(i);
return x;
}
// norm and dot product functions for the MV_Vector<> class
template <class TYPE>
TYPE dot(const MV_Vector<TYPE> &x, const MV_Vector<TYPE> &y)
{
// Check for compatible dimensions:
#ifdef MISE_AU_POINT
if (x.size() != y.size())
{
cout << "Incompatible dimensions in dot(). " << endl;
Sortie(1);
}
#endif
TYPE temp=0.0;
for (int i=0; i<x.size();i++)
temp += x(i)*y(i);
return temp;
}
template <class TYPE>
TYPE norm(const MV_Vector<TYPE> &x)
{
TYPE temp = dot(x,x);
return sqrt(temp);
}
#endif
// _MV_BLAS1_TPL_H_